context.dev 0.0.2

data/lib/context_dev.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+# Standard libraries.
+# rubocop:disable Lint/RedundantRequireStatement
+require "English"
+require "base64"
+require "cgi"
+require "date"
+require "erb"
+require "etc"
+require "json"
+require "net/http"
+require "openssl"
+require "pathname"
+require "rbconfig"
+require "securerandom"
+require "set"
+require "stringio"
+require "time"
+require "uri"
+# rubocop:enable Lint/RedundantRequireStatement
+
+# We already ship the preferred sorbet manifests in the package itself.
+# `tapioca` currently does not offer us a way to opt out of unnecessary compilation.
+if Object.const_defined?(:Tapioca) &&
+   caller.chain([$PROGRAM_NAME]).chain(ARGV).any?(/tapioca/) &&
+   ARGV.none?(/dsl/)
+  return
+end
+
+# Gems.
+require "connection_pool"
+
+# Package files.
+require_relative "context_dev/version"
+require_relative "context_dev/internal/util"
+require_relative "context_dev/internal/type/converter"
+require_relative "context_dev/internal/type/unknown"
+require_relative "context_dev/internal/type/boolean"
+require_relative "context_dev/internal/type/file_input"
+require_relative "context_dev/internal/type/enum"
+require_relative "context_dev/internal/type/union"
+require_relative "context_dev/internal/type/array_of"
+require_relative "context_dev/internal/type/hash_of"
+require_relative "context_dev/internal/type/base_model"
+require_relative "context_dev/internal/type/base_page"
+require_relative "context_dev/internal/type/request_parameters"
+require_relative "context_dev/internal"
+require_relative "context_dev/request_options"
+require_relative "context_dev/file_part"
+require_relative "context_dev/errors"
+require_relative "context_dev/internal/transport/base_client"
+require_relative "context_dev/internal/transport/pooled_net_requester"
+require_relative "context_dev/client"
+require_relative "context_dev/models/brand_ai_product_params"
+require_relative "context_dev/models/brand_ai_product_response"
+require_relative "context_dev/models/brand_ai_products_params"
+require_relative "context_dev/models/brand_ai_products_response"
+require_relative "context_dev/models/brand_ai_query_params"
+require_relative "context_dev/models/brand_ai_query_response"
+require_relative "context_dev/models/brand_fonts_params"
+require_relative "context_dev/models/brand_fonts_response"
+require_relative "context_dev/models/brand_identify_from_transaction_params"
+require_relative "context_dev/models/brand_identify_from_transaction_response"
+require_relative "context_dev/models/brand_prefetch_by_email_params"
+require_relative "context_dev/models/brand_prefetch_by_email_response"
+require_relative "context_dev/models/brand_prefetch_params"
+require_relative "context_dev/models/brand_prefetch_response"
+require_relative "context_dev/models/brand_retrieve_by_email_params"
+require_relative "context_dev/models/brand_retrieve_by_email_response"
+require_relative "context_dev/models/brand_retrieve_by_isin_params"
+require_relative "context_dev/models/brand_retrieve_by_isin_response"
+require_relative "context_dev/models/brand_retrieve_by_name_params"
+require_relative "context_dev/models/brand_retrieve_by_name_response"
+require_relative "context_dev/models/brand_retrieve_by_ticker_params"
+require_relative "context_dev/models/brand_retrieve_by_ticker_response"
+require_relative "context_dev/models/brand_retrieve_naics_params"
+require_relative "context_dev/models/brand_retrieve_naics_response"
+require_relative "context_dev/models/brand_retrieve_params"
+require_relative "context_dev/models/brand_retrieve_response"
+require_relative "context_dev/models/brand_retrieve_simplified_params"
+require_relative "context_dev/models/brand_retrieve_simplified_response"
+require_relative "context_dev/models/brand_screenshot_params"
+require_relative "context_dev/models/brand_screenshot_response"
+require_relative "context_dev/models/brand_styleguide_params"
+require_relative "context_dev/models/brand_styleguide_response"
+require_relative "context_dev/models/brand_web_scrape_html_params"
+require_relative "context_dev/models/brand_web_scrape_html_response"
+require_relative "context_dev/models/brand_web_scrape_images_params"
+require_relative "context_dev/models/brand_web_scrape_images_response"
+require_relative "context_dev/models/brand_web_scrape_md_params"
+require_relative "context_dev/models/brand_web_scrape_md_response"
+require_relative "context_dev/models/brand_web_scrape_sitemap_params"
+require_relative "context_dev/models/brand_web_scrape_sitemap_response"
+require_relative "context_dev/models"
+require_relative "context_dev/resources/brand"

data/manifest.yaml

@@ -0,0 +1,17 @@
+dependencies:
+  - English
+  - base64
+  - cgi
+  - date
+  - erb
+  - etc
+  - json
+  - net/http
+  - openssl
+  - pathname
+  - rbconfig
+  - securerandom
+  - set
+  - stringio
+  - time
+  - uri

data/rbi/context_dev/client.rbi

@@ -0,0 +1,49 @@
+# typed: strong
+
+module ContextDev
+  class Client < ContextDev::Internal::Transport::BaseClient
+    DEFAULT_MAX_RETRIES = 2
+
+    DEFAULT_TIMEOUT_IN_SECONDS = T.let(60.0, Float)
+
+    DEFAULT_INITIAL_RETRY_DELAY = T.let(0.5, Float)
+
+    DEFAULT_MAX_RETRY_DELAY = T.let(8.0, Float)
+
+    sig { returns(String) }
+    attr_reader :api_key
+
+    sig { returns(ContextDev::Resources::Brand) }
+    attr_reader :brand
+
+    # @api private
+    sig { override.returns(T::Hash[String, String]) }
+    private def auth_headers
+    end
+
+    # Creates and returns a new client for interacting with the API.
+    sig do
+      params(
+        api_key: T.nilable(String),
+        base_url: T.nilable(String),
+        max_retries: Integer,
+        timeout: Float,
+        initial_retry_delay: Float,
+        max_retry_delay: Float
+      ).returns(T.attached_class)
+    end
+    def self.new(
+      # Defaults to `ENV["CONTEXT_DEV_API_KEY"]`
+      api_key: ENV["CONTEXT_DEV_API_KEY"],
+      # Override the default base URL for the API, e.g.,
+      # `"https://api.example.com/v2/"`. Defaults to `ENV["CONTEXT_DEV_BASE_URL"]`
+      base_url: ENV["CONTEXT_DEV_BASE_URL"],
+      # Max number of retries to attempt after a failed retryable request.
+      max_retries: ContextDev::Client::DEFAULT_MAX_RETRIES,
+      timeout: ContextDev::Client::DEFAULT_TIMEOUT_IN_SECONDS,
+      initial_retry_delay: ContextDev::Client::DEFAULT_INITIAL_RETRY_DELAY,
+      max_retry_delay: ContextDev::Client::DEFAULT_MAX_RETRY_DELAY
+    )
+    end
+  end
+end

data/rbi/context_dev/errors.rbi

@@ -0,0 +1,205 @@
+# typed: strong
+
+module ContextDev
+  module Errors
+    class Error < StandardError
+      sig { returns(T.nilable(StandardError)) }
+      attr_accessor :cause
+    end
+
+    class ConversionError < ContextDev::Errors::Error
+      sig { returns(T.nilable(StandardError)) }
+      def cause
+      end
+
+      # @api private
+      sig do
+        params(
+          on: T::Class[StandardError],
+          method: Symbol,
+          target: T.anything,
+          value: T.anything,
+          cause: T.nilable(StandardError)
+        ).returns(T.attached_class)
+      end
+      def self.new(on:, method:, target:, value:, cause: nil)
+      end
+    end
+
+    class APIError < ContextDev::Errors::Error
+      sig { returns(URI::Generic) }
+      attr_accessor :url
+
+      sig { returns(T.nilable(Integer)) }
+      attr_accessor :status
+
+      sig { returns(T.nilable(T::Hash[String, String])) }
+      attr_accessor :headers
+
+      sig { returns(T.nilable(T.anything)) }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: T.nilable(Integer),
+          headers: T.nilable(T::Hash[String, String]),
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        headers: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: nil
+      )
+      end
+    end
+
+    class APIConnectionError < ContextDev::Errors::APIError
+      sig { returns(NilClass) }
+      attr_accessor :status
+
+      sig { returns(NilClass) }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          headers: T.nilable(T::Hash[String, String]),
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        headers: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Connection error."
+      )
+      end
+    end
+
+    class APITimeoutError < ContextDev::Errors::APIConnectionError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          headers: T.nilable(T::Hash[String, String]),
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        headers: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Request timed out."
+      )
+      end
+    end
+
+    class APIStatusError < ContextDev::Errors::APIError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          headers: T.nilable(T::Hash[String, String]),
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.for(
+        url:,
+        status:,
+        headers:,
+        body:,
+        request:,
+        response:,
+        message: nil
+      )
+      end
+
+      sig { returns(Integer) }
+      attr_accessor :status
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          headers: T.nilable(T::Hash[String, String]),
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status:,
+        headers:,
+        body:,
+        request:,
+        response:,
+        message: nil
+      )
+      end
+    end
+
+    class BadRequestError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 400
+    end
+
+    class AuthenticationError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 401
+    end
+
+    class PermissionDeniedError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 403
+    end
+
+    class NotFoundError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 404
+    end
+
+    class ConflictError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 409
+    end
+
+    class UnprocessableEntityError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 422
+    end
+
+    class RateLimitError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = 429
+    end
+
+    class InternalServerError < ContextDev::Errors::APIStatusError
+      HTTP_STATUS = T.let((500..), T::Range[Integer])
+    end
+  end
+end

data/rbi/context_dev/file_part.rbi

@@ -0,0 +1,37 @@
+# typed: strong
+
+module ContextDev
+  class FilePart
+    sig { returns(T.any(Pathname, StringIO, IO, String)) }
+    attr_reader :content
+
+    sig { returns(T.nilable(String)) }
+    attr_reader :content_type
+
+    sig { returns(T.nilable(String)) }
+    attr_reader :filename
+
+    # @api private
+    sig { returns(String) }
+    private def read
+    end
+
+    sig { params(a: T.anything).returns(String) }
+    def to_json(*a)
+    end
+
+    sig { params(a: T.anything).returns(String) }
+    def to_yaml(*a)
+    end
+
+    sig do
+      params(
+        content: T.any(Pathname, StringIO, IO, String),
+        filename: T.nilable(T.any(Pathname, String)),
+        content_type: T.nilable(String)
+      ).returns(T.attached_class)
+    end
+    def self.new(content, filename: nil, content_type: nil)
+    end
+  end
+end

data/rbi/context_dev/internal/transport/base_client.rbi

@@ -0,0 +1,300 @@
+# typed: strong
+
+module ContextDev
+  module Internal
+    module Transport
+      # @api private
+      class BaseClient
+        extend ContextDev::Internal::Util::SorbetRuntimeSupport
+
+        abstract!
+
+        RequestComponents =
+          T.type_alias do
+            {
+              method: Symbol,
+              path: T.any(String, T::Array[String]),
+              query:
+                T.nilable(
+                  T::Hash[String, T.nilable(T.any(T::Array[String], String))]
+                ),
+              headers:
+                T.nilable(
+                  T::Hash[
+                    String,
+                    T.nilable(
+                      T.any(
+                        String,
+                        Integer,
+                        T::Array[T.nilable(T.any(String, Integer))]
+                      )
+                    )
+                  ]
+                ),
+              body: T.nilable(T.anything),
+              unwrap:
+                T.nilable(
+                  T.any(
+                    Symbol,
+                    Integer,
+                    T::Array[T.any(Symbol, Integer)],
+                    T.proc.params(arg0: T.anything).returns(T.anything)
+                  )
+                ),
+              page:
+                T.nilable(
+                  T::Class[
+                    ContextDev::Internal::Type::BasePage[
+                      ContextDev::Internal::Type::BaseModel
+                    ]
+                  ]
+                ),
+              stream: T.nilable(T::Class[T.anything]),
+              model: T.nilable(ContextDev::Internal::Type::Converter::Input),
+              options: T.nilable(ContextDev::RequestOptions::OrHash)
+            }
+          end
+
+        RequestInput =
+          T.type_alias do
+            {
+              method: Symbol,
+              url: URI::Generic,
+              headers: T::Hash[String, String],
+              body: T.anything,
+              max_retries: Integer,
+              timeout: Float
+            }
+          end
+
+        # from whatwg fetch spec
+        MAX_REDIRECTS = 20
+
+        PLATFORM_HEADERS = T::Hash[String, String]
+
+        class << self
+          # @api private
+          sig do
+            params(
+              req:
+                ContextDev::Internal::Transport::BaseClient::RequestComponents
+            ).void
+          end
+          def validate!(req)
+          end
+
+          # @api private
+          sig do
+            params(status: Integer, headers: T::Hash[String, String]).returns(
+              T::Boolean
+            )
+          end
+          def should_retry?(status, headers:)
+          end
+
+          # @api private
+          sig do
+            params(
+              request:
+                ContextDev::Internal::Transport::BaseClient::RequestInput,
+              status: Integer,
+              response_headers: T::Hash[String, String]
+            ).returns(ContextDev::Internal::Transport::BaseClient::RequestInput)
+          end
+          def follow_redirect(request, status:, response_headers:)
+          end
+
+          # @api private
+          sig do
+            params(
+              status: T.any(Integer, ContextDev::Errors::APIConnectionError),
+              stream: T.nilable(T::Enumerable[String])
+            ).void
+          end
+          def reap_connection!(status, stream:)
+          end
+        end
+
+        sig { returns(URI::Generic) }
+        attr_reader :base_url
+
+        sig { returns(Float) }
+        attr_reader :timeout
+
+        sig { returns(Integer) }
+        attr_reader :max_retries
+
+        sig { returns(Float) }
+        attr_reader :initial_retry_delay
+
+        sig { returns(Float) }
+        attr_reader :max_retry_delay
+
+        sig { returns(T::Hash[String, String]) }
+        attr_reader :headers
+
+        sig { returns(T.nilable(String)) }
+        attr_reader :idempotency_header
+
+        # @api private
+        sig { returns(ContextDev::Internal::Transport::PooledNetRequester) }
+        attr_reader :requester
+
+        # @api private
+        sig do
+          params(
+            base_url: String,
+            timeout: Float,
+            max_retries: Integer,
+            initial_retry_delay: Float,
+            max_retry_delay: Float,
+            headers:
+              T::Hash[
+                String,
+                T.nilable(
+                  T.any(
+                    String,
+                    Integer,
+                    T::Array[T.nilable(T.any(String, Integer))]
+                  )
+                )
+              ],
+            idempotency_header: T.nilable(String)
+          ).returns(T.attached_class)
+        end
+        def self.new(
+          base_url:,
+          timeout: 0.0,
+          max_retries: 0,
+          initial_retry_delay: 0.0,
+          max_retry_delay: 0.0,
+          headers: {},
+          idempotency_header: nil
+        )
+        end
+
+        # @api private
+        sig { overridable.returns(T::Hash[String, String]) }
+        private def auth_headers
+        end
+
+        # @api private
+        sig { returns(String) }
+        private def user_agent
+        end
+
+        # @api private
+        sig { returns(String) }
+        private def generate_idempotency_key
+        end
+
+        # @api private
+        sig do
+          overridable
+            .params(
+              req:
+                ContextDev::Internal::Transport::BaseClient::RequestComponents,
+              opts: ContextDev::Internal::AnyHash
+            )
+            .returns(ContextDev::Internal::Transport::BaseClient::RequestInput)
+        end
+        private def build_request(req, opts)
+        end
+
+        # @api private
+        sig do
+          params(
+            headers: T::Hash[String, String],
+            retry_count: Integer
+          ).returns(Float)
+        end
+        private def retry_delay(headers, retry_count:)
+        end
+
+        # @api private
+        sig do
+          params(
+            request: ContextDev::Internal::Transport::BaseClient::RequestInput,
+            redirect_count: Integer,
+            retry_count: Integer,
+            send_retry_header: T::Boolean
+          ).returns([Integer, Net::HTTPResponse, T::Enumerable[String]])
+        end
+        def send_request(
+          request,
+          redirect_count:,
+          retry_count:,
+          send_retry_header:
+        )
+        end
+
+        # Execute the request specified by `req`. This is the method that all resource
+        # methods call into.
+        #
+        # @overload request(method, path, query: {}, headers: {}, body: nil, unwrap: nil, page: nil, stream: nil, model: ContextDev::Internal::Type::Unknown, options: {})
+        sig do
+          params(
+            method: Symbol,
+            path: T.any(String, T::Array[String]),
+            query:
+              T.nilable(
+                T::Hash[String, T.nilable(T.any(T::Array[String], String))]
+              ),
+            headers:
+              T.nilable(
+                T::Hash[
+                  String,
+                  T.nilable(
+                    T.any(
+                      String,
+                      Integer,
+                      T::Array[T.nilable(T.any(String, Integer))]
+                    )
+                  )
+                ]
+              ),
+            body: T.nilable(T.anything),
+            unwrap:
+              T.nilable(
+                T.any(
+                  Symbol,
+                  Integer,
+                  T::Array[T.any(Symbol, Integer)],
+                  T.proc.params(arg0: T.anything).returns(T.anything)
+                )
+              ),
+            page:
+              T.nilable(
+                T::Class[
+                  ContextDev::Internal::Type::BasePage[
+                    ContextDev::Internal::Type::BaseModel
+                  ]
+                ]
+              ),
+            stream: T.nilable(T::Class[T.anything]),
+            model: T.nilable(ContextDev::Internal::Type::Converter::Input),
+            options: T.nilable(ContextDev::RequestOptions::OrHash)
+          ).returns(T.anything)
+        end
+        def request(
+          method,
+          path,
+          query: {},
+          headers: {},
+          body: nil,
+          unwrap: nil,
+          page: nil,
+          stream: nil,
+          model: ContextDev::Internal::Type::Unknown,
+          options: {}
+        )
+        end
+
+        # @api private
+        sig { returns(String) }
+        def inspect
+        end
+      end
+    end
+  end
+end