strobe 0.1.3 → 0.1.4

data/lib/strobe/cli/main.rb

@@ -1,5 +1,7 @@
 module Strobe
   class CLI::Main < CLI
+    include CLI::Preview
+
     def help(*args)
       if args.first == "users"
         CLI::Users.start( [ "help" ] + args[1..-1] )
@@ -61,6 +63,10 @@ module Strobe
       end
     end
 
+    action "preview", "preview your application as it would show up on Strobe Platform" do
+      preview_strobe_application
+    end
+
     application_path_option
     method_option "staging",     :type => :boolean, :banner => "deploy to a staging environment"
     method_option "sc-build",    :type => :boolean, :banner => "run `sc-build -c` before deploying"

data/lib/strobe/cli/preview.rb

@@ -10,33 +10,21 @@ module Strobe
       alias wrapped_app app
     end
 
-    class MagicWrapper
-      def initialize(app)
-        @app = app
-      end
-
-      def call(env)
-        puts "ZOMGZOMG, calling: #{env['PATH_INFO']}"
-        @app.call(env)
-      end
-    end
-
   private
 
-    def sproutcore_app
-      gem "sproutcore", ">= 1.4.0"
-      require "sproutcore"
-
-      project = SC::Tools.new.requires_project!
-      SC::Rack::Service.new(*project)
+    def preview_strobe_application
+      is_sproutcore?? preview_sproutcore_application : preview_html_application
     end
 
-    def wrap(app)
-      MagicWrapper.new(app)
+    def preview_html_application
+      app = Rack::Static.new lambda { |e| [] }, :urls => [ '/' ], :root => '.'
+      app = Middleware::Rewrite.new(Middleware::Proxy.new(app))
+      Server.start :app => app, :host => '0.0.0.0', :Port => 9292
     end
 
     def preview_sproutcore_application
-      Server.start :app => wrap(sproutcore_app), :Port => 9292
+      require 'strobe/sproutcore'
+      SC::Tools.start ['server'] + ARGV[1..-1] + [ '--port', '9292' ]
     end
   end
 end

data/lib/strobe/middleware/proxy.rb

@@ -0,0 +1,117 @@
+require 'thread'
+require 'net/http'
+
+module Strobe
+  class Middleware::Proxy
+
+    # A wrapper around the Net/HTTP response body
+    # that allows rack to stream the result down
+    class Body
+      def initialize(queue)
+        @queue = queue
+      end
+
+      def each
+        while chunk = @queue.pop
+          if Exception === chunk
+            raise chunk
+          else
+            yield chunk
+          end
+        end
+      end
+    end
+
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      if env['PATH_INFO'] =~ proxy_path
+        host, port = $1.split(':')
+
+        queue = run_request(env, host, ( port || 80 ).to_i, $2 || '/')
+
+        msg = queue.pop
+
+        if Exception === msg
+          raise msg
+        else
+          [ msg[0], msg[1], Body.new(queue) ]
+        end
+      else
+        @app.call(env)
+      end
+    end
+
+  private
+
+    def proxy_path
+      %r[/_strobe/proxy/([^/]+)(/.*)?$]
+    end
+
+    KEEP = [ 'CONTENT_LENGTH', 'CONTENT_TYPE' ]
+
+    def run_request(env, host, port, path)
+      queue = Queue.new
+
+      if env['CONTENT_LENGTH'] || env['HTTP_TRANSFER_ENCODING']
+        body = env['rack.input']
+      end
+
+      unless env['QUERY_STRING'].blank?
+        path += "?#{env['QUERY_STRING']}"
+      end
+
+      http = Net::HTTP.new(host, port)
+      http.read_timeout = 60
+      http.open_timeout = 60
+
+      request = Net::HTTPGenericRequest.new(
+        env['REQUEST_METHOD'], !!body, true, path,
+        env_to_http_headers(env))
+
+      request.body_stream = body if body
+
+      Thread.new do
+        begin
+          http.request(request) do |response|
+            hdrs = {}
+            response.each_header do |name, val|
+              hdrs[name] = val
+            end
+
+            queue << [ response.code.to_i, hdrs ]
+
+            response.read_body do |chunk|
+              queue << chunk
+            end
+
+            queue << nil
+          end
+        rescue Exception => e
+          queue << e
+        end
+      end
+
+      queue
+    end
+
+    def env_to_http_headers(env)
+      {}.tap do |hdrs|
+        env.each do |name, val|
+          next unless name.is_a?(String)
+          next if     name == 'HTTP_HOST'
+          next unless name =~ /^HTTP_/ || KEEP.include?(name)
+
+          hdrs[ headerize(name) ] = val
+        end
+      end
+    end
+
+    def headerize(str)
+      parts = str.gsub(/^HTTP_/, '').split('_')
+      parts.map! { |p| p.capitalize }.join('-')
+    end
+  end
+end

data/lib/strobe/middleware/rewrite.rb

@@ -0,0 +1,20 @@
+module Strobe
+  class Middleware::Rewrite
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      status, hdrs, body = @app.call(env)
+
+      if status == 404
+        path_info        = env['PATH_INFO']
+        env['PATH_INFO'] = "#{path_info}/index.html".gsub('//', '/')
+
+        return @app.call(env)
+      end
+
+      [ status, hdrs, body ]
+    end
+  end
+end

data/lib/strobe/sproutcore.rb

@@ -0,0 +1,11 @@
+require 'sproutcore'
+
+class SC::Rack::Service
+  def call(env)
+    wrapped_app.call(env)
+  end
+
+  def wrapped_app
+    @wrapped_app ||= Strobe::Middleware::Proxy.new(@app)
+  end
+end

data/lib/strobe.rb

@@ -17,6 +17,11 @@ module Strobe
   autoload :Resources,   'strobe/resources'
   autoload :Validations, 'strobe/validations'
 
+  module Middleware
+    autoload :Proxy,     'strobe/middleware/proxy'
+    autoload :Rewrite,   'strobe/middleware/rewrite'
+  end
+
   module Resource
     autoload :Base,       'strobe/resource/base'
     autoload :Collection, 'strobe/resource/collection'

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 3
-  version: 0.1.3
+  - 4
+  version: 0.1.4
 platform: ruby
 authors: 
 - Yehuda Katz
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-05 00:00:00 -08:00
+date: 2011-01-11 00:00:00 -08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -117,6 +117,8 @@ files:
 - lib/strobe/connection.rb
 - lib/strobe/identity_map.rb
 - lib/strobe/key.rb
+- lib/strobe/middleware/proxy.rb
+- lib/strobe/middleware/rewrite.rb
 - lib/strobe/resource/base.rb
 - lib/strobe/resource/collection.rb
 - lib/strobe/resource/singleton.rb
@@ -129,6 +131,7 @@ files:
 - lib/strobe/resources/team.rb
 - lib/strobe/resources/user.rb
 - lib/strobe/resources.rb
+- lib/strobe/sproutcore.rb
 - lib/strobe/validations.rb
 - lib/strobe.rb
 - bin/strobe