Contents

require 'yaml'
require 'erb'

module SqlMigrations
  # Configuration module
  #
  module Config
    extend self

    def load!(config_file, env = nil)
      @env = (env || ENV['ENV'] || ENV['RAKE_ENV'] || :development).to_sym
      config = get_config_for_env_from_file(config_file)
      @databases = config[:databases]
      @options   = config[:options]
      { databases: @databases, options: @options }
    end

    def env
      @env
    end

    def databases
      get_config_required(:@databases)
    end

    def options
      get_config_optional(:@options)
    end

    private

    def get_config_required(config_variable)
      config_value = instance_variable_get(config_variable)
      if config_value.nil? || config_value.empty?
        raise "No configuration for `#{config_variable.to_s[1..-1]}` !"
      end
      config_value
    end

    def get_config_optional(config_variable)
      config_value = instance_variable_get(config_variable)
      (config_value.nil? || config_value.empty?) ? {} : config_value
    end

    def get_config_for_env_from_file(file)
      yaml_hash = YAML.load(ERB.new(::File.new(file).read).result)
      config = symbolize_keys(yaml_hash)[@env]
      raise LoadError, "No configuration for `#{@env}` environment found !" unless config
      config
    end

    def symbolize_keys(hash)
      hash.each_with_object({}) do |(key, value), new_hash|
        new_key = key.is_a?(String) ? key.to_sym : key
        new_value = value.is_a?(Hash) ? symbolize_keys(value) : value
        new_hash[new_key] = new_value
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
sql_migrations-2.6.1	lib/sql_migrations/config.rb
sql_migrations-2.6.0	lib/sql_migrations/config.rb
sql_migrations-2.5.0	lib/sql_migrations/config.rb
sql_migrations-2.4.0	lib/sql_migrations/config.rb