lib/cuba.rb in cuba-0.1.0 vs lib/cuba.rb in cuba-0.2.0

@@ -1,15 +1,20 @@
 require "cuba/version"
 require "cuba/ron"
 
 module Cuba
+  def self.app
+    @app ||= Rack::Builder.new
+  end
+
+  def self.use(middleware)
+    app.use(middleware)
+  end
+
   def self.define(&block)
-    @app = Rack::Builder.new do
-      use Rack::Session::Cookie
-      run Cuba::Ron.new(&block)
-    end
+    app.run Cuba::Ron.new(&block)
   end
 
   def self.call(env)
-    @app.call(env)
+    app.call(env)
   end
 end