add nanliu/staging to 3rdparty

[dsa-puppet.git] / 3rdparty / modules / staging / lib / puppet / parser / functions / staging_parse.rb

diff --git a/3rdparty/modules/staging/lib/puppet/parser/functions/staging_parse.rb b/3rdparty/modules/staging/lib/puppet/parser/functions/staging_parse.rb
new file mode 100644 (file)
index 0000000..2c77559
--- /dev/null
+++ b/3rdparty/modules/staging/lib/puppet/parser/functions/staging_parse.rb
@@ -0,0 +1,36 @@
+require 'uri'
+
+module Puppet::Parser::Functions
+  newfunction(:staging_parse, :type => :rvalue, :doc => <<-EOS
+Parse filepath to retrieve information about the file.
+    EOS
+  ) do |arguments|
+
+    raise(Puppet::ParseError, "staging_parse(): Wrong number of arguments " +
+      "given (#{arguments.size} for 1, 2, 3)") if arguments.size < 1 || arguments.size > 3
+
+    source    = arguments[0]
+    path      = URI.parse(source.gsub('\\', '/')).path
+
+    raise Puppet::ParseError, "staging_parse(): #{source.inspect} has no URI " +
+      "'path' component" if path.nil?
+
+    info      = arguments[1] ? arguments[1] : 'filename'
+    extension = arguments[2] ? arguments[2] : File.extname(path)
+
+    case info
+    when 'filename'
+      result = File.basename(path)
+    when 'basename'
+      result = File.basename(path, extension)
+    when 'extname'
+      result = File.extname(path)
+    when 'parent'
+      result = File.expand_path(File.join(path, '..'))
+    else
+      raise Puppet::ParseError, "staging_parse(), unknown parse info #{info}."
+    end
+
+    return result
+  end
+end