onering-agent 0.4.3

data/lib/onering/cli.rb

@@ -0,0 +1,92 @@
+module Onering
+  module CLI
+    require 'yaml'
+    require 'multi_json'
+
+    @_args = {}
+
+    class Plugin
+      def self.default_format(output_value=nil, args=nil)
+        if output_value.is_a?(Hash) or 
+           (output_value.is_a?(Array) and output_value.compact.first.is_a?(Hash))
+        then
+          return 'yaml'
+        end
+
+        return nil
+        
+      rescue Exception
+        return nil
+      end
+
+      def self.inherited(subsclass)
+        @_subclasses ||= []
+        @_subclasses << subsclass unless @_subclasses.include?(subsclass)
+      end
+
+      def self.registered_plugins()
+        @_subclasses || []
+      end
+    end
+
+    def self.connect(cliargs)
+      @_args = cliargs
+
+      return Onering::API.new({
+        'autoconnect' => cliargs[:autoconnect],
+        'config' => {
+          'url'            => cliargs[:url],
+          'path'           => cliargs[:path],
+          'source'         => cliargs[:source],
+          'nosslverify'    => cliargs[:nosslverify],
+          'params'         => Hash[(cliargs[:param] || []).collect{|i| i.split('=',2) }],
+          'authentication' => {
+            'type'         => (cliargs[:apikey_given] ? :token : nil),
+            'keyfile'      => (cliargs[:apikey] || cliargs[:sslkey])
+          }.compact
+        }.compact
+      }.compact)
+    end
+
+    def self.output(data, format)
+      return nil if @_args[:quiet]
+      return nil if data.nil?
+
+      Onering::Logger.debug("Outputting data as #{format}:", "Onering::CLI")
+
+      case format
+      when 'text', 'txt'
+        if data.is_a?(Hash)
+          data.coalesce.each do |k,v|
+            puts k.to_s+': '+v.to_s
+          end
+
+        elsif data.is_a?(Array) and data.first.is_a?(Array)
+          puts data.collect{|i| i.map(&:to_s).join(@_args.get(:separator, "\t")) }.sort.join("\n")
+
+        else
+          [*data].each do |d|
+            if d.is_a?(Hash)
+              d.coalesce.each do |k,v|
+                puts k.to_s+': '+v.to_s
+              end
+            else
+              puts d unless d.nil? or d.to_s.empty?
+            end
+          end
+        end
+
+      when 'json'
+        puts MultiJson.dump(data, :pretty => true)
+
+      when 'yaml'
+        puts YAML.dump(data)
+
+      else
+        Onering::Logger.error("Unknown output format #{format.inspect}", "Onering::CLI")
+      end
+
+      return nil
+    end
+  end
+end

data/lib/onering/cli/assets.rb

@@ -0,0 +1,138 @@
+module Onering
+  module CLI
+    class Assets < Plugin
+      def self.configure(global={})
+        @api = Onering::CLI.connect(global).assets
+
+        @opts = ::Trollop::options do
+          banner <<-EOS
+Search for, manipulate, and list values from one or more Onering assets.
+
+Usage:
+    onering [global] assets [options] [subcommands]
+
+Subcommands:
+    find  <urlquery>
+    get   <property> [property2 ..]
+    list  <property> [property2 ..]
+    save
+    set   <property> <value>
+    tag   <tag1>[ tag2 .. tagN]
+    untag <tag1>[ tag2 .. tagN]
+
+Options:
+EOS
+
+          opt :query, "The Onering urlquery to filter devices by", :short => '-f', :type => :string
+          opt :id,    "The node ID of the device to operate on", :short => '-i', :type => :string
+
+        # subcommands
+          stop_on %w{show get set list find save}
+        end
+      end
+
+      def self.run(args)
+        sc = args.shift
+
+        case (sc.downcase.to_sym rescue nil)
+# -----------------------------------------------------------------------------
+        when :show
+          return @api.assets.show(args[0])
+
+# -----------------------------------------------------------------------------
+        when :get
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Assets") unless args.length == 1
+
+          if @opts[:query_given]
+          # doing this until a bulk field query endpoint is built
+            return @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.get_field(i, args[0])}
+
+          elsif @opts[:id_given]
+            return @api.get_field(@opts[:id], args[0])
+
+          end
+
+# -----------------------------------------------------------------------------
+        when :set
+          Onering::Logger.fatal!("Expected 2 parameters, got #{args.length}", "Onering::CLI::Assets") unless args.length == 2
+
+          if @opts[:query]
+          # doing this until a bulk field set endpoint is built
+            return @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.set_field(i, args[0])}
+
+          elsif @opts[:id]
+            return @api.set_field(@opts[:id], args[0], args[1])
+
+          end
+
+# -----------------------------------------------------------------------------
+        when :list
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Assets") unless args.length >= 1
+          return @api.list(args, {
+            :filter => @opts[:query]
+          }.compact)
+
+# -----------------------------------------------------------------------------
+        when :find
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Assets") unless args.length == 1
+          return @api.find(args[0])
+
+# -----------------------------------------------------------------------------
+        when :save
+          rv = @api.save(args[0] || @opts[:id]) do
+          # read from pipe
+            if not STDIN.tty?
+              STDIN.read()
+
+          # read from specified file
+            elsif (File.readable?(args[1]) rescue false)
+              File.read(args[1])
+
+            else
+              Onering::Logger.fatal!("Cannot save data, no input data specified", "Onering::CLI::Assets")
+            end
+          end
+
+          rv.parsed_response
+
+# -----------------------------------------------------------------------------
+        when :tag
+          Onering::Logger.fatal!("Expected 1 parameters, got #{args.length}", "Onering::CLI::Assets") unless args.length > 0
+
+          if @opts[:query]
+          # doing this until a bulk field set endpoint is built
+            @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.get("devices/#{i}/tag/#{args.join('/')}") }
+
+          elsif @opts[:id]
+            @api.get("devices/#{@opts[:id]}/tag/#{args.join('/')}")
+          end
+          
+          return nil
+
+        when :untag
+          Onering::Logger.fatal!("Expected 1 parameters, got #{args.length}", "Onering::CLI::Assets") unless args.length > 0
+
+          if @opts[:query]
+          # doing this until a bulk field set endpoint is built
+            @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.get("devices/#{i}/untag/#{args.join('/')}") }
+
+          elsif @opts[:id]
+            @api.get("devices/#{@opts[:id]}/untag/#{args.join('/')}")
+          end
+
+          return nil
+        else
+          Onering::Logger.fatal!("Unknown subcommand #{sc.inspect}", "Onering::CLI::Assets")
+        end
+      end
+    end
+  end
+end

data/lib/onering/cli/automation.rb

@@ -0,0 +1,62 @@
+module Onering
+  module CLI
+    class Automation < Plugin
+      def self.configure(global={})
+        @requests = Onering::CLI.connect(global).automation_requests
+        @jobs = Onering::CLI.connect(global).automation_jobs
+        @tasks = Onering::CLI.connect(global).automation_tasks
+      end
+
+      def self.run(args)
+        sc = args.shift   # subcommand
+        ssc = args.shift  # sub-subcommand
+
+        case (sc.downcase.to_sym rescue nil)
+# -----------------------------------------------------------------------------
+        when :requests
+          @opts = ::Trollop::options do
+            banner <<-EOS
+Options:
+EOS
+            stop_on %w{show status requeue flush purge}
+          end
+
+          case (ssc.downcase.to_sym rescue nil)
+          when :show
+            return @requests.show(args[0])
+
+          when :requeue
+            return @requests.requeue_all_failed() if args[0].nil?
+            return @requests.requeue(args[0])
+
+          when :status
+            fields = (args.empty? ? ["status"] : args)
+            rv = {}
+            out = @requests.summary(fields)
+
+            _rejigger_hash = Proc.new do |h|
+              [*h].collect{|i|
+                [i['id'], (i['children'].nil? ? i['count'] : Hash[_rejigger_hash.call(i['children'])])]
+              }
+            end
+
+            return Hash[_rejigger_hash.call(out)]
+
+          when :flush
+            return @requests.flush_queue()
+
+          when :purge
+            return @requests.purge()
+          end
+
+# -----------------------------------------------------------------------------
+        when :jobs
+
+
+# -----------------------------------------------------------------------------
+        when :tasks
+        end
+      end
+    end
+  end
+end

data/lib/onering/cli/call.rb

@@ -0,0 +1,53 @@
+module Onering
+  module CLI
+    class Call < Plugin
+      def self.configure(global={})
+        @api = Onering::CLI.connect(global)
+
+        @opts = ::Trollop::options do
+          banner <<-EOS
+Call an arbitrary Onering API endpoint and return the output
+
+Usage:
+    onering call [options] [endpoint]
+
+Examples:
+    # Returns the API status page at path /api/
+    $ onering call /
+
+    # Returns details about the authenticated user
+    $ onering call users/current
+
+    # Delete the device called '0bf29c'
+    $ onering call devices/0bf29c -m delete
+
+Options:
+EOS
+          opt :method, "The HTTP method to use when performing the request", :default => 'get', :short => "-m", :type => :string
+          opt :query,  "A query string attribute to add to the request in the form of NAME=VALUE", :short => '-a', :type => :string, :multi => true
+          opt :header, "A 'Name: Value' header to add to the request", :short => '-H', :type => :string, :multi => true
+        end
+      end
+
+      def self.run(args)
+        data = STDIN.read() unless STDIN.tty?
+
+        headers = {
+          'Content-Type' => 'application/json'
+        }.merge(Hash[@opts[:header].collect{|i|
+            i.split(/[\:=]\s*/,2)
+        }])
+
+        rv = @api.request(@opts[:method], args.first, {
+          :body    => data,
+          :headers => headers,
+          :query   => Hash[@opts[:query].collect{|i|
+            i.split('=',2)
+          }]
+        })
+
+        return (rv.parsed_response rescue rv.response.body)
+      end
+    end
+  end
+end

data/lib/onering/cli/devices.rb

@@ -0,0 +1,98 @@
+module Onering
+  module CLI
+    class Devices < Plugin
+      def self.configure(global={})
+        @api = Onering::CLI.connect(global).assets
+
+        @opts = ::Trollop::options do
+          banner <<-EOS
+[DEPRECATED in 0.1.3] Use the 'assets' plugin from now on.
+
+Usage:
+    onering [global] report [options]
+
+Options:
+EOS
+
+          opt :query, "The Onering urlquery to filter devices by", :short => '-f', :type => :string
+          opt :id,    "The node ID of the device to operate on", :short => '-i', :type => :string
+
+        # subcommands
+          stop_on %w{show get set list find save}
+        end
+      end
+
+      def self.run(args)
+        sc = args.shift
+
+        case (sc.downcase.to_sym rescue nil)
+# -----------------------------------------------------------------------------
+        when :show
+          return @api.devices.show(args[0])
+
+# -----------------------------------------------------------------------------
+        when :get
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Devices") unless args.length == 1
+
+          if @opts[:query_given]
+          # doing this until a bulk field query endpoint is built
+            return @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.get_field(i, args[0])}
+
+          elsif @opts[:id_given]
+            return @api.get_field(@opts[:id], args[0])
+
+          end
+
+# -----------------------------------------------------------------------------
+        when :set
+          Onering::Logger.fatal!("Expected 2 parameters, got #{args.length}", "Onering::CLI::Devices") unless args.length == 2
+
+          if @opts[:query]
+          # doing this until a bulk field set endpoint is built
+            return @api.list('id', {
+              :filter => @opts[:query]
+            }).collect{|i| @api.set_field(i, args[0])}
+
+          elsif @opts[:id]
+            return @api.set_field(@opts[:id], args[0], args[1])
+
+          end
+
+# -----------------------------------------------------------------------------
+        when :list
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Devices") unless args.length >= 1
+          return @api.list(args[0], {
+            :filter => [@opts[:query], args[1]].compact.join('/')
+          }.compact)
+
+# -----------------------------------------------------------------------------
+        when :find
+          Onering::Logger.fatal!("Expected 1 parameter, got #{args.length}", "Onering::CLI::Devices") unless args.length == 1
+          return @api.find(args[0])
+
+# -----------------------------------------------------------------------------
+        when :save
+          rv = @api.save(args[0] || @opts[:id]) do
+          # read from pipe
+            if not STDIN.tty?
+              STDIN.read()
+
+          # read from specified file
+            elsif (File.readable?(args[1]) rescue false)
+              File.read(args[1])
+
+            else
+              Onering::Logger.fatal!("Cannot save data, no input data specified", "Onering::CLI::Devices")
+            end
+          end
+
+          rv.parsed_response
+        else
+          Onering::Logger.fatal!("Unknown subcommand #{sc.inspect}", "Onering::CLI::Devices")
+        end
+      end
+    end
+  end
+end