upguard 0.0.6 → 0.0.17

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')
@@ -25,7 +33,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_url, 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

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')
@@ -19,9 +25,9 @@ module UpGuard
     
     def operating_systems
       obj = http_get("/api/v2/operating_system_families/#{self.id}/operating_systems.json")
-      the_list = OperatingSystemList.new(self.appliance_url, self.api_key, self.sec_key, self.insecure)
+      the_list = OperatingSystemList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
       obj.each do |x|
-        elem = OperatingSystem.new(self.appliance_url, 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 = x["id"] if x.include?("id")
         elem.name = x["name"] if x.include?("name")
         the_list << elem

data/lib/upguard/OperatingSystemFamilyList.rb

@@ -10,12 +10,22 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
-    def initialize(appliance_url, api_key, sec_key, insecure)
-      super(appliance_url, api_key, sec_key, insecure)
+    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)
@@ -24,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

@@ -10,12 +10,22 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
-    def initialize(appliance_url, api_key, sec_key, insecure)
-      super(appliance_url, api_key, sec_key, insecure)
+    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)
@@ -24,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
     
 

data/lib/upguard/Pluggable.rb

@@ -0,0 +1,81 @@
+module UpGuard
+  class Pluggable < BaseObject
+    attr_accessor :description
+    attr_accessor :id
+    attr_accessor :name
+    attr_accessor :operating_system_family
+    attr_accessor :operating_system_family_id
+    attr_accessor :operating_system_id
+    attr_accessor :script
+    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 = nil
+      self.operating_system_family_id = nil
+      self.operating_system_id = nil
+      self.script = nil
+    end
+
+    def from_hash(h)
+      self.description = h['description'] if h.include?('description')
+      self.id = h['id'] if h.include?('id')
+      self.name = h['name'] if h.include?('name')
+      self.operating_system_family = h['operating_system_family'] if h.include?('operating_system_family')
+      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.script = h['script'] if h.include?('script')
+    end
+    def to_hash
+      h = {}
+      h['description'] = self.description
+      h['id'] = self.id
+      h['name'] = self.name
+      h['operating_system_family'] = self.operating_system_family
+      h['operating_system_family_id'] = self.operating_system_family_id
+      h['operating_system_id'] = self.operating_system_id
+      h['script'] = self.script
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+    
+    def load
+      obj = http_get("/api/v2/pluggables/#{self.operating_system_id}.json")
+      from_hash(obj)
+    end
+
+    
+    def save
+      if self.id.to_i == 0
+        return create
+      else
+        return update
+      end
+    end
+
+    
+    def create
+      h = to_hash
+      out = http_post("/api/v2/pluggables.json", h)
+      from_hash(out)
+    end
+
+    
+    def update
+      h = to_hash
+      h.delete("operating_system_family_id")
+      h.delete("operating_system_id")
+      http_put("/api/v2/pluggables/#{self.operating_system_id}.json", h)
+    end
+
+    def delete
+      http_delete("/api/v2/pluggables/#{self.operating_system_id}.json")
+    end
+
+
+  end
+end

data/lib/upguard/PluggableList.rb

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

data/lib/upguard/Policy.rb

@@ -2,14 +2,28 @@ module UpGuard
   class Policy < BaseObject
     attr_accessor :id
     attr_accessor :name
+    attr_accessor :short_description
+    attr_accessor :description
+    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.short_description = nil
+      self.description = nil
+    end
+
     def from_hash(h)
       self.id = h['id'] if h.include?('id')
       self.name = h['name'] if h.include?('name')
+      self.short_description = h['short_description'] if h.include?('short_description')
+      self.description = h['description'] if h.include?('description')
     end
     def to_hash
       h = {}
       h['id'] = self.id
       h['name'] = self.name
+      h['short_description'] = self.short_description
+      h['description'] = self.description
       return h
     end
     def to_json(options = nil)
@@ -48,6 +62,26 @@ module UpGuard
       http_delete("/api/v2/policies/#{self.id}.json")
     end
 
+    
+    def policy_versions
+      obj = http_get("/api/v2/policies/#{self.id}/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.version = x["version"] if x.include?("version")
+        elem.tag = x["tag"] if x.include?("tag")
+        the_list << elem
+      end
+      return the_list
+    end
+
+    def add_file_check(section, file_path, attr, check, expected, absent_pass=false)
+      url = "/api/v2/policies/#{self.id}/add_file_check.json?section=#{section}&file_path=#{file_path}&attr=#{attr}&check=#{check}&expected=#{expected}&absent_pass=#{absent_pass}"
+      obj = http_post(url, nil)
+      return obj
+    end
+
 
   end
 end

data/lib/upguard/PolicyList.rb

@@ -10,12 +10,22 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
-    def initialize(appliance_url, api_key, sec_key, insecure)
-      super(appliance_url, api_key, sec_key, insecure)
+    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)
@@ -24,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["Policy", "UpGuard::Policy"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'Policy' to 'PolicyList'"
+      end
     end
     
 

data/lib/upguard/ScheduledJob.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.data = nil
+      self.id = nil
+      self.source_id = nil
+      self.source_name = nil
+      self.source_type = nil
+      self.status = nil
+    end
+
     def from_hash(h)
       self.data = h['data'] if h.include?('data')
       self.id = h['id'] if h.include?('id')
@@ -60,7 +70,7 @@ module UpGuard
       http_delete("/api/v2/scheduled_jobs/#{self.id}.json")
     end
 
-    def cancel_jobs()
+    def cancel_jobs
       url = "/api/v2/scheduled_jobs/#{self.id}/cancel_jobs.json"
       obj = http_post(url, nil)
       return obj

data/lib/upguard/ScheduledJobList.rb

@@ -10,12 +10,22 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
-    def initialize(appliance_url, api_key, sec_key, insecure)
-      super(appliance_url, api_key, sec_key, insecure)
+    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)
@@ -24,7 +34,11 @@ module UpGuard
 
     
     def <<(obj)
-      @inner_list << obj
+      if ["ScheduledJob", "UpGuard::ScheduledJob"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'ScheduledJob' to 'ScheduledJobList'"
+      end
     end
     
 

data/lib/upguard/SystemMetric.rb

@@ -3,6 +3,13 @@ module UpGuard
     attr_accessor :metric
     attr_accessor :value
     attr_accessor :timestamp
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.metric = nil
+      self.value = nil
+      self.timestamp = nil
+    end
+
     def from_hash(h)
       self.metric = h['metric'] if h.include?('metric')
       self.value = h['value'] if h.include?('value')