upguard 0.0.9 → 0.0.18

data/lib/upguard/NodeGroup.rb

@@ -13,6 +13,23 @@ module UpGuard
     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')

data/lib/upguard/NodeGroupList.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 ["NodeGroup", "UpGuard::NodeGroup"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'NodeGroup' to 'NodeGroupList'"
+      end
     end
     
 
@@ -42,6 +48,17 @@ module UpGuard
     end
     
 
+    
+    def find_by_id(id)
+      @inner_list.each do |x|
+        if x.id.to_s == id.to_s
+          return x
+        end
+      end
+      return nil
+    end
+    
+
 
   end
 end

data/lib/upguard/NodeGroupUser.rb

@@ -2,6 +2,12 @@ module UpGuard
   class NodeGroupUser < BaseObject
     attr_accessor :email
     attr_accessor :user_id
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.email = nil
+      self.user_id = nil
+    end
+
     def from_hash(h)
       self.email = h['email'] if h.include?('email')
       self.user_id = h['user_id'] if h.include?('user_id')

data/lib/upguard/NodeGroupUserList.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 ["NodeGroupUser", "UpGuard::NodeGroupUser"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'NodeGroupUser' to 'NodeGroupUserList'"
+      end
     end
     
 

data/lib/upguard/NodeList.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 ["Node", "UpGuard::Node"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Node' to 'NodeList'"
+      end
     end
     
 

data/lib/upguard/NodeMediumInfo.rb

@@ -0,0 +1,49 @@
+module UpGuard
+  class NodeMediumInfo < 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
+    
+
+    
+    def from_hash(h)
+      @inner_map = h
+    end
+    
+
+
+  end
+end

data/lib/upguard/NodeScan.rb

@@ -0,0 +1,78 @@
+module UpGuard
+  class NodeScan < BaseObject
+    attr_accessor :associated_id
+    attr_accessor :category
+    attr_accessor :created_at
+    attr_accessor :data
+    attr_accessor :id
+    attr_accessor :label
+    attr_accessor :node_id
+    attr_accessor :scan_options
+    attr_accessor :task_id
+    attr_accessor :tsv
+    attr_accessor :updated_at
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.associated_id = nil
+      self.category = nil
+      self.created_at = nil
+      self.data = nil
+      self.id = nil
+      self.label = nil
+      self.node_id = nil
+      self.scan_options = nil
+      self.task_id = nil
+      self.tsv = nil
+      self.updated_at = nil
+    end
+
+    def from_hash(h)
+      self.associated_id = h['associated_id'] if h.include?('associated_id')
+      self.category = h['category'] if h.include?('category')
+      self.created_at = h['created_at'] if h.include?('created_at')
+      self.data = h['data'] if h.include?('data')
+      self.id = h['id'] if h.include?('id')
+      self.label = h['label'] if h.include?('label')
+      self.node_id = h['node_id'] if h.include?('node_id')
+      self.scan_options = h['scan_options'] if h.include?('scan_options')
+      self.task_id = h['task_id'] if h.include?('task_id')
+      self.tsv = h['tsv'] if h.include?('tsv')
+      self.updated_at = h['updated_at'] if h.include?('updated_at')
+    end
+    def to_hash
+      h = {}
+      h['associated_id'] = self.associated_id
+      h['category'] = self.category
+      h['created_at'] = self.created_at
+      h['data'] = self.data
+      h['id'] = self.id
+      h['label'] = self.label
+      h['node_id'] = self.node_id
+      h['scan_options'] = self.scan_options
+      h['task_id'] = self.task_id
+      h['tsv'] = self.tsv
+      h['updated_at'] = self.updated_at
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    
+    def load
+      obj = http_get("/api/v2/node_scans/#{self.id}.json")
+      from_hash(obj)
+    end
+
+    
+    def node
+      obj = http_get("/api/v2/nodes/{node_id}.json")
+      elem = Node.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
+        elem.id = obj["id"]
+        elem.name = obj["name"]
+      return elem
+    end
+
+
+  end
+end

data/lib/upguard/NodeScanList.rb

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

data/lib/upguard/NodeType.rb

@@ -10,7 +10,17 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
+    SERVICE = "SC"
     SERVER = "SV"
+    DATABASE = "DB"
+    DESKTOP = "DT"
+    SWITCH = "SW"
+    FIREWALL = "FW"
+    ROUTER = "RT"
+    SMART_PHONE = "PH"
+    ROBOT = "RB"
+    SAN_STORAGE = "SS"
+    WEBSITE = "WS"
 
   end
 end

data/lib/upguard/OperatingSystem.rb

@@ -4,6 +4,14 @@ module UpGuard
     attr_accessor :id
     attr_accessor :name
     attr_accessor :operating_system_family_id
+    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.operating_system_family_id = 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/OperatingSystemFamily.rb

@@ -2,6 +2,12 @@ module UpGuard
   class OperatingSystemFamily < BaseObject
     attr_accessor :id
     attr_accessor :name
+    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
+    end
+
     def from_hash(h)
       self.id = h['id'] if h.include?('id')
       self.name = h['name'] if h.include?('name')

data/lib/upguard/OperatingSystemFamilyList.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 ["OperatingSystemFamily", "UpGuard::OperatingSystemFamily"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'OperatingSystemFamily' to 'OperatingSystemFamilyList'"
+      end
     end
     
 

data/lib/upguard/OperatingSystemList.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 ["OperatingSystem", "UpGuard::OperatingSystem"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'OperatingSystem' to 'OperatingSystemList'"
+      end
     end