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     Strips directory (and optional suffix) from a filename
   7     DOC
   8              ) do |arguments|
   9
  10     raise(Puppet::ParseError, 'basename(): No arguments given') if arguments.empty?
  11     raise(Puppet::ParseError, "basename(): Too many arguments given (#{arguments.size})") if arguments.size > 2
  12     raise(Puppet::ParseError, 'basename(): Requires string as first argument') unless arguments[0].is_a?(String)
  13
  14     rv = File.basename(arguments[0]) if arguments.size == 1
  15     if arguments.size == 2
  16       raise(Puppet::ParseError, 'basename(): Requires string as second argument') unless arguments[1].is_a?(String)
  17       rv = File.basename(arguments[0], arguments[1])
  18     end
  19
  20     return rv
  21   end
  22 end
  23
  24 # vim: set ts=2 sw=2 et :