puppet-whats 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1f7a7411f919a392627ed06e3f114ffe7da1bf40
+  data.tar.gz: 8d6f155ef1ea1def68cc8f15c83acbb48db66701
+SHA512:
+  metadata.gz: c31f9fa45a2834882ab088dca0bee8f96e600f6192651395cc4284e10327515f5b0e8e67d61c7df4d8a7a4d51edc5d4a1e22efb6afe0ce93cd6f54e25e7b85fc
+  data.tar.gz: 00ad94c6bab5902a9c103bd4c6e8e9b83d936589fe9715ecd648e866c897ad3a8045ddd1f0514a396f8c7f3e790662bc6540ea4a0cdc34295688a94fb6e492e7

data/bin/whats

@@ -0,0 +1,136 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'yaml'
+require 'pp'
+require 'json'
+require 'yaml'
+require 'net/http'
+require 'uri'
+require 'resolv'
+require 'ipaddr'
+require 'terminal-table'
+
+require 'whats'
+
+#SETUP
+default_options={
+  :puppetdb_host => 'puppetdb',
+  :puppetdb_port => '8081',
+  :domains       => ['.local'],
+  :basic_info    => ["hostname","manufacturer","productname","serialnumber","operatingsystem","operatingsystemrelease","architecture","processor0","processorcount","memorysize","kernelrelease","ipaddress","macaddress","virtual","uptime"],
+  :verbose       => false,
+  :yaml          => false,
+  :json          => false,
+  :sort          => false,
+  :all           => false,
+  :config        => '',
+  :quiet         => false,
+
+}
+
+options = default_options
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: whats [options] <hostname>"
+
+  opts.on("-j","--json","JSON output") do |j|
+    options[:json] = j
+  end
+  opts.on("-y","--yaml","YAML output") do |y|
+    options[:yaml] = true
+  end
+  opts.on("-p","--pp","Pretty Print output") do |p|
+    options[:pp] = true
+  end
+  opts.on("-a","--all","Use all facts") do |a|
+    options[:all] = true
+  end
+  opts.on("-c","--config CONFIG","Config file to use") do |c|
+    options.merge!(Hash[YAML::load(open(c)).map { |k, v| [k.to_sym, v] }])
+  options[:config] = c
+  end
+  opts.on("-v","--verbose", "Be verbose") do |v|
+    options[:verbose] = v
+  end
+  opts.on("-q","--quiet", "Be extra quiet, only shown") do |q|
+    options[:quiet] = q
+  end
+  opts.parse!
+end.parse
+
+if ARGV.length != 1
+  puts "Please pass a hostname, see --help"
+  exit
+else
+#see if we were given an ip instead, and if so, get hostname
+  if !(IPAddr.new(ARGV[0]) rescue nil).nil?
+    #got an ip, convert
+    host = Resolv.new.getname ARGV[0]
+        else
+    host = ARGV[0].downcase
+    end
+end
+
+
+
+
+#load config
+if options[:config] == ""
+  config=Whats.config_find()
+  options[:config] = config
+  options.merge!(Hash[YAML::load(open(File.expand_path(config))).map { |k, v| [k.to_sym, v] }])
+  if options[:verbose]
+    pp options
+  end
+else
+  options.merge!(Hash[YAML::load(open(File.expand_path(options[:config]))).map { |k, v| [k.to_sym, v] }])
+  if options[:verbose]
+    pp options
+  end
+end
+#try the name as given first, some hosts have short certnames(sigh)
+
+if !options[:quiet]
+  puts "Searching for #{host}..."
+end
+#detect if we got a short hostname or a fqdn.
+if host.include?('.')
+  fqdn=host
+  if Whats.node_test(options[:puppetdb_host],options[:puppetdb_port],fqdn)
+    facts=Whats.get_facts(options[:puppetdb_host],options[:puppetdb_port],fqdn)
+  else
+    raise ("Node #{fqdn} not found in puppetdb")
+  end
+else
+  #we will need to itterate the available search domains...
+  options[:domains].each do |dom|
+    fqdn="#{host}.#{dom}"
+    if !options[:quiet]
+      puts "Searching for #{fqdn}"
+    end
+    if Whats.node_test(options[:puppetdb_host],options[:puppetdb_port],fqdn)
+      facts=Whats.get_facts(options[:puppetdb_host],options[:puppetdb_port],fqdn)
+      break
+    else
+      next
+    end
+  end
+  if facts == nil
+    raise ("node #{host} not found after serching all domains")
+  end
+end
+
+
+node_facts=Whats.facts_to_hash(facts)
+
+facts2 = Hash.new
+if options[:all] == true
+  facts2 = node_facts
+else
+  options[:basic_info].each do |val|
+    facts2["#{val}"] = node_facts[val]
+  end
+end
+
+Whats.output(options,facts2)

data/lib/whats.rb

@@ -0,0 +1,79 @@
+class Whats
+  
+  def self.config_find()
+    config_search_paths=['~/.whats.yaml','/etc/whats.yaml']
+    config_search_paths.each do |path|
+    #If #{path} is a file and re can read it, return it.
+      if File.exists?(File.expand_path(path))
+        return (path)
+      end
+    end
+    #no config found
+    raise ("No config found!")
+  end
+
+
+
+  #if we passed shortname, get fqdn
+  def self.domain_fix(host,domain,match)
+  #Fix domain if needed
+    if host.match(match)
+      fqdn = host.to_s
+    else
+      fqdn = host.to_s + domain
+    end
+    return(fqdn)
+  end
+
+  #Query to see if the node we're looking for exists
+  def self.node_test(puppetdb_host,puppetdb_port,fqdn)
+    test_uri=URI.parse("http://#{puppetdb_host}:#{puppetdb_port}/pdb/query/v4/nodes/#{fqdn}")
+    test_response = Net::HTTP.get_response(test_uri)
+    test_body=JSON.parse(test_response.body)
+    if test_body.key?("error")
+      return false
+    else
+      return true
+    end
+  end
+
+  #get all facts for the given node
+  def self.get_facts(puppetdb_host,puppetdb_port,fqdn)
+    uri=URI.parse("http://#{puppetdb_host}:#{puppetdb_port}/pdb/query/v4/nodes/#{fqdn}/facts")
+    response = Net::HTTP.get_response(uri)
+    facts=JSON.parse(response.body)
+    return(facts)
+  end
+
+  #Parse the fact output from puppetdb into a simpler hash
+  #
+  # v4 apit returns facts as an array, with each element a hash like
+  #{"certname"=>"some_host",
+  #  "environment"=>"some_env",
+  #  "name"=>"memorysize",
+  #  "value"=>"3.70 GiB"},
+  #
+  def self.facts_to_hash(facts)
+    node_facts=Hash.new
+    facts.each do |fact|
+      node_facts[fact["name"]] = fact["value"]
+    end
+    return node_facts
+  end
+
+#do output
+  def self.output(options,facts2)
+    #output time
+    values=options[:basic_info]
+    if options[:json] == true
+      puts facts2.to_json
+    elsif options[:yaml] == true
+      puts facts2.to_yaml
+    elsif options[:pp] == true
+      pp facts2
+    else
+      table=Terminal::Table.new :headings => ['Fact', 'Value'], :rows => facts2
+      puts table
+    end
+  end
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: puppet-whats
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Matthew Nicholson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-03-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+description: Whats asks pupetdb about systems for you
+email: matthew.a.nicholson@gmail.com
+executables:
+- whats
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/whats
+- lib/whats.rb
+homepage: http://github.com/sjoeboo/whats
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.6
+signing_key: 
+specification_version: 4
+summary: Whats?
+test_files: []