add debug option.

4 files changed, 13 insertions, 15 deletions

diff --git a/lib/ats/amp4e/api.rb b/lib/ats/amp4e/api.rb
index 8ca1c2a..2f1a3e0 100644
--- a/lib/ats/amp4e/api.rb
+++ b/lib/ats/amp4e/api.rb
@@ -9,12 +9,8 @@ module ATS
 
       attr_reader :http, :profile, :configuration
 
-      def initialize(
-        api: HttpAPI.new(headers: HEADERS),
-        profile: :default,
-        configuration: ATS.configuration
-      )
-        @http = api
+      def initialize(profile:, configuration:, debug: false)
+        @http = HttpAPI.new(headers: HEADERS, debug: debug)
         @profile = profile.to_sym
         @configuration = configuration
       end
diff --git a/lib/ats/cli/command.rb b/lib/ats/cli/command.rb
index 77531eb..9fc0721 100644
--- a/lib/ats/cli/command.rb
+++ b/lib/ats/cli/command.rb
@@ -2,6 +2,7 @@ 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|
@@ -13,7 +14,11 @@ module ATS
       protected
 
       def api
-        self.class.constant_name.new(profile: options['profile'])
+        self.class.constant_name.new(
+          profile: options['profile'],
+          configuration: ATS.configuration,
+          debug: options['debug'],
+        )
       end
 
       def print_json(json)
diff --git a/lib/ats/http_api.rb b/lib/ats/http_api.rb
index 82e2d22..662a4cd 100644
--- a/lib/ats/http_api.rb
+++ b/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
 
diff --git a/lib/ats/threat_grid/api.rb b/lib/ats/threat_grid/api.rb
index cb982b9..bef323c 100644
--- a/lib/ats/threat_grid/api.rb
+++ b/lib/ats/threat_grid/api.rb
@@ -10,12 +10,8 @@ module ATS
 
       attr_reader :http, :profile, :configuration
 
-      def initialize(
-        api: HttpAPI.new(headers: HEADERS),
-        profile: :default,
-        configuration: ATS.configuration
-      )
-        @http = api
+      def initialize(profile:, configuration:, debug: false)
+        @http = HttpAPI.new(headers: HEADERS, debug: debug)
         @profile = profile.to_sym
         @configuration = configuration
       end