# frozen_string_literal: true

require "netaddr"

require "sequel/core"

require_relative "config"

require_relative "lib/util"

connect_opts_proc = nil

if Config.clover_database_rds_iam_auth_enabled

require "pg/aws_rds_iam"

driver_options = {aws_rds_iam_auth_token_generator: "default"}

elsif Config.clover_database_gcp_iam_auth_enabled

require_relative "lib/gcp_database_auth" # loaded only on the GCP path

role = GcpDatabaseAuth.url_user(Config.clover_database_url)

connect_opts_proc = GcpDatabaseAuth.connect_opts_proc(role => Config.clover_database_gcp_clover_login_sa)

driver_options = {}

else

driver_options = {}

end

db_ca_bundle_filename = File.join(Dir.pwd, "var", "ca_bundles", "db_ca_bundle.crt")

Util.safe_write_to_file(db_ca_bundle_filename, Config.clover_database_root_certs)

if ENV["SHARED_CONNECTION"] == "1"

max_connections = 1

elsif (process_type = ENV["PROCESS_TYPE"])

max_connections = Config.send(:"db_pool_#{process_type}") - 1

pool_timeout = Config.send(:"database_timeout_#{process_type}")

end

max_connections ||= Config.db_pool - 1

pool_timeout ||= Config.database_timeout

pg_auto_parameterize_min_array_size = 1 if Config.frozen_test?

DB = Sequel.connect(Config.clover_database_url, max_connections:, pool_timeout:, treat_string_list_as_untyped_array: true, pg_auto_parameterize_min_array_size:, driver_options:, connect_opts_proc:)

postgres_monitor_db_ca_bundle_filename = File.join(Dir.pwd, "var", "ca_bundles", "postgres_monitor_db.crt")

Util.safe_write_to_file(postgres_monitor_db_ca_bundle_filename, Config.postgres_monitor_database_root_certs)

POSTGRES_MONITOR_DB = Sequel.connect(Config.postgres_monitor_database_url, max_connections: Config.db_pool_monitor, pool_timeout: Config.database_timeout, driver_options:, connect_opts_proc:, test: false)

# Load Sequel Database/Global extensions here

# DB.extension :date_arithmetic

DB.extension :dataset_run, :pg_array, :pg_json, :pg_auto_parameterize, :pg_auto_parameterize_in_array, :pg_timestamptz, :pg_range, :pg_enum, :pg_loose_count

Sequel.extension :pg_range_ops, :pg_json_ops, :pg_array_ops

DB.extension :lit_require_frozen if Config.unfrozen_test? && !ENV["SKIP_LIT_REQUIRE_FROZEN"]

if Config.development? || Config.unfrozen_test?

DB.extension :pg_auto_parameterize_duplicate_query_detection

else

def DB.ignore_duplicate_queries

yield

end

end

# Replace dangerous (for cidrs) Ruby IPAddr type that is otherwise

# used by sequel_pg. Has come up more than once in the bug tracker:

#

# https://github.com/jeremyevans/sequel_pg/issues?q=inet

# https://github.com/jeremyevans/sequel_pg/issues/30

# https://github.com/jeremyevans/sequel_pg/pull/37

DB.add_conversion_proc(869, NetAddr.method(:parse_ip))

DB.add_conversion_proc(650, NetAddr.method(:parse_net))

DB.register_array_type("inet", oid: 1041, scalar_oid: 869)

DB.register_array_type("cidr", oid: 651, scalar_oid: 650)

require_relative "lib/net_addr_sequel"

DB.extend_datasets(NetAddrSequel::DatasetMethods)

DB.extend(NetAddrSequel::DatabaseMethods)

DB.extension :pg_schema_caching

DB.extension :index_caching

DB.load_schema_cache?("cache/schema.cache")

DB.load_index_cache?("cache/index.cache")

DB.extension :temporarily_release_connection if ENV["SHARED_CONNECTION"] == "1"

DB.extension :query_blocker if Config.frozen_test?

# Will be required separately by web, respirate, and monitor, but too late

# when running frozen tests, so load the extension eagerly in that case

DB.extension :connection_checkout_event_callback if Config.frozen_test?