Merge pull request #718 from sspreitzer/master_glob

Add glob function

3 files changed, 44 insertions, 0 deletions

diff --git a/README.markdown b/README.markdown
index 03daae9..eee46e0 100644
--- a/README.markdown
+++ b/README.markdown
@@ -665,6 +665,17 @@ This is useful if the namespace itself is stored in a string:
 
 *Type*: rvalue.
 
+#### `glob`
+
+Dir#glob wrapper that accepts a string or an array of strings of path patterns.
+Returns an array of strings of matched paths.
+
+  ~~~
+  $confs = glob(['/etc/**/*.conf', '/opt/**/*.conf'])
+  ~~~
+
+*Type*: rvalue.
+
 #### `grep`
 
 Searches through an array and returns any elements that match the provided regular expression. For example, `grep(['aaa','bbb','ccc','aaaddd'], 'aaa')` returns ['aaa','aaaddd']. *Type*: rvalue.
diff --git a/lib/puppet/parser/functions/glob.rb b/lib/puppet/parser/functions/glob.rb
new file mode 100644
index 0000000..54cdda6
--- /dev/null
+++ b/lib/puppet/parser/functions/glob.rb
@@ -0,0 +1,22 @@
+#
+#  glob.rb
+#
+
+module Puppet::Parser::Functions
+  newfunction(:glob, :type => :rvalue, :doc => <<-'EOS'
+    Returns an Array of file entries of a directory or an Array of directories.
+    Uses same patterns as Dir#glob
+    EOS
+  ) do |arguments|
+
+    raise(Puppet::ParseError, "glob(): Wrong number of arguments given " +
+      "(#{arguments.size} for 1)") unless arguments.size == 1
+
+    pattern = arguments[0]
+
+    raise(Puppet::ParseError, 'glob(): Requires either array or string ' +
+      'to work') unless pattern.is_a?(String) || pattern.is_a?(Array)
+
+    Dir.glob(pattern)
+  end
+end
diff --git a/spec/functions/glob_spec.rb b/spec/functions/glob_spec.rb
new file mode 100755
index 0000000..06439da
--- /dev/null
+++ b/spec/functions/glob_spec.rb
@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe 'glob' do
+  it { is_expected.not_to eq(nil) }
+  it { is_expected.to run.with_params().and_raise_error(Puppet::ParseError) }
+  it { is_expected.to run.with_params(1).and_raise_error(Puppet::ParseError) }
+  it { is_expected.to run.with_params('').and_return([]) }
+  it { is_expected.to run.with_params(['']).and_return([]) }
+  it { is_expected.to run.with_params(['', '']).and_return([]) }
+  it { is_expected.to run.with_params(['/etc/xyzxyzxyz', '/etcxyzxyzxyz']).and_return([]) }
+end