zaws 0.0.5 → 0.1.1

data/lib/zaws/external/nessusapi/data/scanners.rb

@@ -0,0 +1,35 @@
+module ZAWS
+  class Nessusapi
+    class Data
+      class Scanners
+
+          def initialize(shellout, nessusapi)
+            @shellout=shellout
+            @nessusapi=nessusapi
+            @instance_hash=nil
+          end
+
+          def validJSON
+            return (@instance_hash.nil?)
+          end
+
+          def load(data, verbose)
+            @instance_raw_data = data
+            verbose.puts(@instance_raw_data) if verbose
+            @instance_hash=data
+          end
+
+          def view(verbose)
+            details = @nessusapi.filestore.retrieve("scanners")
+            if details.nil?
+               load(@nessusapi.resource_scanners.list.execute(verbose),verbose)
+               @nessusapi.filestore.store("scanners",@instance_hash,Time.now + @nessusapi.filestore.timeout)
+            else
+              load(details,verbose)
+            end
+            return @instance_raw_data
+          end
+      end
+    end
+  end
+end

data/lib/zaws/external/nessusapi/nessusapi.rb

@@ -0,0 +1,57 @@
+module ZAWS
+  class Nessusapi
+
+    attr_accessor :home
+
+    def initialize(shellout)
+      @shellout=shellout
+    end
+
+    def filestore
+      @filestore ||= ZAWS::Repository::Filestore.new()
+      @filestore.location="#{@home}/.nessusapi"
+      unless File.directory?(@filestore.location)
+			  FileUtils.mkdir_p(@filestore.location)
+		  end
+      @filestore.timeout = 600
+      return @filestore
+    end
+
+    def remove_creds
+      if File.directory?("#{@home}/.nessusapi")
+        FileUtils.rmtree("#{@home}/.nessusapi")
+      end
+      if File.exist?("#{@home}/.nessus.yml")
+        File.delete("#{@home}/.nessus.yml")
+      end
+    end
+
+    def resource_scanners
+      @_resource_scanners ||= (ZAWS::Nessusapi::Resources::Scanners.new(@shellout, self))
+      return @_resource_scanners
+    end
+
+    def resource_agents
+      @_resource_agents ||= (ZAWS::Nessusapi::Resources::Agents.new(@shellout, self))
+      return @_resource_agents
+    end
+
+    def client
+      fail("Home is null! Make sure its set before getting the client.") if @home== nil
+      creds = ZAWS::Helper::NessusCreds::Creds::YamlFile.new(@home)
+      @_client ||=  (ZAWS::Helper::NessusClient.new(creds))
+    end
+
+    def data_scanners
+      @_data_scanners ||= (ZAWS::Nessusapi::Data::Scanners.new(@shellout, self))
+      return @_data_scanners
+    end
+
+    def data_agents
+      @_data_agents ||= (ZAWS::Nessusapi::Data::Agents.new(@shellout, self))
+      return @_data_agents
+    end
+
+  end
+end
+

data/lib/zaws/external/nessusapi/reources.rb

@@ -0,0 +1,10 @@
+# module ZAWS
+#   class Nessusapi
+#     class Resources
+#       def initialize(shellout, nessusapi)
+#         @shellout=shellout
+#         @nessusapi=nessusapi
+#       end
+#     end
+#   end
+# end

data/lib/zaws/external/nessusapi/resources/agents.rb

@@ -0,0 +1,18 @@
+module ZAWS
+  class Nessusapi
+    class Resources
+      class Agents
+        def initialize(shellout, nessusapi)
+          @shellout=shellout
+          @nessusapi=nessusapi
+        end
+
+        def list
+          @_list ||= (ZAWS::Nessusapi::Resources::Agents::List.new(@shellout, @nessusapi))
+          return @_list
+        end
+
+      end
+    end
+  end
+end

data/lib/zaws/external/nessusapi/resources/agents/list.rb

@@ -0,0 +1,20 @@
+module ZAWS
+  class Nessusapi
+    class Resources
+      class Agents
+        class List
+
+          def initialize(shellout,nessusapi)
+            @shellout=shellout
+            @nessusapi=nessusapi
+          end
+
+          def execute(scanner,verbose=nil)
+            @nessusapi.client.get("/scanners/#{scanner}/agents")
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/zaws/external/nessusapi/resources/scanners.rb

@@ -0,0 +1,18 @@
+module ZAWS
+  class Nessusapi
+    class Resources
+      class Scanners
+        def initialize(shellout, nessusapi)
+          @shellout=shellout
+          @nessusapi=nessusapi
+        end
+
+        def list
+          @_list ||= (ZAWS::Nessusapi::Resources::Scanners::List.new(@shellout, @nessusapi))
+          return @_list
+        end
+
+      end
+    end
+  end
+end

data/lib/zaws/external/nessusapi/resources/scanners/list.rb

@@ -0,0 +1,20 @@
+module ZAWS
+  class Nessusapi
+    class Resources
+      class Scanners
+        class List
+
+          def initialize(shellout,nessusapi)
+            @shellout=shellout
+            @nessusapi=nessusapi
+          end
+
+          def execute(verbose=nil)
+            @nessusapi.client.get("/scanners")
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/zaws/external/newrelicapi/data/servers.rb

@@ -0,0 +1,36 @@
+module ZAWS
+  class Newrelicapi
+    class Data
+      class Servers
+
+        def initialize(shellout, newrelicapi)
+          @shellout=shellout
+          @newrelicapi=newrelicapi
+          @instance_hash=nil
+        end
+
+        def validJSON
+          return (@instance_hash.nil?)
+        end
+
+        def load(data, verbose)
+          @instance_raw_data = data
+          verbose.puts(@instance_raw_data) if verbose
+          @instance_hash=data
+        end
+
+        def view(verbose)
+          details = @newrelicapi.filestore.retrieve("serversjson")
+          if details.nil?
+            load(@newrelicapi.resource_servers.list.execute(verbose),verbose)
+            @newrelicapi.filestore.store("serversjson",@instance_hash,Time.now + @newrelicapi.filestore.timeout)
+          else
+            load(details,verbose)
+          end
+          return @instance_raw_data
+        end
+
+      end
+    end
+  end
+end

data/lib/zaws/external/newrelicapi/newrelic_client.rb

@@ -0,0 +1,64 @@
+require 'excon'
+require 'json'
+
+module ZAWS
+  class Newrelicapi
+    class NewrelicClient
+
+      autoload :NewrelicCreds, "zaws/newrelicapi/newrelic_creds"
+
+      # @param [SumoClient::Creds] creds
+      def initialize(creds)
+        @creds = creds
+      end
+
+      def url
+        @creds.url
+      end
+
+      # Make a GET request expecting a JSON response.
+      def get(path, query = {}, options = {})
+        # Handle nil or empty Array
+        options[:query] = query.to_h if query
+        options[:idempotent] = true
+        json_request('GET', path, options)
+      end
+
+      # Make a POST request expecting a JSON response.
+      def post(path, body, options = {})
+        options[:body] = body
+        json_request('POST', path, options)
+      end
+
+      # Make a PUT request expecting a JSON response.
+      def put(path, body, options = {})
+        options[:body] = body
+        json_request('PUT', path, options)
+      end
+
+      def delete(path, options = {})
+        options[:idempotent] = true
+        json_request('DELETE', path, options)
+      end
+
+      def json_request(method, path, options = {})
+        if options[:body] && !options[:body].instance_of?(String)
+          options[:body] = options[:body].to_json
+        end
+        options[:headers] ||= {}
+        options[:headers]['Content-Type'] = 'application/json'
+        response = request(method, path, options)
+        JSON.parse(response.body) if (response.body.length > 0 && response.headers['content-type'].match(/json/))
+      end
+
+      def request(method, path, options = {})
+        connection = Excon.new(@creds.url)
+        options[:expects] ||= [200]
+        options[:method] = method
+        options[:path] = path
+        options[:headers]['X-api-key'] = "#{@creds.api_key}"
+        connection.request(options)
+      end
+    end
+  end
+end

data/lib/zaws/external/newrelicapi/newrelic_creds.rb

@@ -0,0 +1,60 @@
+require 'yaml'
+
+module ZAWS
+  class Newrelicapi
+    class NewrelicCreds
+        # Simple container for credentials
+        class Creds
+
+          # @param [String] newrelic_url
+          # @param [String] api_key
+          def initialize(newrelic_url, api_key)
+            @creds = {}
+            @creds[:url] = newrelic_url
+            @creds[:api_key] = api_key
+          end
+
+          def url
+            fetch_creds[:url]
+          end
+
+          def api_key
+            fetch_creds[:api_key]
+          end
+
+          private
+
+          def fetch_creds
+            # Nothing to do in the base class
+            @creds
+          end
+        end
+
+        # Subclass that loads creds from a YAML file.
+        class Creds::YamlFile < Creds
+          FILENAME = '.newrelic.yml'
+
+          def initialize(home)
+            @creds_file = File.join(home, FILENAME)
+          end
+
+          private
+
+          def fetch_creds
+            unless @creds
+              fail("Missing file #{@creds_file}") unless File.exist?(@creds_file)
+              file_creds = YAML.load(File.read(@creds_file))
+              file_creds ||= {}
+              ['url', 'apiKey'].each do |key|
+                fail("Missing #{key} value in #{@creds_file}") unless file_creds[key]
+              end
+              @creds = {}
+              @creds[:url] = file_creds['url']
+              @creds[:api_key] = file_creds['apiKey']
+            end
+            @creds
+          end
+        end
+    end
+  end
+end

data/lib/zaws/external/newrelicapi/newrelicapi.rb

@@ -0,0 +1,46 @@
+module ZAWS
+  class Newrelicapi
+
+    attr_accessor :home
+
+    def initialize(shellout)
+      @shellout=shellout
+    end
+
+    def filestore
+      @filestore ||= ZAWS::Repository::Filestore.new()
+      @filestore.location="#{@home}/.newrelicapi"
+      unless File.directory?(@filestore.location)
+        FileUtils.mkdir_p(@filestore.location)
+      end
+      @filestore.timeout = 600
+      return @filestore
+    end
+
+    def remove_creds
+      if File.directory?("#{@home}/.newrelicapi")
+        FileUtils.rmtree("#{@home}/.newrelicapi")
+      end
+      if File.exist?("#{@home}/.newrelic.yml")
+        File.delete("#{@home}/.newrelic.yml")
+      end
+    end
+
+    def resource_servers
+      @_resource_servers ||= (ZAWS::Newrelicapi::Resources::Servers.new(@shellout, self))
+      return @_resource_servers
+    end
+
+    def client
+      fail("Home is null! Make sure its set before getting the client.") if @home== nil
+      creds = ZAWS::Newrelicapi::NewrelicCreds::Creds::YamlFile.new(@home)
+      @_client ||= (ZAWS::Newrelicapi::NewrelicClient.new(creds))
+    end
+
+    def data_servers
+      @_data_servers ||= (ZAWS::Newrelicapi::Data::Servers.new(@shellout, self))
+      return @_data_servers
+    end
+
+  end
+end

data/lib/zaws/external/newrelicapi/resources/servers.rb

@@ -0,0 +1,18 @@
+module ZAWS
+  class Newrelicapi
+    class Resources
+      class Servers
+        def initialize(shellout, newrelicapi)
+          @shellout=shellout
+          @newrelicapi=newrelicapi
+        end
+
+        def list
+          @_list ||= (ZAWS::Newrelicapi::Resources::Servers::List.new(@shellout, @newrelicapi))
+          return @_list
+        end
+
+      end
+    end
+  end
+end

data/lib/zaws/external/newrelicapi/resources/servers/list.rb

@@ -0,0 +1,20 @@
+module ZAWS
+  class Newrelicapi
+    class Resources
+      class Servers
+        class List
+
+          def initialize(shellout,newrelicapi)
+            @shellout=shellout
+            @newrelicapi=newrelicapi
+          end
+
+          def execute(verbose=nil)
+            @newrelicapi.client.get("/v2/servers.json")
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/zaws/external/sumoapi/data/collectors.rb

@@ -0,0 +1,35 @@
+module ZAWS
+  class Sumoapi
+    class Data
+      class Collectors
+
+        def initialize(shellout, sumoapi)
+          @shellout=shellout
+          @sumoapi=sumoapi
+          @instance_hash=nil
+        end
+
+        def validJSON
+          return (@instance_hash.nil?)
+        end
+
+        def load(data, verbose)
+          @instance_raw_data = data
+          verbose.puts(@instance_raw_data) if verbose
+          @instance_hash=data
+        end
+
+        def view(verbose)
+          details = @sumoapi.filestore.retrieve("collectors")
+          if details.nil?
+            load(@sumoapi.resource_collectors.list.execute(verbose),verbose)
+            @sumoapi.filestore.store("collectors",@instance_hash,Time.now + @sumoapi.filestore.timeout)
+          else
+            load(details,verbose)
+          end
+          return @instance_raw_data
+        end
+      end
+    end
+  end
+end