workato-connector-sdk 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 70241ed06aeae85f14f775ff2366ec4edbc54322aa034f0931dd987e9fd1fb2c
-  data.tar.gz: 1a99f10c8c3c4dfe036e591cddb81504980586b0fb48cd319f7cfd5e510a1512
+  metadata.gz: 053e48199f2b1406ef47a84fb43d7b7d0cc74d74ce0f1f1b650c01fc422bc683
+  data.tar.gz: e2155ca61c45230234b105ff0f4ab033205c86b49b35838d88e8a8e473a957b3
 SHA512:
-  metadata.gz: b74c573b4e5c00509b955a14477a132ce01cfa6824382f21e50646bc10ce5ee44f3f2bf9bbf06ad7e565ddedfd0cce8c5017ae31e28e3b470164600abbb29d6d
-  data.tar.gz: 1f1139718e2f18faa63087e3782c42a0bb0b3bae237086044873283bad8128f9ce42b26b3738d0a3a8ff22e529180a6d8df821f7c7eee9da68b85d3b97098340
+  metadata.gz: 4d34393bda1fc5eef5e138c7e742a2a05a7c1ced8fb5a4a8969a1770473c4534cf015c7096cf17a18eb69c669301ca8761ff51a88b3ee8a04ad4550467a23588
+  data.tar.gz: b293f65e21f45ef2bab573d687cd12cd77981a7375bec619ce45561ead9e96f895f633e65910dad7226213daacb38b95453c890b96e7b7a957b5085e5055ff94

data/README.md

@@ -248,8 +248,9 @@ Usage:
   workato edit <PATH>
 
 Options:
-  -k, [--key=KEY]                  # Path to file with encrypt/decrypt key. NOTE: key from WORKATO_CONNECTOR_MASTER_KEY has higher priority
-      [--verbose], [--no-verbose]  
+  -k, [--key=KEY]                  # Path to file with encrypt/decrypt key.
+                                   # NOTE: key from WORKATO_CONNECTOR_MASTER_KEY has higher priority
+      [--verbose], [--no-verbose]
 
 Edit encrypted file, e.g. settings.yaml.enc
 ```
@@ -261,6 +262,7 @@ Edit encrypted file, e.g. settings.yaml.enc
 ### 3.3 workato exec
 ```
 workato help exec
+
 Usage:
   workato exec <PATH>
 
@@ -268,7 +270,8 @@ 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
+  -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
@@ -286,7 +289,7 @@ Options:
       [--redirect-url=REDIRECT_URL]                          # OAuth2 callback url
       [--refresh-token=REFRESH_TOKEN]                        # OAuth2 refresh token
       [--debug], [--no-debug]
-      [--verbose], [--no-verbose]                    
+      [--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.
@@ -394,13 +397,14 @@ 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
+  -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]  
+      [--verbose], [--no-verbose]
 
 Implements OAuth Authorization Code flow
 ```
@@ -420,11 +424,13 @@ Options:
   -l, [--logo=LOGO]                # Path to connector logo: png or jpeg file
   -n, [--notes=NOTES]              # Release notes
   -c, [--connector=CONNECTOR]      # Path to connector source code
-      [--api-email=API_EMAIL]      # Email for accessing Workato API or set WORKATO_API_EMAIL environment variable
-      [--api-token=API_TOKEN]      # Token for accessing Workato API or set WORKATO_API_TOKEN environment variable
-      [--environment=ENVIRONMENT]  # Server to push connector code to
-                                   # Default: live
-                                   # Possible values: preview, preview-eu, live, live-eu
+      [--api-email=API_EMAIL]      # Email for accessing Workato API.
+                                   # If present overrides value from WORKATO_API_EMAIL environment variable.
+      [--api-token=API_TOKEN]      # Token for accessing Workato API.
+                                   # If present overrides value from WORKATO_API_TOKEN environment variable.
+      [--environment=ENVIRONMENT]  # Data center specific URL to push connector code.
+                                   # If present overrides value from WORKATO_BASE_URL environment variable.
+                                   # Examples: 'https://app.workato.com', 'https://app.eu.workato.com'
       [--folder=FOLDER]            # Folder ID if you what to push to folder other than Home
       [--verbose], [--no-verbose]
 

data/lib/workato/cli/edit_command.rb

@@ -1,3 +1,4 @@
+# typed: true
 # frozen_string_literal: true
 
 require 'active_support/encrypted_configuration'
@@ -33,9 +34,9 @@ module Workato
 
       private
 
-      attr_reader :key_path,
-                  :encrypted_file_path,
-                  :encrypted_config
+      attr_reader :key_path
+      attr_reader :encrypted_file_path
+      attr_reader :encrypted_config
 
       def ensure_encryption_key_present
         return if encrypted_config.key.present?

data/lib/workato/cli/exec_command.rb

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 require 'thor'
@@ -24,8 +25,8 @@ module Workato
 
       private
 
-      attr_reader :path,
-                  :options
+      attr_reader :path
+      attr_reader :options
 
       # rubocop:disable Style/GuardClause
       def load_from_default_files
@@ -46,11 +47,9 @@ module Workato
 
       def params
         @params ||= {
-          connector: connector,
           settings: settings,
           input: from_json(options[:input]),
-          output: from_json(options[:input]),
-          webhook_subscribe_output: from_json(options[:webhook_subscribe_output]).presence,
+          webhook_subscribe_output: from_json(options[:webhook_subscribe_output]),
           args: from_json(options[:args]).presence || [],
           extended_input_schema: from_json(options[:extended_input_schema]).presence || [],
           extended_output_schema: from_json(options[:extended_output_schema]).presence || [],
@@ -69,7 +68,7 @@ module Workato
       end
 
       def connector
-        Workato::Connector::Sdk::Connector.from_file(
+        @connector ||= Workato::Connector::Sdk::Connector.from_file(
           options[:connector] || Workato::Connector::Sdk::DEFAULT_CONNECTOR_PATH,
           settings
         )
@@ -85,16 +84,26 @@ module Workato
         )
         @settings = settings_store.read
 
-        Workato::Connector::Sdk::Operation.on_settings_updated = lambda do |message, new_settings|
-          $stdout.pause if verbose?
-          say('')
-          say(message)
-          loop do
-            answer = ask('Updated settings file with new connection attributes? (Yes or No)').to_s.downcase
-            break if %w[n no].include?(answer)
-            break settings_store.update(new_settings) if %w[y yes].include?(answer)
+        Workato::Connector::Sdk::Connection.on_settings_update = lambda do |message, &refresher|
+          begin
+            $stdout.pause if verbose?
+            say('')
+            say(message)
+
+            new_settings = refresher.call
+            break unless new_settings
+
+            loop do
+              answer = ask('Updated settings file with new connection attributes? (Yes or No)').to_s.downcase
+              break new_settings if %w[n no].include?(answer)
+              next unless %w[y yes].include?(answer)
+
+              settings_store.update(new_settings)
+              break new_settings
+            end
+          ensure
+            $stdout.resume if verbose?
           end
-          $stdout.resume if verbose?
         end
 
         @settings
@@ -103,9 +112,9 @@ module Workato
       def from_json(path, parse_json_times: false)
         old_parse_json_times = ActiveSupport.parse_json_times
         ::ActiveSupport.parse_json_times = parse_json_times
-        result = path ? ::ActiveSupport::JSON.decode(File.read(path)) : {}
+        path ? ::ActiveSupport::JSON.decode(File.read(path)) : {}
+      ensure
         ::ActiveSupport.parse_json_times = old_parse_json_times
-        result
       end
 
       def inspect_params(params)
@@ -121,30 +130,13 @@ module Workato
       end
 
       def execute_path
-        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
+        connector.invoke(path, params)
       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)
-      end
-
       def show_output(output)
         if options[:output].present?
           File.open(options[:output], 'w') do |f|

data/lib/workato/cli/generate_command.rb

@@ -1,3 +1,4 @@
+# typed: true
 # frozen_string_literal: true
 
 module Workato

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

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 module Workato

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

@@ -1,3 +1,4 @@
+# typed: true
 # frozen_string_literal: true
 
 module Workato

data/lib/workato/cli/main.rb

@@ -1,3 +1,4 @@
+# typed: true
 # frozen_string_literal: true
 
 require 'thor'
@@ -40,8 +41,8 @@ module Workato
                                  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"
+                          desc: "Path to file with encrypt/decrypt key.\n"\
+                                "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'
@@ -74,7 +75,7 @@ module Workato
 
       method_option :key, type: :string, aliases: '-k',
                           lazy_default: Workato::Connector::Sdk::DEFAULT_MASTER_KEY_PATH,
-                          desc: 'Path to file with encrypt/decrypt key. '\
+                          desc: "Path to file with encrypt/decrypt key.\n"\
                                 "NOTE: key from #{Workato::Connector::Sdk::DEFAULT_MASTER_KEY_ENV} has higher priority"
 
       def edit(path)
@@ -126,17 +127,20 @@ module Workato
                     lazy_default: Workato::Connector::Sdk::DEFAULT_CONNECTOR_PATH
       method_option :api_email,
                     type: :string,
-                    desc: 'Email for accessing Workato API or '\
-                          "set #{Workato::Connector::Sdk::WORKATO_API_EMAIL_ENV} environment variable"
+                    desc: "Email for accessing Workato API.\n"\
+                          "If present overrides value from #{Workato::Connector::Sdk::WORKATO_API_EMAIL_ENV} "\
+                          'environment variable.'
       method_option :api_token,
                     type: :string,
-                    desc: 'Token for accessing Workato API or ' \
-                          "set #{Workato::Connector::Sdk::WORKATO_API_TOKEN_ENV} environment variable"
+                    desc: "Token for accessing Workato API.\n" \
+                          "If present overrides value from #{Workato::Connector::Sdk::WORKATO_API_TOKEN_ENV} "\
+                          'environment variable.'
       method_option :environment,
                     type: :string,
-                    enum: Workato::CLI::PushCommand::ENVIRONMENTS.keys,
-                    default: 'live',
-                    desc: 'Server to push connector code to'
+                    desc: "Data center specific URL to push connector code.\n"\
+                          "If present overrides value from #{Workato::Connector::Sdk::WORKATO_BASE_URL_ENV} "\
+                          "environment variable.\n"\
+                          "Examples: 'https://app.workato.com', 'https://app.eu.workato.com'"
       method_option :folder,
                     type: :string,
                     desc: 'Folder ID if you what to push to folder other than Home'
@@ -167,7 +171,7 @@ module Workato
                     type: :string,
                     aliases: '-k',
                     lazy_default: Workato::Connector::Sdk::DEFAULT_MASTER_KEY_PATH,
-                    desc: 'Path to file with encrypt/decrypt key. '\
+                    desc: "Path to file with encrypt/decrypt key.\n"\
                           "NOTE: key from #{Workato::Connector::Sdk::DEFAULT_MASTER_KEY_ENV} has higher priority"
       method_option :port,
                     type: :string,
@@ -186,6 +190,35 @@ module Workato
           options: options
         ).call
       end
+
+      class << self
+        def print_options(shell, options, group_name = nil)
+          return if options.empty?
+
+          list = []
+          padding = options.map { |o| o.aliases.size }.max.to_i * 4
+
+          options.each do |option|
+            next if option.hide
+
+            description = []
+            description_lines = option.description ? option.description.split("\n") : []
+            first_line = description_lines.shift
+            description << [option.usage(padding), first_line ? "# #{first_line}" : '']
+            description_lines.each do |line|
+              description << ['', "# #{line}"]
+            end
+
+            list.concat(description)
+            list << ['', "# Default: #{option.default}"] if option.show_default?
+            list << ['', "# Possible values: #{option.enum.join(', ')}"] if option.enum
+          end
+
+          shell.say(group_name ? "#{group_name} options:" : 'Options:')
+          shell.print_table(list, indent: 2)
+          shell.say ''
+        end
+      end
     end
   end
 end

data/lib/workato/cli/oauth2_command.rb

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 require 'securerandom'
@@ -55,11 +56,11 @@ module Workato
 
       private
 
-      attr_reader :https,
-                  :base_url,
-                  :redirect_url,
-                  :port,
-                  :options
+      attr_reader :https
+      attr_reader :base_url
+      attr_reader :redirect_url
+      attr_reader :port
+      attr_reader :options
 
       def verbose?
         !!options[:verbose]

data/lib/workato/cli/push_command.rb

@@ -1,3 +1,4 @@
+# typed: true
 # frozen_string_literal: true
 
 require 'uri'
@@ -31,7 +32,9 @@ module Workato
 
       def initialize(options:)
         @options = options
-        @api_base_url = ENVIRONMENTS.fetch(options[:environment])
+        @api_base_url = ENVIRONMENTS.fetch(options[:environment]) do
+          options[:environment].presence || Workato::Connector::Sdk::WORKATO_BASE_URL
+        end
         @api_email = options[:api_email] || ENV[Workato::Connector::Sdk::WORKATO_API_EMAIL_ENV]
         @api_token = options[:api_token] || ENV[Workato::Connector::Sdk::WORKATO_API_TOKEN_ENV]
         @folder_id = options[:folder]
@@ -55,10 +58,10 @@ module Workato
 
       private
 
-      attr_reader :options,
-                  :api_token,
-                  :api_email,
-                  :api_base_url
+      attr_reader :options
+      attr_reader :api_token
+      attr_reader :api_email
+      attr_reader :api_base_url
 
       def verbose?
         @options[:verbose]

data/lib/workato/cli/schema_command.rb

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 module Workato
@@ -8,7 +9,6 @@ module Workato
       SAMPLE_TO_SCHEMA_SUPPORT_TYPES = %w[csv json].freeze
       CSV_SEPARATORS = %w[comma space tab colon semicolon pipe].freeze
 
-      WORKATO_BASE_URL = ENV['WORKATO_BASE_URL'] || 'https://app.workato.com'
       API_GENERATE_SCHEMA_PATH = '/api/sdk/generate_schema'
 
       def initialize(options:)
@@ -31,9 +31,9 @@ module Workato
 
       private
 
-      attr_reader :options,
-                  :api_token,
-                  :api_email
+      attr_reader :options
+      attr_reader :api_token
+      attr_reader :api_email
 
       def verbose?
         @options[:verbose]
@@ -52,7 +52,7 @@ module Workato
       end
 
       def sample_to_schema(sample)
-        url = "#{WORKATO_BASE_URL}#{API_GENERATE_SCHEMA_PATH}/#{sample.delete(:type)}"
+        url = "#{Workato::Connector::Sdk::WORKATO_BASE_URL}#{API_GENERATE_SCHEMA_PATH}/#{sample.delete(:type)}"
         response = RestClient.post(
           url,
           sample.to_json,
@@ -74,8 +74,7 @@ module Workato
         }
       end
 
-      private_constant :API_GENERATE_SCHEMA_PATH,
-                       :WORKATO_BASE_URL
+      private_constant :API_GENERATE_SCHEMA_PATH
     end
   end
 end

data/lib/workato/connector/sdk/account_properties.rb

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 require 'csv'

data/lib/workato/connector/sdk/action.rb

@@ -1,3 +1,4 @@
+# typed: strict
 # frozen_string_literal: true
 
 require_relative './operation'
@@ -7,27 +8,52 @@ module Workato
   module Connector
     module Sdk
       class Action < Operation
+        extend T::Sig
         using BlockInvocationRefinements
 
-        RETRY_DEFAULT_CODES = [429, 500, 502, 503, 504, 507].freeze
-        RETRY_DEFAULT_METHODS = %i[get head].freeze
-        RETRY_DELAY = 5.seconds
+        RETRY_DEFAULT_CODES = T.let([429, 500, 502, 503, 504, 507].freeze, T::Array[Integer])
+        RETRY_DEFAULT_METHODS = T.let(%i[get head].freeze, T::Array[Symbol])
+        RETRY_DELAY = T.let(5, Integer) # seconds
         MAX_RETRIES = 3
 
         MAX_REINVOKES = 5
 
-        def initialize(action:, connection: {}, methods: {}, settings: {}, object_definitions: nil)
+        sig do
+          params(
+            action: SorbetTypes::SourceHash,
+            methods: SorbetTypes::SourceHash,
+            connection: Connection,
+            object_definitions: T.nilable(ObjectDefinitions)
+          ).void
+        end
+        def initialize(action:, methods: {}, connection: Connection.new, object_definitions: nil)
           super(
             operation: action,
             connection: connection,
             methods: methods,
-            settings: settings,
             object_definitions: object_definitions
           )
 
+          @retries_left = T.let(0, Integer)
+          @retry_codes = T.let([], T::Array[Integer])
+          @retry_methods = T.let([], T::Array[String])
+          @retry_matchers = T.let([], T::Array[T.any(Symbol, String, Regexp)])
+
           initialize_retry
         end
 
+        sig do
+          params(
+            settings: T.nilable(SorbetTypes::SettingsHash),
+            input: SorbetTypes::OperationInputHash,
+            extended_input_schema: SorbetTypes::OperationSchema,
+            extended_output_schema: SorbetTypes::OperationSchema,
+            continue: T::Hash[T.any(Symbol, String), T.untyped],
+            block: T.nilable(SorbetTypes::OperationExecuteProc)
+          ).returns(
+            T.untyped
+          )
+        end
         def execute(settings = nil, input = {}, extended_input_schema = [], extended_output_schema = [], continue = {},
                     &block)
           raise InvalidDefinitionError, "'execute' block is required for action" unless block || action[:execute]
@@ -39,7 +65,7 @@ module Workato
 
             reinvoke_sleep if @reinvoke_after
 
-            @reinvoke_after = nil
+            reinvoke_reset
 
             result = super(
               settings,
@@ -52,15 +78,17 @@ module Workato
 
             break result unless @reinvoke_after
 
-            continue = @reinvoke_after[:continue]
+            continue = @reinvoke_after.continue
           end
         rescue RequestError => e
           raise e unless retry?(e)
 
           @retries_left -= 1
-          sleep(RETRY_DELAY) && retry
+          sleep(RETRY_DELAY)
+          retry
         end
 
+        sig { params(input: SorbetTypes::OperationInputHash).returns(T::Hash[T.any(String, Symbol), T.untyped]) }
         def invoke(input = {})
           extended_schema = extended_schema(nil, input)
           config_schema = Schema.new(schema: config_fields_schema)
@@ -72,63 +100,93 @@ module Workato
           apply_output_schema(output, output_schema)
         end
 
+        sig do
+          params(
+            continue: T::Hash[T.untyped, T.untyped],
+            temp_output: T.nilable(T::Hash[T.untyped, T.untyped])
+          ).void
+        end
         def checkpoint!(continue:, temp_output: nil)
           # no-op
         end
 
-        def reinvoke_after(seconds:, continue:, temp_output: nil)
+        sig do
+          params(
+            seconds: Integer,
+            continue: T::Hash[T.untyped, T.untyped],
+            temp_output: T.nilable(T::Hash[T.untyped, T.untyped])
+          ).void
+        end
+        def reinvoke_after(seconds:, continue:, temp_output: nil) # rubocop:disable Lint/UnusedMethodArgument
+          @reinvokes_remaining = T.let(@reinvokes_remaining, T.nilable(Integer))
           @reinvokes_remaining = (@reinvokes_remaining ? @reinvokes_remaining - 1 : reinvoke_limit)
-          @reinvoke_after = {
+          @reinvoke_after = ReinvokeAfter.new(
             seconds: seconds,
-            continue: continue,
-            temp_output: temp_output
-          }
+            continue: continue
+          )
         end
 
         private
 
+        sig { returns(T::Array[T.any(Symbol, String, Regexp, Integer)]) }
         def retry_on_response
           Array(action[:retry_on_response])
         end
 
+        sig { returns(T::Array[T.any(Symbol, String, Regexp, Integer)]) }
         def retry_on_request
           Array(action[:retry_on_request])
         end
 
+        sig { returns(T.nilable(Integer)) }
         def max_retries
           action[:max_retries]
         end
 
+        sig { void }
         def initialize_retry
-          @retries_left = 0
           return if retry_on_response.blank?
 
-          @retry_codes = []
-          @retry_matchers = []
           retry_on_response.each { |m| m.is_a?(::Integer) ? @retry_codes << m : @retry_matchers << m }
           @retry_codes = RETRY_DEFAULT_CODES if @retry_codes.empty?
           @retry_methods = (retry_on_request.presence || RETRY_DEFAULT_METHODS).map(&:to_s).map(&:downcase)
           @retries_left = [[max_retries.is_a?(::Integer) && max_retries || MAX_RETRIES, MAX_RETRIES].min, 0].max
         end
 
+        sig { params(exception: RequestError).returns(T::Boolean) }
         def retry?(exception)
-          return unless @retries_left.positive?
-          return unless @retry_codes.include?(exception.code.to_i)
-          return unless @retry_matchers.empty? || @retry_matchers.any? do |m|
+          return false unless @retries_left.positive?
+          return false unless @retry_codes.include?(exception.code.to_i)
+          return false unless @retry_matchers.empty? || @retry_matchers.any? do |m|
             m === exception.message || m === exception.response
           end
 
           @retry_methods.include?(exception.method.to_s.downcase)
         end
 
+        sig { void }
         def reinvoke_sleep
-          sleep((ENV['WAIT_REINVOKE_AFTER'].presence || @reinvoke_after[:seconds]).to_f)
+          sleep((ENV['WAIT_REINVOKE_AFTER'].presence || T.must(@reinvoke_after).seconds).to_f)
         end
 
+        sig { returns(Integer) }
         def reinvoke_limit
+          @reinvoke_limit = T.let(@reinvoke_limit, T.nilable(Integer))
           @reinvoke_limit ||= (ENV['MAX_REINVOKES'].presence || MAX_REINVOKES).to_i
         end
 
+        sig { void }
+        def reinvoke_reset
+          @reinvoke_after = T.let(nil, T.nilable(ReinvokeAfter))
+        end
+
+        class ReinvokeAfter < T::Struct
+          prop :seconds, T.any(Float, Integer)
+          prop :continue, T::Hash[T.untyped, T.untyped]
+        end
+
+        private_constant :ReinvokeAfter
+
         alias action operation
       end
     end

data/lib/workato/connector/sdk/block_invocation_refinements.rb

@@ -1,3 +1,4 @@
+# typed: false
 # frozen_string_literal: true
 
 module Workato