sysdig 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e3790850725c9f17424b02c5a0338ab5a9b8b757
-  data.tar.gz: b2aded05c614ab191c4ce75632550e8550f6d172
+  metadata.gz: cec8df61179b00aff072de7f58a82be69c8c3d54
+  data.tar.gz: 8b2580b9f30040ce0802abbe7efcc4995c6221a7
 SHA512:
-  metadata.gz: c95da8e9830383b4b3571724226f044f9e87827d455c2eece59a0ff31430de3bafad91647a8b4435320cbaf5cad7433e1b1ce8d08f856c0db7641783e5ad0bb0
-  data.tar.gz: 1ea57891abfef711cc68979bc9061521e54afaec4d943984bef3ed6a93dec08edf1da6dc8b2fbddf279c80b18479d9957471b5873fe645c5f00dd8c44dd463ed
+  metadata.gz: f1d69d7e24782bea00e538acb84e8c1cfca952d23f833f67542712d295f49356e0c4286b20f80027122d5b52130e7438727e50bc6d625c19c4e25b7de5f3dce9
+  data.tar.gz: 0c6d24e5e651494f2837f4cda6b151225290d40182bab38aadd30854a6a79f230c994aca428ebb79f1ba9f510d9eb354dc497bffbfb14f4c15ff1d11698db5ac

data/.gitignore

@@ -8,3 +8,4 @@
 /spec/reports/
 /tmp/
 *.gem
+spec/examples.txt

data/Gemfile

@@ -1,3 +1,8 @@
 source 'https://rubygems.org'
 
 gemspec
+
+group :test do
+  gem 'pry-nav', '~> 0.2'
+  gem 'faker'
+end

data/lib/sysdig.rb

@@ -11,8 +11,7 @@ require "ey/logger/faraday"
 class Sysdig
   include Cistern::Client
 
-  recognizes :url, :adapter, :logger
-  requires :username, :password
+  recognizes :url, :adapter, :logger, :token, :username, :password
 
 end
 
@@ -27,11 +26,20 @@ require 'sysdig/login'
 require 'sysdig/alert'
 require 'sysdig/alerts'
 require 'sysdig/create_alert'
+require 'sysdig/destroy_alert'
 require 'sysdig/get_alert'
 require 'sysdig/get_alerts'
 require 'sysdig/update_alert'
 
-require 'sysdig/get_notification'
-require 'sysdig/get_notifications'
-require 'sysdig/notification'
-require 'sysdig/notifications'
+require 'sysdig/get_alert_notification'
+require 'sysdig/get_alert_notifications'
+require 'sysdig/alert_notification'
+require 'sysdig/alert_notifications'
+
+require 'sysdig/get_user_notifications'
+require 'sysdig/update_user_notifications'
+require 'sysdig/user_notification'
+require 'sysdig/user_notifications'
+require 'sysdig/sns_notification'
+require 'sysdig/email_notification'
+require 'sysdig/pager_duty_notification'

data/lib/sysdig/alert.rb

@@ -19,12 +19,12 @@ class Sysdig::Alert < Sysdig::Model
   attribute :description
   attribute :enabled,            type:  :boolean,     default: true
   attribute :filter,             parser: lambda { |v, _| load_filter(v) }
-  attribute :group_aggregations, type:  :array,       alias:  "groupAggregations"
-  attribute :group_by,           type:  :array,       alias:  "groupBy"
+  attribute :group_aggregations, type:  :array,       alias: "groupAggregations"
+  attribute :group_by,           type:  :array,       alias: "groupBy"
   attribute :group_condition,    alias: "groupCondition"
   attribute :modified_at,        alias: "modifiedOn", parser: method(:epoch_time)
   attribute :name
-  attribute :notification_count, type:  :integer,     alias:  "notificationCount"
+  attribute :notification_count, type:  :integer,     alias: "notificationCount"
   attribute :notify,             parser: lambda { |v, _| Array(v).map { |x| x.upcase } }
   attribute :severity,           type:  :integer
   attribute :segment_by,         alias: "segmentBy"
@@ -34,17 +34,28 @@ class Sysdig::Alert < Sysdig::Model
   attribute :type,               parser: lambda { |v, _| v.to_s.upcase }
   attribute :version,            type:  :integer
 
+  def destroy
+    requires :identity
+
+    service.destroy_alert(identity)
+  end
+
   def save
     params = {
-      "name"        => self.name,
-      "description" => self.description,
-      "enabled"     => self.enabled,
-      "filter"      => self.class.dump_filter(filter || {}),
-      "type"        => self.type,
-      "condition"   => self.condition,
-      "timespan"    => self.timespan,
-      "severity"    => self.severity,
-      "notify"      => self.notify,
+      "condition"         => self.condition,
+      "description"       => self.description,
+      "enabled"           => self.enabled,
+      "filter"            => self.class.dump_filter(filter || {}),
+      "groupAggregations" => self.group_aggregations,
+      "groupBy"           => self.group_by,
+      "groupCondition"    => self.group_condition,
+      "name"              => self.name,
+      "notify"            => self.notify,
+      "segmentBy"         => self.segment_by,
+      "segmentCondition"  => self.segment_condition,
+      "severity"          => self.severity,
+      "timespan"          => self.timespan,
+      "type"              => self.type,
     }
 
     data = (

data/lib/sysdig/{notification.rb → alert_notification.rb}

@@ -1,4 +1,4 @@
-class Sysdig::Notification < Sysdig::Model
+class Sysdig::AlertNotification < Sysdig::Model
 
   identity :id, type: :integer
 

data/lib/sysdig/{notifications.rb → alert_notifications.rb}

@@ -1,6 +1,6 @@
-class Sysdig::Notifications < Sysdig::Collection
+class Sysdig::AlertNotifications < Sysdig::Collection
 
-  model Sysdig::Notification
+  model Sysdig::AlertNotification
 
   def all(options={})
     to   = options[:to]   || Time.now

data/lib/sysdig/create_alert.rb

@@ -1,4 +1,8 @@
-class Sysdig::CreateAlerts < Sysdig::Request
+class Sysdig::CreateAlert < Sysdig::Request
+  def self.params
+    %w[name description enabled filter type condition timespan severity notify segmentBy segmentCondition groupCondition groupBy groupAggregations]
+  end
+
   def real(alert)
     service.request(
       :method => :post,
@@ -6,4 +10,16 @@ class Sysdig::CreateAlerts < Sysdig::Request
       :body   => { "alert" => alert },
     )
   end
+
+  def mock(alert)
+    alert_id = service.serial_id
+    body     = Cistern::Hash.slice(Cistern::Hash.stringify_keys(alert), *self.class.params)
+
+    service.data[:alerts][alert_id] = body.merge!("id" => alert_id)
+
+    service.response(
+      :status => 201,
+      :body   => {"alert" => body},
+    )
+  end
 end

data/lib/sysdig/destroy_alert.rb

@@ -0,0 +1,17 @@
+class Sysdig::DestroyAlert < Sysdig::Request
+  def real(identity)
+    service.request(
+      :method => :delete,
+      :path   => File.join("/api/alerts", identity.to_s)
+    )
+  end
+
+  def mock(identity)
+    service.data[:alerts].fetch(identity)
+    service.data[:alerts].delete(identity)
+
+    service.response(
+      :status => 204,
+    )
+  end
+end

data/lib/sysdig/email_notification.rb

@@ -0,0 +1,15 @@
+class Sysdig::EmailNotification < Sysdig::UserNotification
+
+  type :email
+
+  attribute :recipients, type: :array
+
+  def save
+    notification = service.update_user_notifications("email" => {
+      "enabled"    => self.enabled,
+      "recipients" => self.receipents,
+    }).body.fetch("userNotification")
+
+    merge_attributes(notification.fetch("email"))
+  end
+end

data/lib/sysdig/{get_notification.rb → get_alert_notification.rb}

@@ -1,4 +1,4 @@
-class Sysdig::GetNotification < Sysdig::Request
+class Sysdig::GetAlertNotification < Sysdig::Request
   def real(notification_id)
     service.request(
       :path => "/api/notifications/#{notification_id}",

data/lib/sysdig/{get_notifications.rb → get_alert_notifications.rb}

@@ -1,4 +1,4 @@
-class Sysdig::GetNotifications < Sysdig::Request
+class Sysdig::GetAlertNotifications < Sysdig::Request
   def real(from, to)
     from_i = from.to_i * 1_000_000
     to_i   = to.to_i   * 1_000_000

data/lib/sysdig/get_alerts.rb

@@ -1,8 +1,27 @@
 class Sysdig::GetAlerts < Sysdig::Request
-  def real(params={})
+  def real(from: nil, to: nil)
+    params = {}
+
+    if from
+      params.merge!(from.to_i * 1_000_000)
+    end
+
+    if to
+      params.merge!(to.to_i * 1_000_000)
+    end
+
     service.request(
       :method => :get,
       :path   => "/api/alerts",
+      :params => params,
+    )
+  end
+
+  def mock(from: nil, to: nil)
+    alerts = service.data[:alerts].values
+
+    service.response(
+      :body => { "alerts" => alerts },
     )
   end
 end

data/lib/sysdig/get_user_notifications.rb

@@ -0,0 +1,14 @@
+class Sysdig::GetUserNotifications < Sysdig::Request
+  def real
+    service.request(
+      :method => :get,
+      :path   => "/api/settings/notifications",
+    )
+  end
+
+  def mock
+    service.response(
+      :body => { "userNotification" => service.data[:user_notifications] }
+    )
+  end
+end

data/lib/sysdig/mock.rb

@@ -1,8 +1,30 @@
 class Sysdig::Mock
+  def self.reset!
+    super
+
+    @id = 0
+  end
+
   def self.data
     @@data ||= Hash.new { |h,url|
       h[url] = {
         :alerts => {},
+        :user_notifications => {
+          "email" => {
+            "enabled"    => false,
+            "recipients" => []
+          },
+          "sns" => {
+            "enabled" => false,
+            "topics"  => []
+          },
+          "pagerDuty" => {
+            "enabled"     => false,
+            "integrated"  => false,
+            "resolveOnOk" => false,
+            "connectUrl"  => "https://connect.pagerduty.com/connect?vendor=x&callback=https://app.sysdigcloud.com/api/pagerDuty/callback/y/z"
+          }
+        }
       }
     }
   end
@@ -44,4 +66,9 @@ class Sysdig::Mock
   def data
     self.class.data[self.url]
   end
+
+  def serial_id
+    @id ||= 0
+    @id += 1
+  end
 end

data/lib/sysdig/pager_duty_notification.rb

@@ -0,0 +1,20 @@
+class Sysdig::PagerDutyNotification < Sysdig::UserNotification
+
+  type :pagerDuty
+
+  attribute :enabled,         type:  :boolean
+  attribute :integrated,      type:  :boolean
+  attribute :bind_resolution, type:  :boolean, alias: "resolveOnOk"
+  attribute :connect_url,     alias: "connectUrl"
+
+  def save
+    notification = service.update_user_notifications("pagerDuty" => {
+      "enabled"     => self.enabled,
+      "integrated"  => self.integrated,
+      "resolveOnOk" => self.bind_resolution,
+      "connectUrl"  => self.connect_url,
+    }).body.fetch("userNotification")
+
+    merge_attributes(notification.fetch("pagerDuty"))
+  end
+end

data/lib/sysdig/real.rb

@@ -1,5 +1,6 @@
 class Sysdig::Real
-  attr_reader :url, :path, :connection, :parser, :logger, :adapter, :authentication
+
+  attr_reader :url, :path, :connection, :parser, :logger, :adapter, :authentication, :token
 
   def initialize(options={})
     @url = URI.parse(options[:url] || "https://app.sysdigcloud.com")
@@ -8,6 +9,15 @@ class Sysdig::Real
     connection_options = options[:connection_options] || {}
     logger, @logger    = options[:logger]             || Logger.new(nil)
 
+    @username, @password = *options.values_at(:username, :password)
+
+    unless @username && @password
+      token = @token = options.fetch(:token)
+    end
+
+    @authentication = Mutex.new
+    @authenticated  = false
+
     @connection = Faraday.new({url: @url}.merge(connection_options)) do |builder|
       # response
       builder.response :json
@@ -20,22 +30,22 @@ class Sysdig::Real
         :backoff_factor      => 2
       builder.request :multipart
       builder.request :json
-      builder.use :cookie_jar
+
+      if token
+        builder.authorization :Bearer, token
+      else
+        builder.use :cookie_jar
+      end
 
       builder.use Faraday::Response::RaiseError
       builder.use Ey::Logger::Faraday, format: :machine, device: logger
 
       builder.adapter(*adapter)
     end
-
-    @username, @password = *options.values_at(:username, :password)
-
-    @authentication = Mutex.new
-    @authenticated  = false
   end
 
   def request_with_authentication(options={})
-    if !@authenticated
+    if !@authenticated && token.nil?
       @authentication.synchronize {
         if !@authenticated
           login(@username, @password)

data/lib/sysdig/sns_notification.rb

@@ -0,0 +1,15 @@
+class Sysdig::SnsNotification < Sysdig::UserNotification
+
+  type :sns
+
+  attribute :topics, type: :array
+
+  def save
+    notification = service.update_user_notifications(self.identity => {
+      "enabled" => self.enabled,
+      "topics"  => self.topics,
+    }).body.fetch("userNotification")
+
+    merge_attributes(notification.fetch(self.identity))
+  end
+end

data/lib/sysdig/update_alert.rb

@@ -1,7 +1,6 @@
 class Sysdig::UpdateAlert < Sysdig::Request
+  # @note alert[version] might mean lock version which should be fun
   def real(alert_id, alert)
-    #{"alert":{"version":7,"createdOn":1438965688000,"modifiedOn":1442349562000,"name":"Memory usage critical","description":"Memory utilization is currently >= 98%","enabled":false,"filter":null,"type":"MANUAL","condition":"timeAvg(memory.used.percent) >= 98","timespan":60000000,"severity":2,"notify":["SNS"],"segmentBy":["agent.tag.id"],"segmentCondition":{"type":"ANY"}}}
-    # @todo alert[version] might mean lock version which should be fun
     service.request(
       :method => :put,
       :path   => File.join("/api/alerts", alert_id.to_s),

data/lib/sysdig/update_user_notifications.rb

@@ -0,0 +1,31 @@
+class Sysdig::UpdateUserNotifications < Sysdig::Request
+  def real(notification)
+    service.request(
+      :method => :put,
+      :path   => "/api/settings/notifications",
+      :body   => { "userNotification" => notification },
+    )
+  end
+
+  def mock(notification)
+    user_notification = Cistern::Hash.slice(Cistern::Hash.stringify_keys(notification), "sns", "email", "pagerDuty")
+
+    schema = {
+      "sns"       => %w[enabled topics],
+      "email"     => %w[enabled recipients],
+      "pagerDuty" => %w[enabled integrated resolveOnOk connectUrl],
+    }
+
+    sliced = schema.each_with_object({}) { |(type, keys), r|
+      u = user_notification[type]
+
+      if u
+        r.merge!(type => Cistern::Hash.slice(u, *keys))
+      end
+    }
+
+    service.response(
+      :body => { "userNotification" => service.data[:user_notifications].merge!(sliced) },
+    )
+  end
+end

data/lib/sysdig/user_notification.rb

@@ -0,0 +1,31 @@
+class Sysdig::UserNotification < Sysdig::Model
+
+  def self.types
+    @_types ||= Hash.new
+  end
+
+  def self.type(k=nil)
+    if k.nil?
+      @key
+    else
+      key = k.to_s
+      Sysdig::UserNotification.types[key] = self
+      @key = key
+    end
+  end
+
+  def self.inherited(klass)
+    klass.identity :category
+
+    super
+  end
+
+  attribute :enabled, type: :boolean
+
+  def initialize(attributes={})
+    self.identity = self.class.type
+
+    super
+  end
+
+end

data/lib/sysdig/user_notifications.rb

@@ -0,0 +1,31 @@
+class Sysdig::UserNotifications < Sysdig::Collection
+
+  model Sysdig::UserNotification
+
+  def all(options={})
+    load(service.get_user_notifications.body.fetch("userNotification")).tap { |_| self.records.compact! }
+  end
+
+  def get(type)
+    all.find { |r| r.identity == type.to_s }
+  end
+
+  def new(args)
+    type, attributes = *args
+
+    unless attributes.is_a?(::Hash)
+      raise(ArgumentError.new("Initialization parameters must be an attributes hash, got #{attributes.class} #{attributes.inspect}"))
+    end
+
+    concrete_model = model.types[type]
+
+    if concrete_model
+      concrete_model.new(
+        {
+          :collection => self,
+          :service    => service,
+        }.merge(attributes)
+      )
+    end
+  end
+end

data/lib/sysdig/version.rb

@@ -1,3 +1,3 @@
 class Sysdig
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sysdig
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Josh Lane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-10-20 00:00:00.000000000 Z
+date: 2015-10-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -143,20 +143,28 @@ files:
 - bin/setup
 - lib/sysdig.rb
 - lib/sysdig/alert.rb
+- lib/sysdig/alert_notification.rb
+- lib/sysdig/alert_notifications.rb
 - lib/sysdig/alerts.rb
 - lib/sysdig/create_alert.rb
+- lib/sysdig/destroy_alert.rb
+- lib/sysdig/email_notification.rb
 - lib/sysdig/get_alert.rb
+- lib/sysdig/get_alert_notification.rb
+- lib/sysdig/get_alert_notifications.rb
 - lib/sysdig/get_alerts.rb
-- lib/sysdig/get_notification.rb
-- lib/sysdig/get_notifications.rb
+- lib/sysdig/get_user_notifications.rb
 - lib/sysdig/login.rb
 - lib/sysdig/mock.rb
 - lib/sysdig/model.rb
-- lib/sysdig/notification.rb
-- lib/sysdig/notifications.rb
+- lib/sysdig/pager_duty_notification.rb
 - lib/sysdig/real.rb
 - lib/sysdig/response.rb
+- lib/sysdig/sns_notification.rb
 - lib/sysdig/update_alert.rb
+- lib/sysdig/update_user_notifications.rb
+- lib/sysdig/user_notification.rb
+- lib/sysdig/user_notifications.rb
 - lib/sysdig/version.rb
 - sysdig.gemspec
 homepage: https://github.com/engineyard/sysdig-client