Contents

require 'hashie'
module Figs
  module ENV
    extend self
    
    def env
     @env ||= ::ENV
    end
    
    def set(key,value)
      env[key.to_s] = value.is_a?(String) ? value : YAML::dump(value)
    end
    
    def [](key)
      return demarshall(env[key.to_s])
    end
    
    def []=(key,value)
      set(key, value)
    end
    
    def demarshall(value)
      value.nil? ? nil : YAML::load(value)
    end
    
    def method_missing(method, *args, &block)
      if matches_env?(method) then return env.send(method, *args, &block) end
      
      key, punctuation = extract_key_from_method(method)
      _, value = env.detect { |k, _| k.upcase == key }
      
      value = demarshall(value)

      case punctuation
      when "!" then value || missing_key!(key)
      when "?" then !!value
      when nil then value
      else super
      end
    end

    def extract_key_from_method(method)
      method.to_s.upcase.match(/^(.+?)([!?=])?$/).captures
    end

    def missing_key!(key)
      raise MissingKey.new("Missing required Figaro configuration key #{key.inspect}.")
    end
    
    def respond_to?(method, *)
      return true if matches_env?(method)
      key, punctuation = extract_key_from_method(method)

      case punctuation
      when "!" then env.keys.any? { |k| k.upcase == key } || super
      when "?", nil then true
      else super
      end
    end
    
    def matches_env?(method)
      env.respond_to?(method)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
figs-1.2.2	lib/figs/env.rb
figs-1.2.1	lib/figs/env.rb