# File lib/hammer_cli_foreman/id_resolver.rb, line 81 def initialize(api, searchables) @api = api @searchables = searchables define_id_finders end
# File lib/hammer_cli_foreman/id_resolver.rb, line 118 def puppetclass_ids(options) options[HammerCLI.option_accessor_name("ids")] || find_puppetclasses(options).collect { |c| c['id'] } end
@param mode [Symbol] mode in which ids are searched :single, :multi, nil for old beahvior
# File lib/hammer_cli_foreman/id_resolver.rb, line 88 def scoped_options(scope, options, mode = nil) scoped_options = options.dup resource = HammerCLIForeman.param_to_resource(scope) return scoped_options unless resource option_names = [] if (mode.nil? || mode == :single) option_names << "id" option_names += searchables(resource).map { |s| s.name } end if (mode.nil? || mode == :multi) option_names << "ids" option_names += searchables(resource).map { |s| s.plural_name } end option_names.each do |name| option = HammerCLI.option_accessor_name(name) scoped_option = HammerCLI.option_accessor_name("#{scope}_#{name}") # remove the scope # e.g. option_architecture_id -> option_id if scoped_options[scoped_option] scoped_options[option] = scoped_options.delete(scoped_option) else scoped_options.delete(option) end end scoped_options end
# File lib/hammer_cli_foreman/id_resolver.rb, line 122 def searchables(resource) resource = @api.resource(resource) if resource.is_a? Symbol @searchables.for(resource) end
@param mode [Symbol] mode in which ids are searched :single, :multi, nil for old beahvior
# File lib/hammer_cli_foreman/id_resolver.rb, line 330 def create_search_options(options, resource, mode = nil) searchables(resource).each do |s| value = options[HammerCLI.option_accessor_name(s.name.to_s)] values = options[HammerCLI.option_accessor_name(s.plural_name.to_s)] if value && (mode.nil? || mode == :single) return {:search => "#{s.name} = \"#{value}\""} elsif values && (mode.nil? || mode == :multi) query = values.map{|v| "#{s.name} = \"#{v}\"" }.join(" or ") return {:search => query} end end {} end
# File lib/hammer_cli_foreman/id_resolver.rb, line 310 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
# File lib/hammer_cli_foreman/id_resolver.rb, line 318 def create_smart_variables_search_options(options, mode = nil) search_options = {} value = options[HammerCLI.option_accessor_name('variable')] # handle deprecated --name property value ||= options[HammerCLI.option_accessor_name('name')] search_options[:search] = "key = \"#{value}\"" search_options end
# File lib/hammer_cli_foreman/id_resolver.rb, line 129 def define_id_finders @api.resources.each do |resource| method_name = "#{resource.singular_name}_id" plural_method_name = "#{resource.singular_name}_ids" self.class.send(:define_method, method_name) do |options| get_id(resource.name, options) end unless respond_to?(method_name, true) self.class.send(:define_method, plural_method_name) do |options| get_ids(resource.name, options) end unless respond_to?(plural_method_name, true) end end
@param mode [Symbol] mode in which ids are searched :single, :multi, nil for old beahvior
# File lib/hammer_cli_foreman/id_resolver.rb, line 272 def expected_record_count(options, resource, mode = nil) searchables(resource).each do |s| value = options[HammerCLI.option_accessor_name(s.name.to_s)] unless mode == :multi values = options[HammerCLI.option_accessor_name(s.plural_name.to_s)] unless mode == :single if value return 1 elsif values return values.count end end 0 end
# File lib/hammer_cli_foreman/id_resolver.rb, line 181 def find_puppetclasses(options) resource_name = :puppetclasses resource = @api.resource(resource_name) if (ids = options[HammerCLI.option_accessor_name("ids")]) ids elsif !options_empty?(resource, options) results = resolved_call(resource_name, :index, options, :multi).first.values.flatten raise ResolverError.new(_("one of %s not found.") % resource.name, resource) if results.count < expected_record_count(options, resource, :multi) results else [] end end
# File lib/hammer_cli_foreman/id_resolver.rb, line 196 def find_resource(resource_name, options) results = find_resource_raw(resource_name, options) resource = @api.resource(resource_name) pick_result(results, resource) end
# File lib/hammer_cli_foreman/id_resolver.rb, line 202 def find_resource_raw(resource_name, options) resolved_call(resource_name, :index, options, :single) end
# File lib/hammer_cli_foreman/id_resolver.rb, line 174 def find_resources(resource_name, options) resource = @api.resource(resource_name) results = resolved_call(resource_name, :index, options, :multi) raise ResolverError.new(_("one of %s not found.") % resource.name, resource) if results.count < expected_record_count(options, resource, :multi) results end
# File lib/hammer_cli_foreman/id_resolver.rb, line 144 def get_id(resource_name, options) options[HammerCLI.option_accessor_name("id")] || nil_from_searchables(resource_name, options) || find_resource(resource_name, options)['id'] end
# File lib/hammer_cli_foreman/id_resolver.rb, line 150 def get_ids(resource_name, options) if (ids = options[HammerCLI.option_accessor_name("ids")]) ids elsif (ids = nil_from_searchables(resource_name, options, :plural => true)) ids elsif options_not_set?(@api.resource(resource_name), options) resource = @api.resource(resource_name) raise MissingSearchOptions.new(_("Missing options to search %s") % resource.name, resource) elsif options_empty?(@api.resource(resource_name), options) [] else find_resources(resource_name, options).map{|r| r['id']} end end
# File lib/hammer_cli_foreman/id_resolver.rb, line 165 def nil_from_searchables(resource_name, options, plural = false) resource = @api.resource(resource_name) searchables(resource).each do |s| option_name = plural ? s.plural_name.to_s : s.name.to_s return HammerCLI::NilValue if options[HammerCLI.option_accessor_name(option_name)] == HammerCLI::NilValue end nil end
# File lib/hammer_cli_foreman/id_resolver.rb, line 296 def options_empty?(resource, options) searchables(resource).all? do |s| values = options[HammerCLI.option_accessor_name(s.plural_name.to_s)] values.nil? || (values.respond_to?(:empty?) && values.empty?) end end
# File lib/hammer_cli_foreman/id_resolver.rb, line 303 def options_not_set?(resource, options) searchables(resource).all? do |s| values = options[HammerCLI.option_accessor_name(s.plural_name.to_s)] values.nil? end end
# File lib/hammer_cli_foreman/id_resolver.rb, line 241 def pick_result(results, resource) raise ResolverError.new(_("%s not found.") % resource.singular_name, resource) if results.empty? raise ResolverError.new(_("Found more than one %s.") % resource.singular_name, resource) if results.count > 1 results[0] end
puppet class search results are in non-standard format and needs to be un-hashed first
# File lib/hammer_cli_foreman/id_resolver.rb, line 287 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 = HammerCLIForeman::PuppetClass::ListCommand.unhash_classes(results) pick_result(results, resource)['id'] end
@param mode [Symbol] mode in which ids are searched :single, :multi, nil for old beahvior
# File lib/hammer_cli_foreman/id_resolver.rb, line 208 def resolved_call(resource_name, action_name, options, mode = nil) resource = @api.resource(resource_name) action = resource.action(action_name) search_options = search_options(options, resource, mode) IdParamsFilter.new(:only_required => true).for_action(action).each do |param| scoped_options_params = [param.name.gsub(/_id$/, ""), options] scoped_options_params << mode if method(:scoped_options).arity == -3 search_options[param.name] ||= send(param.name, scoped_options(*scoped_options_params)) end search_options = route_options(options, action).merge(search_options) results = resource.call(action_name, search_options) results = HammerCLIForeman.collection_to_common_format(results) results end
# File lib/hammer_cli_foreman/id_resolver.rb, line 225 def route_options(options, action) return {} if action.routes.any? { |r| r.params_in_path.empty? } route_options = {} action.routes.each do |route| route.params_in_path.each do |param| key = HammerCLI.option_accessor_name(param.to_s) if options[key] route_options[param] ||= options[key] end end end route_options end
@param mode [Symbol] mode in which ids are searched :single, :multi, nil for old beahvior
# File lib/hammer_cli_foreman/id_resolver.rb, line 248 def search_options(options, resource, mode = nil) override_method = "create_#{resource.name}_search_options" search_options = if respond_to?(override_method, true) create_search_options_params = [override_method, options] if method(override_method.to_sym).arity == -2 create_search_options_params << mode else warn "create_*_search_options methods (#{override_method}) without 'mode' parameter are deprecated" end send(*create_search_options_params) else create_search_options_params = [options, resource] if method(:create_search_options).arity == -3 create_search_options_params << mode else warn "create_search_options methods without 'mode' parameter are deprecated" end create_search_options(*create_search_options_params) end raise MissingSearchOptions.new(_("Missing options to search %s.") % resource.singular_name, resource) if search_options.empty? search_options end