hammer_cli_foreman 0.1.4 → 0.2.0

data/lib/hammer_cli_foreman/interface.rb

@@ -0,0 +1,165 @@
+module HammerCLIForeman
+
+  class Interface < HammerCLIForeman::Command
+
+    resource :interfaces
+    desc _("View and manage host's network interfaces")
+
+    def self.format_type(nic)
+      flags = []
+      flags << _('primary') if nic['primary']
+      flags << _('provision') if nic['provision']
+
+      type = nic['type']
+      if !flags.empty?
+        type += " (#{flags.join(', ')})"
+      end
+      type
+    end
+
+    class ListCommand < HammerCLIForeman::ListCommand
+
+      output do
+        field :id, _("Id")
+        field :identifier, _("Identifier")
+        field :_type , _("Type")
+        field :mac , _("MAC address")
+        field :ip , _("IP address")
+        field :name , _("DNS name")
+      end
+
+      def extend_data(nic)
+        nic['_type'] = HammerCLIForeman::Interface.format_type(nic)
+        nic
+      end
+
+      build_options
+    end
+
+
+    class InfoCommand < HammerCLIForeman::InfoCommand
+
+      output do
+        field :id, _("Id")
+        field :identifier, _("Identifier")
+        field :type , _("Type")
+        field :mac , _("MAC address")
+        field :ip , _("IP address")
+        field :name , _("DNS name")
+        field nil, _("Subnet"), Fields::SingleReference, :key => :subnet
+        field nil, _("Domain"), Fields::SingleReference, :key => :domain
+        field :managed, _("Managed"), Fields::Boolean
+        field :primary, _("Primary"), Fields::Boolean
+        field :provision, _("Provision"), Fields::Boolean
+        field :virtual, _("Virtual"), Fields::Boolean
+        field :tag, _("Tag"), nil, :hide_blank => true
+        field :attached_to, _("Attached to"), nil, :hide_blank => true
+
+        label _("BMC"), :hide_blank => true do
+          field :username, _("Username"), nil, :hide_blank => true
+          field :provider, _("Provider"), nil, :hide_blank => true
+        end
+
+        label _("Bond"), :hide_blank => true do
+          field :mode, _("Mode"), nil, :hide_blank => true
+          field :attached_devices, _("Attached devices"), nil, :hide_blank => true
+          field :bond_options, _("Bond options"), nil, :hide_blank => true
+        end
+      end
+
+      build_options
+    end
+
+
+    module InterfaceUpdate
+
+      def self.included(base)
+        base.option "--primary", :flag, _("Should this interface be used for constructing the FQDN of the host? Each managed hosts needs to have one primary interface.")
+        base.option "--provision", :flag, _("Should this interface be used for TFTP of PXELinux (or SSH for image-based hosts)? Each managed hosts needs to have one provision interface.")
+      end
+
+      def get_interfaces(host_id)
+        HammerCLIForeman.foreman_resource!(:interfaces).call(:index, {'host_id' => host_id}, request_headers, request_options)
+      end
+
+      def mandatory_interfaces(host_id, nic_id)
+        mandatory_options = []
+        get_interfaces(host_id)['results'].each do |nic|
+          if (nic['primary'] || nic['provision']) && nic['id'] != nic_id
+            mandatory_options << {
+              'id' => nic['id'],
+              'primary' => nic['primary'],
+              'provision' => nic['provision']
+            }
+          end
+        end
+        mandatory_options
+      end
+
+      def reset_flag(interfaces_params, flag)
+        interfaces_params.each do |nic|
+          nic[flag] = false if nic[flag]
+        end
+      end
+
+      def send_request
+        nic_params = request_params
+        interface = nic_params['interface']
+
+        interface['id'] = nic_params['id'].to_i if nic_params['id']
+        interface['compute_attributes'] = option_compute_attributes
+
+        host_params = {}
+        host_params['id'] = nic_params['host_id']
+        host_params['host'] = {}
+        host_params['host']['interfaces_attributes'] = mandatory_interfaces(nic_params['host_id'], interface['id'])
+
+        reset_flag(host_params['host']['interfaces_attributes'], 'primary') if option_primary?
+        reset_flag(host_params['host']['interfaces_attributes'], 'provision') if option_provision?
+
+        host_params['host']['interfaces_attributes'] += [interface]
+
+        HammerCLIForeman.foreman_resource!(:hosts).call(:update, host_params, request_headers, request_options)
+      end
+    end
+
+
+    class CreateCommand < HammerCLIForeman::CreateCommand
+      success_message _("Interface created")
+      failure_message _("Could not create the interface")
+
+      option "--compute-attributes", "COMPUTE_ATTRS", _("Compute resource specific attributes."),
+        :format => HammerCLI::Options::Normalizers::KeyValueList.new
+
+      include InterfaceUpdate
+
+      build_options :without => [:primary, :provision]
+    end
+
+
+    class UpdateCommand < HammerCLIForeman::UpdateCommand
+      success_message _("Interface updated")
+      failure_message _("Could not update the interface")
+
+      option "--compute-attributes", "COMPUTE_ATTRS", _("Compute resource specific attributes."),
+        :format => HammerCLI::Options::Normalizers::KeyValueList.new
+
+      include InterfaceUpdate
+
+      build_options :without => [:primary, :provision]
+    end
+
+
+    class DeleteCommand < HammerCLIForeman::DeleteCommand
+      success_message _("Interface deleted")
+      failure_message _("Could not delete the interface")
+
+      build_options
+    end
+
+    autoload_subcommands
+  end
+
+end
+
+

data/lib/hammer_cli_foreman/output/fields.rb

@@ -3,6 +3,13 @@ require 'hammer_cli'
 module Fields
 
   class SingleReference < Field
+
+    def display?(value)
+      id_key = "#{parameters[:key]}_id"
+      name_key = "#{parameters[:key]}_name"
+
+      (value[name_key.to_sym] || value[name_key]) && (value[id_key.to_sym] || value[id_key])
+    end
   end
 
   class Reference < Field

data/lib/hammer_cli_foreman/output/formatters.rb

@@ -20,11 +20,9 @@ module HammerCLIForeman::Output
 
         context = field_params[:context] || {}
 
-        if context[:show_ids]
-          "#{name} (id: #{id})" if id && name
-        else
-          "#{name}" if name
-        end
+        out = "#{name}"
+        out += " (id: #{id})" if context[:show_ids] && id
+        out
       end
 
     end

data/lib/hammer_cli_foreman/puppet_class.rb

@@ -1,4 +1,5 @@
 require 'hammer_cli_foreman/smart_class_parameter'
+require 'hammer_cli_foreman/smart_variable'
 
 module HammerCLIForeman
 
@@ -47,8 +48,21 @@ module HammerCLIForeman
 
 
     class SCParamsCommand < HammerCLIForeman::SmartClassParametersBriefList
-      parent_resource :puppetclasses
-      build_options
+      build_options_for :puppetclasses
+
+      def validate_options
+        super
+        validator.any(:option_puppetclass_name, :option_puppetclass_id).required
+      end
+    end
+
+    class SmartVariablesCommand < HammerCLIForeman::SmartVariablesBriefList
+      build_options_for :puppetclasses
+
+      def validate_options
+        super
+        validator.any(:option_puppetclass_name, :option_puppetclass_id).required
+      end
     end
 
 

data/lib/hammer_cli_foreman/settings.rb

@@ -0,0 +1,37 @@
+
+module HammerCLIForeman
+
+  class Settings < HammerCLIForeman::Command
+
+    resource :settings
+
+    class ListCommand < HammerCLIForeman::ListCommand
+
+      output do
+        field :id, _('Id'), Fields::Id
+        field :name, _('Name')
+        field :value, _('Value')
+        field :description, _('Description')
+      end
+
+      def extend_data(data)
+        data['value'] = data['value'].to_s
+        data
+      end
+
+      build_options
+    end
+
+    class UpdateCommand < HammerCLIForeman::UpdateCommand
+      command_name 'set'
+
+      success_message _("Setting [%{name}] updated to [%{value}]")
+      failure_message _("Could not update the setting")
+
+      build_options
+    end
+
+    autoload_subcommands
+  end
+
+end

data/lib/hammer_cli_foreman/smart_class_parameter.rb

@@ -1,6 +1,6 @@
 module HammerCLIForeman
 
-  class SmartClassParametersBriefList < HammerCLIForeman::AssociatedResourceListCommand
+  class SmartClassParametersBriefList < HammerCLIForeman::ListCommand
     resource :smart_class_parameters, :index
     command_name 'sc-params'
 
@@ -20,9 +20,12 @@ module HammerCLIForeman
       HammerCLI::Output::RecordCollection.new(res.uniq, :meta => res.meta)
     end
 
-    def self.build_options(options={})
-      options[:without] ||= [:host_id, :hostgroup_id, :puppetclass_id, :environment_id]
-      super(options)
+    def self.build_options_for(resource)
+      options = {}
+      options[:without] = [:host_id, :puppetclass_id, :environment_id, :hostgroup_id]
+      options[:expand] = {}
+      options[:expand][:except] = ([:hosts, :puppetclasses, :environments, :hostgroups] - [resource])
+      build_options(options)
     end
   end
 
@@ -44,6 +47,11 @@ module HammerCLIForeman
 
       output SmartClassParametersList.output_definition
 
+      def extend_data(res)
+        res['parameter_type'] ||= 'string'
+        res
+      end
+
       build_options
     end
 
@@ -59,14 +67,13 @@ module HammerCLIForeman
           field :validator_rule, _("Rule")
         end
         label _("Override values") do
+          field :merge_overrides, _("Merge overrides"), Fields::Boolean
+          field :avoid_duplicates, _("Avoid duplicates"), Fields::Boolean
           field :override_value_order, _("Order"), Fields::List
-          field :override_values_count, _("Count")
-          collection :override_values, "Values" do
-            label  _("Value") do
+          collection :override_values, _("Values") do
               field :id, _('Id')
               field :match, _('Match')
               field :value, _('Value')
-            end
           end
         end
         HammerCLIForeman::References.environments(self)
@@ -74,11 +81,20 @@ module HammerCLIForeman
       end
 
       def extend_data(res)
-        res['override_value_order'] = res['override_value_order'].split("\n")
+        res['override_value_order'] = res['override_value_order'].split(/\r?\n/)
+        res['parameter_type'] ||= 'string'
         res
       end
 
-      build_options
+      build_options do |options|
+        options.expand.including(:puppetclasses)
+      end
+
+      validate_options do
+        if option(:option_name).exist?
+          any(:option_puppetclass_name, :option_puppetclass_id).required
+        end
+      end
     end
 
     class UpdateCommand < HammerCLIForeman::UpdateCommand
@@ -86,7 +102,10 @@ module HammerCLIForeman
       success_message _("Parameter updated")
       failure_message _("Could not update the parameter")
 
-      build_options :without => [:parameter_type, :validator_type, :override, :required]
+      build_options do |options|
+        options.expand.including(:puppetclasses)
+        options.without(:parameter_type, :validator_type, :override, :required)
+      end
 
       option "--override", "OVERRIDE", _("Override this parameter."),
         :format => HammerCLI::Options::Normalizers::Bool.new
@@ -97,8 +116,53 @@ module HammerCLIForeman
             ['string', 'boolean', 'integer', 'real', 'array', 'hash', 'yaml', 'json'])
       option "--validator-type", "VALIDATOR_TYPE", _("Type of the validator."),
         :format => HammerCLI::Options::Normalizers::Enum.new(['regexp', 'list', ''])
+
+      validate_options do
+        if option(:option_name).exist?
+          any(:option_puppetclass_name, :option_puppetclass_id).required
+        end
+      end
+    end
+
+    class AddOverrideValueCommand < HammerCLIForeman::CreateCommand
+      resource :override_values
+      command_name 'add-override-value'
+
+      success_message _("Override value created")
+      failure_message _("Could not create the override_value")
+
+      build_options do |options|
+        options.without(:smart_variable_id)
+        options.expand.except(:smart_variables)
+        options.expand.including(:puppetclasses)
+      end
+
+      validate_options do
+        if option(:option_smart_class_parameter_name).exist?
+          any(:option_puppetclass_name, :option_puppetclass_id).required
+        end
+      end
     end
 
+    class RemoveOverrideValueCommand < HammerCLIForeman::DeleteCommand
+      resource :override_values
+      command_name 'remove-override-value'
+
+      success_message _("Override value deleted")
+      failure_message _("Could not delete the override value")
+
+      build_options do |options|
+        options.without(:smart_variable_id)
+        options.expand.except(:smart_variables)
+        options.expand.including(:puppetclasses)
+      end
+
+      validate_options do
+        if option(:option_smart_class_parameter_name).exist?
+          any(:option_puppetclass_name, :option_puppetclass_id).required
+        end
+      end
+    end
 
     autoload_subcommands
 

data/lib/hammer_cli_foreman/smart_variable.rb

@@ -0,0 +1,146 @@
+module HammerCLIForeman
+
+  class SmartVariablesBriefList < HammerCLIForeman::ListCommand
+    resource :smart_variables, :index
+    command_name 'smart-variables'
+
+    output do
+      field :id, _("Id")
+
+      field :variable, _("Name")
+      field :default_value, _("Default Value")
+      field :parameter_type, _("Type")
+    end
+
+    def self.build_options_for(resource)
+      options = {}
+      options[:without] = [:host_id, :puppetclass_id, :hostgroup_id]
+      options[:expand] = {}
+      options[:expand][:except] = ([:hosts, :puppetclasses, :hostgroups] - [resource])
+      build_options(options)
+    end
+  end
+
+  class SmartVariablesList < SmartVariablesBriefList
+
+    output do
+      field :puppetclass_name, _("Puppet class")
+      field :puppetclass_id, _("Class Id"), Fields::Id
+    end
+  end
+
+  class SmartVariable < HammerCLIForeman::Command
+
+    resource :smart_variables
+
+    class ListCommand < HammerCLIForeman::ListCommand
+
+      output SmartVariablesList.output_definition
+
+      def extend_data(res)
+        res['parameter_type'] ||= 'string'
+        res
+      end
+
+      build_options
+    end
+
+    class InfoCommand < HammerCLIForeman::InfoCommand
+
+      output ListCommand.output_definition do
+        field :description, _("Description")
+
+        label _("Validator") do
+          field :validator_type, _("Type")
+          field :validator_rule, _("Rule")
+        end
+        label _("Override values") do
+          field :merge_overrides, _("Merge overrides"), Fields::Boolean
+          field :avoid_duplicates, _("Avoid duplicates"), Fields::Boolean
+          field :override_value_order, _("Order"), Fields::List
+          collection :override_values, _("Values") do
+              field :id, _('Id')
+              field :match, _('Match')
+              field :value, _('Value')
+          end
+        end
+        HammerCLIForeman::References.timestamps(self)
+      end
+
+      def extend_data(res)
+        res['override_value_order'] = res['override_value_order'].split(/\r?\n/)
+        res['parameter_type'] ||= 'string'
+        res
+      end
+
+      build_options
+    end
+
+    class CreateCommand < HammerCLIForeman::CreateCommand
+
+      success_message _("Smart variable [%{variable}] created")
+      failure_message _("Could not create the smart variable")
+
+      build_options :without => [:variable_type, :validator_type]
+
+      option "--variable-type", "VARIABLE_TYPE", _("Type of the variable."),
+        :format => HammerCLI::Options::Normalizers::Enum.new(
+            ['string', 'boolean', 'integer', 'real', 'array', 'hash', 'yaml', 'json'])
+      option "--validator-type", "VALIDATOR_TYPE", _("Type of the validator."),
+        :format => HammerCLI::Options::Normalizers::Enum.new(['regexp', 'list', ''])
+    end
+
+    class UpdateCommand < HammerCLIForeman::UpdateCommand
+
+      success_message _("Smart variable [%{variable}] updated")
+      failure_message _("Could not update the smart variable")
+
+      build_options :without => [:variable_type, :validator_type]
+
+      option "--variable-type", "VARIABLE_TYPE", _("Type of the variable."),
+        :format => HammerCLI::Options::Normalizers::Enum.new(
+            ['string', 'boolean', 'integer', 'real', 'array', 'hash', 'yaml', 'json'])
+      option "--validator-type", "VALIDATOR_TYPE", _("Type of the validator."),
+        :format => HammerCLI::Options::Normalizers::Enum.new(['regexp', 'list', ''])
+    end
+
+
+    class DeleteCommand < HammerCLIForeman::DeleteCommand
+
+      success_message _("Smart variable [%{variable}] deleted")
+      failure_message _("Could not delete the smart variable")
+
+      build_options
+    end
+
+
+    class AddOverrideValueCommand < HammerCLIForeman::CreateCommand
+      resource :override_values
+      command_name 'add-override-value'
+
+      success_message _("Override value created")
+      failure_message _("Could not create the override_value")
+
+      build_options do |options|
+        options.without(:smart_class_parameter_id)
+        options.expand.except(:smart_class_parameters)
+      end
+    end
+
+    class RemoveOverrideValueCommand < HammerCLIForeman::DeleteCommand
+      resource :override_values
+      command_name 'remove-override-value'
+
+      success_message _("Override value deleted")
+      failure_message _("Could not delete the override value")
+
+      build_options do |options|
+        options.without(:smart_class_parameter_id)
+        options.expand.except(:smart_class_parameters)
+      end
+    end
+
+    autoload_subcommands
+
+  end
+end