lib/rock_config/config.rb in rock_config-0.0.3 vs lib/rock_config/config.rb in rock_config-0.0.4

@@ -2,20 +2,34 @@
   class Config
     def initialize(hash)
       @hash = hash
     end
 
-    def method_missing(*args, &block)
-      if value = @hash[args.first.to_s]
-        if Hash === value
-          Config.new(value)
-        else
-          value
-        end
-      end
+    def [](key)
+      fetch(key)
     end
 
+    def method_missing(name, *args, &block)
+      fetch(name.to_s)
+    end
+
     def raw
       @hash.dup
+    end
+
+    private
+
+    def fetch(key, default = nil)
+      if value = @hash[key.to_s]
+        return value_or_config(value)
+      end
+    end
+
+    def value_or_config(value)
+      if Hash === value
+        Config.new(value)
+      else
+        value
+      end
     end
   end
 end