lib/fake_sqs.rb in fake_sqs-0.0.9 vs lib/fake_sqs.rb in fake_sqs-0.0.10

@@ -6,28 +6,60 @@
 require 'fake_sqs/queue_factory'
 require 'fake_sqs/queues'
 require 'fake_sqs/responder'
 require 'fake_sqs/server'
 require 'fake_sqs/version'
-require 'fake_sqs/web_interface'
+require 'fake_sqs/memory_database'
+require 'fake_sqs/file_database'
 
 module FakeSQS
 
+  def self.to_rack(options)
+
+    require 'fake_sqs/web_interface'
+    app = FakeSQS::WebInterface
+
+    if (log = options[:log])
+      $stdout.reopen(log, "w:utf-8")
+      $stderr.reopen(log, "w:utf-8")
+      app.enable :logging
+    end
+
+    if options[:verbose]
+      require 'fake_sqs/show_output'
+      app.use FakeSQS::ShowOutput
+      app.enable :logging
+    end
+
+    if options[:daemonize]
+      require 'fake_sqs/daemonize'
+      Daemonize.new(options).call
+    end
+
+    app.set :port, options[:port] if options[:port]
+    app.set :bind, options[:host] if options[:host]
+    app.set :server, options[:server] if options[:server]
+    server = FakeSQS.server(port: options[:port], host: options[:host])
+    app.set :api, FakeSQS.api(server: server, database: options[:database])
+    app
+  end
+
   def self.server(options = {})
     Server.new(options)
   end
 
   def self.api(options = {})
+    db = database_for(options.fetch(:database) { ":memory:" })
     API.new(
       server: options.fetch(:server),
-      queues: queues,
+      queues: queues(db),
       responder: responder
     )
   end
 
-  def self.queues
-    Queues.new(queue_factory: queue_factory)
+  def self.queues(database)
+    Queues.new(queue_factory: queue_factory, database: database)
   end
 
   def self.responder
     Responder.new
   end
@@ -40,8 +72,16 @@
     Message
   end
 
   def self.queue
     Queue
+  end
+
+  def self.database_for(name)
+    if name == ":memory:"
+      MemoryDatabase.new
+    else
+      FileDatabase.new(name)
+    end
   end
 
 end