New params_lookup function. Moved url_parse from stdlib42

Author: alvagante

lib/puppet/parser/functions/params_lookup.rb

@@ -0,0 +1,53 @@
+#
+# params_lookup.rb
+#
+# This function lookups for a variable value in various locations
+# following this order
+# - Hiera backend, if present
+# - ::varname (if option abs_lookup is true)
+# - ::modulename_varname
+# - ::modulename::params::varname
+# - The value set with the "default" option
+#
+# It's based on a suggestion of Dan Bode on how to better manage
+# Example42 NextGen modules params lookups.
+# Code has been written with the decisive help of Brice Figureau,
+# Peter Meier and Ohad Levy during the Fosdem 2012 days (thanks guys)
+#
+# Alessandro Franceschi al@lab42.it
+# 
+module Puppet::Parser::Functions
+  newfunction(:params_lookup, :type => :rvalue, :doc => <<-EOS
+This fuction looks for the given variable name in a set of different sources:
+- Hiera, if available ('varname' if option abs_lookup is true)
+- Hiera, if available ('modulename_varname')
+- ::varname (if option abs_lookup is true)
+- ::modulename_varname
+- ::modulename::params::varname
+- The value set with the "default" option
+    EOS
+  ) do |arguments|
+
+    raise(Puppet::ParseError, "params_lookup(): Define at least the variable name " +
+      "given (#{arguments.size} for 1)") if arguments.size < 1
+
+    value = :undef
+    var_name = arguments[0]
+    options = { 'default' => :undef, 'abs_lookup' => false }.merge(arguments[1] || {}) 
+    module_name = parent_module_name
+    if Puppet::Parser::Functions.function('hiera')
+      value = function_hiera("#{var_name}",:undef) if options["abs_lookup"]
+      value = function_hiera("#{module_name}_#{var_name}",:undef) 
+    end
+    if value == :undef
+      value = undef_as(:undef, lookupvar("::#{var_name}")) if options["abs_lookup"]
+      value = undef_as(:undef, lookupvar("::#{module_name}_#{var_name}")) 
+      value = undef_as(:undef, lookupvar("::#{module_name}::params::#{var_name}")) if value == :undef
+      value = undef_as(:undef, options["default"]) if value == :undef
+    end
+    
+    return value
+  end
+end
+
+# vim: set ts=2 sw=2 et :

lib/puppet/parser/functions/url_parse.rb

@@ -0,0 +1,44 @@
+require 'uri'
+
+Puppet::Parser::Functions::newfunction(:url_parse, :type => :rvalue, :doc => <<-'ENDHEREDOC') do |args|
+    Returns information about an url
+    
+    This function expects two arguments, an URL and the part of the url you want to retrieve. 
+
+    Example:
+    $source_filename = url_parse($source_url,path)
+
+    Given an url like: https://my_user:my_pass@www.example.com:8080/path/to/file.php?id=1&ret=0
+    You obtain the following results according to the second argument:
+    scheme   : https
+    userinfo : my_user:my_pass
+    user     : my_user
+    password : my_pass
+    host     : www.example.com
+    port     : 8080
+    path     : /path/to/file.php
+    query    : id=1&ret=0
+    filename : file.php
+    filetype : php
+    filedir  : file
+   
+ 
+  ENDHEREDOC
+  raise ArgumentError, ("url_parse(): wrong number of arguments (#{args.length}; must be 2)") if args.length != 2
+  url=URI.parse args[0]
+  case args[1]
+    when 'scheme' then url.scheme
+    when 'userinfo' then url.userinfo
+    when 'user' then url.user
+    when 'password' then url.password
+    when 'host' then url.host
+    when 'port' then url.port
+    when 'path' then url.path
+    when 'query' then url.query
+    when 'filename' then File.basename url.path
+    when 'filetype' then File.extname url.path
+    when 'filedir' then (File.basename url.path).chomp(File.extname(url.path))
+    else url
+  end
+end
+