ZOHOCRMSDK2_0_docgo 6.2.0

data/src/com/zoho/api/authenticator/oauth_token.rb

@@ -0,0 +1,285 @@
+require 'uri'
+require 'net/http'
+require 'json'
+require 'monitor'
+
+require_relative 'token'
+require_relative '../../crm/api/util/constants'
+require_relative '../logger/sdk_logger'
+require_relative '../../crm/api/exception/sdk_exception'
+
+module ZOHOCRMSDK
+  module Authenticator
+    # This class gets and refreshes the tokens based on the expiry time.
+    class OAuthToken < Token
+      attr_accessor :client_id, :client_secret, :redirect_url, :grant_token, :refresh_token, :access_token, :user_mail, :id, :expires_in
+      @@sync_lock = Monitor.new
+
+      # Creates an OAuthToken class instance with the specified parameters.
+      # @param client_id [string] A String containing the OAuth client id.
+      # @param client_secret [string] A String containing the OAuth client secret.
+      # @param grant_token [string]  A String containing the grant token.
+      # @param refresh_token [string] A String containing the refresh token.
+      # @param redirect_url [string]  A String containing the OAuth redirect URL.
+      # @param id [string] A String containing ID
+      
+      def initialize(client_id: nil, client_secret: nil, grant_token: nil, refresh_token: nil, redirect_url: nil, id: nil, access_token: nil)
+        error = {}
+
+        if grant_token.nil? && refresh_token.nil? && id.nil? && access_token.nil?
+          raise SDKException.new(Constants::MANDATORY_VALUE_ERROR, Constants::MANDATORY_KEY_ERROR, Constants::OAUTH_MANDATORY_KEYS)
+        end
+
+        if id.nil? and access_token.nil?
+          unless client_id.is_a?(String)
+            error[Constants::ERROR_HASH_FIELD] = Constants::CLIENT_ID
+
+            error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+            error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+            raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+          end
+
+          unless client_secret.is_a?(String)
+            error[Constants::ERROR_HASH_FIELD] = Constants::CLIENT_SECRET
+
+            error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+            error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+            raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+          end
+
+          if !grant_token.nil? && !grant_token.is_a?(String)
+            error[Constants::ERROR_HASH_FIELD] = Constants::GRANT_TOKEN
+
+            error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+            error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+            raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+          end
+
+          if !refresh_token.nil? && !refresh_token.is_a?(String)
+            error[Constants::ERROR_HASH_FIELD] = Constants::REFRESH_TOKEN
+
+            error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+            error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+            raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+          end
+
+          if !redirect_url.nil? && !redirect_url.is_a?(String)
+            error[Constants::ERROR_HASH_FIELD] = Constants::REDIRECT_URL
+
+            error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+            error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+            raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+          end
+        end
+
+        if !id.nil? && !id.is_a?(String)
+          error[Constants::ERROR_HASH_FIELD] = Constants::ID
+
+          error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+          error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+          raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+
+        end
+
+        if !access_token.nil? && !access_token.is_a?(String)
+          error[Constants::ERROR_HASH_FIELD] = Constants::ACCESS_TOKEN
+
+          error[Constants::ERROR_HASH_EXPECTED_TYPE] = String
+
+          error[Constants::ERROR_HASH_CLASS] = OAuthToken.class
+
+          raise SDKException.new(Constants::INPUT_ERROR, nil, error, nil)
+        end
+
+        @client_id = client_id
+
+        @client_secret = client_secret
+
+        @redirect_url = redirect_url
+
+        @refresh_token = refresh_token
+
+        @grant_token = grant_token
+
+        @access_token = access_token
+
+        @expires_in = nil
+
+        @user_mail = nil
+
+        @id = id
+      end
+      
+      def authenticate(url_connection)
+        @@sync_lock.synchronize do
+          initializer = Initializer.get_initializer
+          store = initializer.store
+          user = initializer.user
+
+          oauth_token = nil
+          if @access_token.nil?
+            if !@id.nil?
+              oauth_token = initializer.store.get_token_by_id(@id, self)
+            else
+              oauth_token = initializer.store.get_token(user, self)
+            end
+          else
+            oauth_token = self
+          end
+
+          token = ""
+
+          if oauth_token.nil?
+            token = @refresh_token.nil? ? generate_access_token(user, store).access_token : refresh_access_token(user, store).access_token
+          elsif !oauth_token.expires_in.nil? && oauth_token.expires_in.to_f - get_current_time_in_millis < 5000
+            SDKLog::SDKLogger.info(Constants::REFRESH_TOKEN_MESSAGE)
+            token = oauth_token.refresh_access_token(user, store).access_token
+          else
+            token = oauth_token.access_token
+          end
+          url_connection.headers[Constants::AUTHORIZATION] = Constants::OAUTH_HEADER_PREFIX + token
+        end
+      end
+
+      def remove
+        Initializer.get_initializer.store.delete_token(self)
+        true
+      rescue StandardError
+        false
+      end
+
+      def generate_access_token(user, store)
+        json_body = {}
+
+        json_body[Constants::CLIENT_ID] = @client_id
+
+        json_body[Constants::CLIENT_SECRET] = @client_secret
+  
+        json_body[Constants::REDIRECT_URI] = @redirect_url if !@redirect_url.nil?
+
+        json_body[Constants::GRANT_TYPE] = Constants::GRANT_TYPE_AUTH_CODE
+
+        json_body[Constants::CODE] = @grant_token
+
+        response = get_response_from_server(json_body)
+
+        begin
+          parse_response(response)
+
+          generate_id()
+
+          store.save_token(user, self)
+        rescue SDKException => e
+          raise e
+        rescue StandardError => e
+          raise SDKException.new(nil, Constants::SAVE_TOKEN_ERROR, nil, e)
+        end
+        self
+      end
+
+      def refresh_access_token(user, store)
+        json_body = {}
+
+        json_body[Constants::CLIENT_ID] = @client_id
+
+        json_body[Constants::CLIENT_SECRET] = @client_secret
+
+        json_body[Constants::GRANT_TYPE] = Constants::REFRESH_TOKEN
+
+        json_body[Constants::REFRESH_TOKEN] = @refresh_token
+
+        response = get_response_from_server(json_body)
+
+        begin
+          parse_response(response)
+
+          if @id.nil?
+            generate_id()
+          end
+
+          store.save_token(user, self)
+        rescue SDKException => e
+          raise e
+        rescue StandardError => e
+          raise SDKException.new(nil, Constants::SAVE_TOKEN_ERROR, nil, e)
+        end
+        self
+      end
+
+      def parse_response(response)
+        response_json = JSON.parse(response)
+
+        unless response_json.key? Constants::ACCESS_TOKEN
+          raise SDKException.new(Constants::INVALID_CLIENT_ERROR, response_json[Constants::ERROR_KEY], nil, nil)
+        end
+
+        @access_token = response_json[Constants::ACCESS_TOKEN]
+
+        @expires_in = get_token_expiry_time(response_json)
+
+        @refresh_token = response_json[Constants::REFRESH_TOKEN] if response_json.key? Constants::REFRESH_TOKEN
+        self
+      end
+
+      def get_response_from_server(json_body)
+        accounts_url = Initializer.get_initializer.environment.accounts_url
+
+        url = URI(accounts_url)
+
+        http = Net::HTTP.new(url.host, url.port)
+
+        http.use_ssl = true
+
+        request = Net::HTTP::Post.new(url.request_uri)
+        request.add_field(Constants::USER_AGENT_KEY, Constants::USER_AGENT)
+        request.set_form(json_body, Constants::APPLICATION_FORM_URLENCODED)
+
+        response = http.request(request)
+
+        response.body
+      rescue StandardError => e
+        raise SDKException.new(nil, nil, nil, e)
+      end
+
+      def get_token_expiry_time(response_json)
+        expires_in = response_json[Constants::EXPIRES_IN]
+
+        expires_in *= 1000 unless response_json.key? Constants::EXPIRES_IN_SEC
+
+        expires_in += get_current_time_in_millis
+
+        expires_in
+      end
+
+      def get_current_time_in_millis
+        (Time.now.to_f * 1000).to_i
+      end
+
+      def generate_id
+        id = ""
+        email = Initializer.get_initializer.user.email
+        environment = Initializer.get_initializer.environment.name
+        id = id + (Constants::RUBY) + email[0..email.index(Constants::AT) - 1] + Constants::UNDERSCORE
+        id = id + environment + Constants::UNDERSCORE + @refresh_token[@refresh_token.length-4..@refresh_token.length-1]
+        @id = id
+      end
+    end
+  end
+end

data/src/com/zoho/api/authenticator/store/db_store.rb

@@ -0,0 +1,194 @@
+require_relative 'token_store'
+require_relative '../oauth_token'
+require_relative '../../../crm/api/util/constants'
+
+module ZOHOCRMSDK
+  module Store
+    # This class stores the user token details to the MySQL DataBase.
+    class DBStore < TokenStore
+      attr_reader :user_name, :password, :port_number, :host, :database_name, :table_name
+
+      # Creates an DBStore class instance with the specified parameters.
+      # @param host [string] A String containing the DataBase host name.
+      # @param database_name A String containing the DataBase name.
+      # @param table_name A String containing the table name.
+      # @param user_name A String containing the DataBase user name.
+      # @param password A String containing the DataBase password.
+      # @param port_number A String containing the DataBase port number.
+      def initialize(host: Constants::MYSQL_HOST, database_name: Constants::MYSQL_DATABASE_NAME ,table_name: Constants::MYSQL_TABLE_NAME,user_name: Constants::MYSQL_USER_NAME,password: '',port_number: Constants::MYSQL_PORT_NUMBER )
+        require 'mysql2' unless Object.const_defined?(:Mysql2)
+
+        @host = host
+        @database_name = database_name
+        @table_name = table_name
+        @user_name = user_name
+        @password = password
+        @port_number = port_number
+      end
+
+      def get_token(user, token)
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+
+        query = construct_db_query(user.email, token, false)
+
+        rs = con.query(query)
+
+        con.close
+
+        rs.each do |row|
+          oauthtoken = token
+
+          oauthtoken.client_id = row[Constants::CLIENT_ID]
+
+          oauthtoken.client_secret = row[Constants::CLIENT_SECRET]
+
+          oauthtoken.grant_token = row[Constants::GRANT_TOKEN]
+
+          oauthtoken.redirect_url = row[Constants::REDIRECT_URL]
+
+          oauthtoken.refresh_token = row[Constants::REFRESH_TOKEN]
+
+          oauthtoken.id = row[Constants::ID]
+
+          oauthtoken.user_mail = row[Constants::USER_MAIL]
+
+          oauthtoken.access_token = row[Constants::ACCESS_TOKEN]
+
+          oauthtoken.expires_in = row[Constants::EXPIRY_TIME]
+
+          return oauthtoken
+        end
+        nil
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::GET_TOKEN_DB_ERROR, nil, e)
+      end
+
+      def save_token(user, token)
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+        if token.is_a? Authenticator::OAuthToken
+          query = "insert into #{token.table_name}(id,user_mail,client_id,client_secret,refresh_token,access_token,grant_token,expiry_time,redirect_url) values ('#{token.id}','#{user.email}','#{token.client_id}','#{token.client_secret}','#{token.refresh_token}','#{token.access_token}','#{token.grant_token}','#{token.expires_in}','#{token.redirect_url}')"
+          token.user_mail = user.email
+          delete_token(token)
+          con.query(query)
+        end
+
+        con.close
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::SAVE_TOKEN_DB_ERROR, nil, e)
+      end
+
+      def delete_token(token)
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+        if token.is_a? Authenticator::OAuthToken
+          query = construct_db_query(token.user_mail, token, true)
+
+          con.query(query)
+        end
+
+        con.close
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::DELETE_TOKEN_DB_ERROR, nil, e)
+      end
+
+      def delete_tokens
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+
+        query = "delete * from #{@table_name}"
+
+        con.query(query)
+
+        con.close
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::DELETE_TOKENS_DB_ERROR, nil, e)
+      end
+
+      def get_tokens
+        tokens = []
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+
+        query = "select * from #{@table_name}"
+
+        rs = con.query(query)
+
+        con.close
+
+        rs.each do |row|
+
+
+          oauthtoken = Authenticator::OAuthToken.new(client_id: row[Constants::CLIENT_ID],client_secret: row[Constants::CLIENT_SECRET],grant_token: grant_token,refresh_token: row[Constants::REFRESH_TOKEN])
+
+          oauthtoken.id = row[Constants::ID]
+
+          oauthtoken.user_mail = row[Constants::USER_MAIL]
+
+          oauthtoken.access_token = row[Constants::ACCESS_TOKEN]
+
+          oauthtoken.redirect_url = row[Constants::REDIRECT_URL]
+
+          oauthtoken.expires_in = row[Constants::EXPIRY_TIME]
+
+          tokens.push(oauthtoken)
+        end
+        tokens
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::GET_TOKENS_DB_ERROR, nil, e)
+      end
+
+      def construct_db_query(email, token, is_delete)
+        raise SDKException.new(Constants::USER_MAIL_NULL_ERROR, Constants::USER_MAIL_NULL_ERROR_MESSAGE) if email.nil?
+
+        query = is_delete ? 'delete from ' : 'select * from '
+
+        query += "#{@table_name} where user_mail='#{email}' and client_id='#{token.client_id}' and "
+
+        query += if token.grant_token.nil?
+                  "refresh_token='#{token.refresh_token}'"
+                else
+                  "grant_token='#{token.grant_token}'"
+                end
+
+        query
+      end
+      def get_token_by_id(id,token)
+        con = Mysql2::Client.new(host: @host, username: @user_name, password: @password, database: @database_name, port: @port_number)
+        
+        if token.is_a? Authenticator::OAuthToken
+          query = "select * from #{@table_name} where id='#{id}'" 
+
+          rs = con.query(query)
+
+          con.close
+
+          rs.each do |row|
+            if id == row[Constants::ID]
+              oauthtoken = token
+
+              oauthtoken.client_id = row[Constants::CLIENT_ID]
+
+              oauthtoken.client_secret = row[Constants::CLIENT_SECRET]
+
+              oauthtoken.grant_token = row[Constants::GRANT_TOKEN]
+
+              oauthtoken.redirect_url = row[Constants::REDIRECT_URL]
+
+              oauthtoken.refresh_token = row[Constants::REFRESH_TOKEN]
+
+              oauthtoken.id = row[Constants::ID]
+
+              oauthtoken.user_mail = row[Constants::USER_MAIL]
+
+              oauthtoken.access_token = row[Constants::ACCESS_TOKEN]
+
+              oauthtoken.expires_in = row[Constants::EXPIRY_TIME]
+
+              return oauthtoken
+            end
+          end
+        end
+        nil
+      rescue Mysql2::Error => e
+        raise SDKException.new(Constants::TOKEN_STORE, Constants::GET_TOKEN_BY_ID_DB_ERROR, nil, e)
+      end
+    end
+  end
+end