src/derails/passive_aggressive/database_configurations/url_config.rb
# frozen_string_literal: true

module PassiveAggressive
  class DatabaseConfigurations
    # = Active Record Database Url Config
    #
    # A +UrlConfig+ object is created for each database configuration
    # entry that is created from a URL. This can either be a URL string
    # or a hash with a URL in place of the config hash.
    #
    # A URL config:
    #
    #   postgres://localhost/foo
    #
    # Becomes:
    #
    #   #<PassiveAggressive::DatabaseConfigurations::UrlConfig:0x00007fdc3238f340
    #     @env_name="default_env", @name="primary",
    #     @config={adapter: "postgresql", database: "foo", host: "localhost"},
    #     @url="postgres://localhost/foo">
    #
    # See PassiveAggressive::DatabaseConfigurations for more info.
    #
    class UrlConfig < HashConfig
      attr_reader :url

      # Initialize a new +UrlConfig+ object
      #
      # ==== Options
      #
      # * <tt>:env_name</tt> - The \Rails environment, i.e. "development".
      # * <tt>:name</tt> - The db config name. In a standard two-tier
      #   database configuration this will default to "primary". In a multiple
      #   database three-tier database configuration this corresponds to the name
      #   used in the second tier, for example "primary_readonly".
      # * <tt>:url</tt> - The database URL.
      # * <tt>:config</tt> - The config hash. This is the hash that contains the
      #   database adapter, name, and other important information for database
      #   connections.
      def initialize(env_name, name, url, configuration_hash = {})
        super(env_name, name, configuration_hash)

        @url = url
        @configuration_hash = @configuration_hash.merge(build_url_hash)

        if @configuration_hash[:schema_dump] == "false"
          @configuration_hash[:schema_dump] = false
        end

        query_cache = parse_query_cache
        @configuration_hash[:query_cache] = query_cache unless query_cache.nil?

        to_boolean!(@configuration_hash, :replica)
        to_boolean!(@configuration_hash, :database_tasks)

        @configuration_hash.freeze
      end

      private
        def parse_query_cache
          case value = @configuration_hash[:query_cache]
          when /\A\d+\z/
            value.to_i
          when "false"
            false
          else
            value
          end
        end

        def to_boolean!(configuration_hash, key)
          if configuration_hash[key].is_a?(String)
            configuration_hash[key] = configuration_hash[key] != "false"
          end
        end

        # Return a Hash that can be merged into the main config that represents
        # the passed in url
        def build_url_hash
          if url.nil? || url.start_with?("jdbc:", "http:", "https:")
            { url: url }
          else
            ConnectionUrlResolver.new(url).to_hash
          end
        end
    end
  end
end
