clc-cheffish 0.8.clc

data/lib/chef/provider/chef_group.rb

@@ -0,0 +1,78 @@
+require 'cheffish/chef_provider_base'
+require 'chef/resource/chef_group'
+require 'chef/chef_fs/data_handler/group_data_handler'
+
+class Chef::Provider::ChefGroup < 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 group #{new_resource.name} at #{rest.url}" ] + differences
+        converge_by description do
+          rest.put("groups/#{new_resource.name}", normalize_for_put(new_json))
+        end
+      end
+    else
+      description = [ "create group #{new_resource.name} at #{rest.url}" ] + differences
+      converge_by description do
+        rest.post("groups", normalize_for_post(new_json))
+      end
+    end
+  end
+
+  action :delete do
+    if current_resource_exists?
+      converge_by "delete group #{new_resource.name} at #{rest.url}" do
+        rest.delete("groups/#{new_resource.name}")
+      end
+    end
+  end
+
+  def load_current_resource
+    begin
+      @current_resource = json_to_resource(rest.get("groups/#{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['users']   |= new_resource.users
+    json['clients'] |= new_resource.clients
+    json['groups']  |= new_resource.groups
+    json['users']   -= new_resource.remove_users
+    json['clients'] -= new_resource.remove_clients
+    json['groups']  -= new_resource.remove_groups
+    json
+  end
+
+  #
+  # Helpers
+  #
+
+  def resource_class
+    Chef::Resource::ChefGroup
+  end
+
+  def data_handler
+    Chef::ChefFS::DataHandler::GroupDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'groupname' => :name
+    }
+  end
+end

data/lib/chef/provider/chef_mirror.rb

@@ -0,0 +1,138 @@
+require 'chef/provider/lwrp_base'
+require 'chef/chef_fs/file_pattern'
+require 'chef/chef_fs/file_system'
+require 'chef/chef_fs/parallelizer'
+require 'chef/chef_fs/file_system/chef_server_root_dir'
+require 'chef/chef_fs/file_system/chef_repository_file_system_root_dir'
+
+class Chef::Provider::ChefMirror < Chef::Provider::LWRPBase
+
+  def whyrun_supported?
+    true
+  end
+
+  action :upload do
+    copy_to(local_fs, remote_fs)
+  end
+
+  action :download do
+    copy_to(remote_fs, local_fs)
+  end
+
+  def copy_to(src_root, dest_root)
+    if new_resource.concurrency && new_resource.concurrency <= 0
+      raise "chef_mirror.concurrency must be above 0!  Was set to #{new_resource.concurrency}"
+    end
+    # Honor concurrency
+    Chef::ChefFS::Parallelizer.threads = (new_resource.concurrency || 10) - 1
+
+    # We don't let the user pass absolute paths; we want to reserve those for
+    # multi-org support (/organizations/foo).
+    if new_resource.path[0] == '/'
+      raise "Absolute paths in chef_mirror not yet supported."
+    end
+    # Copy!
+    path = Chef::ChefFS::FilePattern.new("/#{new_resource.path}")
+    ui = CopyListener.new(self)
+    error = Chef::ChefFS::FileSystem.copy_to(path, src_root, dest_root, nil, options, ui, proc { |p| p.path })
+
+    if error
+      raise "Errors while copying:#{ui.errors.map { |e| "#{e}\n" }.join('')}"
+    end
+  end
+
+  def local_fs
+    # If chef_repo_path is set to a string, put it in the form it usually is in
+    # chef config (:chef_repo_path, :node_path, etc.)
+    path_config = new_resource.chef_repo_path
+    if path_config.is_a?(Hash)
+      chef_repo_path = path_config.delete(:chef_repo_path)
+    elsif path_config
+      chef_repo_path = path_config
+      path_config = {}
+    else
+      chef_repo_path = Chef::Config.chef_repo_path
+      path_config = Chef::Config
+    end
+    chef_repo_path = Array(chef_repo_path).flatten
+
+    # Go through the expected object paths and figure out the local paths for each.
+    case repo_mode
+    when 'hosted_everything'
+      object_names = %w(acls clients cookbooks containers data_bags environments groups nodes roles)
+    else
+      object_names = %w(clients cookbooks data_bags environments nodes roles users)
+    end
+
+    object_paths = {}
+    object_names.each do |object_name|
+      variable_name = "#{object_name[0..-2]}_path" # cookbooks -> cookbook_path
+      if path_config[variable_name.to_sym]
+        paths = Array(path_config[variable_name.to_sym]).flatten
+      else
+        paths = chef_repo_path.map { |path| ::File.join(path, object_name) }
+      end
+      object_paths[object_name] = paths.map { |path| ::File.expand_path(path) }
+    end
+
+    # Set up the root dir
+    Chef::ChefFS::FileSystem::ChefRepositoryFileSystemRootDir.new(object_paths)
+  end
+
+  def remote_fs
+    config = {
+      :chef_server_url => new_resource.chef_server[:chef_server_url],
+      :node_name => new_resource.chef_server[:options][:client_name],
+      :client_key => new_resource.chef_server[:options][:client_key],
+      :repo_mode => repo_mode
+    }
+    Chef::ChefFS::FileSystem::ChefServerRootDir.new("remote", config)
+  end
+
+  def repo_mode
+    new_resource.chef_server[:chef_server_url] =~ /\/organizations\// ? 'hosted_everything' : 'everything'
+  end
+
+  def options
+    result = {
+      :purge => new_resource.purge,
+      :freeze => new_resource.freeze,
+      :diff => new_resource.no_diff,
+      :dry_run => whyrun_mode?
+    }
+    result[:diff] = !result[:diff]
+    result[:repo_mode] = repo_mode
+    result[:concurrency] = new_resource.concurrency if new_resource.concurrency
+    result
+  end
+
+  def load_current_resource
+  end
+
+  class CopyListener
+    def initialize(mirror)
+      @mirror = mirror
+      @errors = []
+    end
+
+    attr_reader :mirror
+    attr_reader :errors
+
+    # TODO output is not *always* indicative of a change.  We may want to give
+    # ChefFS the ability to tell us that info.  For now though, assuming any output
+    # means change is pretty damn close to the truth.
+    def output(str)
+      mirror.converge_by str do
+      end
+    end
+    def warn(str)
+      mirror.converge_by "WARNING: #{str}" do
+      end
+    end
+    def error(str)
+      mirror.converge_by "ERROR: #{str}" do
+      end
+      @errors << str
+    end
+  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)
+    # Preserve tags even if "attributes" was overwritten directly
+    json['normal']['tags'] = current_json['normal']['tags'] unless json['normal']['tags']
+    # Apply modifiers
+    json['run_list'] = apply_run_list_modifiers(new_resource.run_list_modifiers, new_resource.run_list_removers, json['run_list'])
+    json['normal'] = apply_modifiers(new_resource.attribute_modifiers, json['normal'])
+    # Preserve default/override/automatic even when "complete true"
+    json['default'] = current_json['default']
+    json['override'] = current_json['override']
+    json['automatic'] = current_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,
+      'normal' => :attributes
+    }
+  end
+end

data/lib/chef/provider/chef_organization.rb

@@ -0,0 +1,150 @@
+require 'cheffish/chef_provider_base'
+require 'chef/resource/chef_organization'
+require 'chef/chef_fs/data_handler/data_handler_base'
+
+class Chef::Provider::ChefOrganization < 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 organization #{new_resource.name} at #{rest.url}" ] + differences
+        converge_by description do
+          rest.put("#{rest.root_url}/organizations/#{new_resource.name}", normalize_for_put(new_json))
+        end
+      end
+    else
+      description = [ "create organization #{new_resource.name} at #{rest.url}" ] + differences
+      converge_by description do
+        rest.post("#{rest.root_url}/organizations", normalize_for_post(new_json))
+      end
+    end
+
+    # Revoke invites and memberships when asked
+    invites_to_remove.each do |user|
+      if outstanding_invites.has_key?(user)
+        converge_by "revoke #{user}'s invitation to organization #{new_resource.name}" do
+          rest.delete("#{rest.root_url}/organizations/#{new_resource.name}/association_requests/#{outstanding_invites[user]}")
+        end
+      end
+    end
+    members_to_remove.each do |user|
+      if existing_members.include?(user)
+        converge_by "remove #{user} from organization #{new_resource.name}" do
+          rest.delete("#{rest.root_url}/organizations/#{new_resource.name}/users/#{user}")
+        end
+      end
+    end
+
+    # Invite and add members when asked
+    new_resource.invites.each do |user|
+      if !existing_members.include?(user) && !outstanding_invites.has_key?(user)
+        converge_by "invite #{user} to organization #{new_resource.name}" do
+          rest.post("#{rest.root_url}/organizations/#{new_resource.name}/association_requests", { 'user' => user })
+        end
+      end
+    end
+    new_resource.members.each do |user|
+      if !existing_members.include?(user)
+        converge_by "Add #{user} to organization #{new_resource.name}" do
+          rest.post("#{rest.root_url}/organizations/#{new_resource.name}/users/#{user}", {})
+        end
+      end
+    end
+  end
+
+  def existing_members
+    @existing_members ||= rest.get("#{rest.root_url}/organizations/#{new_resource.name}/users").map { |u| u['user']['username'] }
+  end
+
+  def outstanding_invites
+    @outstanding_invites ||= begin
+      invites = {}
+      rest.get("#{rest.root_url}/organizations/#{new_resource.name}/association_requests").each do |r|
+        invites[r['username']] = r['id']
+      end
+      invites
+    end
+  end
+
+  def invites_to_remove
+    if new_resource.complete
+      if new_resource.invites_specified? || new_resource.members_specified?
+        outstanding_invites.keys - (new_resource.invites | new_resource.members)
+      else
+        []
+      end
+    else
+      new_resource.remove_members
+    end
+  end
+
+  def members_to_remove
+    if new_resource.complete
+      if new_resource.members_specified?
+        existing_members - (new_resource.invites | new_resource.members)
+      else
+        []
+      end
+    else
+      new_resource.remove_members
+    end
+  end
+
+  action :delete do
+    if current_resource_exists?
+      converge_by "delete organization #{new_resource.name} at #{rest.url}" do
+        rest.delete("#{rest.root_url}/organizations/#{new_resource.name}")
+      end
+    end
+  end
+
+  def load_current_resource
+    begin
+      @current_resource = json_to_resource(rest.get("#{rest.root_url}/organizations/#{new_resource.name}"))
+    rescue Net::HTTPServerException => e
+      if e.response.code == "404"
+        @current_resource = not_found_resource
+      else
+        raise
+      end
+    end
+  end
+
+  #
+  # Helpers
+  #
+
+  def resource_class
+    Chef::Resource::ChefOrganization
+  end
+
+  def data_handler
+    OrganizationDataHandler.new
+  end
+
+  def keys
+    {
+      'name' => :name,
+      'full_name' => :full_name
+    }
+  end
+
+  class OrganizationDataHandler < Chef::ChefFS::DataHandler::DataHandlerBase
+    def normalize(organization, entry)
+      # Normalize the order of the keys for easier reading
+      normalize_hash(organization, {
+        'name' => remove_dot_json(entry.name),
+        'full_name' => remove_dot_json(entry.name),
+        'org_type' => 'Business',
+        'clientname' => "#{remove_dot_json(entry.name)}-validator",
+        'billing_plan' => 'platform-free'
+      })
+    end
+  end
+end