summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRichard Soderberg <rsoderberg@mozilla.com>2013-03-26 15:45:40 -0700
committerRichard Soderberg <rsoderberg@mozilla.com>2013-03-26 15:46:55 -0700
commit88a93ac6cdf38045e1cf29325a70e5e4143016b3 (patch)
tree528c93aaa1c89dbdf1205fe4a69965de2d3d0874
parent05273419e1c8b34115ede15b1d8a8739f6a0db00 (diff)
downloadpuppet-stdlib-88a93ac6cdf38045e1cf29325a70e5e4143016b3.tar.gz
puppet-stdlib-88a93ac6cdf38045e1cf29325a70e5e4143016b3.tar.bz2
add suffix function to accompany the prefix function
-rw-r--r--README.markdown13
-rw-r--r--lib/puppet/parser/functions/suffix.rb45
-rw-r--r--spec/unit/puppet/parser/functions/suffix_spec.rb19
3 files changed, 77 insertions, 0 deletions
diff --git a/README.markdown b/README.markdown
index 121b784..2f93db4 100644
--- a/README.markdown
+++ b/README.markdown
@@ -661,6 +661,19 @@ Would result in: "aaa"
- *Type*: rvalue
+suffix
+------
+This function applies a suffix to all elements in an array.
+
+*Examples:*
+
+ suffix(['a','b','c'], 'p')
+
+Will return: ['ap','bp','cp']
+
+
+- *Type*: rvalue
+
swapcase
--------
This function will swap the existing case of a string.
diff --git a/lib/puppet/parser/functions/suffix.rb b/lib/puppet/parser/functions/suffix.rb
new file mode 100644
index 0000000..5018280
--- /dev/null
+++ b/lib/puppet/parser/functions/suffix.rb
@@ -0,0 +1,45 @@
+#
+# suffix.rb
+#
+
+module Puppet::Parser::Functions
+ newfunction(:suffix, :type => :rvalue, :doc => <<-EOS
+This function applies a suffix to all elements in an array.
+
+*Examples:*
+
+ suffix(['a','b','c'], 'p')
+
+Will return: ['ap','bp','cp']
+ EOS
+ ) do |arguments|
+
+ # Technically we support two arguments but only first is mandatory ...
+ raise(Puppet::ParseError, "suffix(): Wrong number of arguments " +
+ "given (#{arguments.size} for 1)") if arguments.size < 1
+
+ array = arguments[0]
+
+ unless array.is_a?(Array)
+ raise(Puppet::ParseError, 'suffix(): Requires array to work with')
+ end
+
+ suffix = arguments[1] if arguments[1]
+
+ if suffix
+ unless suffix.is_a?(String)
+ raise(Puppet::ParseError, 'suffix(): Requires string to work with')
+ end
+ end
+
+ # Turn everything into string same as join would do ...
+ result = array.collect do |i|
+ i = i.to_s
+ suffix ? i + suffix : i
+ end
+
+ return result
+ end
+end
+
+# vim: set ts=2 sw=2 et :
diff --git a/spec/unit/puppet/parser/functions/suffix_spec.rb b/spec/unit/puppet/parser/functions/suffix_spec.rb
new file mode 100644
index 0000000..c28f719
--- /dev/null
+++ b/spec/unit/puppet/parser/functions/suffix_spec.rb
@@ -0,0 +1,19 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+describe "the suffix function" do
+ let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+ it "should exist" do
+ Puppet::Parser::Functions.function("suffix").should == "function_suffix"
+ end
+
+ it "should raise a ParseError if there is less than 1 arguments" do
+ lambda { scope.function_suffix([]) }.should( raise_error(Puppet::ParseError))
+ end
+
+ it "should return a suffixed array" do
+ result = scope.function_suffix([['a','b','c'], 'p'])
+ result.should(eq(['ap','bp','cp']))
+ end
+end