workato-connector-sdk 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fcda7a1bda0d422376d4f8345bb63c8e7016d62432cb7d5b8838689347f7f18d
-  data.tar.gz: 8e9a38245c57162d41dbdc6ebd7385f464dc3807848decf02eb295b5a8a6d490
+  metadata.gz: b958407bc907daf631e2bd91ecb5c002f911906e3bf8a88c00aedd612fe80bbe
+  data.tar.gz: 1445969d82d648667e1de62809722625d06e40a30bb4bec565f56f1b7ce113da
 SHA512:
-  metadata.gz: fdc43f6ba0e21396c896afc879883b145306a6f03a1f45701b734d066f6431d66b34dc105d5a79bbfcb3e7b1409a2caab954e04e62e1ae395bbd06632e8a8a3f
-  data.tar.gz: 22fcdc88c6632b992481dd5f6b910ba8ec85b9ba72ea1a3a4345ee3d476aed7b534556b8ac8bc1cfae0f68b4f3be293328739d62d7cd248d371f1301f4e470e9
+  metadata.gz: 4ad8b4ef4ed273ad65573acb5a4e45dd1e0772b24a4de82cc4c40a3fe0a926c8f93f067eab0df2a6f23b0f94317f2d4bcaad588d203f9323417d7a973084a1fe
+  data.tar.gz: 879463eea7f3e0b9ef01ce1a5019eafbc8af18610306541ff97f6b62520869e0db8d1de1c0a61b9bb3c233c0932dc780e1ec04fd61704350ece80e9fbf9c764d

data/README.md

@@ -15,6 +15,7 @@ This guide below showcases how you can do the following things:
 1. Install [RVM ("Ruby Version Manager")](http://rvm.io/) or a Ruby manager of your choice. You can find more at [here](https://www.ruby-lang.org/en/documentation/installation/)
 2. Choose between Ruby versions `2.4.10` `2.5.X` or `2.7.X`. Our preferred version is `2.7.X`.
 3. Verify you're running a valid ruby version. Do this by running either `ruby -v` or the commands within your version manager. i.e., `rvm current` if you have installed RVM.
+4. For Windows you need tzinfo-data gem installed as well. `gem install tzinfo-data`
    
 ```bash
 ruby -v
@@ -38,7 +39,8 @@ Commands:
   workato generate <SUBCOMMAND>  # Generates code from template
   workato help [COMMAND]         # Describe available commands or one specific command
   workato new <CONNECTOR_PATH>   # Inits new connector folder
-  workato push <FOLDER>          # Upload and release connector's code
+  workato oauth2                 # Implements OAuth Authorization Code flow
+  workato push                   # Upload and release connector's code
 
 Options:
   [--verbose], [--no-verbose]  
@@ -124,11 +126,11 @@ You don't need to create this file. It'll be created later on. This file just ho
 
 ### 2.5 README.MD
 This file shows up on your Github project (or other Git software you use). Use it to document what your connector does! **Not created via `workato new <PATH>` commands.**
-When you use the `workato push <FOLDER>` command to sync your connector with your Workato workspace, this is the default file for your connector's description.
+When you use the `workato push` command to sync your connector with your Workato workspace, this is the default file for your connector's description.
 
 ### 2.6 logo.png
 The logo of your connector. **Not created via `workato new <PATH>` commands.**
-When you use the `workato push <FOLDER>` command to sync your connector with your Workato workspace, this is the default file for your connector's logo.
+When you use the `workato push` command to sync your connector with your Workato workspace, this is the default file for your connector's logo.
 
 ### 2.7 connector.rb
 Well, this is your actual connector code. This file should be a replica of your connector code in Workato.
@@ -223,7 +225,8 @@ Commands:
   workato generate <SUBCOMMAND>  # Generates code from template
   workato help [COMMAND]         # Describe available commands or one specific command
   workato new <CONNECTOR_PATH>   # Inits new connector folder
-  workato push <FOLDER>          # Upload and release connector's code
+  workato oauth2                 # Implements OAuth Authorization Code flow
+  workato push                   # Upload and release connector's code
 
 Options:
   [--verbose], [--no-verbose]  
@@ -258,7 +261,6 @@ Edit encrypted file, e.g. settings.yaml.enc
 ### 3.3 workato exec
 ```
 workato help exec
-
 Usage:
   workato exec <PATH>
 
@@ -269,6 +271,7 @@ Options:
   -k, [--key=KEY]                                        # Path to file with encrypt/decrypt key. NOTE: key from WORKATO_CONNECTOR_MASTER_KEY has higher priority
   -i, [--input=INPUT]                                    # Path to file with input JSON
       [--closure=CLOSURE]                                # Path to file with next poll closure JSON
+      [--continue=CONTINUE]                              # Path to file with next multistep action continue closure JSON
   -a, [--args=ARGS]                                      # Path to file with method arguments JSON
       [--extended-input-schema=EXTENDED_INPUT_SCHEMA]    # Path to file with extended input schema definition JSON
       [--extended-output-schema=EXTENDED_OUTPUT_SCHEMA]  # Path to file with extended output schema definition JSON
@@ -278,8 +281,11 @@ Options:
       [--webhook-headers=WEBHOOK_HEADERS]                # Path to file with webhook headers JSON
       [--webhook-url=WEBHOOK_URL]                        # Webhook URL for automatic webhook subscription
   -o, [--output=OUTPUT]                                  # Write output to JSON file
-      [--debug], [--no-debug]
-      [--verbose], [--no-verbose]
+      [--oauth2-code=OAUTH2_CODE]                        # OAuth2 code exchange to tokens pair
+      [--redirect-url=REDIRECT_URL]                      # OAuth2 callback url
+      [--refresh-token=REFRESH_TOKEN]                    # OAuth2 refresh token
+      [--debug], [--no-debug]                            
+      [--verbose], [--no-verbose]                        
 
 Description:
   The 'workato exec' executes connector's lambda block at <PATH>. Lambda's parameters can be provided if needed, see options part.
@@ -357,12 +363,36 @@ Please select default HTTP mocking behavior suitable for your project?
 
 - `simple` means your HTTP requests will be stored in plain text.
 
-### 3.6 workato push
+### 3.6 workato oauth2
+```
+workato help oauth2
+
+Usage:
+  workato oauth2
+
+Options:
+  -c, [--connector=CONNECTOR]      # Path to connector source code
+  -s, [--settings=SETTINGS]        # Path to plain or encrypted file with connection configs, passwords, tokens, secrets etc
+  -n, [--connection=CONNECTION]    # Connection name if settings file contains multiple settings
+  -k, [--key=KEY]                  # Path to file with encrypt/decrypt key. NOTE: key from WORKATO_CONNECTOR_MASTER_KEY has higher priority
+      [--port=PORT]                # Listen requests on specific port
+                                   # Default: 45555
+      [--ip=IP]                    # Listen requests on specific interface
+                                   # Default: 127.0.0.1
+      [--https], [--no-https]      # Start HTTPS server using self-signed certificate
+      [--verbose], [--no-verbose]  
+
+Implements OAuth Authorization Code flow
+```
+
+Use this to implement the OAuth2 Authorization code grant flow for applicable connectors. Applicable connectors are ones where the connection hash has `type: 'oauth2`. For more information, check out our guide on our [main docs site](https://docs.workato.com/developing-connectors/sdk/guides/authentication/oauth/auth-code.html#how-to-guide-oauth-2-0-authorization-code-variant).
+
+### 3.7 workato push
 ```
 workato help push
 
 Usage:
-  workato push <FOLDER>
+  workato push
 
 Options:
   -t, [--title=TITLE]              # Connector title on the Workato Platform
@@ -375,7 +405,8 @@ Options:
       [--environment=ENVIRONMENT]  # Server to push connector code to
                                    # Default: live
                                    # Possible values: preview, preview-eu, live, live-eu
-      [--verbose], [--no-verbose]  
+      [--folder=FOLDER]            # Folder ID if you what to push to folder other than Home
+      [--verbose], [--no-verbose]
 
 Upload and release connector's code
 ```
@@ -488,7 +519,7 @@ workato exec test #Output of the test: lambda function should be shown
 > *Note*: You may also see a intermediary command from the Gem asking if you'd like to refresh your access tokens. This is done when HTTP requests are made which have a response that triggers the `refresh_on` block. Selecting yes would cause the Gem to update your settings file with the latest auth credentials.
 
 ### 4.3 Example: Testing your connection on CLI - OAuth 2 - auth code grant flows
-For auth code grant flows, the Workato Gem doesn't simulate the browser popup so you'll need to supply the access token and refresh tokens directly!
+For auth code grant flows, the Workato Gem allows you to simulate the OAuth2 flow using the `workato oauth2` command.
 
 ```ruby
 {
@@ -572,27 +603,24 @@ For auth code grant flows, the Workato Gem doesn't simulate the browser popup so
 and a `settings.yaml.enc` or `settings.yaml` file with the following details
 
 ```yaml
-My Valid Connection:
-  access_token: "valid_access_token"
-  refresh_token: "valid_refresh_token"
-  user_key: "valid_user_key"
-  client_id: "valid_client_id"
-  client_secret: "valid_client_secret"
-Invalid Access Token:
-  access_token: "invalid_access_token"
-  refresh_token: "valid_refresh_token"
-  user_key: "valid_user_key"
-  client_id: "valid_client_id"
-  client_secret: "valid_client_secret"
+client_id: valid_client_id
+client_secret: valid_client_secret
 ```
 
-You can now run the following commands to verify that the `test:` lambda function you have defined is working:
+You can now run the following commands to go through the OAuth2 Authorization code flow which includes a browser popup.
 ```bash
-workato exec test  --connection='My Valid Connection' #Output of the test: lambda function should be shown
-workato exec test  --connection='Invalid Access Token' --verbose #You should see a see a set of request where the test is tried, access token receives 401 and the sequence to refresh the gem using the refresh key is used! This allows you to test how your connector refreshes a connection.
+workato oauth2 
 ```
 
-> *Note*: `--verbose` can be used to details everything, including the HTTP requests. You may also see a intermediary command from the Gem asking if you'd like to refresh your access tokens. This is done when HTTP requests are made which have a response that triggers the `refresh_on` block. Selecting yes would cause the Gem to update your settings file with the latest auth credentials.
+https://user-images.githubusercontent.com/25265275/137942408-812fa6ad-353f-4ea2-bf37-f804e2ff7b04.mov
+
+
+> *Note*: `--verbose` can be used to detail everything, including the HTTP requests.
+
+
+Now after you've successfully gone through the flow, you may be use the same `workato exec test` command to verify you're applying your token properly in your requests! Depending on when you received your token, you may also see a intermediary command from the Gem asking if you'd like to refresh your access tokens (if it has expired). This is done when HTTP requests are made which have a response that triggers the `refresh_on` block. Selecting "Yes" would cause the Gem to update your settings file with the latest auth credentials.
+
+Take note, you may also use `workato exec` to execute lambdas in your `authorization` hash like `acquire` and `refresh`. **That said, we highlight recommend you use `workato exec test` and `workato oauth2` which handle the updating of your `settings.yaml` file automatically.**
 
 ### Example: Testing a sample action on CLI
 Continuing from the previous example, let's take a look at a simple action and invoke the individual lambda functions.
@@ -631,7 +659,7 @@ Continuing from the previous example, let's take a look at a simple action and i
       end,
       
       
-      execute: lambda do |connection, input, input_schema, output_schema|
+      execute: lambda do |connection, input, input_schema, output_schema, closure|
         get("/api/v2/customers",input)
       end,
       
@@ -1086,8 +1114,7 @@ jobs:
     #   env: 
     #     WORKATO_API_EMAIL: ${{ secrets.WORKATO_DEV_ENVIRONMENT_API_EMAIL}}
     #     WORKATO_API_TOKEN: ${{ secrets.WORKATO_DEV_ENVIRONMENT_API_TOKEN}} 
-    #     WORKATO_API_FOLDER: ${{ secrets.WORKATO_DEV_ENVIRONMENT_API_HOME_FOLDER}} 
-    #   run: workato push $WORKATO_API_FOLDER
+    #   run: workato push
 ```
 
 You may also add more Github actions for rubocop to automate this.

data/exe/workato

@@ -2,4 +2,15 @@
 # frozen_string_literal: true
 
 require 'workato/cli/main'
-Workato::CLI::Main.start
+
+begin
+  Workato::CLI::Main.start
+rescue Workato::CLI::ExecCommand::DebugExceptionError => e
+  raise e.cause
+rescue SystemExit, SignalException
+  raise
+rescue Exception => e # rubocop:disable Lint/RescueException
+  puts ''
+  puts e.message
+  exit(false)
+end

data/lib/workato/cli/edit_command.rb

@@ -28,7 +28,7 @@ module Workato
 
         puts 'File encrypted and saved.'
       rescue ActiveSupport::MessageEncryptor::InvalidMessage
-        puts "Couldn't decrypt #{encrypted_file_path}. Perhaps you passed the wrong key?"
+        raise "Couldn't decrypt #{encrypted_file_path}. Perhaps you passed the wrong key?"
       end
 
       private
@@ -63,8 +63,6 @@ module Workato
         yield
       rescue Interrupt
         puts 'Aborted changing file: nothing saved.'
-      rescue ActiveSupport::EncryptedFile::MissingKeyError => e
-        puts e.message
       end
     end
   end

data/lib/workato/cli/exec_command.rb

@@ -1,13 +1,14 @@
 # frozen_string_literal: true
 
 require 'thor'
-require 'ruby-progressbar'
 
 module Workato
   module CLI
     class ExecCommand
       include Thor::Shell
 
+      DebugExceptionError = Class.new(StandardError)
+
       def initialize(path:, options:)
         @path = path
         @options = options
@@ -55,10 +56,14 @@ module Workato
           extended_output_schema: from_json(options[:extended_output_schema]).presence || [],
           config_fields: from_json(options[:config_fields]),
           closure: from_json(options[:closure], parse_json_times: true).presence,
+          continue: from_json(options[:continue], parse_json_times: true),
           payload: from_json(options[:webhook_payload]),
           params: from_json(options[:webhook_params]),
           headers: from_json(options[:webhook_headers]),
           webhook_url: options[:webhook_url],
+          oauth2_code: options[:oauth2_code],
+          redirect_url: options[:redirect_url],
+          refresh_token: options[:refresh_token],
           recipe_id: SecureRandom.uuid
         }
       end
@@ -119,16 +124,25 @@ module Workato
         methods = path.split('.')
         method = methods.pop
         object = methods.inject(params[:connector]) { |obj, m| obj.public_send(m) }
+        output = invoke_method(object, method)
+        if output.respond_to?(:invoke)
+          invoke_method(output, :invoke)
+        else
+          output
+        end
+      rescue Exception => e # rubocop:disable Lint/RescueException
+        raise DebugExceptionError, e if options[:debug]
+
+        raise
+      end
+
+      def invoke_method(object, method)
         parameters = object.method(method).parameters.reject { |p| p[0] == :block }.map(&:second)
         args = params.values_at(*parameters)
         if parameters.last == :args
           args = args.take(args.length - 1) + Array.wrap(args.last).flatten(1)
         end
         object.public_send(method, *args)
-      rescue Exception => e # rubocop:disable Lint/RescueException
-        raise e if options[:debug]
-
-        e.message
       end
 
       def show_output(output)
@@ -149,6 +163,8 @@ module Workato
       def with_progress
         return yield unless verbose?
 
+        require 'ruby-progressbar'
+
         say('')
 
         old_stdout = $stdout

data/lib/workato/cli/generate_command.rb

@@ -70,8 +70,7 @@ module Workato
       def ensure_connector_source
         return if connector
 
-        error "Can't find connector source code"
-        exit
+        raise "Can't find connector source code"
       end
 
       def create_spec_file(type, name)

data/lib/workato/cli/generators/connector_generator.rb

@@ -30,7 +30,7 @@ module Workato
         def create_spec_files
           template('.rspec.erb', '.rspec')
 
-          @vcr_record_mode = ask(<<~HELP)
+          say(<<~HELP)
             Please select default HTTP mocking behavior suitable for your project?
 
             1 - secure. Cause an error to be raised for any unknown requests, all request recordings are encrypted.
@@ -39,8 +39,11 @@ module Workato
                         Example: VCR_RECORD_MODE=once bundle exec rspec spec/actions/test_action_spec.rb
 
             2 - simple. Record new interaction if it is a new request, requests are stored as plain text and expose secret tokens.
+
           HELP
 
+          @vcr_record_mode = ask('Your choice:')
+
           MasterKeyGenerator.new.call if @vcr_record_mode == '1'
 
           template('spec/spec_helper.rb.erb', 'spec/spec_helper.rb')

data/lib/workato/cli/main.rb

@@ -6,6 +6,7 @@ require_relative './exec_command'
 require_relative './edit_command'
 require_relative './generate_command'
 require_relative './push_command'
+require_relative './oauth2_command'
 require_relative './generators/connector_generator'
 require_relative './generators/master_key_generator'
 
@@ -42,6 +43,7 @@ module Workato
                           "NOTE: key from #{Workato::Connector::Sdk::DEFAULT_MASTER_KEY_ENV} has higher priority"
       method_option :input, type: :string, aliases: '-i', desc: 'Path to file with input JSON'
       method_option :closure, type: :string, desc: 'Path to file with next poll closure JSON'
+      method_option :continue, type: :string, desc: 'Path to file with next multistep action continue closure JSON'
       method_option :args, type: :string, aliases: '-a', desc: 'Path to file with method arguments JSON'
       method_option :extended_input_schema, type: :string,
                                             desc: 'Path to file with extended input schema definition JSON'
@@ -53,6 +55,9 @@ module Workato
       method_option :webhook_headers, type: :string, desc: 'Path to file with webhook headers JSON'
       method_option :webhook_url, type: :string, desc: 'Webhook URL for automatic webhook subscription'
       method_option :output, type: :string, aliases: '-o', desc: 'Write output to JSON file'
+      method_option :oauth2_code, type: :string, desc: 'OAuth2 code exchange to tokens pair'
+      method_option :redirect_url, type: :string, desc: 'OAuth2 callback url'
+      method_option :refresh_token, type: :string, desc: 'OAuth2 refresh token'
 
       method_option :debug, type: :boolean
 
@@ -95,7 +100,7 @@ module Workato
       desc 'generate <SUBCOMMAND>', 'Generates code from template'
       subcommand('generate', GenerateCommand)
 
-      desc 'push <FOLDER>', "Upload and release connector's code"
+      desc 'push', "Upload and release connector's code"
       method_option :title,
                     type: :string,
                     aliases: '-t',
@@ -130,10 +135,52 @@ module Workato
                     enum: Workato::CLI::PushCommand::ENVIRONMENTS.keys,
                     default: 'live',
                     desc: 'Server to push connector code to'
+      method_option :folder,
+                    type: :string,
+                    desc: 'Folder ID if you what to push to folder other than Home'
 
-      def push(folder)
+      def push
         PushCommand.new(
-          folder: folder,
+          options: options
+        ).call
+      end
+
+      desc 'oauth2', 'Implements OAuth Authorization Code flow'
+
+      method_option :connector,
+                    type: :string,
+                    aliases: '-c',
+                    desc: 'Path to connector source code',
+                    lazy_default: Workato::Connector::Sdk::DEFAULT_CONNECTOR_PATH
+      method_option :settings,
+                    type: :string,
+                    aliases: '-s',
+                    desc: 'Path to plain or encrypted file with connection configs, passwords, tokens, secrets etc',
+                    lazy_default: Workato::Connector::Sdk::DEFAULT_ENCRYPTED_SETTINGS_PATH
+      method_option :connection,
+                    type: :string,
+                    aliases: '-n',
+                    desc: 'Connection name if settings file contains multiple settings'
+      method_option :key,
+                    type: :string,
+                    aliases: '-k',
+                    lazy_default: Workato::Connector::Sdk::DEFAULT_MASTER_KEY_PATH,
+                    desc: 'Path to file with encrypt/decrypt key. '\
+                          "NOTE: key from #{Workato::Connector::Sdk::DEFAULT_MASTER_KEY_ENV} has higher priority"
+      method_option :port,
+                    type: :string,
+                    desc: 'Listen requests on specific port',
+                    default: Workato::CLI::OAuth2Command::DEFAULT_PORT
+      method_option :ip,
+                    type: :string,
+                    desc: 'Listen requests on specific interface',
+                    default: Workato::CLI::OAuth2Command::DEFAULT_ADDRESS
+      method_option :https,
+                    type: :boolean,
+                    desc: 'Start HTTPS server using self-signed certificate'
+
+      def oauth2
+        OAuth2Command.new(
           options: options
         ).call
       end

data/lib/workato/cli/oauth2_command.rb

@@ -0,0 +1,180 @@
+# frozen_string_literal: true
+
+require 'workato/web/app'
+
+module Workato
+  module CLI
+    class OAuth2Command
+      include Thor::Shell
+
+      AWAIT_CODE_TIMEOUT_INTERVAL = 180 # seconds
+      AWAIT_CODE_SLEEP_INTERVAL = 5 # seconds
+
+      DEFAULT_ADDRESS = '127.0.0.1'
+      DEFAULT_PORT = '45555'
+
+      def initialize(options: {})
+        @options = options
+        @port = options[:port] || DEFAULT_PORT
+        @https = options[:https].is_true?
+        @base_url = "#{https ? 'https' : 'http'}://localhost:#{port}"
+        @redirect_url = "#{base_url}#{Workato::Web::App::CALLBACK_PATH}"
+      end
+
+      def call
+        require_gems
+        start_webrick
+
+        say 'Local server is running. Allow following redirect_url in your OAuth2 provider:'
+        say "\n"
+        say redirect_url
+        say ''
+
+        say_status :success, "Open #{authorize_url} in browser"
+        Launchy.open(authorize_url) do |exception|
+          raise "Attempted to open #{authorize_url} and failed because #{exception}"
+        end
+
+        code = await_code
+        say_status :success, "Receive OAuth2 code=#{code}"
+
+        tokens = acquire_token(code)
+        say_status :success, 'Receive OAuth2 tokens'
+        jj tokens if verbose?
+
+        settings_store.update(tokens)
+        say_status :success, 'Update settings file'
+      rescue Timeout::Error
+        raise "Have not received callback from OAuth2 provider in #{AWAIT_CODE_TIMEOUT_INTERVAL} seconds. Aborting!"
+      rescue Errno::EADDRINUSE
+        raise "Port #{port} already in use. Try to use different port with --port=#{rand(10_000..65_664)}"
+      ensure
+        stop_webrick
+      end
+
+      private
+
+      attr_reader :https,
+                  :base_url,
+                  :redirect_url,
+                  :port,
+                  :options
+
+      def verbose?
+        !!options[:verbose]
+      end
+
+      def require_gems
+        require 'oauth2'
+        require 'launchy'
+        require 'rack'
+      end
+
+      def start_webrick
+        @thread = Thread.start do
+          Rack::Handler::WEBrick.run(
+            Workato::Web::App.new,
+            {
+              Port: port,
+              BindAddress: options[:ip] || DEFAULT_ADDRESS,
+              SSLEnable: https,
+              SSLVerifyClient: OpenSSL::SSL::VERIFY_NONE,
+              SSLCertName: [%w[CN localhost]]
+            }.tap do |server_options|
+              unless verbose?
+                server_options[:AccessLog] = []
+                server_options[:Logger] = WEBrick::Log.new($stderr, 0)
+              end
+            end
+          )
+        end
+        @thread.abort_on_exception = true
+      end
+
+      def stop_webrick
+        Rack::Handler::WEBrick.shutdown
+        @thread.exit
+      end
+
+      def client
+        @client ||= OAuth2::Client.new(
+          connector.connection.authorization.client_id(settings),
+          connector.connection.authorization.client_secret(settings),
+          site: connector.connection.base_uri(settings),
+          token_url: connector.connection.authorization.token_url(settings),
+          redirect_uri: redirect_url
+        )
+      end
+
+      def authorize_url
+        return @authorize_url if defined?(@authorize_url)
+
+        @authorize_url =
+          if (authorization_url = connector.connection.authorization.authorization_url(settings))
+            params = {
+              client_id: connector.connection.authorization.client_id(settings),
+              redirect_uri: redirect_url
+            }
+            uri = URI(authorization_url)
+            uri.query = params.with_indifferent_access.merge(Rack::Utils.parse_nested_query(uri.query || '')).to_param
+            uri.to_s
+          end
+      end
+
+      def settings_store
+        @settings_store ||= Workato::Connector::Sdk::Settings.new(
+          path: options[:settings],
+          name: options[:connection],
+          key_path: options[:key]
+        )
+      end
+
+      def settings
+        @settings ||= settings_store.read
+      end
+
+      def connector
+        @connector ||= Workato::Connector::Sdk::Connector.from_file(
+          options[:connector] || Workato::Connector::Sdk::DEFAULT_CONNECTOR_PATH
+        )
+      end
+
+      def await_code
+        code_uri = URI("#{base_url}#{Workato::Web::App::CODE_PATH}")
+
+        Timeout.timeout(AWAIT_CODE_TIMEOUT_INTERVAL) do
+          loop do
+            response = get(code_uri) rescue nil
+            break response if response.present?
+
+            sleep(AWAIT_CODE_SLEEP_INTERVAL)
+          end
+        end
+      end
+
+      def acquire_token(code)
+        if connector.source.dig(:connection, :authorization, :acquire)
+          tokens, _, extra_settings = connector.connection.authorization.acquire(settings, await_code, redirect_url)
+          tokens ||= {}
+          extra_settings ||= {}
+          extra_settings.merge(tokens)
+        else
+          client.auth_code.get_token(code).to_hash
+        end
+      end
+
+      def get(uri)
+        http = Net::HTTP.new(uri.host, uri.port)
+        if https
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        end
+
+        request = Net::HTTP::Get.new(uri.request_uri)
+
+        response = http.request(request)
+        response.body
+      end
+    end
+  end
+end