try again, with puppetforge modules, correctly included now

[dsa-puppet.git] / 3rdparty / modules / openstacklib / lib / puppet / parser / functions / os_database_connection.rb

diff --git a/3rdparty/modules/openstacklib/lib/puppet/parser/functions/os_database_connection.rb b/3rdparty/modules/openstacklib/lib/puppet/parser/functions/os_database_connection.rb
new file mode 100644 (file)
index 0000000..8764f1b
--- /dev/null
+++ b/3rdparty/modules/openstacklib/lib/puppet/parser/functions/os_database_connection.rb
@@ -0,0 +1,72 @@
+require 'puppet/parser/functions'
+
+Puppet::Parser::Functions.newfunction(:os_database_connection,
+                                      :type => :rvalue,
+                                      :doc => <<-EOS
+This function builds a os_database_connection string from various parameters.
+EOS
+) do |arguments|
+
+  require 'uri'
+
+  if (arguments.size != 1) then
+    raise(Puppet::ParseError, "os_database_connection(): Wrong number of arguments " +
+      "given (#{arguments.size} for 1)")
+  end
+
+  v = arguments[0]
+  klass = v.class
+
+  unless klass == Hash
+    raise(Puppet::ParseError, "os_database_connection(): Requires an hash, got #{klass}")
+  end
+
+  v.keys.each do |key|
+    unless (v[key].class == String) or (v[key] == :undef)
+      raise(Puppet::ParseError, "os_database_connection(): #{key} should be a String")
+    end
+  end
+
+  parts = {}
+
+  unless v.include?('dialect')
+    raise(Puppet::ParseError, 'os_database_connection(): dialect is required')
+  end
+
+  if v.include?('host')
+    parts[:host] = v['host']
+  end
+
+  unless v.include?('database')
+    raise(Puppet::ParseError, 'os_database_connection(): database is required')
+  end
+
+  if v.include?('port')
+    if v.include?('host')
+      parts[:port] = v['port'].to_i
+    else
+      raise(Puppet::ParseError, 'os_database_connection(): host is required with port')
+    end
+  end
+
+  if v.include?('username') and (v['username'] != :undef) and (v['username'].to_s != '')
+    parts[:userinfo] = URI.escape(v['username'])
+    if v.include?('password') and (v['password'] != :undef) and (v['password'].to_s != '')
+      parts[:userinfo] += ":#{URI.escape(v['password'])}"
+    end
+  end
+
+  if v.include?('charset')
+    parts[:query] = "charset=#{v['charset']}"
+  end
+
+  parts[:scheme] = v['dialect']
+
+  if v.include?('host')
+    parts[:path] = "/#{v['database']}"
+  else
+    parts[:path] = "///#{v['database']}"
+  end
+
+  URI::Generic.build(parts).to_s
+end