# rubocop:disable Naming/UncommunicativeMethodParamName

require "forwardable"
require "logger"
require "expectation"

require_relative "sql/version"
require_relative "sql/fragment"
require_relative "sql/helpers"
require_relative "sql/result"
require_relative "sql/config"
require_relative "sql/logging"
require_relative "sql/connection"

module Simple
  # The Simple::SQL module
  module SQL
    extend self

    extend Forwardable
    delegate [:ask, :all, :each, :exec, :locked, :lock!, :print, :transaction, :wait_for_notify, :estimate_cost] => :default_connection
    delegate [:reflection] => :default_connection
    delegate [:duplicate] => :default_connection
    delegate [:insert] => :default_connection
    delegate [:scope] => :default_connection

    delegate [:logger, :logger=] => ::Simple::SQL::Logging

    def escape_string(s)
      expect! s => [Symbol, String, nil]

      return "NULL" unless s

      "'#{PG::Connection.escape_string(s)}'"
    end

    # connects to the database specified via the url parameter. If called
    # without argument it tries to determine a DATABASE_URL from either the
    # environment setting (DATABASE_URL) or from a config/database.yml file,
    # taking into account the RAILS_ENV and RACK_ENV settings.
    #
    # Returns the connection object.
    def connect(database_url = :auto)
      Connection.create(database_url)
    end

    # deprecated
    def configuration
      Config.parse_url(Config.determine_url)
    end

    # -- default connection ---------------------------------------------------

    DEFAULT_CONNECTION_KEY = :"Simple::SQL.default_connection"

    # returns the default connection.
    def default_connection
      Thread.current[DEFAULT_CONNECTION_KEY] ||= connect(:auto)
    end

    # connects to the database specified via the url parameter, and sets
    # Simple::SQL's default connection.
    #
    # \see connect, default_connection
    def connect!(database_url = :auto)
      disconnect!
      Thread.current[DEFAULT_CONNECTION_KEY] ||= connect(database_url)
    end

    # disconnects the current default connection.
    def disconnect!
      connection = Thread.current[DEFAULT_CONNECTION_KEY]
      return unless connection

      connection.disconnect!
      Thread.current[DEFAULT_CONNECTION_KEY] = nil
    end
  end
end