upguard 0.0.6 → 0.0.17

data/lib/upguard/BaseObject.rb

@@ -3,19 +3,19 @@ require 'json'
 module UpGuard
   class BaseObject
     attr_accessor :appliance_url
-    attr_accessor :api_key
+    attr_accessor :appliance_api_key
     attr_accessor :sec_key
     attr_accessor :insecure
     def from_hash(h)
       self.appliance_url = h['appliance_url'] if h.include?('appliance_url')
-      self.api_key = h['api_key'] if h.include?('api_key')
+      self.appliance_api_key = h['appliance_api_key'] if h.include?('appliance_api_key')
       self.sec_key = h['sec_key'] if h.include?('sec_key')
       self.insecure = h['insecure'] if h.include?('insecure')
     end
     def to_hash
       h = {}
       h['appliance_url'] = self.appliance_url
-      h['api_key'] = self.api_key
+      h['appliance_api_key'] = self.appliance_api_key
       h['sec_key'] = self.sec_key
       h['insecure'] = self.insecure
       return h
@@ -24,21 +24,21 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
-    def initialize(appliance_url, api_key, sec_key, insecure = false)
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
       if appliance_url.to_s.start_with?("http://") || appliance_url.to_s.start_with?("https://")
         # all good
       else
         appliance_url = "https://" + appliance_url
       end
       self.appliance_url = appliance_url
-      self.api_key = api_key
+      self.appliance_api_key = appliance_api_key
       self.sec_key = sec_key
       self.insecure = insecure
     end
 
     def make_headers
       return {
-        "Authorization" => "Token token=\"#{self.api_key}#{self.sec_key}\"",
+        "Authorization" => "Token token=\"#{self.appliance_api_key}#{self.sec_key}\"",
         "Content-Type" => "application/json"
       }
     end
@@ -62,9 +62,10 @@ module UpGuard
                    :verify => (self.insecure == false),
                    :body => body.to_json
                  )
-      if ["200", "201"].include?(response.code.to_s) == false
+      if ["200", "201", "204"].include?(response.code.to_s) == false
         raise response.body
       end
+      return true if response.code.to_s == "204"
       obj = JSON.parse(response.body)
       return obj
     end

data/lib/upguard/ChangeRequest.rb

@@ -0,0 +1,55 @@
+module UpGuard
+  class ChangeRequest < BaseObject
+    attr_accessor :data
+    attr_accessor :ended_at
+    attr_accessor :external_id
+    attr_accessor :id
+    attr_accessor :planned_end_at
+    attr_accessor :planned_start_at
+    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.data = nil
+      self.ended_at = nil
+      self.external_id = nil
+      self.id = nil
+      self.planned_end_at = nil
+      self.planned_start_at = nil
+      self.short_description = nil
+      self.started_at = nil
+      self.url = nil
+    end
+
+    def from_hash(h)
+      self.data = h['data'] if h.include?('data')
+      self.ended_at = h['ended_at'] if h.include?('ended_at')
+      self.external_id = h['external_id'] if h.include?('external_id')
+      self.id = h['id'] if h.include?('id')
+      self.planned_end_at = h['planned_end_at'] if h.include?('planned_end_at')
+      self.planned_start_at = h['planned_start_at'] if h.include?('planned_start_at')
+      self.short_description = h['short_description'] if h.include?('short_description')
+      self.started_at = h['started_at'] if h.include?('started_at')
+      self.url = h['url'] if h.include?('url')
+    end
+    def to_hash
+      h = {}
+      h['data'] = self.data
+      h['ended_at'] = self.ended_at
+      h['external_id'] = self.external_id
+      h['id'] = self.id
+      h['planned_end_at'] = self.planned_end_at
+      h['planned_start_at'] = self.planned_start_at
+      h['short_description'] = self.short_description
+      h['started_at'] = self.started_at
+      h['url'] = self.url
+      return h
+    end
+    def to_json(options = nil)
+      h = to_hash
+      return h.to_json(options)
+    end
+
+  end
+end

data/lib/upguard/ChangeRequestList.rb

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

data/lib/upguard/ConnectionManager.rb

@@ -1,18 +1,39 @@
 module UpGuard
   class ConnectionManager < BaseObject
+    attr_accessor :agent_version
+    attr_accessor :agent_type
     attr_accessor :channels
     attr_accessor :connection_manager_group_id
     attr_accessor :created_at
     attr_accessor :id
+    attr_accessor :ip_address
     attr_accessor :hostname
     attr_accessor :last_contact
     attr_accessor :stats
     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.agent_version = nil
+      self.agent_type = nil
+      self.channels = nil
+      self.connection_manager_group_id = nil
+      self.created_at = nil
+      self.id = nil
+      self.ip_address = nil
+      self.hostname = nil
+      self.last_contact = nil
+      self.stats = nil
+      self.updated_at = nil
+    end
+
     def from_hash(h)
+      self.agent_version = h['agent_version'] if h.include?('agent_version')
+      self.agent_type = h['agent_type'] if h.include?('agent_type')
       self.channels = h['channels'] if h.include?('channels')
       self.connection_manager_group_id = h['connection_manager_group_id'] if h.include?('connection_manager_group_id')
       self.created_at = h['created_at'] if h.include?('created_at')
       self.id = h['id'] if h.include?('id')
+      self.ip_address = h['ip_address'] if h.include?('ip_address')
       self.hostname = h['hostname'] if h.include?('hostname')
       self.last_contact = h['last_contact'] if h.include?('last_contact')
       self.stats = h['stats'] if h.include?('stats')
@@ -20,10 +41,13 @@ module UpGuard
     end
     def to_hash
       h = {}
+      h['agent_version'] = self.agent_version
+      h['agent_type'] = self.agent_type
       h['channels'] = self.channels
       h['connection_manager_group_id'] = self.connection_manager_group_id
       h['created_at'] = self.created_at
       h['id'] = self.id
+      h['ip_address'] = self.ip_address
       h['hostname'] = self.hostname
       h['last_contact'] = self.last_contact
       h['stats'] = self.stats
@@ -37,7 +61,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_url, 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

data/lib/upguard/ConnectionManagerGroup.rb

@@ -1,20 +1,28 @@
 module UpGuard
   class ConnectionManagerGroup < BaseObject
+    attr_accessor :api_key
     attr_accessor :id
     attr_accessor :name
-    attr_accessor :organisation_id
     attr_accessor :status
+    def initialize(appliance_url, appliance_api_key, sec_key, insecure = false)
+      super(appliance_url, appliance_api_key, sec_key, insecure)
+      self.api_key = nil
+      self.id = nil
+      self.name = nil
+      self.status = nil
+    end
+
     def from_hash(h)
+      self.api_key = h['api_key'] if h.include?('api_key')
       self.id = h['id'] if h.include?('id')
       self.name = h['name'] if h.include?('name')
-      self.organisation_id = h['organisation_id'] if h.include?('organisation_id')
       self.status = h['status'] if h.include?('status')
     end
     def to_hash
       h = {}
+      h['api_key'] = self.api_key
       h['id'] = self.id
       h['name'] = self.name
-      h['organisation_id'] = self.organisation_id
       h['status'] = self.status
       return h
     end
@@ -22,12 +30,14 @@ module UpGuard
       h = to_hash
       return h.to_json(options)
     end
+    STATUS_DISABLED = 0
+    STATUS_ACTIVE = 1
     
     def connection_managers
-      obj = http_get("todo")
-      the_list = ConnectionManagerList.new(self.appliance_url, self.api_key, self.sec_key, self.insecure)
+      obj = http_get("/api/v2/connection_manager_groups/#{self.id}/connection_managers.json")
+      the_list = ConnectionManagerList.new(self.appliance_url, self.appliance_api_key, self.sec_key, self.insecure)
       obj.each do |x|
-        elem = ConnectionManager.new(self.appliance_url, self.api_key, self.sec_key, self.insecure)
+        elem = ConnectionManager.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/ConnectionManagerGroupList.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 ["ConnectionManagerGroup", "UpGuard::ConnectionManagerGroup"].include?(obj.class.name)
+        @inner_list << obj
+      else
+        raise "Can only append 'ConnectionManagerGroup' to 'ConnectionManagerGroupList'"
+      end
     end
     
 

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

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')
@@ -82,9 +96,9 @@ module UpGuard
     
     def nodes
       obj = http_get("/api/v2/environments/#{self.id}/nodes.json")
-      the_list = NodeList.new(self.appliance_url, self.api_key, self.sec_key, self.insecure)
+      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.api_key, self.sec_key, self.insecure)
+        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

data/lib/upguard/EnvironmentList.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 ["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)
@@ -25,7 +37,7 @@ module UpGuard
     
     def environment
       obj = http_get("/api/v2/environments/#{self.environment_id}.json")
-      elem = Environment.new(self.appliance_url, 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"]
         elem.organisation_id = obj["organisation_id"]