Daemon working with beanstalk, spec for custom plugin

bin/stasis

@@ -7,8 +7,8 @@ require 'slop'
 
 opts = Slop.parse :help => true do
   on :a, :auto, 'Auto-regenerate'
+  on :b, :beanstalk, 'Beanstalk addresses (comma-separated, default: localhost:11300)', :optional => true, :as => Array, :default => "localhost:11300"
   on :o, :only, 'Only generate specific files (comma-separated)', :as => Array
-  on :p, :port, 'Port (daemon mode)', :optional => true, :as => Integer, :default => 7000
 end
 
 args = []
@@ -20,8 +20,8 @@ if opts.auto?
   Stasis::Auto.new(dir)
 elsif opts.only?
   Stasis.new(dir).generate(:only => opts[:only])
-elsif opts.port?
-  Stasis::Daemon.new(dir, opts[:port])
+elsif opts.beanstalk?
+  Stasis::Daemon.new(opts)
 else
   Stasis.new(dir).generate
 end

config/gemsets.yml

@@ -1,4 +1,5 @@
 stasis:
+  beanstalk-client: ~>1.1.0
   directory_watcher: ~>1.4.0
   rake: >=0.8.7
   rocco: ~>0.6

config/gemspec.yml

@@ -7,6 +7,7 @@ homepage: http://wintoni.us
 summary: Markup-agnostic static site generator
 description: A markup-agnostic static site generator.
 dependencies:
+  - beanstalk-client
   - directory_watcher
   - slop
   - tilt

lib/stasis.rb

@@ -26,7 +26,7 @@
 
 # Activate the [Tilt][ti] gem.
 
-Stasis::Gems.activate %w(tilt yajl-ruby)
+Stasis::Gems.activate %w(tilt)
 
 # Add the project directory to the load paths.
 

lib/stasis/daemon.rb

@@ -1,56 +1,36 @@
-Stasis::Gems.activate %w(eventmachine yajl-ruby)
-require 'eventmachine'
+Stasis::Gems.activate %w(beanstalk-client yajl-ruby)
+
+require 'beanstalk-client'
 require 'yajl'
 
 class Stasis
   class Daemon
 
-    def initialize(dir, port)
-      @port = port
-
-      last_retry = nil
-      retries = 0
+    def initialize(options={})
+      puts "\nStarting Stasis daemon (beanstalk @ #{options[:beanstalk].join(', ')})..."
 
-      puts "\nStarting Stasis daemon on port #{port}..."
+      beanstalk = Beanstalk::Pool.new(options[:beanstalk])
 
       begin
-        EM.epoll if EM.epoll?
-        EM.run do
-          EM.start_server '0.0.0.0', port, Tcp
+        while true
+          sleep(1.0 / 1000.0)
+          job = beanstalk.reserve
+          data = Yajl::Parser.parse(job.body)
+          puts data.inspect
+          job.delete
         end
       rescue Interrupt
         shut_down
       rescue Exception => e
         puts "\nError: #{e.message}"
         puts "\t#{e.backtrace.join("\n\t")}"
-
-        if retries >= 10 && Time.now - last_retry < 10
-          shut_down
-        else
-          retries += 1
-          last_retry = Time.now
-        end
+        retry
       end
     end
 
     def shut_down
-      puts "\nShutting down Stasis daemon on port #{@port}..."
+      puts "\nShutting down Stasis daemon..."
       exit
     end
-
-    module Tcp
-      def self.parser
-        @parser ||= Yajl::Parser.new(:symbolize_keys => true)
-      end
-
-      def parser
-        self.class.parser
-      end
-
-      def receive_data(data)
-        puts parser.parse(data).inspect
-        send_data "OK\n"
-      end
-    end
   end
 end

spec/fixtures/project/plugin.rb

@@ -4,7 +4,7 @@ class Plugin < Stasis::Plugin
   def plugin(controller, controllers, paths)
     if controller._[:root] == controller._[:dir]
       controller.before("custom_plugin.html") do
-        instead render(:text => 'root')
+        instead "pass"
       end
     end
     [ controller, controllers, paths ]

spec/stasis_spec.rb

@@ -10,6 +10,10 @@
     $files['not_dynamic.html'].should =~ /pass/
   end
 
+  it "should use the custom plugin" do
+    $files['custom_plugin.html'].should =~ /pass/
+  end
+
   describe 'generate with :only option' do
 
     before(:each) do