testdroid-api-client 0.2.2 → 0.5.0

data/lib/testdroid_api/admin.rb

@@ -0,0 +1,25 @@
+module TestdroidAPI
+  class Admin < CloudResource
+    def initialize(uri, client, params = {})
+      super uri, client, "admin", params
+      @uri, @client = uri, client
+    end
+
+    def device_statuses
+      AdminDeviceStatuses.new(@uri + "/device/statuses", @client)
+    end
+
+    def device_problems
+      AdminDeviceProblems.new(@uri + "/device-problems", @client)
+    end
+
+    def device_models
+      AdminDeviceModels.new(@uri + "/device-models", @client)
+    end
+
+    def devices
+      AdminDevices.new(@uri + "/devices", @client)
+    end
+
+  end
+end

data/lib/testdroid_api/admin_device_models.rb

@@ -0,0 +1,10 @@
+module TestdroidAPI
+  class AdminDeviceModels < CloudListResource
+  end
+  class AdminDeviceModel < CloudResource
+    def initialize(uri, client, params = {})
+      super uri, client, "adminDeviceModel", params
+      @uri, @client = uri, client
+    end
+  end
+end

data/lib/testdroid_api/admin_device_problems.rb

@@ -0,0 +1,9 @@
+module TestdroidAPI
+  class AdminDeviceProblems < CloudListResource
+  end
+  class AdminDeviceProblem < CloudResource
+    def initialize(uri, client, params = {})
+      super uri, client, "adminDeviceProblem", params
+    end
+  end
+end

data/lib/testdroid_api/admin_device_statuses.rb

@@ -0,0 +1,12 @@
+module TestdroidAPI
+  class AdminDeviceStatuses < CloudListResource
+    def initialize(uri, client)
+      super uri, client, "AdminDeviceStatus"
+    end
+  end
+  class AdminDeviceStatus < CloudResource
+    def initialize(uri, client, params = {})
+      super uri, client, "adminDeviceStatus", params
+    end
+  end
+end

data/lib/testdroid_api/admin_devices.rb

@@ -0,0 +1,10 @@
+module TestdroidAPI
+  class AdminDevices < CloudListResource
+  end
+  class AdminDevice < CloudResource
+    def initialize(uri, client, params = {})
+      super uri, client, "adminDevice", params
+      @uri, @client = uri, client
+    end
+  end
+end

data/lib/testdroid_api/apikey_client.rb

@@ -0,0 +1,120 @@
+module TestdroidAPI
+  class ApikeyClient < Client
+
+    attr_reader :config
+    attr_accessor :logger
+
+    API_VERSION = 'api/v2'
+    CLOUD_ENDPOINT = 'https://cloud.bitbar.com'
+
+    def initialize(api_key, cloud_url = CLOUD_ENDPOINT, logger = nil)
+      # Instance variables
+      @api_key = api_key
+      @cloud_url = cloud_url
+      @logger = logger
+
+      if @logger.nil?
+        @logger = Logger.new(STDOUT)
+        @logger.info("Logger is not defined => output to STDOUT")
+      end
+    end
+
+    def authorize
+
+      if @cloud_user.nil?
+        @cloud_user = TestdroidAPI::User.new("/#{API_VERSION}/me", self).refresh
+        @cloud_user = TestdroidAPI::User.new("/#{API_VERSION}/users/#{@cloud_user.id}", self).refresh
+
+      end
+      @cloud_user
+    end
+
+    # Basic methods
+
+    def request_factory(method, uri, http_params = {}, headers = {})
+      default_http_params = {
+          :method => method,
+          :url => uri,
+          :user => @api_key,
+          :password => "",
+          :headers => headers
+      }
+      request_http_params = default_http_params.deep_merge!(http_params)
+
+      RestClient::Request.new(request_http_params)
+    end
+
+    def get(uri, params = {}, http_params = {})
+      uri = @cloud_url + uri
+      begin
+        http_params = http_params.deep_merge({:headers => {:params => params}})
+        request = self.request_factory(:get, uri, http_params)
+        resp = request.execute
+      rescue => e
+        @logger.error "Failed to get resource #{uri} #{e}"
+        return nil
+      end
+      JSON.parse(resp.body)
+    end
+
+    def post(uri, params = {}, http_params = {})
+      uri = @cloud_url + uri
+      begin
+        payload = params.fetch(:body, params)
+        headers = params.fetch(:headers, {})
+        http_params = http_params.deep_merge({:payload => payload})
+        request = self.request_factory(:post, uri, http_params, headers)
+        resp = request.execute
+      rescue => e
+        @logger.error "Failed to post resource #{uri} #{e}"
+        return nil
+      end
+
+      if resp.body.nil? || resp.body.length == 0
+        return nil
+      end
+
+      JSON.parse(resp.body)
+    end
+
+    def delete(uri)
+      uri = @cloud_url + uri
+      begin
+        request = self.request_factory(:delete, uri)
+        resp = request.execute
+      rescue => e
+        @logger.error "Failed to delete resource #{uri} #{e}"
+        return nil
+      end
+
+      if resp.code != 204
+        @logger.error "Failed to delete resource #{uri} #{e}"
+      else
+        @logger.info "response: #{resp.code}"
+      end
+    end
+
+    def upload(uri, file_name)
+      self.post(uri, {
+          :multipart => true,
+          :file => ::File.new(file_name, 'rb')
+      })
+    end
+
+    def download(uri, file_name, params = {}, http_params = {})
+      begin
+        ::File.open(file_name, "w+b") do |file|
+          full_uri = uri.start_with?(@cloud_url) ? uri : @cloud_url + uri
+          http_params = http_params.deep_merge({:headers => {:params => params}})
+          request = self.request_factory(:get, full_uri, http_params)
+          resp = request.execute
+          file.write(resp.body)
+        end
+      rescue => e
+        @logger.error "Failed to get resource #{uri} #{e}"
+        return nil
+      end
+    end
+
+  end
+end

data/lib/testdroid_api/client.rb

@@ -1,126 +1,161 @@
 module TestdroidAPI
 
-	class Client
-		attr_reader :config
-		attr_accessor :logger
-		attr_reader :token
-
-		API_VERSION = 'api/v2'
-		CLOUD_ENDPOINT='https://cloud.testdroid.com'
-		ACCEPT_HEADERS={'Accept' => 'application/json'}
-
-		def initialize(username, password, cloud_url = CLOUD_ENDPOINT, logger = nil)
-			# Instance variables
-			@username = username
-			@password = password
-			@cloud_url = cloud_url
-			@logger = logger
-
-			if @logger.nil?
-				@logger = Logger.new(STDOUT)
-				@logger.info("Logger is not defined => output to STDOUT")
-			end
-		end
-		def label_groups
-			label_groups = TestdroidAPI::LabelGroups.new( "/#{API_VERSION}/label-groups", self )
-			label_groups.list
-			label_groups
-		end
-		def devices
-			devices = TestdroidAPI::Devices.new( "/#{API_VERSION}/devices", self )
-			devices.list
-			devices
-		end
-		def authorize
-
-			@client = OAuth2::Client.new('testdroid-cloud-api', nil, :site => @cloud_url, :authorize_url    => 'oauth/authorize',
-			:token_url        => 'oauth/token',  :headers => ACCEPT_HEADERS)  do |faraday|
-				faraday.request  :multipart
-				faraday.request  :url_encoded
-				faraday.response :logger, @logger
-				faraday.adapter  Faraday.default_adapter
-			end
-
-			@token = @client.password.get_token(@username, @password, :headers => ACCEPT_HEADERS)
-
-			if (@cloud_user.nil?)
-				@cloud_user = TestdroidAPI::User.new( "/#{API_VERSION}/me", self ).refresh
-				@cloud_user = TestdroidAPI::User.new( "/#{API_VERSION}/users/#{@cloud_user.id}", self ).refresh
-
-			end
-			@cloud_user
-		end
-		def upload(uri, filename, file_type)
-			begin
-				@token = @token.refresh!(:headers => ACCEPT_HEADERS) if @token.expired?
-				connection = @token.client.connection
-				payload = {:file  => Faraday::UploadIO.new(filename, file_type) }
-				headers = ACCEPT_HEADERS.merge(@token.headers)
-				response = connection.post(@cloud_url+"#{uri}",payload, headers)
-			rescue => e
-				@logger.error e
-				return nil
-			end
-			JSON.parse(response.body)
-		end
-		def post(uri, params)
-
-			@token = @token.refresh!(:headers => ACCEPT_HEADERS) if  @token.expired?
-
-			begin
-				resp = @token.post("#{@cloud_url}#{uri}", params.merge(:headers => ACCEPT_HEADERS))
-			rescue => e
-				@logger.error "Failed to post resource #{uri} #{e}"
-				return nil
-			end
-			
-			if resp.body.nil? || resp.body.length == 0
-				return nil
-			end
-			
-			JSON.parse(resp.body)
-		end
-		def get(uri, params={})
-
-			@token = @token.refresh!(:headers => ACCEPT_HEADERS) if  @token.expired?
-
-			begin
-				resp = @token.get(@cloud_url+"#{uri}", params.merge(:headers => ACCEPT_HEADERS))
-			rescue => e
-				@logger.error "Failed to get resource #{uri} #{e}"
-				return nil
-			end
-			JSON.parse(resp.body)
-		end
-		def delete(uri)
-
-			@token = @token.refresh!(:headers => ACCEPT_HEADERS) if  @token.expired?
-
-			begin
-				resp = @token.delete(@cloud_url+"#{uri}",  :headers => ACCEPT_HEADERS )
-			rescue => e
-				@logger.error "Failed to delete resource #{uri} #{e}"
-				return nil
-			end
-
-			if (resp.status != 204)
-				@logger.error "Failed to delete resource #{uri} #{e}"
-				return nil
-			else
-				@logger.info "response: #{resp.status}"
-			end
-		end
-		def download(uri, file_name)
-			begin
-				@token = @token.refresh!(:headers => ACCEPT_HEADERS) if  @token.expired?
-				File.open(file_name, "w+b") do |file|
-					resp = @token.get("#{@cloud_url}/#{uri}", :headers => ACCEPT_HEADERS)
-					file.write(resp.body)
-				end
-			rescue => e
-				@logger.error "Failed to get resource #{uri} #{e}"
-				return nil
-			end
-		end
-	end
+  class Client
+    attr_reader :config
+    attr_accessor :logger
+    attr_reader :token
+
+    API_VERSION = 'api/v2'
+    CLOUD_ENDPOINT = 'https://cloud.bitbar.com'
+
+    def initialize(username, password, cloud_url = CLOUD_ENDPOINT, logger = nil)
+      # Instance variables
+      @username = username
+      @password = password
+      @cloud_url = cloud_url
+      @logger = logger
+
+      if @logger.nil?
+        @logger = Logger.new(STDOUT)
+        @logger.info("Logger is not defined => output to STDOUT")
+      end
+    end
+
+    def authorize
+      @client = OAuth2::Client.new(
+          'testdroid-cloud-api', nil, :site => @cloud_url, :token_url => 'oauth/token') do |faraday|
+        faraday.request :multipart
+        faraday.request :url_encoded
+        faraday.response :logger, @logger
+        faraday.adapter Faraday.default_adapter
+      end
+
+      @token = @client.password.get_token(@username, @password)
+
+      if @cloud_user.nil?
+        @cloud_user = TestdroidAPI::User.new("/#{API_VERSION}/me", self).refresh
+        @cloud_user = TestdroidAPI::User.new("/#{API_VERSION}/users/#{@cloud_user.id}", self).refresh
+
+      end
+      @cloud_user
+    end
+
+    def mime_for(path)
+      mime = MIME::Types.type_for path
+      mime.empty? ? 'text/plain' : mime[0].content_type
+    end
+
+    def upload(uri, filename)
+      begin
+        @token = @token.refresh! if @token.expired?
+        connection = @token.client.connection
+        payload = {:file => Faraday::UploadIO.new(filename, mime_for(filename))}
+        response = connection.post(@cloud_url + "#{uri}", payload, @token.headers)
+      rescue => e
+        @logger.error e
+        return nil
+      end
+      JSON.parse(response.body)
+    end
+
+    def post(uri, params = {})
+
+      @token = @token.refresh! if @token.expired?
+
+      begin
+        resp = @token.post("#{@cloud_url}#{uri}", params)
+      rescue => e
+        @logger.error "Failed to post resource #{uri} #{e}"
+        return nil
+      end
+
+      if resp.body.nil? || resp.body.length == 0
+        return nil
+      end
+
+      JSON.parse(resp.body)
+    end
+
+    def get(uri, params = {})
+
+      @token = @token.refresh! if @token.expired?
+
+      begin
+        resp = @token.get(@cloud_url + "#{uri}", :params => params)
+      rescue => e
+        @logger.error "Failed to get resource #{uri} #{e}"
+        return nil
+      end
+      JSON.parse(resp.body)
+    end
+
+    def delete(uri)
+
+      @token = @token.refresh! if @token.expired?
+
+      begin
+        resp = @token.delete(@cloud_url + "#{uri}")
+      rescue => e
+        @logger.error "Failed to delete resource #{uri} #{e}"
+        return nil
+      end
+
+      if resp.status != 204
+        @logger.error "Failed to delete resource #{uri} #{e}"
+      else
+        @logger.info "response: #{resp.status}"
+      end
+    end
+
+    def download(uri, file_name)
+      begin
+        @token = @token.refresh! if @token.expired?
+        ::File.open(file_name, "w+b") do |file|
+          full_uri = uri.start_with?(@cloud_url) ? uri : @cloud_url + uri
+          resp = @token.get(full_uri)
+          file.write(resp.body)
+        end
+      rescue => e
+        @logger.error "Failed to get resource #{uri} #{e}"
+        return nil
+      end
+    end
+
+# Resources
+
+# public read-only
+
+    def info
+      TestdroidAPI::CloudResource.new("/#{API_VERSION}/info", self, "info")
+    end
+
+    def devices
+      TestdroidAPI::Devices.new("/#{API_VERSION}/devices", self)
+    end
+
+    def label_groups
+      TestdroidAPI::LabelGroups.new("/#{API_VERSION}/label-groups", self)
+    end
+
+# user read-write
+
+    def me
+      TestdroidAPI::User.new("/#{API_VERSION}/me", self).load
+    end
+
+    def properties
+      TestdroidAPI::Properties.new("/#{API_VERSION}/properties", self)
+    end
+
+    def device_session_connections
+      TestdroidAPI::DeviceSessionConnections.new("/#{API_VERSION}/device-session-connections", self)
+    end
+
+# admin only
+
+    def admin
+      TestdroidAPI::Admin.new("/#{API_VERSION}/admin", self)
+    end
+
+  end
 end