require 'rails'

module ActiveRecord
  module ConnectionAdapters
    module CipherStashPG
      class Railtie < ::Rails::Railtie
        rake_tasks do
          load "cipherstash_tasks.rake"
        end

        initializer "postgres_cipherstash_adapter.configure" do
          if defined?(Rails.application.credentials)
            cs_credentials = Rails.application.credentials.try(:cipherstash)

            client_id = cs_credentials&.fetch(:client_id, nil)
            client_key = cs_credentials&.fetch(:client_key, nil)
            workspace_id = cs_credentials&.fetch(:workspace_id, nil)
            client_access_key = cs_credentials&.fetch(:client_access_key, nil)
            vitur_host = cs_credentials&.fetch(:vitur_host, nil)

            unless client_id.nil?
              ENV["CS_CLIENT_ID"] = client_id
            end

            unless client_key.nil?
              ENV["CS_CLIENT_KEY"] = client_key
            end

            unless workspace_id.nil?
              ENV["CS_WORKSPACE_ID"] = workspace_id
            end

            unless client_access_key.nil?
              ENV["CS_CLIENT_ACCESS_KEY"] = client_access_key
            end

            unless vitur_host.nil?
              ENV["CS_VITUR_HOST"] = vitur_host
            end
          end
        end
      end
    end
  end
end