hammer_cli_foreman 0.0.10 → 0.0.11

data/lib/hammer_cli_foreman/exception_handler.rb

@@ -1,13 +1,15 @@
 require 'hammer_cli/exception_handler'
+require 'hammer_cli_foreman/exceptions'
 
 module HammerCLIForeman
   class ExceptionHandler < HammerCLI::ExceptionHandler
 
     def mappings
       super + [
+        [HammerCLIForeman::OperationNotSupportedError, :handle_unsupported_operation],
         [RestClient::Forbidden, :handle_forbidden],
         [RestClient::UnprocessableEntity, :handle_unprocessable_entity],
-        [ArgumentError, :handle_argument_error]
+        [ArgumentError, :handle_argument_error],
       ]
     end
 
@@ -28,12 +30,20 @@ module HammerCLIForeman
       HammerCLI::EX_USAGE
     end
 
+
     def handle_forbidden(e)
       print_error "Forbidden - server refused to process the request"
       log_full_error e
       HammerCLI::EX_NOPERM
     end
 
+
+    def handle_unsupported_operation(e)
+      print_error e.message
+      log_full_error e
+      HammerCLI::EX_UNAVAILABLE
+    end
+
   end
 end
 

data/lib/hammer_cli_foreman/exceptions.rb

@@ -0,0 +1,9 @@
+require 'hammer_cli_foreman/exceptions'
+
+module HammerCLIForeman
+
+  class OperationNotSupportedError < StandardError; end
+
+end
+
+

data/lib/hammer_cli_foreman/host.rb

@@ -4,6 +4,7 @@ require 'hammer_cli_foreman/commands'
 require 'hammer_cli_foreman/parameter'
 require 'hammer_cli_foreman/report'
 require 'hammer_cli_foreman/puppet_class'
+require 'hammer_cli_foreman/smart_class_parameter'
 
 module HammerCLIForeman
 
@@ -59,8 +60,12 @@ module HammerCLIForeman
       params['host']['host_parameters_attributes'] = parameter_attributes
       params['host']['compute_attributes'] = compute_attributes || {}
       params['host']['compute_attributes']['volumes_attributes'] = nested_attributes(volume_list)
-      params['host']['compute_attributes']['interfaces_attributes'] = nested_attributes(interface_list)
-      params['host']['compute_attributes']['nics_attributes'] = nested_attributes(interface_list)
+      if compute_resource_id
+        params['host']['compute_attributes']['interfaces_attributes'] = nested_attributes(interface_list)
+        params['host']['compute_attributes']['nics_attributes'] = nested_attributes(interface_list)
+      else
+        params['host']['interfaces_attributes'] = nested_attributes(interface_list)
+      end
 
       params
     end
@@ -87,11 +92,12 @@ module HammerCLIForeman
   end
 
 
-  class Host < HammerCLI::AbstractCommand
+  class Host < HammerCLI::Apipie::Command
+
+    resource ForemanApi::Resources::Host
+
     class ListCommand < HammerCLIForeman::ListCommand
       # FIXME: list compute resource (model)
-      resource ForemanApi::Resources::Host, "index"
-
       output do
         from "host" do
           field :id, "Id"
@@ -109,8 +115,6 @@ module HammerCLIForeman
 
     class InfoCommand < HammerCLIForeman::InfoCommand
 
-      resource ForemanApi::Resources::Host, "show"
-
       def retrieve_data
         host = super
         host["host"]["environment_name"] = host["host"]["environment"]["environment"]["name"] rescue nil
@@ -167,13 +171,39 @@ module HammerCLIForeman
     end
 
 
-    class StatusCommand < HammerCLIForeman::InfoCommand
+    class StatusCommand < HammerCLI::Apipie::ReadCommand
+
+      identifiers :id, :name
 
       command_name "status"
-      resource ForemanApi::Resources::Host, "status"
 
-      def print_data(records)
-        print_message records["status"]
+      output do
+        field :status, "Status"
+        field :power, "Power"
+      end
+
+      def retrieve_data
+        {
+          :status => get_status,
+          :power => get_power_status
+        }
+      end
+
+      def get_status
+        params = {
+          'id' => get_identifier[0],
+        }
+        status = resource.call(:status, params)[0]
+        status["status"]
+      end
+
+      def get_power_status
+        params = {
+          'id' => get_identifier[0],
+          'power_action' => :state
+        }
+        status = resource.call(:power, params)[0]
+        status["power"]
       end
     end
 
@@ -181,7 +211,7 @@ module HammerCLIForeman
     class PuppetRunCommand < HammerCLIForeman::InfoCommand
 
       command_name "puppetrun"
-      resource ForemanApi::Resources::Host, "puppetrun"
+      action "puppetrun"
 
       def print_data(records)
         print_message 'Puppet run triggered'
@@ -260,7 +290,7 @@ module HammerCLIForeman
 
       success_message "Host created"
       failure_message "Could not create the host"
-      resource ForemanApi::Resources::Host, "create"
+      action "create"
 
       include HammerCLIForeman::CommonHostUpdateOptions
 
@@ -278,7 +308,6 @@ module HammerCLIForeman
 
       success_message "Host updated"
       failure_message "Could not update the host"
-      resource ForemanApi::Resources::Host, "update"
 
       include HammerCLIForeman::CommonHostUpdateOptions
     end
@@ -288,7 +317,6 @@ module HammerCLIForeman
 
       success_message "Host deleted"
       failure_message "Could not delete the host"
-      resource ForemanApi::Resources::Host, "destroy"
 
       apipie_options
     end
@@ -339,10 +367,73 @@ module HammerCLIForeman
       end
     end
 
+
+    class StartCommand < HammerCLI::Apipie::WriteCommand
+
+      identifiers :id, :name
+      action "power"
+
+      command_name "start"
+      desc "Power a host on"
+      success_message "The host is starting."
+
+      def power_action
+        :start
+      end
+    end
+
+
+    class StopCommand < HammerCLI::Apipie::WriteCommand
+
+      option '--force', :flag, "Force turning off a host"
+
+      identifiers :id, :name
+      action "power"
+
+      command_name "stop"
+      desc "Power a host off"
+
+      def power_action
+        if force?
+          :cycle
+        else
+          :stop
+        end
+      end
+
+      def success_message
+        if force?
+          "Power off forced."
+        else
+          "Powering the host off."
+        end
+      end
+    end
+
+    class RebootCommand < HammerCLI::Apipie::WriteCommand
+
+      identifiers :id, :name
+      action "power"
+
+      command_name "reboot"
+      desc "Reboot a host"
+      success_message "Host reboot started."
+
+      def power_action
+        :soft
+      end
+    end
+
+    class SCParamsCommand < HammerCLIForeman::SmartClassParametersList
+
+      apipie_options :without => [:host_id, :hostgroup_id, :puppetclass_id, :environment_id]
+      option ['--id', '--name'], 'HOST_ID', 'host id/name', 
+              :attribute_name => :host_id, :required => true
+    end
+
     autoload_subcommands
   end
 
 end
 
 HammerCLI::MainCommand.subcommand 'host', "Manipulate Foreman's hosts.", HammerCLIForeman::Host
-

data/lib/hammer_cli_foreman/hostgroup.rb

@@ -2,6 +2,7 @@ require 'hammer_cli'
 require 'foreman_api'
 require 'hammer_cli_foreman/commands'
 require 'hammer_cli_foreman/parameter'
+require 'hammer_cli_foreman/smart_class_parameter'
 
 module HammerCLIForeman
 
@@ -137,6 +138,15 @@ module HammerCLIForeman
       end
     end
 
+
+    class SCParamsCommand < HammerCLIForeman::SmartClassParametersList
+
+      apipie_options :without => [:host_id, :hostgroup_id, :puppetclass_id, :environment_id]
+      option ['--id', '--name'], 'HOSTGROUP_ID', 'hostgroup id/name', 
+            :attribute_name => :hostgroup_id, :required => true
+    end
+
+
     autoload_subcommands
   end
 

data/lib/hammer_cli_foreman/puppet_class.rb

@@ -1,6 +1,7 @@
 require 'hammer_cli'
 require 'foreman_api'
 require 'hammer_cli_foreman/commands'
+require 'hammer_cli_foreman/smart_class_parameter'
 
 module HammerCLIForeman
 
@@ -46,10 +47,18 @@ module HammerCLIForeman
     end
 
 
+    class SCParamsCommand < HammerCLIForeman::SmartClassParametersBriefList
+
+      apipie_options :without => [:host_id, :hostgroup_id, :puppetclass_id, :environment_id]
+      option ['--id', '--name'], 'PUPPET_CLASS_ID', 'puppet class id/name', 
+              :attribute_name => :puppetclass_id, :required => true
+    end
+
+
     autoload_subcommands
   end
 
 end
 
-HammerCLI::MainCommand.subcommand 'puppet_class', "Browse and read reports.", HammerCLIForeman::PuppetClass
+HammerCLI::MainCommand.subcommand 'puppet_class', "Search Foreman's puppet modules.", HammerCLIForeman::PuppetClass
 

data/lib/hammer_cli_foreman/resource_supported_test.rb

@@ -1,14 +1,14 @@
 module HammerCLIForeman
 
+  class OperationNotSupportedError < StandardError; end
+
   module ResourceSupportedTest
 
     def execute
       if resource_supported?
         super
       else
-        error = "The server does not support such operation."
-        HammerCLI::Output::Output.print_error error, nil, context, :adapter => adapter
-        1
+        raise OperationNotSupportedError, "The server does not support such operation."
       end
     end
 

data/lib/hammer_cli_foreman/smart_class_parameter.rb

@@ -0,0 +1,110 @@
+require 'hammer_cli'
+require 'foreman_api'
+require 'hammer_cli_foreman/commands'
+
+module HammerCLIForeman
+
+  class SmartClassParametersBriefList < HammerCLIForeman::ListCommand
+    resource ForemanApi::Resources::SmartClassParameter, "index"
+    command_name 'sc_params'
+
+    output do
+      field :id, "Id"
+
+      field :parameter, "Parameter"
+      field :default_value, "Default Value"
+      field :override, "Override"
+    end
+
+    def retrieve_data
+      res = super
+      # FIXME: API returns doubled records, probably just if filtered by puppetclasses
+      # it seems group by environment is missing
+      # having the uniq to fix that
+      res["smart_class_parameters"].uniq 
+    end
+  end
+
+  class SmartClassParametersList < SmartClassParametersBriefList
+
+    output do 
+      from :puppetclass do
+        field :name, "Puppet class"
+        field :id, "Class Id", Fields::Id
+      end
+    end
+  end 
+
+  class SmartClassParameter < HammerCLI::Apipie::Command
+
+    resource ForemanApi::Resources::SmartClassParameter
+
+    class ListCommand < HammerCLIForeman::SmartClassParametersList
+      command_name 'list'
+      apipie_options
+    end
+
+    class InfoCommand < HammerCLIForeman::InfoCommand
+
+      output ListCommand.output_definition do
+        field :description, "Description"
+        field :parameter_type, "Type"
+        field :required, "Required"
+        field :_environments, "Environments", Fields::List
+        field :_environment_ids, "Environment Ids", Fields::List
+        label "Validator" do
+          field :validator_type, "Type"
+          field :validator_rule, "Rule"
+        end
+        label "Override values" do
+          field :override_value_order, "Order", Fields::List
+          field :override_values_count, "Count"
+          collection :override_values, "Values" do
+            label  "Value" do
+              field :id, 'Id'
+              field :match, 'Match'
+              field :value, 'Value' 
+            end
+          end
+        end
+        field :created_at, "Created at", Fields::Date
+        field :updated_at, "Updated at", Fields::Date
+      end
+
+      def retrieve_data
+        res = super['smart_class_parameter']
+        res['override_value_order'] = res['override_value_order'].split("\n") 
+        res['_environments'] = res['environments'].map { |e| e['environment']['name']}
+        res['_environment_ids'] = res['environments'].map { |e| e['environment']['id']}
+        res
+      end
+
+    end
+
+    class UpdateCommand < HammerCLIForeman::UpdateCommand
+      # identifiers :id, :name
+
+      success_message "Parameter updated"
+      failure_message "Could not update the parameter"
+
+      apipie_options :without => [:parameter_type, :validator_type, :id, :override, :required]
+
+      option "--override", "OVERRIDE", "Override this parameter.",
+        :format => HammerCLI::Options::Normalizers::Bool.new
+      option "--required", "REQUIRED", "This parameter is required.",
+        :format => HammerCLI::Options::Normalizers::Bool.new
+      option "--parameter-type", "PARAMETER_TYPE", "Type of the parameter.",
+        :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
+
+
+    autoload_subcommands
+
+  end
+
+  HammerCLI::MainCommand.subcommand 'sc_param', "Manipulate Foreman's smart class parameters.", HammerCLIForeman::SmartClassParameter
+
+end

data/lib/hammer_cli_foreman/version.rb

@@ -1,5 +1,5 @@
 module HammerCLIForeman
   def self.version
-    @version ||= Gem::Version.new '0.0.10'
+    @version ||= Gem::Version.new '0.0.11'
   end
 end

data/test/unit/apipie_resource_mock.rb

@@ -0,0 +1,88 @@
+class ApipieResourceMock
+
+  def initialize(resource)
+    @return_values = {}
+    @resource = resource
+    @resource.doc["methods"].each do |method|
+      stub_method(method["name"], get_return_value(method))
+    end
+  end
+
+  def doc
+    @resource.doc
+  end
+
+  def new(attrs)
+    return self
+  end
+
+  def expects_with(method_name, params, return_value=nil)
+    return_value ||= return_value_for(method_name)
+    self.expects(method_name).with(params).returns(return_value)
+  end
+
+  def stub_method(method_name, return_value)
+    self.stubs(method_name.to_s).returns([return_value, return_value.to_s])
+    @return_values[method_name.to_s] = return_value
+  end
+
+  private
+
+  def return_value_for(method_name)
+    @return_values[method_name.to_s]
+  end
+
+  def get_return_value(method)
+    return nil if method["examples"].empty?
+
+    #parse actual json from the example string
+    #examples are in format:
+    # METHOD /api/some/route
+    # <input params in json, multiline>
+    # RETURN_CODE
+    # <output in json, multiline>
+    parse_re = /.*(\n\d+\n)(.*)/m
+    json_string = method["examples"][0][parse_re, 2]
+    response = JSON.parse(json_string) rescue json_string
+
+    response
+  end
+
+end
+
+
+class ApipieDisabledResourceMock
+
+  def initialize(resource)
+    @resource = resource
+    @resource.doc["methods"].each do |method|
+      self.stubs(method["name"]).raises(RestClient::ResourceNotFound)
+    end
+  end
+
+  def doc
+    @resource.doc
+  end
+
+  def new(attrs)
+    return self
+  end
+
+end
+
+def mock_resource_method(method, response)
+  resource_mock = ApipieResourceMock.new(cmd.class.resource.resource_class)
+  resource_mock.stubs(method).returns(response)
+  cmd.class.resource resource_mock
+end
+
+module ResourceMocks
+
+  def self.smart_class_parameter
+    sc_params = ApipieResourceMock.new(ForemanApi::Resources::SmartClassParameter)
+    sc_params.stubs(:index).returns([{ 'smart_class_parameters' => [ {} ] }, ""])
+    sc_params.stubs(:show).returns([{ 'smart_class_parameter' => { 'override_value_order' => '', 'environments' => [] }}, ""])
+    sc_params
+  end
+
+end