ats 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a1072c017d8e77937188e39e7f019b5176505e1b27041f53672d60e1c339745f
-  data.tar.gz: f0e796f9ae6feedd3e79ea67bc0a11bdecd68a8a59429d881a1eafc7f5aec21c
+  metadata.gz: 226e424b41b47a9bed3a8e3e02daa4c8e6747d44cd9c3c0d25e9bbf0f7b977d3
+  data.tar.gz: d67d968d01a44e95a75d1c2541a29319c19d1d7b2d1b05c924c03479ec533c9c
 SHA512:
-  metadata.gz: d70bac3a644e5546a20ca47466c97749ea0947a787908de7aca141c71bb47c809d9c2741e1e225d056b1bb44574f3f9909028b68c3f7a479319cb457a2054a23
-  data.tar.gz: e274f5b2e9d4c7d7d37ad42265a9ae7f7f3c8f00b9cd4dba09556527c7fd246d0d6324c321979d3e0c9b43734e431cce4b0a07c2db2001596469960b21c969a9
+  metadata.gz: 6e8e7e9a31b025f787681b847fc1d41f08c634ba579facd07d3dd9e468dcaac90a5c0337a7afff9ebadfa7ede6f4d086177ea20400a9879aeb668835c023c840
+  data.tar.gz: 57eababf9c32f5749f6d43a237f7e04ecc3a235e442700449cb36aed4cfea50b4c7764aa1120d4f053fa6658fcdbe804b2858ac23fe111fe834a43e16c8d71bc

data/lib/ats.rb

@@ -27,6 +27,7 @@ module ATS
 
     def configure
       yield configuration
+      configuration
     end
 
     def configuration

data/lib/ats/amp4e/api.rb

@@ -7,16 +7,16 @@ module ATS
         'User-Agent' => "RubyGems/ATS #{ATS::VERSION}",
       }.freeze
 
-      attr_reader :http, :profile, :configuration
+      attr_reader :http, :port, :host, :scheme, :client_id, :client_secret
 
-      def initialize(
-        api: HttpAPI.new(headers: HEADERS),
-        profile: :default,
-        configuration: ATS.configuration
-      )
-        @http = api
-        @profile = profile.to_sym
+      def initialize(configuration:, debug: false)
+        @http = HttpAPI.new(headers: HEADERS, debug: debug)
         @configuration = configuration
+        @port = configuration[:port]
+        @scheme = configuration[:scheme]
+        @host = configuration[:host]
+        @client_id = configuration[:client_id]
+        @client_secret = configuration[:client_secret]
       end
 
       def computers
@@ -47,26 +47,6 @@ module ATS
         URI::Generic.build(host: host, port: port, scheme: scheme, path: "/v#{version}/#{relative_url}")
       end
 
-      def client_id
-        configuration[profile][:amp4e][:client_id]
-      end
-
-      def client_secret
-        configuration[profile][:amp4e][:client_secret]
-      end
-
-      def host
-        configuration[profile][:amp4e][:host]
-      end
-
-      def scheme
-        configuration[profile][:amp4e][:scheme]
-      end
-
-      def port
-        configuration[profile][:amp4e][:port]
-      end
-
       def headers
         { AUTHORIZATION: "Basic #{Base64.strict_encode64("#{client_id}:#{client_secret}")}" }
       end

data/lib/ats/cli.rb

@@ -1,8 +1,12 @@
 require 'ats'
-
 require 'thor'
-require 'ats/cli/threat_grid'
-require 'ats/cli/amp4e/command'
+require 'ats/cli/command'
+require 'ats/cli/threat_grid/organizations'
+require 'ats/cli/threat_grid/samples'
+require 'ats/cli/threat_grid/search'
+require 'ats/cli/threat_grid/users'
+require 'ats/cli/threat_grid/whoami'
+require 'ats/cli/threat_grid/application'
 require 'ats/cli/amp4e/computers'
 require 'ats/cli/amp4e/events'
 require 'ats/cli/amp4e/groups'
@@ -28,7 +32,7 @@ module ATS
       desc 'setup', 'Initialize the .atsrc file.'
       def setup(configuration = ATS.configuration)
         say "Current Configuration:", :green
-        say JSON.pretty_generate(configuration.configuration), :green
+        say JSON.pretty_generate(configuration.to_h), :green
 
         configuration.config_files.each do |file|
           if File.exist?(file)
@@ -49,7 +53,7 @@ module ATS
               port: 443,
               scheme: 'https',
             },
-            threat_grid: {
+            threatgrid: {
               api_key: '',
               host: 'panacea.threatgrid.com',
               port: 443,

data/lib/ats/cli/command.rb

@@ -0,0 +1,46 @@
+module ATS
+  module CLI
+    class Command < Thor
+      class_option :profile, default: :default, required: false
+      class_option :debug, default: false, required: false
+
+      def self.printable_commands(*args)
+        super.map do |x|
+          x[0] = x[0].gsub(/^ats/, "ats #{service_name.downcase}")
+          x
+        end
+      end
+
+      protected
+
+      def api
+        self.class.constant_name.new(
+          configuration: configuration.fetch(profile)[self.class.service_name.downcase.to_sym],
+          debug: configuration.debug,
+        )
+      end
+
+      def print_json(json)
+        say JSON.pretty_generate(json), :green
+      end
+
+      def configuration
+        ATS.configure do |x|
+          x.debug = options['debug']
+        end
+      end
+
+      def profile
+        options['profile'].to_sym
+      end
+
+      def self.constant_name
+        Kernel.const_get("ATS::#{service_name}::API")
+      end
+
+      def self.service_name
+        name.split("::")[2]
+      end
+    end
+  end
+end

data/lib/ats/configuration.rb

@@ -1,26 +1,35 @@
 module ATS
   class Configuration
-    attr_accessor :logger, :config_files, :configuration
+    attr_accessor :logger, :config_files, :debug
 
     def initialize
       @logger = Logger.new(STDOUT)
       @logger.level = Logger::DEBUG
+      @debug = false
       @config_files = [
         File.join(Dir.home, ".atsrc"),
         File.expand_path('.atsrc'),
         ENV['ATSRC']
       ].compact
+    end
+
+    def fetch(key)
+      hash.fetch(key.to_sym)
+    end
 
-      @configuration = load_configuration
+    def to_h
+      hash.dup
     end
 
-    def [](key)
-      @configuration[key.to_sym]
+    private
+
+    def hash
+      @hash ||= load_configuration
     end
 
     def load_configuration(files = config_files)
       files.inject({}) do |memo, file|
-        logger.debug("Loading... #{file}")
+        logger.debug("Searching for #{file}...") if debug
         memo.merge!(YAML.load(IO.read(file))) if File.exist?(file)
         memo
       end

data/lib/ats/http_api.rb

@@ -1,7 +1,8 @@
 module ATS
   class HttpAPI
-    def initialize(headers: {})
+    def initialize(headers: {}, debug: false)
       @default_headers = headers
+      @debug = debug
     end
 
     def execute(uri, request)
@@ -44,7 +45,7 @@ module ATS
       http = Net::HTTP.new(uri.host, uri.port)
       http.read_timeout = 30
       http.use_ssl = uri.scheme == "https"
-      http.set_debug_output(ATS.logger)
+      http.set_debug_output(ATS.logger) if @debug
       http
     end
 

data/lib/ats/threat_grid/api.rb

@@ -1,5 +1,3 @@
-require 'ats/http_api'
-
 module ATS
   module ThreatGrid
     class API
@@ -8,16 +6,15 @@ module ATS
         'User-Agent' => "RubyGems/ATS #{ATS::VERSION}",
       }.freeze
 
-      attr_reader :http, :profile, :configuration
+      attr_reader :http, :port, :host, :scheme, :api_key
 
-      def initialize(
-        api: HttpAPI.new(headers: HEADERS),
-        profile: :default,
-        configuration: ATS.configuration
-      )
-        @http = api
-        @profile = profile.to_sym
+      def initialize(configuration:, debug: false)
+        @http = HttpAPI.new(headers: HEADERS, debug: debug)
         @configuration = configuration
+        @port = configuration[:port]
+        @scheme = configuration[:scheme]
+        @host = configuration[:host]
+        @api_key = configuration[:api_key]
       end
 
       def whoami
@@ -55,22 +52,6 @@ module ATS
       def build_uri(relative_url, version:)
         URI::Generic.build(host: host, port: port, scheme: scheme, path: "/api/v#{version}/#{relative_url}")
       end
-
-      def api_key
-        configuration[profile][:threat_grid][:api_key]
-      end
-
-      def host
-        configuration[profile][:threat_grid][:host]
-      end
-
-      def scheme
-        configuration[profile][:threat_grid][:scheme]
-      end
-
-      def port
-        configuration[profile][:threat_grid][:port]
-      end
     end
   end
 end

data/lib/ats/version.rb

@@ -1,3 +1,3 @@
 module ATS
-  VERSION = "0.1.7"
+  VERSION = "0.1.8"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ats
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - mokha
@@ -88,14 +88,12 @@ files:
 - lib/ats/amp4e/policies.rb
 - lib/ats/cli.rb
 - lib/ats/cli/amp4e/application.rb
-- lib/ats/cli/amp4e/command.rb
 - lib/ats/cli/amp4e/computers.rb
 - lib/ats/cli/amp4e/events.rb
 - lib/ats/cli/amp4e/groups.rb
 - lib/ats/cli/amp4e/policies.rb
-- lib/ats/cli/threat_grid.rb
+- lib/ats/cli/command.rb
 - lib/ats/cli/threat_grid/application.rb
-- lib/ats/cli/threat_grid/command.rb
 - lib/ats/cli/threat_grid/organizations.rb
 - lib/ats/cli/threat_grid/samples.rb
 - lib/ats/cli/threat_grid/search.rb

data/lib/ats/cli/amp4e/command.rb

@@ -1,26 +0,0 @@
-module ATS
-  module CLI
-    module AMP4E
-      class Command < Thor
-        class_option :profile, default: :default, required: false
-
-        def self.printable_commands(*args)
-          super.map do |x|
-            x[0] = x[0].gsub(/^ats/, 'ats amp4e')
-            x
-          end
-        end
-
-        protected
-
-        def api
-          ATS::AMP4E::API.new(profile: options['profile'])
-        end
-
-        def print_json(json)
-          say JSON.pretty_generate(json), :green
-        end
-      end
-    end
-  end
-end

data/lib/ats/cli/threat_grid.rb

@@ -1,15 +0,0 @@
-require 'ats/cli/threat_grid/command'
-require 'ats/cli/threat_grid/organizations'
-require 'ats/cli/threat_grid/samples'
-require 'ats/cli/threat_grid/search'
-require 'ats/cli/threat_grid/users'
-require 'ats/cli/threat_grid/whoami'
-
-require 'ats/cli/threat_grid/application'
-
-module ATS
-  module CLI
-    module ThreatGrid
-    end
-  end
-end

data/lib/ats/cli/threat_grid/command.rb

@@ -1,26 +0,0 @@
-module ATS
-  module CLI
-    module ThreatGrid
-      class Command < Thor
-        class_option :profile, default: :default, required: false
-
-        def self.printable_commands(*args)
-          super.map do |x|
-            x[0] = x[0].gsub(/^ats/, 'ats threatgrid')
-            x
-          end
-        end
-
-        protected
-
-        def api
-          ATS::ThreatGrid::API.new(profile: options['profile'])
-        end
-
-        def print_json(json)
-          say JSON.pretty_generate(json), :green
-        end
-      end
-    end
-  end
-end