add amp4e events command.

6 files changed, 38 insertions, 0 deletions

diff --git a/lib/ats.rb b/lib/ats.rb
index ccd264f..d0257a5 100644
--- a/lib/ats.rb
+++ b/lib/ats.rb
@@ -10,6 +10,7 @@ require 'ats/version'
 
 require 'ats/amp4e/api'
 require 'ats/amp4e/computers'
+require 'ats/amp4e/events'
 require 'ats/threat_grid/api'
 require 'ats/threat_grid/organizations'
 require 'ats/threat_grid/samples'
diff --git a/lib/ats/amp4e/api.rb b/lib/ats/amp4e/api.rb
index 202df1a..f1ad63e 100644
--- a/lib/ats/amp4e/api.rb
+++ b/lib/ats/amp4e/api.rb
@@ -23,6 +23,10 @@ module ATS
         ATS::AMP4E::Computers.new(self)
       end
 
+      def events
+        ATS::AMP4E::Events.new(self)
+      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)
diff --git a/lib/ats/amp4e/events.rb b/lib/ats/amp4e/events.rb
new file mode 100644
index 0000000..4e06e21
--- /dev/null
+++ b/lib/ats/amp4e/events.rb
@@ -0,0 +1,15 @@
+module ATS
+  module AMP4E
+    class Events
+      attr_reader :api
+
+      def initialize(api)
+        @api = api
+      end
+
+      def list
+        api.get("events")
+      end
+    end
+  end
+end
diff --git a/lib/ats/cli.rb b/lib/ats/cli.rb
index 682bc50..3abf11c 100644
--- a/lib/ats/cli.rb
+++ b/lib/ats/cli.rb
@@ -4,6 +4,7 @@ require 'thor'
 require 'ats/cli/threat_grid'
 require 'ats/cli/amp4e/command'
 require 'ats/cli/amp4e/computers'
+require 'ats/cli/amp4e/events'
 require 'ats/cli/amp4e/application'
 
 module ATS
diff --git a/lib/ats/cli/amp4e/application.rb b/lib/ats/cli/amp4e/application.rb
index 34e5693..e1ed61a 100644
--- a/lib/ats/cli/amp4e/application.rb
+++ b/lib/ats/cli/amp4e/application.rb
@@ -6,6 +6,9 @@ module ATS
 
         desc 'computers SUBCOMMAND ...ARGS', 'interact with the Threat Grid API'
         subcommand :computers, ATS::CLI::AMP4E::Computers
+
+        desc 'events SUBCOMMAND ...ARGS', 'interact with the Threat Grid API'
+        subcommand :events, ATS::CLI::AMP4E::Events
       end
     end
   end
diff --git a/lib/ats/cli/amp4e/events.rb b/lib/ats/cli/amp4e/events.rb
new file mode 100644
index 0000000..aed9d85
--- /dev/null
+++ b/lib/ats/cli/amp4e/events.rb
@@ -0,0 +1,14 @@
+module ATS
+  module CLI
+    module AMP4E
+      class Events < Command
+        class_option :profile, default: :default, required: false
+
+        desc 'list', 'list'
+        def list
+          print_json api.events.list
+        end
+      end
+    end
+  end
+end