upguard 0.0.9 → 0.0.18

data/lib/upguard/Environment.rb

@@ -10,6 +10,20 @@ module UpGuard
     attr_accessor :updated_at
     attr_accessor :created_at
     attr_accessor :weight
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.description = nil
+      self.id = nil
+      self.name = nil
+      self.node_rules = nil
+      self.short_description = nil
+      self.updated_by = nil
+      self.created_by = nil
+      self.updated_at = nil
+      self.created_at = nil
+      self.weight = nil
+    end
+
     def from_hash(h)
       self.description = h['description'] if h.include?('description')
       self.id = h['id'] if h.include?('id')

data/lib/upguard/EnvironmentList.rb

@@ -15,15 +15,17 @@ module UpGuard
       @inner_list = []
     end
     
-    def count
-      return @inner_list.count
-    end
-    
+
     def [](idx)
       return @inner_list[idx]
     end
     
 
+    def count
+      return @inner_list.count
+    end
+    
+
     
     def each(&block)
       @inner_list.each(&block)
@@ -32,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["Environment", "UpGuard::Environment"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Environment' to 'EnvironmentList'"
+      end
     end
     
 

data/lib/upguard/Event.rb

@@ -4,11 +4,22 @@ module UpGuard
     attr_accessor :type_id
     attr_accessor :environment_id
     attr_accessor :created_at
+    attr_accessor :variables
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.id = nil
+      self.type_id = nil
+      self.environment_id = nil
+      self.created_at = nil
+      self.variables = EventVariables.new(appliance_url, appliance_api_key, sec_key, insecure)
+    end
+
     def from_hash(h)
       self.id = h['id'] if h.include?('id')
       self.type_id = h['type_id'] if h.include?('type_id')
       self.environment_id = h['environment_id'] if h.include?('environment_id')
       self.created_at = h['created_at'] if h.include?('created_at')
+      self.variables = h['variables'] if h.include?('variables')
     end
     def to_hash
       h = {}
@@ -16,6 +27,7 @@ module UpGuard
       h['type_id'] = self.type_id
       h['environment_id'] = self.environment_id
       h['created_at'] = self.created_at
+      h['variables'] = self.variables
       return h
     end
     def to_json(options = nil)

data/lib/upguard/EventAction.rb

@@ -6,6 +6,16 @@ module UpGuard
     attr_accessor :type
     attr_accessor :variables
     attr_accessor :view
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.id = nil
+      self.name = nil
+      self.status = nil
+      self.type = nil
+      self.variables = nil
+      self.view = nil
+    end
+
     def from_hash(h)
       self.id = h['id'] if h.include?('id')
       self.name = h['name'] if h.include?('name')

data/lib/upguard/EventActionList.rb

@@ -15,15 +15,17 @@ module UpGuard
       @inner_list = []
     end
     
-    def count
-      return @inner_list.count
-    end
-    
+
     def [](idx)
       return @inner_list[idx]
     end
     
 
+    def count
+      return @inner_list.count
+    end
+    
+
     
     def each(&block)
       @inner_list.each(&block)
@@ -32,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["EventAction", "UpGuard::EventAction"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'EventAction' to 'EventActionList'"
+      end
     end
     
 

data/lib/upguard/EventList.rb

@@ -0,0 +1,53 @@
+module UpGuard
+  class EventList < BaseObject
+    def from_hash(h)
+    end
+    def to_hash
+      h = {}
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      @inner_list = []
+    end
+    
+
+    def [](idx)
+      return @inner_list[idx]
+    end
+    
+
+    def count
+      return @inner_list.count
+    end
+    
+
+    
+    def each(&block)
+      @inner_list.each(&block)
+    end
+    
+
+    
+    def <<(obj)
+      if ["Event", "UpGuard::Event"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Event' to 'EventList'"
+      end
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/EventVariables.rb

@@ -0,0 +1,43 @@
+module UpGuard
+  class EventVariables < BaseObject
+    def from_hash(h)
+    end
+    def to_hash
+      h = {}
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      @inner_map = {}
+    end
+    
+
+    def [](key)
+      return @inner_map[key]
+    end
+    
+
+    def []=(key, value)
+      @inner_map[key] = value
+    end
+    
+
+    
+    def to_json(options = nil)
+      return @inner_map.to_json(options)
+    end
+    
+
+    
+    def to_hash
+      return @inner_map
+    end
+    
+
+
+  end
+end

data/lib/upguard/ExternalEvent.rb

@@ -0,0 +1,36 @@
+module UpGuard
+  class ExternalEvent < BaseObject
+    attr_accessor :type
+    attr_accessor :variables
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.type = nil
+      self.variables = EventVariables.new(appliance_url, appliance_api_key, sec_key, insecure)
+    end
+
+    def from_hash(h)
+      self.type = h['type'] if h.include?('type')
+      self.variables = h['variables'] if h.include?('variables')
+    end
+    def to_hash
+      h = {}
+      h['type'] = self.type
+      h['variables'] = self.variables
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    
+    def create
+      h = {}
+      h["variables"] = self.variables.to_hash
+      h["variables"]["type"] = self.type
+      http_post("/api/v2/events.json", h)
+    end
+    
+
+
+  end
+end

data/lib/upguard/Incident.rb

@@ -6,6 +6,16 @@ module UpGuard
     attr_accessor :short_description
     attr_accessor :started_at
     attr_accessor :url
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.ended_at = nil
+      self.external_id = nil
+      self.id = nil
+      self.short_description = nil
+      self.started_at = nil
+      self.url = nil
+    end
+
     def from_hash(h)
       self.ended_at = h['ended_at'] if h.include?('ended_at')
       self.external_id = h['external_id'] if h.include?('external_id')

data/lib/upguard/IncidentList.rb

@@ -15,15 +15,17 @@ module UpGuard
       @inner_list = []
     end
     
-    def count
-      return @inner_list.count
-    end
-    
+
     def [](idx)
       return @inner_list[idx]
     end
     
 
+    def count
+      return @inner_list.count
+    end
+    
+
     
     def each(&block)
       @inner_list.each(&block)
@@ -32,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["Incident", "UpGuard::Incident"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Incident' to 'IncidentList'"
+      end
     end
     
 

data/lib/upguard/Job.rb

@@ -6,6 +6,16 @@ module UpGuard
     attr_accessor :source_name
     attr_accessor :source_type
     attr_accessor :status
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.id = nil
+      self.organisation_id = nil
+      self.source_id = nil
+      self.source_name = nil
+      self.source_type = nil
+      self.status = nil
+    end
+
     def from_hash(h)
       self.id = h['id'] if h.include?('id')
       self.organisation_id = h['organisation_id'] if h.include?('organisation_id')

data/lib/upguard/JobList.rb

@@ -15,15 +15,17 @@ module UpGuard
       @inner_list = []
     end
     
-    def count
-      return @inner_list.count
-    end
-    
+
     def [](idx)
       return @inner_list[idx]
     end
     
 
+    def count
+      return @inner_list.count
+    end
+    
+
     
     def each(&block)
       @inner_list.each(&block)
@@ -32,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["Job", "UpGuard::Job"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Job' to 'JobList'"
+      end
     end
     
 

data/lib/upguard/MediumType.rb

@@ -10,7 +10,10 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
+    AGENT = 1
     SSH = 3
+    WINRM = 7
+    DATABASE = 11
 
   end
 end

data/lib/upguard/Node.rb

@@ -1,5 +1,6 @@
 module UpGuard
   class Node < BaseObject
+    attr_accessor :connection_manager_group_id
     attr_accessor :environment_id
     attr_accessor :external_id
     attr_accessor :id
@@ -7,13 +8,42 @@ module UpGuard
     attr_accessor :last_scan_status
     attr_accessor :last_scan_status_string
     attr_accessor :mac_address
+    attr_accessor :medium_hostname
+    attr_accessor :medium_info
+    attr_accessor :medium_port
+    attr_accessor :medium_type
+    attr_accessor :medium_username
     attr_accessor :name
     attr_accessor :node_type
     attr_accessor :online
     attr_accessor :operating_system_family_id
     attr_accessor :operating_system_id
     attr_accessor :short_description
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.connection_manager_group_id = nil
+      self.environment_id = nil
+      self.external_id = nil
+      self.id = nil
+      self.ip_address = nil
+      self.last_scan_status = nil
+      self.last_scan_status_string = nil
+      self.mac_address = nil
+      self.medium_hostname = nil
+      self.medium_info = NodeMediumInfo.new(appliance_url, appliance_api_key, sec_key, insecure)
+      self.medium_port = nil
+      self.medium_type = nil
+      self.medium_username = nil
+      self.name = nil
+      self.node_type = nil
+      self.online = nil
+      self.operating_system_family_id = nil
+      self.operating_system_id = nil
+      self.short_description = nil
+    end
+
     def from_hash(h)
+      self.connection_manager_group_id = h['connection_manager_group_id'] if h.include?('connection_manager_group_id')
       self.environment_id = h['environment_id'] if h.include?('environment_id')
       self.external_id = h['external_id'] if h.include?('external_id')
       self.id = h['id'] if h.include?('id')
@@ -21,6 +51,11 @@ module UpGuard
       self.last_scan_status = h['last_scan_status'] if h.include?('last_scan_status')
       self.last_scan_status_string = h['last_scan_status_string'] if h.include?('last_scan_status_string')
       self.mac_address = h['mac_address'] if h.include?('mac_address')
+      self.medium_hostname = h['medium_hostname'] if h.include?('medium_hostname')
+      self.medium_info = h['medium_info'] if h.include?('medium_info')
+      self.medium_port = h['medium_port'] if h.include?('medium_port')
+      self.medium_type = h['medium_type'] if h.include?('medium_type')
+      self.medium_username = h['medium_username'] if h.include?('medium_username')
       self.name = h['name'] if h.include?('name')
       self.node_type = h['node_type'] if h.include?('node_type')
       self.online = h['online'] if h.include?('online')
@@ -30,6 +65,7 @@ module UpGuard
     end
     def to_hash
       h = {}
+      h['connection_manager_group_id'] = self.connection_manager_group_id
       h['environment_id'] = self.environment_id
       h['external_id'] = self.external_id
       h['id'] = self.id
@@ -37,6 +73,11 @@ module UpGuard
       h['last_scan_status'] = self.last_scan_status
       h['last_scan_status_string'] = self.last_scan_status_string
       h['mac_address'] = self.mac_address
+      h['medium_hostname'] = self.medium_hostname
+      h['medium_info'] = self.medium_info
+      h['medium_port'] = self.medium_port
+      h['medium_type'] = self.medium_type
+      h['medium_username'] = self.medium_username
       h['name'] = self.name
       h['node_type'] = self.node_type
       h['online'] = self.online
@@ -118,12 +159,29 @@ module UpGuard
       return elem
     end
 
-    def start_scan()
+    def start_scan
       url = "/api/v2/nodes/#{self.id}/start_scan.json"
       obj = http_post(url, nil)
       return obj
     end
 
+    def node_scans
+      url = "/api/v2/node_scans.json?node_id=#{self.id}&per_page=5000"
+      obj = http_get(url)
+      the_list = NodeScanList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = NodeScan.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = x["id"]
+        elem.node_id = x["node_id"]
+        elem.created_at = x["created_at"]
+        elem.updated_at = x["updated_at"]
+        elem.scan_options = x["scan_options"]
+        the_list << elem
+      end
+      return the_list
+    end
+    
+
     
     def node_groups
       obj = http_get("/api/v2/nodes/#{self.id}/node_groups.json")