Contents

require "enumerator"

module SettingsEnvLoader
  VERSION = "0.1.2"

  # Merges ENV and modify the hash directly
  def merge_env(prefix = nil, separator = '_')
    Merger.new(prefix, separator, self).merge
    self
  end

  # Iterates the Hash as ENV key value pairs
  def each_env(prefix = nil, separator = '_', &block)
    enumerator = Enumerator.new do |y|
      Exporter.new(prefix, separator, self).each(y)
    end

    enumerator.each(&block) if block
    enumerator
  end

  class Merger
    def initialize(prefix, separator, hash)
      @prefix = prefix
      @separator = separator
      @hash = hash
    end

    def merge
      @hash.each do |key, value|
        env_name = [@prefix, key].compact.join(@separator).upcase

        if value.is_a?(Hash)
          Merger.new(env_name, @separator, value).merge
        else
          env_value = ENV[env_name]
          if env_value
            @hash[key] = convert(env_value, value)
          end
        end
      end
    end

    def convert(value, origin_value)
      case origin_value
      when Symbol
        value.to_sym
      when Fixnum
        value.to_i
      when Float
        value.to_f
      when Array
        Array(YAML.load(value))
      else
        value
      end
    end
  end

  class Exporter
    def initialize(prefix, separator, hash)
      @prefix = prefix
      @separator = separator
      @hash = hash
    end

    def each(y)
      @hash.each do |key, value|
        env_name = [@prefix, key].compact.join(@separator).upcase
        if value.is_a?(Hash)
          Exporter.new(env_name, @separator, value).each(y)
        else
          y << [env_name, value.to_s]
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
settings-env-loader-0.1.2	lib/settings-env-loader.rb