3rdparty/modules/stdlib/lib/puppet/parser/functions/basename.rb

   1 #
   2 # basename.rb
   3 #
   4 module Puppet::Parser::Functions
   5   newfunction(:basename, :type => :rvalue, :doc => <<-DOC
   6     @summary
   7       Strips directory (and optional suffix) from a filename
   8
   9     @return [String] The stripped filename
  10     DOC
  11              ) do |arguments|
  12
  13     raise(Puppet::ParseError, 'basename(): No arguments given') if arguments.empty?
  14     raise(Puppet::ParseError, "basename(): Too many arguments given (#{arguments.size})") if arguments.size > 2
  15     raise(Puppet::ParseError, 'basename(): Requires string as first argument') unless arguments[0].is_a?(String)
  16
  17     rv = File.basename(arguments[0]) if arguments.size == 1
  18     if arguments.size == 2
  19       raise(Puppet::ParseError, 'basename(): Requires string as second argument') unless arguments[1].is_a?(String)
  20       rv = File.basename(arguments[0], arguments[1])
  21     end
  22
  23     return rv
  24   end
  25 end
  26
  27 # vim: set ts=2 sw=2 et :