node-deploy.rb in capistrano-node-deploy-1.1.0

- old
+ new

@@ -1,5 +1,6 @@
+require "digest/md5"
 require "railsless-deploy"
 require "multi_json"
 
 UPSTART_TEMPLATE = <<EOD
 #!upstart
@@ -11,69 +12,88 @@
 
 respawn
 respawn limit 99 5
 
 script
-    exec su -c 'NODE_ENV=production {{node_binary}} {{current_path}}/{{app_command}}' deploy 2>&1 >> {{shared_path}}/{{application}}.log
+    exec sudo -u {{node_user}} NODE_ENV={{node_env}} {{node_binary}} {{current_path}}/{{app_command}} 2>> {{shared_path}}/{{node_env}}.err.log 1>> {{shared_path}}/{{node_env}}.out.log
 end script
 EOD
 
 def remote_file_exists?(full_path)
   'true' ==  capture("if [ -e #{full_path} ]; then echo 'true'; fi").strip
 end
 
+def remote_file_content_same_as?(full_path, content)
+  Digest::MD5.hexdigest(content) == capture("md5sum #{full_path} | awk '{ print $1 }'").strip
+end
+
+def remote_file_differs?(full_path, content)
+  exists = remote_file_exists?(full_path)
+  !exists || exists && !remote_file_content_same_as?(full_path, content)
+end
+
+def generate_upstart_config
+  UPSTART_TEMPLATE.gsub(/\{\{(.*?)\}\}/) { eval($1) }
+end
+
 Capistrano::Configuration.instance(:must_exist).load do |configuration|
   before "deploy", "deploy:create_release_dir"
   before "deploy", "node:check_upstart_config"
-  after "deploy:create_symlink", "node:install_packages"
-  after "deploy:create_symlink", "node:restart"
+  after "deploy:finalize_update", "node:install_packages"
+  after "deploy:finalize_update", "node:restart"
   after "deploy:rollback", "node:restart"
 
   package_json = MultiJson.load(File.open("package.json").read) rescue {}
 
   set :application, package_json["name"] unless defined? application
   set :app_command, package_json["main"] || "index.js" unless defined? app_command
+
   set :node_binary, "/usr/bin/node" unless defined? node_binary
+  set :node_env, "production" unless defined? node_env
+  set :node_user, "deploy" unless defined? node_user
 
+  set :upstart_job_name, "#{application}-#{node_env}" unless defined? upstart_job_name
+  set :upstart_file_path, "/etc/init/#{upstart_job_name}.conf" unless defined? upstart_file_path
+
   namespace :node do
     desc "Check required packages and install if packages are not installed"
     task :install_packages do
       run "mkdir -p #{shared_path}/node_modules"
       run "cp #{current_path}/package.json #{shared_path}"
       run "cd #{shared_path} && npm install"
       run "ln -s #{shared_path}/node_modules #{current_path}/node_modules"
     end
 
     task :check_upstart_config do
-      create_upstart_config unless remote_file_exists?("/etc/init/#{application}.conf")
+      create_upstart_config if remote_file_differs?(upstart_file_path, generate_upstart_config)
     end
 
     desc "Create upstart script for this node app"
     task :create_upstart_config do
       config_file_path = "/etc/init/#{application}.conf"
       temp_config_file_path = "#{shared_path}/#{application}.conf"
 
       # Generate and upload the upstart script
-      put UPSTART_TEMPLATE.gsub(/\{\{(.*?)\}\}/) { eval($1) }, temp_config_file_path
+      put generate_upstart_config, temp_config_file_path
 
       # Copy the script into place and make executable
-      sudo "cp #{temp_config_file_path} #{config_file_path}"
-      sudo "chmod +x #{config_file_path}"
+      sudo "cp #{temp_config_file_path} #{upstart_file_path}"
     end
 
     desc "Start the node application"
     task :start do
-      sudo "start #{application}"
+      sudo "start #{upstart_job_name}"
     end
 
     desc "Stop the node application"
     task :stop do
-      sudo "stop #{application}"
+      sudo "stop #{upstart_job_name}"
     end
 
     desc "Restart the node application"
     task :restart do
-      sudo "restart #{application} || sudo start #{application}"
+      sudo "stop #{upstart_job_name}"
+      sudo "start #{upstart_job_name}"
     end
   end
 
   namespace :deploy do
     task :create_release_dir, :except => {:no_release => true} do
\ No newline at end of file