cheffish 0.1

data/lib/chef/provider/chef_environment.rb

@@ -0,0 +1,78 @@
+require 'cheffish/chef_provider_base'
+require 'chef/resource/chef_environment'
+require 'chef/chef_fs/data_handler/environment_data_handler'
+
+class Chef::Provider::ChefEnvironment < Cheffish::ChefProviderBase
+
+  def whyrun_supported?
+    true
+  end
+
+  action :create do
+    differences = json_differences(current_json, new_json)
+
+    if current_resource_exists?
+      if differences.size > 0
+        description = [ "update environment #{new_resource.name} at #{rest.url}" ] + differences
+        converge_by description do
+          rest.put("environments/#{new_resource.name}", normalize_for_put(new_json))
+        end
+      end
+    else
+      description = [ "create environment #{new_resource.name} at #{rest.url}" ] + differences
+      converge_by description do
+        rest.post("environments", normalize_for_post(new_json))
+      end
+    end
+  end
+
+  action :delete do
+    if current_resource_exists?
+      converge_by "delete environment #{new_resource.name} at #{rest.url}" do
+        rest.delete("environments/#{new_resource.name}")
+      end
+    end
+  end
+
+  def load_current_resource
+    begin
+      @current_resource = json_to_resource(rest.get("environments/#{new_resource.name}"))
+    rescue Net::HTTPServerException => e
+      if e.response.code == "404"
+        @current_resource = not_found_resource
+      else
+        raise
+      end
+    end
+  end
+
+  def augment_new_json(json)
+    # Apply modifiers
+    json['default_attributes'] = apply_modifiers(new_resource.default_attribute_modifiers, json['default_attributes'])
+    json['override_attributes'] = apply_modifiers(new_resource.override_attribute_modifiers, json['override_attributes'])
+    json
+  end
+
+  #
+  # Helpers
+  #
+
+  def resource_class
+    Chef::Resource::ChefEnvironment
+  end
+
+  def data_handler
+    Chef::ChefFS::DataHandler::EnvironmentDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'description' => :description,
+      'cookbook_versions' => :cookbook_versions,
+      'default_attributes' => :default_attributes,
+      'override_attributes' => :override_attributes
+    }
+  end
+
+end

data/lib/chef/provider/chef_node.rb

@@ -0,0 +1,82 @@
+require 'cheffish/chef_provider_base'
+require 'chef/resource/chef_node'
+require 'chef/chef_fs/data_handler/node_data_handler'
+
+class Chef::Provider::ChefNode < Cheffish::ChefProviderBase
+
+  def whyrun_supported?
+    true
+  end
+
+  action :create do
+    differences = json_differences(current_json, new_json)
+
+    if current_resource_exists?
+      if differences.size > 0
+        description = [ "update node #{new_resource.name} at #{rest.url}" ] + differences
+        converge_by description do
+          rest.put("nodes/#{new_resource.name}", normalize_for_put(new_json))
+        end
+      end
+    else
+      description = [ "create node #{new_resource.name} at #{rest.url}" ] + differences
+      converge_by description do
+        rest.post("nodes", normalize_for_post(new_json))
+      end
+    end
+  end
+
+  action :delete do
+    if current_resource_exists?
+      converge_by "delete node #{new_resource.name} at #{rest.url}" do
+        rest.delete("nodes/#{new_resource.name}")
+      end
+    end
+  end
+
+  def load_current_resource
+    begin
+      @current_resource = json_to_resource(rest.get("nodes/#{new_resource.name}"))
+    rescue Net::HTTPServerException => e
+      if e.response.code == "404"
+        @current_resource = not_found_resource
+      else
+        raise
+      end
+    end
+  end
+
+  def augment_new_json(json)
+    # Apply modifiers
+    json['run_list'] = apply_run_list_modifiers(new_resource.run_list_modifiers, new_resource.run_list_removers, json['run_list'])
+    json['default'] = apply_modifiers(new_resource.default_modifiers, json['default'])
+    json['normal'] = apply_modifiers(new_resource.normal_modifiers, json['normal'])
+    json['override'] = apply_modifiers(new_resource.override_modifiers, json['override'])
+    json['automatic'] = apply_modifiers(new_resource.automatic_modifiers, json['automatic'])
+    json
+  end
+
+  #
+  # Helpers
+  #
+
+  def resource_class
+    Chef::Resource::ChefNode
+  end
+
+  def data_handler
+    Chef::ChefFS::DataHandler::NodeDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'chef_environment' => :chef_environment,
+      'run_list' => :run_list,
+      'default' => :default_attributes,
+      'normal' => :normal_attributes,
+      'override' => :override_attributes,
+      'automatic' => :automatic_attributes
+    }
+  end
+end

data/lib/chef/provider/chef_role.rb

@@ -0,0 +1,79 @@
+require 'cheffish/chef_provider_base'
+require 'chef/resource/chef_role'
+require 'chef/chef_fs/data_handler/role_data_handler'
+
+class Chef::Provider::ChefRole < Cheffish::ChefProviderBase
+
+  def whyrun_supported?
+    true
+  end
+
+  action :create do
+    differences = json_differences(current_json, new_json)
+
+    if current_resource_exists?
+      if differences.size > 0
+        description = [ "update role #{new_resource.name} at #{rest.url}" ] + differences
+        converge_by description do
+          rest.put("roles/#{new_resource.name}", normalize_for_put(new_json))
+        end
+      end
+    else
+      description = [ "create role #{new_resource.name} at #{rest.url}" ] + differences
+      converge_by description do
+        rest.post("roles", normalize_for_post(new_json))
+      end
+    end
+  end
+
+  action :delete do
+    if current_resource_exists?
+      converge_by "delete role #{new_resource.name} at #{rest.url}" do
+        rest.delete("roles/#{new_resource.name}")
+      end
+    end
+  end
+
+  def load_current_resource
+    begin
+      @current_resource = json_to_resource(rest.get("roles/#{new_resource.name}"))
+    rescue Net::HTTPServerException => e
+      if e.response.code == "404"
+        @current_resource = not_found_resource
+      else
+        raise
+      end
+    end
+  end
+
+  def augment_new_json(json)
+    # Apply modifiers
+    json['run_list'] = apply_run_list_modifiers(new_resource.run_list_modifiers, new_resource.run_list_removers, json['run_list'])
+    json['default_attributes'] = apply_modifiers(new_resource.default_attribute_modifiers, json['default_attributes'])
+    json['override_attributes'] = apply_modifiers(new_resource.override_attribute_modifiers, json['override_attributes'])
+    json
+  end
+
+  #
+  # Helpers
+  #
+
+  def resource_class
+    Chef::Resource::ChefRole
+  end
+
+  def data_handler
+    Chef::ChefFS::DataHandler::RoleDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'description' => :description,
+      'run_list' => :run_list,
+      'env_run_lists' => :env_run_lists,
+      'default_attributes' => :default_attributes,
+      'override_attributes' => :override_attributes
+    }
+  end
+end

data/lib/chef/provider/chef_user.rb

@@ -0,0 +1,48 @@
+require 'cheffish/actor_provider_base'
+require 'chef/resource/chef_user'
+require 'chef/chef_fs/data_handler/user_data_handler'
+
+class Chef::Provider::ChefUser < Cheffish::ActorProviderBase
+
+  def whyrun_supported?
+    true
+  end
+
+  action :create do
+    create_actor
+  end
+
+  action :delete do
+    delete_actor
+  end
+
+  #
+  # Helpers
+  #
+  # Gives us new_json, current_json, not_found_json, etc.
+
+  def actor_type
+    'user'
+  end
+
+  def resource_class
+    Chef::Resource::ChefUser
+  end
+
+  def data_handler
+    Chef::ChefFS::DataHandler::UserDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'admin' => :admin,
+      'email' => :email,
+      'password' => :password,
+      'external_authentication_uid' => :external_authentication_uid,
+      'recovery_authentication_enabled' => :recovery_authentication_enabled,
+      'public_key' => :source_key
+    }
+  end
+
+end

data/lib/chef/provider/private_key.rb

@@ -0,0 +1,160 @@
+require 'chef/provider/lwrp_base'
+require 'openssl'
+require 'cheffish/key_formatter'
+
+class Chef::Provider::PrivateKey < Chef::Provider::LWRPBase
+
+  action :create do
+    create_key(false)
+  end
+
+  action :regenerate do
+    create_key(true)
+  end
+
+  action :delete do
+    if Array(current_resource.action) == [ :create ]
+      converge_by "delete private key #{new_resource.path}" do
+        ::File.unlink(new_resource.path)
+      end
+    end
+  end
+
+  use_inline_resources
+
+  def whyrun_supported?
+    true
+  end
+
+  def create_key(regenerate)
+    final_private_key = nil
+    if new_source_key
+      #
+      # Create private key from source
+      #
+      desired_output = encode_private_key(new_source_key)
+      if Array(current_resource.action) == [ :delete ] || desired_output != IO.read(new_resource.path)
+        converge_by "reformat key at #{new_resource.source_key_path} to #{new_resource.format} private key #{new_resource.path} (#{new_resource.pass_phrase ? ", #{new_resource.cipher} password" : ""})" do
+          IO.write(new_resource.path, desired_output)
+        end
+      end
+
+      final_private_key = new_source_key
+
+    else
+      #
+      # Create private key file
+      #
+      if Array(current_resource.action) == [ :delete ] || regenerate ||
+        (new_resource.regenerate_if_different &&
+          (current_resource.size != new_resource.size ||
+           current_resource.type != new_resource.type))
+        action = (current_resource.path == :none || ::File.exists?(current_resource.path)) ? "overwrite" : "create"
+        converge_by "#{action} #{new_resource.type} private key #{new_resource.path} (#{new_resource.size} bits#{new_resource.pass_phrase ? ", #{new_resource.cipher} password" : ""})" do
+          case new_resource.type
+          when :rsa
+            if new_resource.exponent
+              final_private_key = OpenSSL::PKey::RSA.generate(new_resource.size, new_resource.exponent)
+            else
+              final_private_key = OpenSSL::PKey::RSA.generate(new_resource.size)
+            end
+          when :dsa
+            final_private_key = OpenSSL::PKey::DSA.generate(new_resource.size)
+          end
+
+          if new_resource.path != :none
+            write_private_key(final_private_key)
+          end
+        end
+      else
+        # Warn if existing key has different characteristics than expected
+        if current_resource.size != new_resource.size
+          Chef::Log.warn("Mismatched key size!  #{current_resource.path} is #{current_resource.size} bytes, desired is #{new_resource.size} bytes.  Use action :regenerate to force key regeneration.")
+        elsif current_resource.type != new_resource.type
+          Chef::Log.warn("Mismatched key type!  #{current_resource.path} is #{current_resource.type}, desired is #{new_resource.type} bytes.  Use action :regenerate to force key regeneration.")
+        end
+
+        final_private_key = current_private_key
+
+        if current_resource.format != new_resource.format
+          converge_by "change format of #{new_resource.type} private key #{new_resource.path} from #{current_resource.format} to #{new_resource.format}" do
+            write_private_key(current_private_key)
+          end
+        end
+      end
+    end
+
+    if new_resource.public_key_path
+      public_key_path = new_resource.public_key_path
+      public_key_format = new_resource.public_key_format
+      Cheffish.inline_resource(self) do
+        public_key public_key_path do
+          source_key final_private_key
+          format public_key_format
+        end
+      end
+    end
+
+    if new_resource.after
+      new_resource.after.call(new_resource, final_private_key)
+    end
+  end
+
+  def encode_private_key(key)
+    key_format = {}
+    key_format[:format] = new_resource.format if new_resource.format
+    key_format[:pass_phrase] = new_resource.pass_phrase if new_resource.pass_phrase
+    key_format[:cipher] = new_resource.cipher if new_resource.cipher
+    Cheffish::KeyFormatter.encode(key, key_format)
+  end
+
+  def write_private_key(key)
+    IO.write(new_resource.path, encode_private_key(key))
+    # TODO permissions on file?
+  end
+
+  def new_source_key
+    @new_source_key ||= begin
+      if new_resource.source_key.is_a?(String)
+        source_key, source_key_format = Cheffish::KeyFormatter.decode(new_resource.source_key, new_resource.source_key_pass_phrase)
+        source_key
+      elsif new_resource.source_key
+        new_resource.source_key
+      elsif new_resource.source_key_path
+        source_key, source_key_format = Cheffish::KeyFormatter.decode(IO.read(new_resource.source_key_path), new_resource.source_key_pass_phrase, new_resource.source_key_path)
+        source_key
+      else
+        nil
+      end
+    end
+  end
+
+  attr_reader :current_private_key
+
+  def load_current_resource
+    resource = Chef::Resource::PrivateKey.new(new_resource.name)
+
+    if new_resource.path != :none && ::File.exist?(new_resource.path)
+      resource.path new_resource.path
+
+      begin
+        key, key_format = Cheffish::KeyFormatter.decode(IO.read(new_resource.path), new_resource.pass_phrase, new_resource.path)
+        if key
+          @current_private_key = key
+          resource.format key_format[:format]
+          resource.type key_format[:type]
+          resource.size key_format[:size]
+          resource.exponent key_format[:exponent]
+          resource.pass_phrase key_format[:pass_phrase]
+          resource.cipher key_format[:cipher]
+        end
+      rescue
+        # If there's an error reading, we assume format and type are wrong and don't futz with them
+      end
+    else
+      resource.action :delete
+    end
+
+    @current_resource = resource
+  end
+end