added config class

2 files changed, 22 insertions, 5 deletions

diff --git a/lib/config.rb b/lib/config.rb
new file mode 100644
index 0000000..dded140
--- /dev/null
+++ b/lib/config.rb
@@ -0,0 +1,18 @@
+require 'yaml'
+
+module LeapCA
+  class Config
+    def initialize(filename)
+      file = File.new(filename, 'r')
+      @hash = YAML::load(file)
+    end
+
+    def server
+      @hash['server']
+    end
+
+    def database
+      @hash['database']
+    end
+  end
+end
diff --git a/lib/couch_stream.rb b/lib/couch_stream.rb
index bed296a..081688a 100644
--- a/lib/couch_stream.rb
+++ b/lib/couch_stream.rb
@@ -1,7 +1,6 @@
 class CouchStream
-  def initialize(server, db)
-    @server = server
-    @db = db
+  def initialize(config)
+    @config = config
   end
 
   def get(path, options)
@@ -11,11 +10,11 @@ class CouchStream
       yield(hash)
     end
   end
-  
+
   protected
 
   def url_for(path, options = {})
-    url = @server + @db + '/' + path
+    url = [@config.server, @config.database, path].join('/')
     url += '?' if options.any?
     url += options.map {|k,v| "#{k}=#{v}"}.join('&')
   end