adding restart_command to deploy resource

README.rdoc

@@ -6,13 +6,17 @@ A gem that provides resources and providers for deploying ruby web apps from che
 Uses the same directory layout as capistrano and steals the git remote cached deploy strategy 
 from cap and adapts it to work without cap and under chef
 
-  deploy "/data/myapp" do
-    repo "git://github.com/engineyard/rack-app.git"
-    branch "HEAD"
-    enable_submodules true
-    shallow_clone true
-    action :manage
-  end
+ deploy "/data/#{app}" do
+   repo "git://github.com/engineyard/rack-app.git"
+   branch "HEAD"
+   user "ez"
+   enable_submodules true
+   migrate true
+   migration_command "rake db:migrate"
+   environment "production"
+   shallow_clone true
+   action :manage
+ end
 
 
 Currently only supports doing the cached git checkout, rsyncing it into place and doing the symlink dance.

lib/chef-deploy.rb

@@ -6,6 +6,9 @@
 #   branch "HEAD"
 #   user "ez"
 #   enable_submodules true
+#   migrate true
+#   migration_command "rake db:migrate"
+#   environment "production"
 #   shallow_clone true
 #   action :manage
 # end
@@ -34,6 +37,14 @@ def repo(arg=nil)
         )
       end
 
+      def restart_command(arg=nil)
+        set_or_return(
+          :restart_command,
+          arg,
+          :kind_of => [ String ]
+        )
+      end
+
       def migrate(arg=nil)
         set_or_return(
           :migrate,
@@ -129,6 +140,7 @@ def action_manage
         Chef::Log.level(:debug)
         Chef::Log.info "Running a new deploy\nto: #{@new_resource.name}\nrepo: #{@new_resource.repo}"
         dep = CachedDeploy.new  :user       => @new_resource.user,
+                                :restart_command => @new_resource.restart_command,
                                 :repository => @new_resource.repo,
                                 :environment => @new_resource.environment,
                                 :migration_command => @new_resource.migration_command,

lib/chef-deploy/cached_deploy.rb

@@ -1,10 +1,12 @@
 # stolen wholesale from capistrano, thanks Jamis!
 
+class ChefDeployFailure < StandardError
+end
+
 class CachedDeploy
   # Executes the SCM command for this strategy and writes the REVISION
   # mark file to each host.
   def deploy
-    @buffer = []
     @configuration[:release_path] = "#{@configuration[:deploy_to]}/releases/#{Time.now.utc.strftime("%Y%m%d%H%M%S")}"
     @configuration[:revision] ||= source.query_revision('HEAD') {|cmd| run cmd}
     Chef::Log.info "updating the cached checkout"
@@ -16,7 +18,14 @@ def deploy
     callback(:before_symlink)
     symlink
     callback(:before_restart)
-    @buffer
+    restart
+  end
+
+  def restart
+    unless @configuration[:restart_command].empty?
+      Chef::Log.info "restarting app: #{latest_release}"
+      Chef::Log.info run(@configuration[:restart_command])
+    end
   end
 
   # before_symlink
@@ -101,7 +110,7 @@ def symlink
 
   def run(cmd)
     res = `#{cmd}`
-    @buffer << res
+    raise ChefDeployFailure unless $? == 0
     res
   end