upguard 0.0.4 → 0.0.16

data/lib/upguard/IncidentList.rb

@@ -0,0 +1,53 @@
+module UpGuard
+  class IncidentList < 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 ["Incident", "UpGuard::Incident"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Incident' to 'IncidentList'"
+      end
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/Job.rb

@@ -0,0 +1,70 @@
+module UpGuard
+  class Job < BaseObject
+    attr_accessor :id
+    attr_accessor :organisation_id
+    attr_accessor :source_id
+    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')
+      self.source_id = h['source_id'] if h.include?('source_id')
+      self.source_name = h['source_name'] if h.include?('source_name')
+      self.source_type = h['source_type'] if h.include?('source_type')
+      self.status = h['status'] if h.include?('status')
+    end
+    def to_hash
+      h = {}
+      h['id'] = self.id
+      h['organisation_id'] = self.organisation_id
+      h['source_id'] = self.source_id
+      h['source_name'] = self.source_name
+      h['source_type'] = self.source_type
+      h['status'] = self.status
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    
+    def tasks
+      obj = http_get("/api/v2/jobs/{job_id}/tasks.json")
+      the_list = TaskList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = Task.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.agent_name = x["agent_name"] if x.include?("agent_name")
+        elem.connection_manager_id = x["connection_manager_id"] if x.include?("connection_manager_id")
+        elem.created_at = x["created_at"] if x.include?("created_at")
+        elem.id = x["id"] if x.include?("id")
+        elem.job_id = x["job_id"] if x.include?("job_id")
+        elem.label = x["label"] if x.include?("label")
+        elem.log = x["log"] if x.include?("log")
+        elem.node_id = x["node_id"] if x.include?("node_id")
+        elem.node_session_id = x["node_session_id"] if x.include?("node_session_id")
+        elem.payload = x["payload"] if x.include?("payload")
+        elem.report = x["report"] if x.include?("report")
+        elem.sequence = x["sequence"] if x.include?("sequence")
+        elem.status = x["status"] if x.include?("status")
+        elem.step_description = x["step_description"] if x.include?("step_description")
+        elem.updated_at = x["updated_at"] if x.include?("updated_at")
+        elem.uuid = x["uuid"] if x.include?("uuid")
+        the_list << elem
+      end
+      return the_list
+    end
+
+
+  end
+end

data/lib/upguard/JobList.rb

@@ -0,0 +1,53 @@
+module UpGuard
+  class JobList < 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 ["Job", "UpGuard::Job"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Job' to 'JobList'"
+      end
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/MediumType.rb

@@ -6,11 +6,14 @@ module UpGuard
       h = {}
       return h
     end
-    def to_json
+    def to_json(options = nil)
       h = to_hash
-      return h.to_json
+      return h.to_json(options)
     end
+    AGENT = 1
     SSH = 3
+    WINRM = 7
+    DATABASE = 11
 
   end
 end

data/lib/upguard/Node.rb

@@ -3,45 +3,92 @@ module UpGuard
     attr_accessor :connection_manager_group_id
     attr_accessor :environment_id
     attr_accessor :external_id
-    attr_accessor :hostname
     attr_accessor :id
+    attr_accessor :ip_address
+    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.hostname = h['hostname'] if h.include?('hostname')
       self.id = h['id'] if h.include?('id')
+      self.ip_address = h['ip_address'] if h.include?('ip_address')
+      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')
       self.operating_system_family_id = h['operating_system_family_id'] if h.include?('operating_system_family_id')
       self.operating_system_id = h['operating_system_id'] if h.include?('operating_system_id')
+      self.short_description = h['short_description'] if h.include?('short_description')
     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['hostname'] = self.hostname
       h['id'] = self.id
+      h['ip_address'] = self.ip_address
+      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
       h['operating_system_family_id'] = self.operating_system_family_id
       h['operating_system_id'] = self.operating_system_id
+      h['short_description'] = self.short_description
       return h
     end
-    def to_json
+    def to_json(options = nil)
       h = to_hash
-      return h.to_json
+      return h.to_json(options)
     end
     
     def load
@@ -79,7 +126,7 @@ module UpGuard
     
     def connection_manager_group
       obj = http_get("/api/v2/connection_manager_groups/{connection_manager_group_id}.json")
-      elem = ConnectionManagerGroup.new(self.appliance_hostname, self.api_key, self.sec_key, self.insecure)
+      elem = ConnectionManagerGroup.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
         elem.id = obj["id"]
         elem.name = obj["name"]
       return elem
@@ -88,7 +135,7 @@ module UpGuard
     
     def environment
       obj = http_get("/api/v2/environments/#{self.environment_id}.json")
-      elem = Environment.new(self.appliance_hostname, self.api_key, self.sec_key, self.insecure)
+      elem = Environment.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
         elem.id = obj["id"]
         elem.name = obj["name"]
       return elem
@@ -97,7 +144,7 @@ module UpGuard
     
     def operating_system
       obj = http_get("/api/v2/operating_systems/#{self.operating_system_id}.json")
-      elem = OperatingSystem.new(self.appliance_hostname, self.api_key, self.sec_key, self.insecure)
+      elem = OperatingSystem.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
         elem.id = obj["id"]
         elem.name = obj["name"]
       return elem
@@ -106,7 +153,7 @@ module UpGuard
     
     def operating_system_family
       obj = http_get("/api/v2/operating_system_families/#{self.operating_system_family_id}.json")
-      elem = OperatingSystemFamily.new(self.appliance_hostname, self.api_key, self.sec_key, self.insecure)
+      elem = OperatingSystemFamily.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
         elem.id = obj["id"]
         elem.name = obj["name"]
       return elem
@@ -118,6 +165,36 @@ module UpGuard
       return obj
     end
 
+    
+    def node_groups
+      obj = http_get("/api/v2/nodes/#{self.id}/node_groups.json")
+      the_list = NodeGroupList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = NodeGroup.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = x["id"] if x.include?("id")
+        elem.name = x["name"] if x.include?("name")
+        elem.description = x["description"] if x.include?("description")
+        elem.node_rules = x["node_rules"] if x.include?("node_rules")
+        the_list << elem
+      end
+      return the_list
+    end
+
+    
+    def raw_files
+      obj = http_get("/api/v2/nodes/#{self.id}/raw_files.json")
+      the_list = RawFileList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = RawFile.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = x["id"] if x.include?("id")
+        elem.node_scan_id = x["node_scan_id"] if x.include?("node_scan_id")
+        elem.created_at = x["created_at"] if x.include?("created_at")
+        elem.updated_at = x["updated_at"] if x.include?("updated_at")
+        the_list << elem
+      end
+      return the_list
+    end
+
 
   end
 end

data/lib/upguard/NodeGroup.rb

@@ -1,20 +1,70 @@
 module UpGuard
   class NodeGroup < BaseObject
+    attr_accessor :created_at
+    attr_accessor :created_by
+    attr_accessor :description
+    attr_accessor :diff_notify
+    attr_accessor :external_id
     attr_accessor :id
     attr_accessor :name
+    attr_accessor :node_rules
+    attr_accessor :scan_options
+    attr_accessor :search_query
+    attr_accessor :status
+    attr_accessor :updated_at
+    attr_accessor :updated_by
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.created_at = nil
+      self.created_by = nil
+      self.description = nil
+      self.diff_notify = nil
+      self.external_id = nil
+      self.id = nil
+      self.name = nil
+      self.node_rules = nil
+      self.scan_options = nil
+      self.search_query = nil
+      self.status = nil
+      self.updated_at = nil
+      self.updated_by = nil
+    end
+
     def from_hash(h)
+      self.created_at = h['created_at'] if h.include?('created_at')
+      self.created_by = h['created_by'] if h.include?('created_by')
+      self.description = h['description'] if h.include?('description')
+      self.diff_notify = h['diff_notify'] if h.include?('diff_notify')
+      self.external_id = h['external_id'] if h.include?('external_id')
       self.id = h['id'] if h.include?('id')
       self.name = h['name'] if h.include?('name')
+      self.node_rules = h['node_rules'] if h.include?('node_rules')
+      self.scan_options = h['scan_options'] if h.include?('scan_options')
+      self.search_query = h['search_query'] if h.include?('search_query')
+      self.status = h['status'] if h.include?('status')
+      self.updated_at = h['updated_at'] if h.include?('updated_at')
+      self.updated_by = h['updated_by'] if h.include?('updated_by')
     end
     def to_hash
       h = {}
+      h['created_at'] = self.created_at
+      h['created_by'] = self.created_by
+      h['description'] = self.description
+      h['diff_notify'] = self.diff_notify
+      h['external_id'] = self.external_id
       h['id'] = self.id
       h['name'] = self.name
+      h['node_rules'] = self.node_rules
+      h['scan_options'] = self.scan_options
+      h['search_query'] = self.search_query
+      h['status'] = self.status
+      h['updated_at'] = self.updated_at
+      h['updated_by'] = self.updated_by
       return h
     end
-    def to_json
+    def to_json(options = nil)
       h = to_hash
-      return h.to_json
+      return h.to_json(options)
     end
     
     def load
@@ -51,14 +101,53 @@ module UpGuard
     
     def nodes
       obj = http_get("/api/v2/node_groups/#{self.id}/nodes.json")
-      list = []
+      the_list = NodeList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = Node.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = x["id"] if x.include?("id")
+        elem.name = x["name"] if x.include?("name")
+        the_list << elem
+      end
+      return the_list
+    end
+
+    
+    def policy_versions
+      obj = http_get("/api/v2/node_groups/#{self.id}/policy_versions.json")
+      the_list = PolicyVersionList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+      obj.each do |x|
+        elem = PolicyVersion.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = x["id"] if x.include?("id")
+        elem.name = x["name"] if x.include?("name")
+        elem.version = x["version"] if x.include?("version")
+        the_list << elem
+      end
+      return the_list
+    end
+
+    
+    def users
+      obj = http_get("/api/v2/node_groups/#{self.id}/users.json")
+      the_list = NodeGroupUserList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
       obj.each do |x|
-        elem = Node.new(self.appliance_hostname, self.api_key, self.sec_key, self.insecure)
-        elem.id = x["id"]
-        elem.name = x["name"]
-        list << elem
+        elem = NodeGroupUser.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.email = x["email"] if x.include?("email")
+        elem.user_id = x["user_id"] if x.include?("user_id")
+        the_list << elem
       end
-      return list
+      return the_list
+    end
+
+    def add_user(user_id)
+      url = "/api/v2/node_groups/#{self.id}/add_users.json?user_id=#{user_id}"
+      obj = http_post(url, nil)
+      return obj
+    end
+
+    def remove_user(user_id)
+      url = "/api/v2/node_groups/#{self.id}/remove_user.json?user_id=#{user_id}"
+      obj = http_put(url, nil)
+      return obj
     end