From 20e0e0709021b7ba4a819f1324526d98bcb5dcc3 Mon Sep 17 00:00:00 2001
From: Jaka Hudoklin <jakahudoklin@gmail.com>
Date: Wed, 2 Jan 2013 12:10:43 +0000
Subject: Add getparam function to get defined resource parameters

As far as i know there's no other puppet-dsl-like way to get parameter of
defined resource, so that's why i implemented getparam function, which takes
resource reference and parameter name and returns parameter value.

    Here's another example why this function is really useful:

    define config($path, $config_param1, $config_param2) { }

    define example_resource($config) {
        $path = getparam($config, "path")

        notice("Path is $path")
    }

    define example_resource2($example_resource, $config = getparam($example_resource, "config")) {
        $config_param1 = getparam($config, "config_param1")

        notice("Config parameter is $config_param1")
    }

    define example_resource3($example_resource, $config = getparam($example_resource, "config")) {
        $config_param2 = getparam($config, "config_param2")

        notice("Config parameter is $config_param2")
    }

    class test_getparam {

        config { "config_instance":
            path => "/some/config/path",
            config_param1 => "someconfigtext1",
            config_param2 => "someconfigtext2",
        }

        example_resource { "example_resource_instance":
            config => Config["config_instance"]
        }

        example_resource2 { "example_resource_instance":
            example_resource => Example_resource["example_resource_instance"]
        }

        example_resource3 { "example_resource_instance":
            example_resource => Example_resource2["example_resource_instance"]
        }
    }

    class { "test_getparam": }
---
 README.markdown                         | 19 ++++++++++++++++++
 lib/puppet/parser/functions/getparam.rb | 33 ++++++++++++++++++++++++++++++++
 spec/functions/getparam_spec.rb         | 34 +++++++++++++++++++++++++++++++++
 3 files changed, 86 insertions(+)
 create mode 100644 lib/puppet/parser/functions/getparam.rb
 create mode 100644 spec/functions/getparam_spec.rb

diff --git a/README.markdown b/README.markdown
index 21f7e7f..922383c 100644
--- a/README.markdown
+++ b/README.markdown
@@ -209,6 +209,25 @@ Example:
 
 - *Type*: rvalue
 
+getparam
+--------
+
+Takes a resource reference and name of the parameter and returns
+value of resource's parameter.
+
+For example:
+
+    define example_resource($param) {
+    }
+
+    example_resource { "example_resource_instance":
+        param => "param_value"
+    }
+
+    getparam(Example_resource["example_resource_instance"], "param")
+
+- *Type*: rvalue
+
 getvar
 ------
 Lookup a variable in a remote namespace.
diff --git a/lib/puppet/parser/functions/getparam.rb b/lib/puppet/parser/functions/getparam.rb
new file mode 100644
index 0000000..0962656
--- /dev/null
+++ b/lib/puppet/parser/functions/getparam.rb
@@ -0,0 +1,33 @@
+# Test whether a given class or definition is defined
+require 'puppet/parser/functions'
+
+Puppet::Parser::Functions.newfunction(:getparam,
+                                      :type => :rvalue,
+                                      :doc => <<-'ENDOFDOC'
+Takes a resource reference and name of the parameter and
+returns value of resource's parameter.
+
+*Examples:*
+
+    define example_resource($param) {
+    }
+
+    example_resource { "example_resource_instance":
+        param => "param_value"
+    }
+
+    getparam(Example_resource["example_resource_instance"], "param")
+
+Would return: param_value
+ENDOFDOC
+) do |vals|
+  reference, param = vals
+  raise(ArgumentError, 'Must specify a reference') unless reference
+  raise(ArgumentError, 'Must specify name of a parameter') unless param and param.instance_of? String
+
+  if resource = findresource(reference.to_s)
+    return resource[param] if resource[param]
+  end
+
+  return ''
+end
diff --git a/spec/functions/getparam_spec.rb b/spec/functions/getparam_spec.rb
new file mode 100644
index 0000000..d9c50a6
--- /dev/null
+++ b/spec/functions/getparam_spec.rb
@@ -0,0 +1,34 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+require 'rspec-puppet'
+describe 'getparam' do
+  describe 'when a resource is not specified' do
+    it do
+      should run.with_params().and_raise_error(ArgumentError)
+      should run.with_params('User[dan]').and_raise_error(ArgumentError)
+      should run.with_params('User[dan]', {}).and_raise_error(ArgumentError)
+      should run.with_params('User[dan]', '').and_return('')
+    end
+  end
+  describe 'when compared against a resource with no params' do
+    let :pre_condition do
+      'user { "dan": }'
+    end
+    it do
+      should run.with_params('User[dan]', 'shell').and_return('')
+    end
+  end
+
+  describe 'when compared against a resource with params' do
+    let :pre_condition do
+      'user { "dan": ensure => present, shell => "/bin/sh", managehome => false}'
+    end
+    it do
+      should run.with_params('User[dan]', 'shell').and_return('/bin/sh')
+      should run.with_params('User[dan]', '').and_return('')
+      should run.with_params('User[dan]', 'ensure').and_return('present')
+      should run.with_params('User[dan]', 'managehome').and_return(false)
+    end
+  end
+end
-- 
cgit v1.2.3