mms-api 0.0.9 → 0.0.10

data/lib/mms/client.rb

@@ -1,49 +1,23 @@
-require 'singleton'
-require "net/http"
-require "uri"
-require 'net/http/digest_auth'
-require 'json'
-
 module MMS
 
   class Client
 
-    include Singleton
-
     attr_accessor :username
     attr_accessor :apikey
+    attr_accessor :url
 
-    attr_accessor :api_protocol
-    attr_accessor :api_host
-    attr_accessor :api_port
-    attr_accessor :api_path
-    attr_accessor :api_version
-
-    def initialize
-      @username, @apikey = nil
-
-      @api_protocol = 'https'
-      @api_host = 'mms.mongodb.com'
-      @api_port = '443'
-      @api_path = '/api/public'
-      @api_version = 'v1.0'
-    end
-
-    def auth_setup(username, apikey)
+    def initialize(username = nil, apikey = nil, url = nil)
       @username = username
       @apikey = apikey
-    end
-
-    def site
-      [@api_protocol, '://', @api_host, ':', @api_port, @api_path, '/',  @api_version].join.to_s
+      @url = url.nil? ? 'https://mms.mongodb.com:443/api/public/v1.0' : url
     end
 
     def get(path)
-      _get site + path, @username, @apikey
+      _get(@url + path, @username, @apikey)
     end
 
     def post(path, data)
-      _post site + path, data, @username, @apikey
+      _post(@url + path, data, @username, @apikey)
     end
 
     private
@@ -70,8 +44,12 @@ module MMS
       response = http.request(req)
       response_json = JSON.parse(response.body)
 
-      unless response_json['error'].nil?
-       response_json = nil
+      unless response.code == 200 or response_json['error'].nil?
+        msg = "http 'get' error for url `#{url}`"
+        msg = response_json['detail'] unless response_json['detail'].nil?
+
+        raise MMS::AuthError.new(msg, req, response) if response.code == '401'
+        raise MMS::ApiError.new(msg, req, response)
       end
 
       (response_json.nil? or response_json['results'].nil?) ? response_json : response_json['results']
@@ -88,19 +66,23 @@ module MMS
       http = Net::HTTP.new uri.host, uri.port
       http.use_ssl = true
 
-      req = Net::HTTP::Post.new uri.request_uri, {'Content-Type' =>'application/json'}
+      req = Net::HTTP::Post.new uri.request_uri, {'Content-Type' => 'application/json'}
       res = http.request req
 
       auth = digest_auth.auth_header uri, res['WWW-Authenticate'], 'POST'
-      req = Net::HTTP::Post.new uri.request_uri, {'Content-Type' =>'application/json'}
+      req = Net::HTTP::Post.new uri.request_uri, {'Content-Type' => 'application/json'}
       req.add_field 'Authorization', auth
       req.body = data.to_json
 
       response = http.request req
       response_json = JSON.parse response.body
 
-      unless response_json['error'].nil?
-        response_json = nil
+      unless response.code == 200 or response_json['error'].nil?
+        msg = "http 'get' error for url `#{url}`"
+        msg = response_json['detail'] unless response_json['detail'].nil?
+
+        raise MMS::AuthError.new(msg, req, response) if response.code == '401'
+        raise MMS::ApiError.new(msg, req, response)
       end
 
       (response_json.nil? or response_json['results'].nil?) ? response_json : response_json['results']

data/lib/mms/config.rb

@@ -0,0 +1,57 @@
+module MMS
+
+  class Config
+
+    default = {
+        username: proc {
+          nil
+        },
+        apikey: proc {
+          nil
+        },
+        apiurl: proc {
+          [api_protocol, '://', api_host, ':', api_port, api_path, '/', api_version].join.to_s
+        },
+        limit: proc {
+          10
+        },
+        api_protocol: proc {
+          'https'
+        },
+        api_host: proc {
+          'mms.mongodb.com'
+        },
+        api_port: proc {
+          '443'
+        },
+        api_path: proc {
+          '/api/public'
+        },
+        api_version: proc {
+          'v1.0'
+        },
+        default_group_id: proc {
+          nil
+        },
+        default_cluster_id: proc {
+          nil
+        },
+        config_path: proc {
+          Dir.home + '/.mms-api'
+        }
+    }
+
+    default.each do |key, value|
+      define_method(key) do
+        if default[key].equal?(value)
+          default[key] = instance_eval(&value)
+        end
+        default[key]
+      end
+      define_method("#{key}=") do |value|
+        default[key] = value
+      end
+    end
+
+  end
+end

data/lib/mms/errors.rb

@@ -0,0 +1,41 @@
+module MMS
+
+  class RuntimeError < StandardError;
+
+    def initialize(message, request, response)
+      super(message)
+
+      @request = request
+      @response = response
+    end
+
+    attr_reader :request
+    attr_reader :response
+
+  end
+
+  class ResourceError < StandardError
+
+    def initialize(message, resource)
+      super(message)
+
+      @resource = resource
+    end
+
+    attr_reader :resource
+
+  end
+
+  class ApiError < RuntimeError
+
+    def initialize(message, request, response)
+      super("API Response error! Code: #{response.code}, body: #{response.body}", request, response)
+    end
+
+  end
+
+  class AuthError < RuntimeError; end
+
+  class ConfigError < StandardError; end
+
+end

data/lib/mms/resource.rb

@@ -19,6 +19,10 @@ module MMS
       end
     end
 
+    def to_hash
+      _to_hash
+    end
+
     def reload
       @data = _load(@id)
       save @data unless @data.nil? or @data.empty?
@@ -42,5 +46,29 @@ module MMS
       from_hash data
       MMS::Cache.instance.set "Class::#{self.class.name}:#{@id}", data
     end
+
+    def table_row
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
+
+    def table_section
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
+
+    def self.table_header
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
+
+    def _load(id)
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
+
+    def _from_hash(data)
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
+
+    def _to_hash
+      raise("`#{__method__}` is not implemented for `#{self.class.name}`")
+    end
   end
 end

data/lib/mms/resource/alert.rb

@@ -0,0 +1,83 @@
+module MMS
+
+  class Resource::Alert < Resource
+
+    @client = nil
+
+    attr_accessor :name
+    attr_accessor :group
+
+    attr_accessor :type_name
+    attr_accessor :event_type_name
+    attr_accessor :status
+    attr_accessor :acknowledged_until
+    attr_accessor :created
+    attr_accessor :updated
+    attr_accessor :resolved
+    attr_accessor :last_notified
+    attr_accessor :current_value
+
+    def initialize(client, data)
+      id = data['id']
+      group_id = data['groupId']
+
+      raise MMS::ResourceError.new('`Id` for alert resource must be defined', self) if id.nil?
+      raise MMS::ResourceError.new('`groupId` for alert resource must be defined', self) if group_id.nil?
+
+      @client = client
+
+      @group = MMS::Resource::Group.new(client, {'id' => group_id})
+
+      super id, data
+    end
+
+    def ack(time, description)
+      data = {
+          :acknowledgedUntil => time,
+          :acknowledgementComment => description
+      }
+      alert = @client.post '/groups/' + @group.id + '/alerts/' + @id, data
+      !alert.nil?
+    end
+
+    def table_row
+      [@status, @group.name, @type_name, @event_type_name, @created, @updated, @resolved, @last_notified, JSON.dump(@current_value)]
+    end
+
+    def table_section
+      rows = []
+      rows << table_row
+      rows << [{:value => "AlertId: #{@id}   GroupId: #{@group.id}", :colspan => 9, :alignment => :left}]
+      rows << :separator
+      rows
+    end
+
+    def self.table_header
+      ['Status', 'Group', 'Type', 'Event name', 'Created', 'Updated', 'Resolved', 'Last notified', 'Value']
+    end
+
+    private
+
+    def _load(id)
+      @client.get '/groups/' + @group.id + '/alerts/' + id.to_s
+    end
+
+    def _from_hash(data)
+      @type_name = data['typeName']
+      @event_type_name = data['eventTypeName']
+      @status = data['status']
+      @acknowledged_until = data['acknowledgedUntil']
+      @created = data['created']
+      @updated = data['updated']
+      @resolved = data['resolved']
+      @last_notified = data['lastNotified']
+      @current_value = data['currentValue']
+      @name = @type_name
+    end
+
+    def _to_hash
+      @data
+    end
+
+  end
+end

data/lib/mms/resource/cluster.rb

@@ -2,6 +2,8 @@ module MMS
 
   class Resource::Cluster < Resource
 
+    @client = nil
+
     attr_accessor :name
     attr_accessor :group
     attr_accessor :shard_name
@@ -12,39 +14,109 @@ module MMS
     attr_accessor :snapshots
     attr_accessor :restorejobs
 
-    def initialize(id, group_id, data = nil)
+    def initialize(client, data)
+      id = data['id']
+      group_id = data['groupId']
+
+      raise MMS::ResourceError.new('`Id` for cluster resource must be defined', self) if id.nil?
+      raise MMS::ResourceError.new('`groupId` for cluster resource must be defined', self) if group_id.nil?
+
       @snapshots = []
       @restorejobs = []
 
-      @group = MMS::Resource::Group.new(group_id)
+      @client = client
+
+      @group = MMS::Resource::Group.new(client, {'id' => group_id})
 
       super id, data
     end
 
     def snapshot(id)
-      MMS::Resource::Snapshot.new id, @id, @group.id
+      MMS::Resource::Snapshot.new(@client, {'id' => id, 'clusterId' => @id, 'groupId' => @group.id})
     end
 
-    def snapshots(page = 1, limit = 1000)
+    def snapshots(page = 1, limit = 10)
       if @snapshots.empty?
-        MMS::Client.instance.get('/groups/' + @group.id + '/clusters/' + @id + '/snapshots?pageNum=' + page.to_s + '&itemsPerPage=' + limit.to_s).each do |snapshot|
-          @snapshots.push MMS::Resource::Snapshot.new(snapshot['id'], snapshot['clusterId'], snapshot['groupId'], snapshot)
+        @client.get('/groups/' + @group.id + '/clusters/' + @id + '/snapshots?pageNum=' + page.to_s + '&itemsPerPage=' + limit.to_s).each do |snapshot|
+          @snapshots.push MMS::Resource::Snapshot.new(@client, snapshot)
         end
       end
       @snapshots
     end
 
-    def restorejobs(page = 1, limit = 1000)
+    def restorejobs(page = 1, limit = 10)
       if @restorejobs.empty?
-        MMS::Client.instance.get('/groups/' + @group.id + '/clusters/' + @id + '/restoreJobs?pageNum=' + page.to_s + '&itemsPerPage=' + limit.to_s).each do |job|
-          @restorejobs.push MMS::Resource::RestoreJob.new(job['id'], job['clusterId'], job['groupId'], job)
+        @client.get('/groups/' + @group.id + '/clusters/' + @id + '/restoreJobs?pageNum=' + page.to_s + '&itemsPerPage=' + limit.to_s).each do |job|
+
+          if job['snapshotId'].nil? and job['clusterId'].nil?
+            raise MMS::ResourceError.new("RestoreJob `#{job['id']}` with status `#{job['statusName']}` has no `clusterId` and no `snapshotId`.", self)
+          elsif job['clusterId'].nil?
+            snapshot = @group.findSnapshot(job['snapshotId'])
+            job['clusterId'] = snapshot.cluster.id unless snapshot.nil?
+          end
+
+          @restorejobs.push MMS::Resource::RestoreJob.new(@client, job)
         end
       end
       @restorejobs
     end
 
+    def create_restorejob(point_in_time = nil)
+      data = {
+          'timestamp' => {
+              'date' => point_in_time,
+              'increment' => 0
+          }
+      }
+      jobs = @client.post('/groups/' + @group.id + '/clusters/' + @id + '/restoreJobs', data)
+
+      if jobs.nil?
+        raise MMS::ResourceError.new("Cannot create job from snapshot `#{self.id}`", self)
+      end
+
+      job_list = []
+      # work around due to bug in MMS API; cannot read restoreJob using provided info.
+      # The config-server RestoreJob and Snapshot has no own ClusterId to be accessed.
+      tries = 5
+      while tries > 0
+        begin
+          restore_jobs = restorejobs
+          tries = 0
+        rescue Exception => e
+          tries-=1;
+          raise MMS::ResourceError.new(e.message, self) if tries < 1
+
+          STDERR.puts e.message
+          STDERR.puts 'Sleeping for 5 seconds. Trying again...'
+          sleep(5)
+        end
+      end
+
+      jobs.each do |job|
+        _list = restore_jobs.select { |restorejob| restorejob.id == job['id'] }
+        _list.each do |restorejob|
+          job_list.push restorejob
+        end
+      end
+      job_list
+    end
+
+    def table_row
+      [@group.name, @name, @shard_name, @replicaset_name, @type_name, @last_heartbeat, @id]
+    end
+
+    def table_section
+      [table_row]
+    end
+
+    def self.table_header
+      ['Group', 'Cluster', 'Shard name', 'Replica name', 'Type', 'Last heartbeat', 'Cluster Id']
+    end
+
+    private
+
     def _load(id)
-      MMS::Client.instance.get '/groups/' + @group.id + '/clusters/' + id.to_s
+      @client.get('/groups/' + @group.id + '/clusters/' + id.to_s)
     end
 
     def _from_hash(data)
@@ -54,5 +126,10 @@ module MMS
       @type_name = data['typeName']
       @last_heartbeat = data['lastHeartbeat']
     end
+
+    def _to_hash
+      @data
+    end
+
   end
 end