hammer_cli_foreman_puppet 0.0.1

data/lib/hammer_cli_foreman_puppet/config_group.rb

@@ -0,0 +1,45 @@
+module HammerCLIForemanPuppet
+  class ConfigGroup < HammerCLIForemanPuppet::Command
+    resource :config_groups
+
+    class ListCommand < HammerCLIForemanPuppet::ListCommand
+      output do
+        field :id, _('ID')
+        field :name, _('Name')
+      end
+
+      build_options
+    end
+
+    class InfoCommand < HammerCLIForemanPuppet::InfoCommand
+      output ListCommand.output_definition do
+        HammerCLIForemanPuppet::PuppetReferences.puppetclasses(self)
+      end
+
+      build_options
+    end
+
+    class CreateCommand < HammerCLIForemanPuppet::CreateCommand
+      success_message _('Config group created.')
+      failure_message _('Could not create the config group')
+
+      build_options
+    end
+
+    class UpdateCommand < HammerCLIForemanPuppet::UpdateCommand
+      success_message _('Config group updated.')
+      failure_message _('Could not update the config group')
+
+      build_options
+    end
+
+    class DeleteCommand < HammerCLIForemanPuppet::DeleteCommand
+      success_message _('Config group has been deleted.')
+      failure_message _('Could not delete the config group')
+
+      build_options
+    end
+
+    autoload_subcommands
+  end
+end

data/lib/hammer_cli_foreman_puppet/discovery.rb

@@ -0,0 +1,11 @@
+require 'hammer_cli_foreman_discovery/discovery'
+require 'hammer_cli_foreman_puppet/command_extensions/discovery'
+
+module HammerCLIForemanPuppet
+  HammerCLIForemanDiscovery::DiscoveredHost::ProvisionCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::Provision.new
+  )
+  HammerCLIForemanDiscovery::DiscoveredHost::ProvisionCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new
+  )
+end

data/lib/hammer_cli_foreman_puppet/environment.rb

@@ -0,0 +1,59 @@
+module HammerCLIForemanPuppet
+  class PuppetEnvironment < HammerCLIForemanPuppet::Command
+    resource :environments
+
+    class ListCommand < HammerCLIForemanPuppet::ListCommand
+      include EnvironmentNameMapping
+      output do
+        field :id, _('Id')
+        field :name, _('Name')
+      end
+
+      build_options
+    end
+
+    class InfoCommand < HammerCLIForemanPuppet::InfoCommand
+      include EnvironmentNameMapping
+      output ListCommand.output_definition do
+        HammerCLIForemanPuppet::PuppetReferences.puppetclasses(self)
+        HammerCLIForeman::References.taxonomies(self)
+        HammerCLIForeman::References.timestamps(self)
+      end
+
+      build_options
+    end
+
+    class CreateCommand < HammerCLIForemanPuppet::CreateCommand
+      include EnvironmentNameMapping
+      success_message _("Environment created.")
+      failure_message _("Could not create the environment")
+
+      build_options
+    end
+
+    class UpdateCommand < HammerCLIForemanPuppet::UpdateCommand
+      include EnvironmentNameMapping
+      success_message _("Environment updated.")
+      failure_message _("Could not update the environment")
+
+      build_options
+    end
+
+    class DeleteCommand < HammerCLIForemanPuppet::DeleteCommand
+      include EnvironmentNameMapping
+      success_message _("Environment deleted.")
+      failure_message _("Could not delete the environment")
+
+      build_options
+    end
+
+    class SCParamsCommand < HammerCLIForemanPuppet::SmartClassParametersList
+      include EnvironmentNameMapping
+      build_options_for :environments
+
+      extend_with(HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new)
+    end
+
+    autoload_subcommands
+  end
+end

data/lib/hammer_cli_foreman_puppet/environment_name_mapping.rb

@@ -0,0 +1,20 @@
+module HammerCLIForemanPuppet
+  module EnvironmentNameMapping
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def resource_name_mapping
+        mapping = Command.resource_name_mapping
+        mapping[:environment] = :puppet_environment
+        mapping[:environments] = :puppet_environments
+        mapping
+      end
+
+      def resource_alias_name_mapping
+        super.merge(HammerCLIForemanPuppet::RESOURCE_ALIAS_NAME_MAPPING.dup)
+      end
+    end
+  end
+end

data/lib/hammer_cli_foreman_puppet/host.rb

@@ -0,0 +1,70 @@
+require 'hammer_cli_foreman/host'
+require 'hammer_cli_foreman_puppet/class'
+require 'hammer_cli_foreman_puppet/host'
+require 'hammer_cli_foreman_puppet/command_extensions/host'
+
+module HammerCLIForemanPuppet
+  class Host < HammerCLIForemanPuppet::Command
+
+    class PuppetClassesCommand < HammerCLIForemanPuppet::ListCommand
+      include EnvironmentNameMapping
+      command_name "puppet-classes"
+      resource :puppetclasses
+
+      output HammerCLIForemanPuppet::PuppetClass::ListCommand.output_definition
+
+      def send_request
+        HammerCLIForemanPuppet::PuppetClass::ListCommand.unhash_classes(super)
+      end
+
+      build_options do |o|
+        o.without(:hostgroup_id, :environment_id)
+        o.expand.only(:hosts)
+      end
+    end
+
+    class SCParamsCommand < HammerCLIForemanPuppet::SmartClassParametersList
+      build_options_for :hosts
+      command_name "sc-params"
+
+      def validate_options
+        super
+        validator.any(:option_host_name, :option_host_id).required
+      end
+    end
+
+    class InfoCommand < HammerCLIForemanPuppet::InfoCommand
+      include EnvironmentNameMapping
+      output do
+        field nil, _("Puppet Environment"), Fields::SingleReference, :key => :environment
+        field nil, _("Puppet CA Proxy"), Fields::SingleReference, :key => :puppet_ca_proxy
+        field nil, _("Puppet Master Proxy"), Fields::SingleReference, :key => :puppet_proxy
+      end
+    end
+  end
+
+  HammerCLIForeman::Host.subcommand 'puppet-classes',
+                                     HammerCLIForemanPuppet::Host::PuppetClassesCommand.desc,
+                                     HammerCLIForemanPuppet::Host::PuppetClassesCommand
+
+  HammerCLIForeman::Host.subcommand 'sc-params',
+                                     HammerCLIForemanPuppet::Host::SCParamsCommand.desc,
+                                     HammerCLIForemanPuppet::Host::SCParamsCommand
+
+  HammerCLIForeman::Host::ListCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new
+  )
+  HammerCLIForeman::Host::CreateCommand.include(HammerCLIForemanPuppet::EnvironmentNameMapping)
+  HammerCLIForeman::Host::CreateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new,
+    HammerCLIForemanPuppet::CommandExtensions::HostPuppetProxy.new
+  )
+  HammerCLIForeman::Host::UpdateCommand.include(HammerCLIForemanPuppet::EnvironmentNameMapping)
+  HammerCLIForeman::Host::UpdateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new,
+    HammerCLIForemanPuppet::CommandExtensions::HostPuppetProxy.new
+  )
+  HammerCLIForeman::Host::InfoCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::Host.new
+  )
+end

data/lib/hammer_cli_foreman_puppet/hostgroup.rb

@@ -0,0 +1,76 @@
+require 'hammer_cli_foreman/hostgroup'
+require 'hammer_cli_foreman_puppet/references'
+require 'hammer_cli_foreman_puppet/command_extensions/hostgroup.rb'
+
+module HammerCLIForemanPuppet
+  class Hostgroup < HammerCLIForemanPuppet::Command
+
+    class PuppetClassesCommand < HammerCLIForemanPuppet::ListCommand
+      command_name "puppet-classes"
+      resource :puppetclasses
+
+      output HammerCLIForemanPuppet::PuppetClass::ListCommand.output_definition
+
+      def send_request
+        HammerCLIForemanPuppet::PuppetClass::ListCommand.unhash_classes(super)
+      end
+
+      build_options do |o|
+        o.without(:host_id, :environment_id)
+        o.expand.only(:hostgroups)
+      end
+    end
+
+    class PuppetSCParamsCommand < HammerCLIForemanPuppet::SmartClassParametersList
+      build_options_for :hostgroups
+      command_name "sc-params"
+
+      def validate_options
+        super
+        validator.any(:option_hostgroup_name, :option_hostgroup_id).required
+      end
+    end
+
+    class ListCommand < HammerCLIForemanPuppet::InfoCommand
+      output do
+        field nil, _("Puppet Environment"), Fields::SingleReference, :key => :environment
+      end
+    end
+
+    class InfoCommand < HammerCLIForemanPuppet::InfoCommand
+      output do
+        field nil, _("Puppet Environment"), Fields::SingleReference, :key => :environment
+        field nil, _("Puppet CA Proxy"), Fields::SingleReference, :key => :puppet_ca_proxy
+        field nil, _("Puppet Master Proxy"), Fields::SingleReference, :key => :puppet_proxy
+        HammerCLIForemanPuppet::References.puppetclasses(self)
+      end
+    end
+  end
+
+  HammerCLIForeman::Hostgroup.subcommand 'puppet-classes',
+                                          HammerCLIForemanPuppet::Hostgroup::PuppetClassesCommand.desc,
+                                          HammerCLIForemanPuppet::Hostgroup::PuppetClassesCommand
+
+  HammerCLIForeman::Hostgroup.subcommand 'sc-params',
+                                          HammerCLIForemanPuppet::Hostgroup::PuppetSCParamsCommand.desc,
+                                          HammerCLIForemanPuppet::Hostgroup::PuppetSCParamsCommand
+
+  HammerCLIForeman::Hostgroup::CreateCommand.include(HammerCLIForemanPuppet::EnvironmentNameMapping)
+  HammerCLIForeman::Hostgroup::CreateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new,
+    HammerCLIForemanPuppet::CommandExtensions::HostgroupPuppetProxy.new
+  )
+  HammerCLIForeman::Hostgroup::UpdateCommand.include(HammerCLIForemanPuppet::EnvironmentNameMapping)
+  HammerCLIForeman::Hostgroup::UpdateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironment.new,
+    HammerCLIForemanPuppet::CommandExtensions::HostgroupPuppetProxy.new
+  )
+  HammerCLIForeman::Hostgroup::InfoCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::HostgroupInfo.new
+  )
+  HammerCLIForeman::Hostgroup::ListCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::HostgroupList.new
+  )
+#TODO - adding puppet class options
+#TODO - resolver
+end

data/lib/hammer_cli_foreman_puppet/id_resolver.rb

@@ -0,0 +1,73 @@
+module HammerCLIForemanPuppet
+  module PuppetIdResolver
+    # puppet class search results are in non-standard format
+    # and needs to be un-hashed first
+    def puppetclass_id(options)
+      return options[HammerCLI.option_accessor_name('id')] if options[HammerCLI.option_accessor_name('id')]
+
+      resource = @api.resource(:puppetclasses)
+      results = find_resource_raw(:puppetclasses, options)
+      require('hammer_cli_foreman_puppet/class')
+      results = HammerCLIForemanPuppet::PuppetClass::ListCommand.unhash_classes(results)
+      pick_result(results, resource)['id']
+    end
+
+    def puppetclass_ids(options)
+      resource_name = :puppetclasses
+      resource = @api.resource(resource_name)
+      results = if (ids = options[HammerCLI.option_accessor_name("ids")])
+        ids
+      elsif (ids = nil_from_searchables(resource_name, options, :plural => true))
+        ids
+      elsif options_not_set?(resource, options)
+        raise HammerCLIForeman::MissingSearchOptions.new(_("Missing options to search %s") % resource.name, resource)
+      elsif options_empty?(resource, options)
+        []
+      else
+        require('hammer_cli_foreman_puppet/class')
+        results = HammerCLIForemanPuppet::PuppetClass::ListCommand.unhash_classes(
+          resolved_call(resource_name, :index, options, :multi)
+        )
+        raise HammerCLIForeman::ResolverError.new(_("one of %s not found.") % resource.name, resource) if results.count < expected_record_count(options, resource, :multi)
+          results.map { |r| r['id'] }
+      end
+    end
+
+    def puppet_environment_id(options)
+      get_id(:environments, options)
+    end
+
+    def puppet_environment_ids(options)
+      get_ids(:environments, options)
+    end
+
+    def create_smart_class_parameters_search_options(options, _mode = nil)
+      search_options = {}
+      value = options[HammerCLI.option_accessor_name('name')]
+      search_options[:search] = "key = \"#{value}\""
+      search_options[:puppetclass_id] = puppetclass_id(scoped_options('puppetclass', options))
+      search_options
+    end
+  end
+
+  class Searchables < HammerCLIForeman::Searchables
+    SEARCHABLES = {
+      :environment =>        [s_name(_('Puppet environment name'))],
+      :puppet_environment => [s_name(_('Puppet environment name'))],
+      :puppetclass =>      [ s_name(_("Puppet class name")) ],
+      :smart_class_parameter => [ s_name(_("Smart class parameter name"), :editable => false) ],
+    }.freeze
+
+    DEFAULT_SEARCHABLES = [s_name(_("Name to search by"))].freeze
+
+    def for(resource)
+      SEARCHABLES[resource.singular_name.to_sym] || super(resource)
+    end
+  end
+
+  class IdResolver < HammerCLIForeman::IdResolver
+  end
+  # Temp workaround for resolvers in case of multiple plugins
+  # extensions of the same command
+  HammerCLIForeman::IdResolver.include HammerCLIForemanPuppet::PuppetIdResolver
+end

data/lib/hammer_cli_foreman_puppet/location.rb

@@ -0,0 +1,35 @@
+require 'hammer_cli_foreman/location'
+require 'hammer_cli_foreman_puppet/references'
+require 'hammer_cli_foreman_puppet/command_extensions/location'
+
+module HammerCLIForemanPuppet
+  module LocationAddAssociatedCommand
+    def self.included(base)
+      HammerCLIForemanPuppet::AssociatingCommands::PuppetEnvironment.extend_command(base)
+      base.create_subcommand
+    end
+  end
+
+  class Location < HammerCLIForemanPuppet::Command
+    class InfoCommand < HammerCLIForemanPuppet::InfoCommand
+      output do
+        HammerCLIForemanPuppet::References.environments(self)
+      end
+    end
+  end
+
+  HammerCLIForeman::Location::CreateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironments.new
+  )
+  HammerCLIForeman::Location::UpdateCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::PuppetEnvironments.new
+  )
+  HammerCLIForeman::Location::InfoCommand.extend_with(
+    HammerCLIForemanPuppet::CommandExtensions::LocationInfo.new
+  )
+
+  HammerCLIForeman::Location.class_eval do
+    extend HammerCLIForemanPuppet::AssociatingCommands::ExtendCommands
+    include HammerCLIForemanPuppet::LocationAddAssociatedCommand
+  end
+end

data/lib/hammer_cli_foreman_puppet/option_sources.rb

@@ -0,0 +1 @@
+require 'hammer_cli_foreman_puppet/option_sources/puppet_environment_params'

data/lib/hammer_cli_foreman_puppet/option_sources/puppet_environment_params.rb

@@ -0,0 +1,60 @@
+
+module HammerCLIForemanPuppet
+  module OptionSources
+    class PuppetEnvironmentParams < HammerCLI::Options::Sources::Base
+      def initialize(command)
+        @command = command
+      end
+
+      def get_options(_defined_options, result)
+        if result['option_environment_id'].nil? &&
+           (@command.respond_to?(:option_environment_id) ||
+           @command.respond_to?(:option_environment_name))
+          put_puppet_environment_id(result)
+        end
+        if result['option_environment_ids'].nil? &&
+           (@command.respond_to?(:option_environment_ids) ||
+           @command.respond_to?(:option_environment_names))
+          put_puppet_environment_ids(result)
+        end
+        result
+      end
+
+      private
+
+      def put_puppet_environment_id(result)
+        if result['option_environment_name'].nil?
+          if @command.option_environment_id
+            result['option_environment_id'] = @command.option_environment_id
+          elsif @command.option_environment_name
+            result['option_environment_name'] = @command.option_environment_name
+            result['option_environment_id'] = @command.resolver.puppet_environment_id(
+              @command.resolver.scoped_options('environment', result, :single)
+            )
+          end
+        else
+          result['option_environment_id'] = @command.resolver.puppet_environment_id(
+            @command.resolver.scoped_options('environment', result, :single)
+          )
+        end
+      end
+
+      def put_puppet_environment_ids(result)
+        if result['option_environment_names'].nil?
+          if @command.option_environment_ids
+            result['option_environment_ids'] = @command.option_environment_ids
+          elsif @command.option_environment_names
+            result['option_environment_names'] = @command.option_environment_names
+            result['option_environment_ids'] = @command.resolver.puppet_environment_ids(
+              @command.resolver.scoped_options('environment', result, :multi)
+            )
+          end
+        else
+          result['option_environment_ids'] = @command.resolver.puppet_environment_ids(
+            @command.resolver.scoped_options('environment', result, :multi)
+          )
+        end
+      end
+    end
+  end
+end