ats 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 226e424b41b47a9bed3a8e3e02daa4c8e6747d44cd9c3c0d25e9bbf0f7b977d3
-  data.tar.gz: d67d968d01a44e95a75d1c2541a29319c19d1d7b2d1b05c924c03479ec533c9c
+  metadata.gz: 1f2fe2728729b93415b0a3207e83702f05b99d6bdd4171881cd92b3c7e135549
+  data.tar.gz: a078e537c475aca4ecb294278f862aba09964e5067793e84b4747b057f1e0666
 SHA512:
-  metadata.gz: 6e8e7e9a31b025f787681b847fc1d41f08c634ba579facd07d3dd9e468dcaac90a5c0337a7afff9ebadfa7ede6f4d086177ea20400a9879aeb668835c023c840
-  data.tar.gz: 57eababf9c32f5749f6d43a237f7e04ecc3a235e442700449cb36aed4cfea50b4c7764aa1120d4f053fa6658fcdbe804b2858ac23fe111fe834a43e16c8d71bc
+  metadata.gz: 36443b75271928b0cc6adc098a064b39b554b4d53b677b91b7569e7330ef010ecfacf8e641e8234eb5e59eac97c4ffbf0592ce2fd526faccf01462b18b58ac75
+  data.tar.gz: 3f0657e29f705f3076acef287c6984edf1a11d7a9618bc4b82feb9cd3a9ec6c15a4e38984aad34aba0029e89b0af0bde4aae40b7fa117baa2e4e052c0cd07889

data/lib/ats/cli.rb

@@ -1,6 +1,7 @@
 require 'ats'
 require 'thor'
 require 'ats/cli/command'
+require 'ats/cli/setup'
 require 'ats/cli/threat_grid/organizations'
 require 'ats/cli/threat_grid/samples'
 require 'ats/cli/threat_grid/search'
@@ -24,46 +25,13 @@ module ATS
       desc 'amp4e SUBCOMMAND ...ARGS', 'interact with the AMP for Endpoints API'
       subcommand 'amp4e', AMP4E::Application
 
+      desc 'setup', 'setup'
+      subcommand :setup, ATS::CLI::Setup
+
       desc 'version', 'Display the current version'
       def version
         say ATS::VERSION
       end
-
-      desc 'setup', 'Initialize the .atsrc file.'
-      def setup(configuration = ATS.configuration)
-        say "Current Configuration:", :green
-        say JSON.pretty_generate(configuration.to_h), :green
-
-        configuration.config_files.each do |file|
-          if File.exist?(file)
-            say "Found #{file}. Nothing to do. Good bye!", :green
-            exit 0
-          end
-        end
-
-        say "Configuration file not found."
-        new_file = configuration.config_files.first
-        say "New file created at #{new_file}."
-        yaml = YAML.dump({
-          default: {
-            amp4e: {
-              client_id: '',
-              client_secret: '',
-              host: 'api.amp.cisco.com',
-              port: 443,
-              scheme: 'https',
-            },
-            threatgrid: {
-              api_key: '',
-              host: 'panacea.threatgrid.com',
-              port: 443,
-              scheme: 'https',
-            },
-          }
-        })
-        say yaml, :yellow
-        IO.write(new_file, yaml)
-      end
     end
   end
 end

data/lib/ats/cli/setup.rb

@@ -0,0 +1,49 @@
+module ATS
+  module CLI
+    class Setup < Command
+      default_command :setup
+
+      desc 'setup', 'Initialize the .atsrc file.'
+      def setup
+        say "Current Configuration:", :green
+        say JSON.pretty_generate(configuration.to_h), :green
+
+        configuration.config_files.each do |file|
+          if File.exist?(file)
+            say "Found #{file}. Nothing to do. Good bye!", :green
+            exit 0
+          end
+        end
+
+        say "Configuration file not found."
+        new_file = configuration.config_files.first
+        say "New file created at #{new_file}."
+        yaml = YAML.dump(default_settings)
+        say yaml, :yellow
+        IO.write(new_file, yaml)
+      end
+
+      private
+
+      def default_settings
+        {
+          default: {
+            amp4e: {
+              client_id: '',
+              client_secret: '',
+              host: 'api.amp.cisco.com',
+              port: 443,
+              scheme: 'https',
+            },
+            threatgrid: {
+              api_key: '',
+              host: 'panacea.threatgrid.com',
+              port: 443,
+              scheme: 'https',
+            },
+          }
+        }
+      end
+    end
+  end
+end

data/lib/ats/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ats
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - mokha
@@ -93,6 +93,7 @@ files:
 - lib/ats/cli/amp4e/groups.rb
 - lib/ats/cli/amp4e/policies.rb
 - lib/ats/cli/command.rb
+- lib/ats/cli/setup.rb
 - lib/ats/cli/threat_grid/application.rb
 - lib/ats/cli/threat_grid/organizations.rb
 - lib/ats/cli/threat_grid/samples.rb