ats 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1f2fe2728729b93415b0a3207e83702f05b99d6bdd4171881cd92b3c7e135549
-  data.tar.gz: a078e537c475aca4ecb294278f862aba09964e5067793e84b4747b057f1e0666
+  metadata.gz: 7eb27ec01faf12a9d4ccf0ca36588f91b72305ce29c6dc6034fe9089f546bc53
+  data.tar.gz: 941b422bf5a5e71ca0004dc6ad9b2d516a54b83db255fcb24e292844fe130322
 SHA512:
-  metadata.gz: 36443b75271928b0cc6adc098a064b39b554b4d53b677b91b7569e7330ef010ecfacf8e641e8234eb5e59eac97c4ffbf0592ce2fd526faccf01462b18b58ac75
-  data.tar.gz: 3f0657e29f705f3076acef287c6984edf1a11d7a9618bc4b82feb9cd3a9ec6c15a4e38984aad34aba0029e89b0af0bde4aae40b7fa117baa2e4e052c0cd07889
+  metadata.gz: 165b94fb99a63af982466640a85abddcef4e32a2a23369e88371e09f19f8be124249b7891e542228f408ed292b09bf3acd11f3614f72aa3bf5695228a9e958da
+  data.tar.gz: 78d3cfecdebf5b0e0ec64aac27319c2670e3fb98d46f6b6e4ed770d64b6e67d0dfa957e22988c0a1e918e670ac918af42fe288b81d981bf0a00b7ddbdbd1c416

data/ats-cli.gemspec

@@ -22,6 +22,7 @@ Gem::Specification.new do |spec|
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
+  spec.required_ruby_version = '>= 2.5.0'
 
   spec.add_dependency 'thor', '~> 0.20'
   spec.add_development_dependency "bundler", "~> 1.16"

data/lib/ats.rb

@@ -18,6 +18,8 @@ require 'ats/threat_grid/organizations'
 require 'ats/threat_grid/samples'
 require 'ats/threat_grid/search'
 require 'ats/threat_grid/users'
+require 'ats/shiro/api'
+require 'ats/shiro/tokens'
 
 module ATS
   class << self

data/lib/ats/cli.rb

@@ -13,6 +13,9 @@ require 'ats/cli/amp4e/events'
 require 'ats/cli/amp4e/groups'
 require 'ats/cli/amp4e/policies'
 require 'ats/cli/amp4e/application'
+require 'ats/cli/shiro/tokens'
+require 'ats/cli/shiro/versions'
+require 'ats/cli/shiro/application'
 
 module ATS
   module CLI
@@ -25,6 +28,9 @@ module ATS
       desc 'amp4e SUBCOMMAND ...ARGS', 'interact with the AMP for Endpoints API'
       subcommand 'amp4e', AMP4E::Application
 
+      desc 'shiro SUBCOMMAND ...ARGS', 'interact with the AMP for Endpoints API'
+      subcommand 'shiro', Shiro::Application
+
       desc 'setup', 'setup'
       subcommand :setup, ATS::CLI::Setup
 

data/lib/ats/cli/setup.rb

@@ -5,6 +5,10 @@ module ATS
 
       desc 'setup', 'Initialize the .atsrc file.'
       def setup
+        say "Default Configuration:", :green
+        yaml = YAML.dump(default_settings)
+        say yaml, :yellow
+
         say "Current Configuration:", :green
         say JSON.pretty_generate(configuration.to_h), :green
 
@@ -18,9 +22,8 @@ module ATS
         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)
+        File.chmod(0600, new_file)
       end
 
       private
@@ -41,6 +44,12 @@ module ATS
               port: 443,
               scheme: 'https',
             },
+            shiro: {
+              bearer_token: '',
+              host: 'auth.amp.cisco.com',
+              port: 443,
+              scheme: 'https',
+            },
           }
         }
       end

data/lib/ats/cli/shiro/application.rb

@@ -0,0 +1,13 @@
+module ATS
+  module CLI
+    module Shiro
+      class Application < Thor
+        desc 'tokens SUBCOMMAND ...ARGS', 'interact with the Auth API'
+        subcommand :tokens, ATS::CLI::Shiro::Tokens
+
+        desc 'version SUBCOMMAND ...ARGS', 'interact with the Auth API'
+        subcommand :version, ATS::CLI::Shiro::Versions
+      end
+    end
+  end
+end

data/lib/ats/cli/shiro/tokens.rb

@@ -0,0 +1,12 @@
+module ATS
+  module CLI
+    module Shiro
+      class Tokens < Command
+        desc 'introspect <TOKEN>', 'introspection endpoint'
+        def introspect(token)
+          print_json api.tokens.introspect(token)
+        end
+      end
+    end
+  end
+end

data/lib/ats/cli/shiro/versions.rb

@@ -0,0 +1,14 @@
+module ATS
+  module CLI
+    module Shiro
+      class Versions < Command
+        default_command :show
+
+        desc 'show', ''
+        def show
+          print_json api.version
+        end
+      end
+    end
+  end
+end

data/lib/ats/http_api.rb

@@ -1,3 +1,5 @@
+require 'openssl'
+
 module ATS
   class HttpAPI
     def initialize(headers: {}, debug: false)
@@ -45,7 +47,10 @@ 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) if @debug
+      if debug?
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        http.set_debug_output(ATS.logger)
+      end
       http
     end
 
@@ -66,5 +71,9 @@ module ATS
         x.body = JSON.generate(body)
       end
     end
+
+    def debug?
+      @debug
+    end
   end
 end

data/lib/ats/shiro/api.rb

@@ -0,0 +1,52 @@
+module ATS
+  module Shiro
+    class API
+      HEADERS = {
+        'Content-Type' => 'application/json',
+        'Accept' => 'application/json',
+        'User-Agent' => "RubyGems/ATS #{ATS::VERSION}",
+      }.freeze
+
+      attr_reader :http, :port, :host, :scheme, :bearer_token
+
+      def initialize(configuration:, debug: false)
+        @http = HttpAPI.new(headers: HEADERS, debug: debug)
+        @configuration = configuration
+        @port = configuration[:port]
+        @scheme = configuration[:scheme]
+        @host = configuration[:host]
+        @bearer_token = configuration[:bearer_token]
+      end
+
+      def tokens
+        ATS::Shiro::Tokens.new(self)
+      end
+
+      def version
+        get("version")
+      end
+
+      def get(url, params: {}, version: 1)
+        http.get(build_uri(url, version: version), headers: headers, body: params) do |request, response|
+          JSON.parse(response.body, symbolize_names: true)
+        end
+      end
+
+      def post(url, params: {}, version: 1)
+        http.post(build_uri(url, version: version), headers: headers, body: params) do |request, response|
+          JSON.parse(response.body, symbolize_names: true)
+        end
+      end
+
+      private
+
+      def build_uri(relative_url, version:)
+        URI::Generic.build(host: host, port: port, scheme: scheme, path: "/api/v#{version}/#{relative_url}")
+      end
+
+      def headers
+        { AUTHORIZATION: "Bearer #{bearer_token}" }
+      end
+    end
+  end
+end

data/lib/ats/shiro/tokens.rb

@@ -0,0 +1,15 @@
+module ATS
+  module Shiro
+    class Tokens
+      attr_reader :api
+
+      def initialize(api)
+        @api = api
+      end
+
+      def introspect(token)
+        api.post("tokens/introspect", params: { token: token })
+      end
+    end
+  end
+end

data/lib/ats/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ats
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - mokha
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-03-19 00:00:00.000000000 Z
+date: 2018-03-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -94,6 +94,9 @@ files:
 - lib/ats/cli/amp4e/policies.rb
 - lib/ats/cli/command.rb
 - lib/ats/cli/setup.rb
+- lib/ats/cli/shiro/application.rb
+- lib/ats/cli/shiro/tokens.rb
+- lib/ats/cli/shiro/versions.rb
 - lib/ats/cli/threat_grid/application.rb
 - lib/ats/cli/threat_grid/organizations.rb
 - lib/ats/cli/threat_grid/samples.rb
@@ -102,6 +105,8 @@ files:
 - lib/ats/cli/threat_grid/whoami.rb
 - lib/ats/configuration.rb
 - lib/ats/http_api.rb
+- lib/ats/shiro/api.rb
+- lib/ats/shiro/tokens.rb
 - lib/ats/threat_grid/api.rb
 - lib/ats/threat_grid/organizations.rb
 - lib/ats/threat_grid/samples.rb
@@ -120,7 +125,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.5.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="