choria-mcorpc-support 2.22.1 → 2.23.0.pre

data/lib/mcollective/application/find.rb

@@ -1,4 +1,4 @@
-class MCollective::Application::Find<MCollective::Application
+class MCollective::Application::Find < MCollective::Application # rubocop:disable Style/ClassAndModuleChildren
   description "Find hosts using the discovery system matching filter criteria"
 
   def main
@@ -12,12 +12,12 @@ class MCollective::Application::Find<MCollective::Application
 
     discoverytime = Time.now - starttime
 
-    STDERR.puts if options[:verbose]
+    $stderr.puts if options[:verbose]
 
     nodes.each {|c| puts c}
 
-    STDERR.puts "\nDiscovered %s nodes in %.2f seconds using the %s discovery plugin" % [nodes.size, discoverytime, mc.client.discoverer.discovery_method] if options[:verbose]
+    warn "\nDiscovered %s nodes in %.2f seconds using the %s discovery plugin" % [nodes.size, discoverytime, mc.client.discoverer.discovery_method] if options[:verbose]
 
-    nodes.size > 0 ? exit(0) : exit(1)
+    !nodes.empty? ? exit(0) : exit(1)
   end
 end

data/lib/mcollective/application/help.rb

@@ -1,10 +1,10 @@
 module MCollective
-  class Application::Help<Application
+  class Application::Help < Application # rubocop:disable Style/ClassAndModuleChildren
     description "Application list and help"
     usage "rpc help [application name]"
 
     def post_option_parser(configuration)
-      configuration[:application] = ARGV.shift if ARGV.size > 0
+      configuration[:application] = ARGV.shift unless ARGV.empty?
     end
 
     def main
@@ -17,7 +17,7 @@ module MCollective
         Applications.list.sort.each do |app|
           begin
             puts "  %-15s %s" % [app, Applications[app].application_description]
-          rescue
+          rescue # rubocop:disable Lint/SuppressedException
           end
         end
 

data/lib/mcollective/application/inventory.rb

@@ -1,344 +1,6 @@
-class MCollective::Application::Inventory<MCollective::Application
+class MCollective::Application::Inventory < MCollective::Application # rubocop:disable Style/ClassAndModuleChildren
   description "General reporting tool for nodes, collectives and subcollectives"
 
-  option :script,
-         :description    => "Script to run",
-         :arguments      => ["--script SCRIPT"]
-
-  option :collectives,
-         :description    => "List all known collectives",
-         :arguments      => ["--list-collectives", "--lc"],
-         :default        => false,
-         :type           => :bool
-
-  option :collectivemap,
-         :description    => "Create a DOT graph of all collectives",
-         :arguments      => ["--collective-graph MAP", "--cg MAP", "--map MAP"]
-
-  def post_option_parser(configuration)
-    configuration[:node] = ARGV.shift if ARGV.size > 0
-  end
-
-  def validate_configuration(configuration)
-    unless configuration[:node] || configuration[:script] || configuration[:collectives] || configuration[:collectivemap]
-      raise "Need to specify either a node name, script to run or other options"
-    end
-  end
-
-  # Get all the known collectives and nodes that belong to them
-  def get_collectives
-    util = rpcclient("rpcutil")
-    util.progress = false
-
-    collectives = {}
-    nodes = 0
-    total = 0
-
-    util.collective_info do |r, cinfo|
-      begin
-        if cinfo[:data] && cinfo[:data][:collectives]
-          cinfo[:data][:collectives].each do |collective|
-            collectives[collective] ||= []
-            collectives[collective]  << cinfo[:sender]
-          end
-
-          nodes += 1
-          total += 1
-        end
-      end
-    end
-
-    {:collectives => collectives, :nodes => nodes, :total_nodes => total}
-  end
-
-  # Writes a crude DOT graph to a file
-  def collectives_map(file)
-    File.open(file, "w") do |graph|
-      puts "Retrieving collective info...."
-      collectives = get_collectives
-
-      graph.puts 'graph {'
-
-      collectives[:collectives].keys.sort.each do |collective|
-        graph.puts '   subgraph "%s" {' % [ collective ]
-
-        collectives[:collectives][collective].each do |member|
-          graph.puts '      "%s" -- "%s"' % [ member, collective ]
-        end
-
-        graph.puts '   }'
-      end
-
-      graph.puts '}'
-
-      puts "Graph of #{collectives[:total_nodes]} nodes has been written to #{file}"
-    end
-  end
-
-  # Prints a report of all known sub collectives
-  def collectives_report
-    collectives = get_collectives
-
-    puts "   %-30s %s" % [ "Collective", "Nodes" ]
-    puts "   %-30s %s" % [ "==========", "=====" ]
-
-    collectives[:collectives].sort_by {|key,count| count.size}.each do |collective|
-      puts "   %-30s %d" % [ collective[0], collective[1].size ]
-    end
-
-    puts
-    puts "   %30s %d" % [ "Total nodes:", collectives[:nodes] ]
-    puts
-  end
-
-  def node_inventory
-    node = configuration[:node]
-
-    util = rpcclient("rpcutil")
-    util.identity_filter node
-    util.progress = false
-
-    nodestats = util.custom_request("daemon_stats", {}, node, {"identity" => node}).first
-
-    unless nodestats
-      STDERR.puts "Did not receive any results from node #{node}"
-      exit 1
-    end
-
-    unless nodestats[:statuscode] == 0
-      STDERR.puts "Failed to retrieve daemon_stats from #{node}: #{nodestats[:statusmsg]}"
-    else
-      util.custom_request("inventory", {}, node, {"identity" => node}).each do |resp|
-        unless resp[:statuscode] == 0
-          STDERR.puts "Failed to retrieve inventory for #{node}: #{resp[:statusmsg]}"
-          next
-        end
-
-        data = resp[:data]
-
-        begin
-          puts "Inventory for #{resp[:sender]}:"
-          puts
-
-          nodestats = nodestats[:data]
-
-          puts "   Server Statistics:"
-          puts "                      Version: #{nodestats[:version]}"
-          puts "                   Start Time: #{Time.at(nodestats[:starttime])}"
-          puts "                  Config File: #{nodestats[:configfile]}"
-          puts "                  Collectives: #{data[:collectives].join(', ')}" if data.include?(:collectives)
-          puts "              Main Collective: #{data[:main_collective]}" if data.include?(:main_collective)
-          puts "                   Process ID: #{nodestats[:pid]}"
-          puts "               Total Messages: #{nodestats[:total]}"
-          puts "      Messages Passed Filters: #{nodestats[:passed]}"
-          puts "            Messages Filtered: #{nodestats[:filtered]}"
-          puts "             Expired Messages: #{nodestats[:ttlexpired]}"
-          puts "                 Replies Sent: #{nodestats[:replies]}"
-          puts "         Total Processor Time: #{nodestats[:times][:utime]} seconds"
-          puts "                  System Time: #{nodestats[:times][:stime]} seconds"
-
-          puts
-
-          puts "   Agents:"
-          if data[:agents].size > 0
-            data[:agents].sort.in_groups_of(3, "") do |agents|
-              puts "      %-15s %-15s %-15s" % agents
-            end
-          else
-            puts "      No agents installed"
-          end
-
-          puts
-
-          puts "   Data Plugins:"
-          if data[:data_plugins].size > 0
-            data[:data_plugins].sort.in_groups_of(3, "") do |plugins|
-              puts "      %-15s %-15s %-15s" % plugins.map{|p| p.gsub("_data", "")}
-            end
-          else
-            puts "      No data plugins installed"
-          end
-
-          puts
-
-          puts "   Configuration Management Classes:"
-          if data[:classes].size > 0
-            field_size = MCollective::Util.field_size(data[:classes], 30)
-            fields_num = MCollective::Util.field_number(field_size)
-            format = "   " + (" %-#{field_size}s" * fields_num)
-
-            data[:classes].sort.in_groups_of(fields_num, "") do |klasses|
-              puts format % klasses
-            end
-          else
-            puts "      No classes applied"
-          end
-
-          puts
-
-          puts "   Facts:"
-          if data[:facts].size > 0
-            data[:facts].sort_by{|f| f[0]}.each do |f|
-              puts "      #{f[0]} => #{f[1]}"
-            end
-          else
-            puts "      No facts known"
-          end
-
-          break
-        rescue Exception => e
-          STDERR.puts "Failed to display node inventory: #{e.class}: #{e}"
-        end
-      end
-    end
-
-    halt util.stats
-  end
-
-  # Helpers to create a simple DSL for scriptlets
-  def format(fmt)
-    @fmt = fmt
-  end
-
-  def fields(&blk)
-    @flds = blk
-  end
-
-  def identity
-    @node[:identity]
-  end
-
-  def facts
-    @node[:facts]
-  end
-
-  def classes
-    @node[:classes]
-  end
-
-  def agents
-    @node[:agents]
-  end
-
-  def page_length(len)
-    @page_length = len
-  end
-
-  def page_heading(fmt)
-    @page_heading = fmt
-  end
-
-  def page_body(fmt)
-    @page_body = fmt
-  end
-
-  # Expects a simple printf style format and apply it to
-  # each node:
-  #
-  #    inventory do
-  #        format "%s:\t\t%s\t\t%s"
-  #
-  #        fields { [ identity, facts["serialnumber"], facts["productname"] ] }
-  #    end
-  def inventory(&blk)
-    raise "Need to give a block to inventory" unless block_given?
-
-    blk.call if block_given?
-
-    raise "Need to define a format" if @fmt.nil?
-    raise "Need to define inventory fields" if @flds.nil?
-
-    util = rpcclient("rpcutil")
-    util.progress = false
-
-    util.inventory do |t, resp|
-      @node = {:identity => resp[:sender],
-        :facts    => resp[:data][:facts],
-        :classes  => resp[:data][:classes],
-        :agents   => resp[:data][:agents]}
-
-      puts @fmt % @flds.call
-    end
-  end
-
-  # Use the ruby formatr gem to build reports using Perls formats
-  #
-  # It is kind of ugly but brings a lot of flexibility in report
-  # writing without building an entire reporting language.
-  #
-  # You need to have formatr installed to enable reports like:
-  #
-  #    formatted_inventory do
-  #        page_length 20
-  #
-  #        page_heading <<TOP
-  #
-  #                Node Report @<<<<<<<<<<<<<<<<<<<<<<<<<
-  #                            time
-  #
-  #    Hostname:         Customer:     Distribution:
-  #    -------------------------------------------------------------------------
-  #    TOP
-  #
-  #        page_body <<BODY
-  #
-  #    @<<<<<<<<<<<<<<<< @<<<<<<<<<<<< @<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
-  #    identity,    facts["customer"], facts["lsbdistdescription"]
-  #                                    @<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
-  #                                    facts["processor0"]
-  #    BODY
-  #    end
-  def formatted_inventory(&blk)
-    require 'formatr'
-
-    raise "Need to give a block to formatted_inventory" unless block_given?
-
-    blk.call if block_given?
-
-    raise "Need to define page body format" if @page_body.nil?
-
-    body_fmt = FormatR::Format.new(@page_heading, @page_body)
-    body_fmt.setPageLength(@page_length)
-    time = Time.now
-
-    util = rpcclient("rpcutil")
-    util.progress = false
-
-    util.inventory do |t, resp|
-      @node = {:identity => resp[:sender],
-        :facts    => resp[:data][:facts],
-        :classes  => resp[:data][:classes],
-        :agents   => resp[:data][:agents]}
-
-      body_fmt.printFormat(binding)
-    end
-  rescue Exception => e
-    STDERR.puts "Could not create report: #{e.class}: #{e}"
-    exit 1
-  end
-
-  @fmt = nil
-  @flds = nil
-  @page_heading = nil
-  @page_body = nil
-  @page_length = 40
-
-  def main
-    if configuration[:script]
-      if File.exist?(configuration[:script])
-        eval(File.read(configuration[:script]))
-      else
-        raise "Could not find script to run: #{configuration[:script]}"
-      end
-
-    elsif configuration[:collectivemap]
-      collectives_map(configuration[:collectivemap])
-
-    elsif configuration[:collectives]
-      collectives_report
-
-    else
-      node_inventory
-    end
-  end
+  external(:command => "choria", :args => ["inventory"])
+  external_help(:command => "choria", :args => ["inventory", "--help"])
 end

data/lib/mcollective/application/ping.rb

@@ -1,82 +1,8 @@
-# encoding: utf-8
 module MCollective
-  class Application::Ping<Application
+  class Application::Ping < Application # rubocop:disable Style/ClassAndModuleChildren
     description "Ping all nodes"
 
-    option :graph,
-           :description    => "Shows a graph of ping distribution",
-           :arguments      => ["--graph", "-g"],
-           :default        => false,
-           :type           => :bool
-
-    # Convert the times structure into a array representing
-    # buckets of responses in 50 ms intervals.  Return a small
-    # sparkline graph using UTF8 characters
-    def spark(resp_times)
-      return "" unless configuration[:graph] || Config.instance.pluginconf["rpc.graph"]
-
-      ticks=%w[▁ ▂ ▃ ▄ ▅ ▆ ▇]
-
-      histo = {}
-
-      # round each time to its nearest 50ms
-      # and keep a count for each 50ms
-      resp_times.each do |time|
-        time = Integer(time + 50 - (time % 50))
-        histo[time] ||= 0
-        histo[time] += 1
-      end
-
-      # set the 50ms intervals that saw no traffic to 0
-      ((histo.keys.max - histo.keys.min) / 50).times do |i|
-        time = (i * 50) + histo.keys.min
-        histo[time] = 0 unless histo[time]
-      end
-
-      # get a numerically sorted list of times
-      histo = histo.keys.sort.map{|k| histo[k]}
-
-      range = histo.max - histo.min
-      scale = ticks.size - 1
-      distance = histo.max.to_f / scale
-
-      histo.map do |val|
-        tick = (val / distance).round
-        tick = 0 if tick < 0
-
-        ticks[tick]
-      end.join
-    end
-
-    def main
-      # If the user did not override the default timeout include the discovery timeout
-      if options[:timeout] == 5
-        discovery_timeout = options[:disctimeout] || Config.instance.discovery_timeout || 0
-        options[:timeout] = options[:timeout] + discovery_timeout
-      end
-      client = MCollective::Client.new(options)
-
-      start = Time.now.to_f
-      times = []
-
-      client.req("ping", "discovery") do |resp|
-        times << (Time.now.to_f - start) * 1000
-
-        puts "%-40s time=%.2f ms" % [resp[:senderid], times.last]
-      end
-
-      puts("\n\n---- ping statistics ----")
-
-      if times.size > 0
-        sum = times.inject(0){|acc,i|acc +i}
-        avg = sum / times.length.to_f
-
-        puts "%d replies max: %.2f min: %.2f avg: %.2f %s" % [times.size, times.max, times.min, avg, spark(times)]
-      else
-        puts("No responses received")
-      end
-
-      halt client.stats
-    end
+    external(:command => "choria", :args => ["ping"])
+    external_help(:command => "choria", :args => ["ping", "--help"])
   end
 end