+++ /dev/null
-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