upguard 0.0.3 → 0.0.14

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
+      return @inner_map.to_json
+    end
+    
+
+    
+    def to_hash
+      return self.inner_map
+    end
+    
+
+    
+    def from_hash(h)
+      self.inner_map = h
+    end
+    
+
+
+  end
+end

data/lib/upguard/NodeType.rb

@@ -6,11 +6,21 @@ 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
+    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

@@ -18,14 +18,14 @@ module UpGuard
       h['operating_system_family_id'] = self.operating_system_family_id
       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 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

data/lib/upguard/OperatingSystemFamily.rb

@@ -12,21 +12,21 @@ module UpGuard
       h['name'] = self.name
       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 operating_systems
       obj = http_get("/api/v2/operating_system_families/#{self.id}/operating_systems.json")
-      list = []
+      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_hostname, self.api_key, self.sec_key, self.insecure)
-        elem.id = x["id"]
-        elem.name = x["name"]
-        list << elem
+        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
       end
-      return list
+      return the_list
     end
 
 

data/lib/upguard/OperatingSystemFamilyList.rb

@@ -0,0 +1,52 @@
+module UpGuard
+  class OperatingSystemFamilyList < 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 obj.class.name != "OperatingSystemFamily"
+        raise "Can only append 'OperatingSystemFamily' to 'OperatingSystemFamilyList'
+      end
+      @inner_list << obj
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/OperatingSystemList.rb

@@ -0,0 +1,52 @@
+module UpGuard
+  class OperatingSystemList < 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 obj.class.name != "OperatingSystem"
+        raise "Can only append 'OperatingSystem' to 'OperatingSystemList'
+      end
+      @inner_list << obj
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/Policy.rb

@@ -0,0 +1,79 @@
+module UpGuard
+  class Policy < BaseObject
+    attr_accessor :id
+    attr_accessor :name
+    attr_accessor :short_description
+    attr_accessor :description
+    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)
+      h = to_hash
+      return h.to_json(options)
+    end
+    
+    def load
+      obj = http_get("/api/v2/policies/#{self.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/policies.json", h)
+      from_hash(out)
+    end
+
+    
+    def update
+      h = to_hash
+      http_put("/api/v2/policies/#{self.id}.json", h)
+    end
+
+    def delete
+      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

@@ -0,0 +1,52 @@
+module UpGuard
+  class PolicyList < 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 obj.class.name != "Policy"
+        raise "Can only append 'Policy' to 'PolicyList'
+      end
+      @inner_list << obj
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end

data/lib/upguard/ScheduledJob.rb

@@ -1,11 +1,13 @@
 module UpGuard
   class ScheduledJob < BaseObject
+    attr_accessor :data
     attr_accessor :id
     attr_accessor :source_id
     attr_accessor :source_name
     attr_accessor :source_type
     attr_accessor :status
     def from_hash(h)
+      self.data = h['data'] if h.include?('data')
       self.id = h['id'] if h.include?('id')
       self.source_id = h['source_id'] if h.include?('source_id')
       self.source_name = h['source_name'] if h.include?('source_name')
@@ -14,6 +16,7 @@ module UpGuard
     end
     def to_hash
       h = {}
+      h['data'] = self.data
       h['id'] = self.id
       h['source_id'] = self.source_id
       h['source_name'] = self.source_name
@@ -21,13 +24,13 @@ module UpGuard
       h['status'] = self.status
       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
-      obj = http_get("/api/v2/scheduledu_jobs/#{self.id}.json")
+      obj = http_get("/api/v2/scheduled_jobs/#{self.id}.json")
       from_hash(obj)
     end
 

data/lib/upguard/ScheduledJobList.rb

@@ -0,0 +1,52 @@
+module UpGuard
+  class ScheduledJobList < 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 obj.class.name != "ScheduledJob"
+        raise "Can only append 'ScheduledJob' to 'ScheduledJobList'
+      end
+      @inner_list << obj
+    end
+    
+
+    
+    def to_json
+      return @inner_list.to_json
+    end
+    
+
+
+  end
+end