|
#!/usr/bin/env ruby
|
|
# Copyright (C) 2011 Vijay Brian Gupta brian.gupta@brandorr.com
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
|
|
# USA.
|
|
|
|
require "optparse"
|
|
|
|
collections_filterable = [
|
|
:architectures,
|
|
:common_parameters,
|
|
:config_templates,
|
|
:domains,
|
|
:environments,
|
|
:fact_values,
|
|
:hosts,
|
|
:hostgroups,
|
|
:media,
|
|
:puppetclasses,
|
|
:reports,
|
|
:roles,
|
|
:settings,
|
|
:lookup_keys,
|
|
:dashboard,
|
|
:operatingsystems,
|
|
:subnets,
|
|
:ptables,
|
|
:users
|
|
]
|
|
|
|
collections = [
|
|
:auth_source_ldaps,
|
|
:hypervisors,
|
|
:lookup_values,
|
|
:smart_proxies,
|
|
:statistics,
|
|
:usergroups,
|
|
]
|
|
|
|
collections_not_implemented = [ :audits, :bookmarks ]
|
|
|
|
@foreman_user = ENV['FOREMAN_USER'] if ENV['FOREMAN_USER']
|
|
@foreman_pass = ENV['FOREMAN_PASSWORD'] if ENV['FOREMAN_PASSWORD']
|
|
@foreman_url = ENV['FOREMAN_SERVER'] if ENV['FOREMAN_SERVER']
|
|
|
|
options = {}
|
|
OptionParser.new do |opts|
|
|
opts.banner = "Usage: " + File.basename($0) + " [options] ..."
|
|
|
|
options[:debug] = false
|
|
opts.on( "-d", "--debug", "Output more information" ) do
|
|
options[:debug] = true
|
|
end
|
|
options[:username] = false
|
|
opts.on( '-u', '--user USER', "Foreman user") do |f|
|
|
options[:username] = f
|
|
end
|
|
options[:password] = false
|
|
opts.on( "-p", "--pass PASSWORD", "Foreman password" ) do |f|
|
|
options[:password] = f
|
|
end
|
|
options[:server] = false
|
|
opts.on( "-s", "--server URL", "Foreman Server URL" ) do |f|
|
|
options[:server] = f
|
|
end
|
|
options[:json] = false
|
|
opts.on( "--json", "JSON output" ) do
|
|
options[:json] = true
|
|
end
|
|
options[:yaml] = false
|
|
opts.on( "--yaml", "YAML output" ) do
|
|
options[:yaml] = true
|
|
end
|
|
options[:status] = false
|
|
opts.on( "--status", "Foreman status" ) do
|
|
options[:status] = true
|
|
end
|
|
# options[:dashboard] = false
|
|
# opts.on( "--dashboard", "Foreman dashboard" ) do |f|
|
|
# options[:dashboard] = f
|
|
# end
|
|
options[:custom] = false
|
|
opts.on( "--custom COLLECTION", "Custom COLLECTION string, see: http://theforeman.org/projects/foreman/wiki/API" ) do |f|
|
|
options[:custom] = f
|
|
end
|
|
collections_filterable.each do |collection|
|
|
options[collection] = false
|
|
opts.on("--#{collection} [filter]", "Retrieve a list of #{collection}") do |f|
|
|
options[collection] = f || true
|
|
end
|
|
end
|
|
collections.each do |collection|
|
|
options[collection] = false
|
|
opts.on("--#{collection}", "Retrieve a list of #{collection}") do
|
|
options[collection] = true
|
|
end
|
|
end
|
|
collections_not_implemented.each do |collection|
|
|
options[collection] = false
|
|
opts.on("--#{collection}", "Not implemented") do
|
|
options[collection] = true
|
|
end
|
|
end
|
|
opts.on_tail("-h", "--help", "Show this message") do
|
|
puts opts
|
|
puts ""
|
|
puts " ENVIRONMENT VARIABLES:"
|
|
puts ""
|
|
puts " FOREMAN_SERVER Foreman Server URL"
|
|
puts " FOREMAN_USER Foreman user"
|
|
puts " FOREMAN_PASSWORD Foreman password"
|
|
puts ""
|
|
puts " CLI options take precedence over ENVIRONMENT VARIABLES"
|
|
puts ""
|
|
puts " FILTERS:"
|
|
puts ""
|
|
puts " Please see:"
|
|
puts " http://theforeman.org/projects/foreman/wiki/Search_API"
|
|
puts ""
|
|
puts " Examples:"
|
|
puts " --hosts \"domain = domain.com AND class = squid\""
|
|
puts " --hosts \"domain = domain.com AND facts.architecture = x86_64 AND \\"
|
|
puts " class = module::class"
|
|
puts " --classes \"name = squid\""
|
|
puts " --domains \"name = domain.com\""
|
|
puts ""
|
|
exit 1
|
|
end
|
|
end.parse!
|
|
if options.values.uniq == [false]
|
|
warn "Use -h, --help for usage."
|
|
exit 1
|
|
end
|
|
|
|
require "rubygems"
|
|
require "rest_client"
|
|
require "json"
|
|
require "yaml"
|
|
|
|
@foreman_user = options.delete(:username) if options[:username]
|
|
@foreman_pass = options.delete(:password) if options[:password]
|
|
@foreman_url = options.delete(:server) if options[:server]
|
|
@usejson = options.delete(:json) if options[:json]
|
|
@useyaml = options.delete(:yaml) if options[:yaml]
|
|
@custom = options.delete(:custom) if options[:custom]
|
|
@status = options.delete(:status) if options[:status]
|
|
|
|
RestClient.log = 'stdout' if options.delete(:debug)
|
|
|
|
def get_collection(path, options = {})
|
|
JSON.parse(RestClient::Request.new({:method => :get,
|
|
:url => "#{@foreman_url}/#{path.to_s.chomp('/')}",
|
|
:user => @foreman_user, :password => @foreman_pass,
|
|
:headers => { :accept => :json, :content_type => :json }}.merge(options)).execute.to_s)
|
|
end
|
|
|
|
def search_collection(path,search)
|
|
get_collection(path, :url => "#{@foreman_url}/#{path}?search=#{URI.encode(search)}")
|
|
end
|
|
|
|
def format
|
|
return "json" if @usejson
|
|
return "yaml" if @useyaml
|
|
end
|
|
|
|
def print_response response
|
|
puts(case format
|
|
when "json"
|
|
JSON.pretty_generate(response)
|
|
when "yaml"
|
|
YAML.dump(response)
|
|
else
|
|
r = response.first
|
|
if r.is_a?(Hash)
|
|
# we simply return host list
|
|
if r.first[0] == "host"
|
|
response.map{|h| h["host"]["name"]}
|
|
else
|
|
response.map{|o| o.inspect}
|
|
end
|
|
else
|
|
response.join("\n")
|
|
end
|
|
end)
|
|
end
|
|
|
|
options.each do |collection, filter|
|
|
next unless filter
|
|
if filter
|
|
print_response(get_collection(collection))
|
|
else
|
|
print_response(search_collection(collection, filter))
|
|
end
|
|
end
|
|
|
|
print_response(get_collection(@custom)) if @custom
|
|
|
|
puts JSON.pretty_generate(get_collection("status")) if @status
|
