Prepare Type & Provider

Signed-off-by: Emilien Macchi <emilien.macchi@enovance.com>

lib/puppet/provider/trove.rb

@@ -0,0 +1,150 @@
+require 'csv'
+require 'puppet/util/inifile'
+
+class Puppet::Provider::Trove < Puppet::Provider
+
+  def self.conf_filename
+    '/etc/trove/trove.conf'
+  end
+
+  def self.withenv(hash, &block)
+    saved = ENV.to_hash
+    hash.each do |name, val|
+      ENV[name.to_s] = val
+    end
+
+    yield
+  ensure
+    ENV.clear
+    saved.each do |name, val|
+      ENV[name] = val
+    end
+  end
+
+  def self.trove_credentials
+    @trove_credentials ||= get_trove_credentials
+  end
+
+  def self.get_trove_credentials
+    auth_keys = ['auth_host', 'auth_port', 'auth_protocol',
+                 'admin_tenant_name', 'admin_user', 'admin_password']
+    conf = trove_conf
+    if conf and conf['keystone_authtoken'] and
+        auth_keys.all?{|k| !conf['keystone_authtoken'][k].nil?}
+      return Hash[ auth_keys.map \
+                   { |k| [k, conf['keystone_authtoken'][k].strip] } ]
+    else
+      raise(Puppet::Error, "File: #{conf_filename} does not contain all \
+required sections.  Trove types will not work if trove is not \
+correctly configured.")
+    end
+  end
+
+  def trove_credentials
+    self.class.trove_credentials
+  end
+
+  def self.auth_endpoint
+    @auth_endpoint ||= get_auth_endpoint
+  end
+
+  def self.get_auth_endpoint
+    q = trove_credentials
+    "#{q['auth_protocol']}://#{q['auth_host']}:#{q['auth_port']}/v2.0/"
+  end
+
+  def self.trove_conf
+    return @trove_conf if @trove_conf
+    @trove_conf = Puppet::Util::IniConfig::File.new
+    @trove_conf.read(conf_filename)
+    @trove_conf
+  end
+
+  def self.auth_trove(*args)
+    q = trove_credentials
+    authenv = {
+      :OS_AUTH_URL    => self.auth_endpoint,
+      :OS_USERNAME    => q['admin_user'],
+      :OS_TENANT_NAME => q['admin_tenant_name'],
+      :OS_PASSWORD    => q['admin_password']
+    }
+    begin
+      withenv authenv do
+        trove(args)
+      end
+    rescue Exception => e
+      if (e.message =~ /\[Errno 111\] Connection refused/) or
+          (e.message =~ /\(HTTP 400\)/)
+        sleep 10
+        withenv authenv do
+          trove(args)
+        end
+      else
+       raise(e)
+      end
+    end
+  end
+
+  def auth_trove(*args)
+    self.class.auth_trove(args)
+  end
+
+  def self.reset
+    @trove_conf        = nil
+    @trove_credentials = nil
+  end
+
+  def self.list_trove_resources(type)
+    ids = []
+    list = auth_trove("#{type}-list", '--format=csv',
+                        '--column=id', '--quote=none')
+    (list.split("\n")[1..-1] || []).compact.collect do |line|
+      ids << line.strip
+    end
+    return ids
+  end
+
+  def self.get_trove_resource_attrs(type, id)
+    attrs = {}
+    net = auth_trove("#{type}-show", '--format=shell', id)
+    last_key = nil
+    (net.split("\n") || []).compact.collect do |line|
+      if line.include? '='
+        k, v = line.split('=', 2)
+        attrs[k] = v.gsub(/\A"|"\Z/, '')
+        last_key = k
+      else
+        # Handle the case of a list of values
+        v = line.gsub(/\A"|"\Z/, '')
+        attrs[last_key] = [attrs[last_key], v]
+      end
+    end
+    return attrs
+  end
+
+  def self.get_tenant_id(catalog, name)
+    instance_type = 'keystone_tenant'
+    instance = catalog.resource("#{instance_type.capitalize!}[#{name}]")
+    if ! instance
+      instance = Puppet::Type.type(instance_type).instances.find do |i|
+        i.provider.name == name
+      end
+    end
+    if instance
+      return instance.provider.id
+    else
+      fail("Unable to find #{instance_type} for name #{name}")
+    end
+  end
+
+  def self.parse_creation_output(data)
+    hash = {}
+    data.split("\n").compact.each do |line|
+      if line.include? '='
+        hash[line.split('=').first] = line.split('=', 2)[1].gsub(/\A"|"\Z/, '')
+      end
+    end
+    hash
+  end
+
+end

lib/puppet/provider/trove_config/ini_setting.rb

@@ -0,0 +1,27 @@
+Puppet::Type.type(:trove_config).provide(
+  :ini_setting,
+  :parent => Puppet::Type.type(:ini_setting).provider(:ruby)
+) do
+
+  def section
+    resource[:name].split('/', 2).first
+  end
+
+  def setting
+    resource[:name].split('/', 2).last
+  end
+
+  def separator
+    '='
+  end
+
+  def self.file_path
+    '/etc/trove/trove.conf'
+  end
+
+  # added for backwards compatibility with older versions of inifile
+  def file_path
+    self.class.file_path
+  end
+
+end

lib/puppet/type/trove_config.rb

@@ -0,0 +1,42 @@
+Puppet::Type.newtype(:trove_config) do
+
+  ensurable
+
+  newparam(:name, :namevar => true) do
+    desc 'Section/setting name to manage from /etc/trove/trove.conf'
+    newvalues(/\S+\/\S+/)
+  end
+
+  newproperty(:value) do
+    desc 'The value of the setting to be defined.'
+    munge do |value|
+      value = value.to_s.strip
+      value.capitalize! if value =~ /^(true|false)$/i
+      value
+    end
+
+    def is_to_s( currentvalue )
+      if resource.secret?
+        return '[old secret redacted]'
+      else
+        return currentvalue
+      end
+    end
+
+    def should_to_s( newvalue )
+      if resource.secret?
+        return '[new secret redacted]'
+      else
+        return newvalue
+      end
+    end
+  end
+
+  newparam(:secret, :boolean => true) do
+    desc 'Whether to hide the value from Puppet logs. Defaults to `false`.'
+
+    newvalues(:true, :false)
+
+    defaultto false
+  end
+end