databricks_sql 0.1.0

data/lib/databricks_sql/type_coercer.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require "time"
+
+module DatabricksSql
+  module TypeCoercer
+    module_function
+
+    def coerce_rows(rows, column_schema)
+      return rows if column_schema.nil? || column_schema.empty?
+
+      rows.map { |row| coerce_row(row, column_schema) }
+    end
+
+    def coerce_row(row, column_schema)
+      row.each_with_object({}) do |(key, value), coerced|
+        target_type = column_schema[key.to_s] || column_schema[key.to_sym]
+        coerced[key] = coerce_value(value, target_type)
+      end
+    end
+
+    def coerce_value(value, target_type)
+      return value if value.nil? || target_type.nil?
+
+      type = target_type.to_s.downcase
+
+      case type
+      when "string"
+        value.to_s
+      when "integer", "int"
+        Integer(value)
+      when "float", "double", "decimal"
+        Float(value)
+      when "boolean", "bool"
+        coerce_boolean(value)
+      when "time", "datetime", "timestamp"
+        Time.parse(value.to_s)
+      else
+        value
+      end
+    rescue ArgumentError, TypeError
+      value
+    end
+
+    def coerce_boolean(value)
+      return value if [true, false].include?(value)
+
+      case value.to_s.strip.downcase
+      when "true", "1", "yes", "y"
+        true
+      when "false", "0", "no", "n"
+        false
+      else
+        value
+      end
+    end
+    private_class_method :coerce_boolean
+  end
+end

data/lib/databricks_sql/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module DatabricksSql
+  VERSION = "0.1.0"
+end

data/lib/databricks_sql.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative "databricks_sql/version"
+require_relative "databricks_sql/errors"
+require_relative "databricks_sql/configuration"
+require_relative "databricks_sql/result"
+require_relative "databricks_sql/type_coercer"
+require_relative "databricks_sql/external_link_handler"
+require_relative "databricks_sql/client"
+
+module DatabricksSql
+  class << self
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+    end
+
+    def reset_configuration!
+      @configuration = Configuration.new
+    end
+
+    def client(**overrides)
+      Client.new(**overrides)
+    end
+  end
+end
+
+Databricks = DatabricksSql unless Object.const_defined?(:Databricks)

data/sig/databricks_sql.rbs

@@ -0,0 +1,182 @@
+module DatabricksSql
+  VERSION: String
+
+  class << self
+    def configuration: () -> Configuration
+    def configure: () { (Configuration) -> untyped } -> Configuration
+    def reset_configuration!: () -> Configuration
+    def client: (**untyped overrides) -> Client
+  end
+
+  class Configuration
+    attr_accessor host: String?
+    attr_accessor token: String?
+    attr_accessor warehouse_id: String?
+    attr_accessor timeout: Numeric?
+    attr_accessor open_timeout: Numeric?
+    attr_accessor statement_path: String?
+    attr_accessor external_link_require_https: bool?
+    attr_accessor external_link_allowed_hosts: Array[String]?
+    def initialize: () -> void
+  end
+
+  class Error < ::StandardError
+  end
+
+  class ConfigurationError < Error
+  end
+
+  class ConnectionError < Error
+  end
+
+  class ParseError < Error
+  end
+
+  class TimeoutError < Error
+    attr_reader timeout_seconds: Numeric
+    def initialize: (String message, Numeric timeout_seconds) -> void
+  end
+
+  class HTTPError < Error
+    attr_reader status_code: Integer
+    attr_reader response_body: String?
+    def initialize: (String message, status_code: Integer, ?response_body: String?) -> void
+  end
+
+  class AuthenticationError < HTTPError
+  end
+
+  class AuthorizationError < HTTPError
+  end
+
+  class NotFoundError < HTTPError
+  end
+
+  class RateLimitError < HTTPError
+    attr_reader retry_after_seconds: Integer?
+    def initialize: (String message, status_code: Integer, ?response_body: String?, ?retry_after_seconds: Integer?) -> void
+  end
+
+  class ServerError < HTTPError
+  end
+
+  class ExecutionError < Error
+    attr_reader statement_id: String?
+    attr_reader status: String?
+    def initialize: (String message, statement_id: String?, status: String?) -> void
+  end
+
+  class Result
+    attr_reader statement_id: String?
+    attr_reader status: String
+    attr_reader disposition: String
+    attr_reader format: String
+    attr_reader columns: Array[String]
+    attr_reader rows: Array[untyped]
+    attr_reader manifest: Hash[untyped, untyped]?
+    attr_reader raw_response: Hash[untyped, untyped]
+    attr_reader next_chunk_internal_link: String?
+
+    def initialize: (
+      statement_id: String?,
+      status: String,
+      disposition: String,
+      format: String,
+      columns: Array[String],
+      rows: Array[untyped],
+      manifest: Hash[untyped, untyped]?,
+      raw_response: Hash[untyped, untyped],
+      ?next_chunk_internal_link: String?
+    ) -> void
+
+    def success?: () -> bool
+  end
+
+  class Client
+    def initialize: (
+      ?host: String?,
+      ?token: String?,
+      ?warehouse_id: String?,
+      ?timeout: Numeric,
+      ?open_timeout: Numeric,
+      ?statement_path: String
+    ) -> void
+
+    def execute_statement: (
+      statement: String,
+      ?parameters: untyped,
+      ?format: String,
+      ?disposition: String,
+      ?catalog: String?,
+      ?schema: String?,
+      ?column_schema: Hash[untyped, untyped]?,
+      ?wait_timeout: String,
+      ?on_wait_timeout: String,
+      ?poll_interval: Numeric,
+      ?max_wait: Numeric,
+      ?request_timeout: Numeric?,
+      ?cancel_on_timeout: bool,
+      ?auto_fetch_chunks: bool,
+      ?byte_limit: Integer?,
+      ?row_limit: Integer?,
+      ?query_tags: Array[Hash[untyped, untyped]]?
+    ) -> Result
+
+    def execute_statement_async: (
+      statement: String,
+      ?parameters: untyped,
+      ?format: String,
+      ?disposition: String,
+      ?catalog: String?,
+      ?schema: String?,
+      ?wait_timeout: String,
+      ?on_wait_timeout: String,
+      ?request_timeout: Numeric?,
+      ?byte_limit: Integer?,
+      ?row_limit: Integer?,
+      ?query_tags: Array[Hash[untyped, untyped]]?
+    ) -> Hash[untyped, untyped]
+
+    def get_statement: (statement_id: String, ?request_timeout: Numeric?) -> Hash[untyped, untyped]
+    def get_statement_chunk: (
+      statement_id: String,
+      chunk_index: Integer,
+      ?row_offset: Integer?,
+      ?request_timeout: Numeric?
+    ) -> Hash[untyped, untyped]
+    def get_statement_chunk_by_link: (
+      next_chunk_internal_link: String,
+      ?request_timeout: Numeric?
+    ) -> Hash[untyped, untyped]
+    def fetch_next_chunk: (
+      statement_id: String,
+      next_chunk_internal_link: String,
+      columns: Array[String],
+      ?disposition: String,
+      ?column_schema: Hash[untyped, untyped]?,
+      ?request_timeout: Numeric?
+    ) -> Result
+    def cancel_statement: (statement_id: String, ?request_timeout: Numeric?) -> Hash[untyped, untyped]?
+
+    def wait_for_statement: (
+      statement_id: String,
+      ?format: String,
+      ?disposition: String,
+      ?column_schema: Hash[untyped, untyped]?,
+      ?poll_interval: Numeric,
+      ?max_wait: Numeric,
+      ?request_timeout: Numeric?,
+      ?cancel_on_timeout: bool,
+      ?auto_fetch_chunks: bool
+    ) -> Result
+  end
+end
+
+module Databricks
+  class << self
+    def configuration: () -> DatabricksSql::Configuration
+    def configure: () { (DatabricksSql::Configuration) -> untyped } -> DatabricksSql::Configuration
+    def reset_configuration!: () -> DatabricksSql::Configuration
+    def client: (**untyped overrides) -> DatabricksSql::Client
+  end
+end