ssp 0.5 → 0.6

data/lib/ssp/application/chef.rb

@@ -1,6 +1,7 @@
 require 'chef/config'
 require 'chef/search/query'
 require 'chef/node'
+require 'chef/data_bag'
 
 class SSP::App::Chef < Thor
   namespace :chef
@@ -9,19 +10,37 @@ class SSP::App::Chef < Thor
                :desc => "Location of the knife configuration file",
                :default => File.join(ENV['HOME'], '.chef', 'knife.rb')
 
+  class_option :ssh_user, :aliases => "-u",
+               :desc => "The ssh username",
+               :default => "root"
+  class_option :ssh_password, :aliases => "-p",
+               :desc => "The ssh password"
 
   desc "client [NODES...]", "Runs chef-client on multiple nodes"
-  method_option :ssh_user, :aliases => "-u",
-                :desc => "The ssh username",
-                :default => "root"
-  method_option :ssh_password, :aliases => "-p",
-                :desc => "The ssh password"
   def client(*nodes)
     chef_config
     command = options[:ssh_user] == "root" ? "chef-client" : "sudo chef-client"
     ssh_run nodes.join(" "), command
   end
 
+  desc "deploy APP", "Runs chef-client with deployment enabled for the application on the right node"
+  method_options %w( production -P ) => false, %w( migrate -m ) => true,
+                 %w( rollback -r ) => false, %w( force -f ) => false
+  def deploy(app)
+    if options[:ssh_user] == "root"
+      raise Thor::Error, "You cannot deploy directly with the root user, use the -u option to specify your user"
+    end
+    chef_config
+    node = "#{options[:production] ? "#{app}-app-01" : "sage"}.sspti.me"
+    command = "sudo chef-client -V -l debug -L /tmp/#{app}_deploy_chef_run.#{Time.now.strftime("%Y%m%d%H%M")}.log | grep '\\[[^]]*\\] [A-Z]*:' | grep -v DEBUG:"
+
+    env = options[:production] ? "production" : "staging"
+    deploy_action = options[:rollback] ? "rollback" : options[:force] ? "force_deploy" : "deploy"
+    with_updated_bag(app, env, :deploy_action => deploy_action, :migrate => options[:migrate]) do
+      ssh_run(node, command)
+    end
+  end
+
   private
     def ssh_run(host, command, user = options[:ssh_user], pass = options[:ssh_password], manual = !host.index(':'))
       ssh = ::Chef::Knife::Ssh.new
@@ -52,4 +71,26 @@ class SSP::App::Chef < Thor
       end
       ::Chef::Config
     end
+
+    def with_updated_bag(app, env, attributes={})
+      bag = Chef::DataBagItem.new
+      bag.data_bag("apps")
+      bag.raw_data = Chef::DataBagItem.load("apps", app)
+
+      original_values = {}
+      attributes.each do |key, value|
+        key = key.to_s
+        bag.raw_data[key] ||= {}
+        original_values[key] = bag.raw_data[key].dup
+        bag.raw_data[key][env] = value
+      end
+      bag.save
+
+      yield
+
+      attributes.each do |key, value|
+        bag.raw_data[key.to_s] = original_values[key.to_s]
+      end
+      bag.save
+    end
 end

data/lib/ssp/version.rb

@@ -1,3 +1,3 @@
 module SSP
-  VERSION = "0.5"
+  VERSION = "0.6"
 end

metadata

@@ -4,8 +4,8 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 5
-  version: "0.5"
+  - 6
+  version: "0.6"
 platform: ruby
 authors: 
 - "L\xC3\xA1szl\xC3\xB3 B\xC3\xA1csi"
@@ -13,7 +13,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-10-14 00:00:00 +02:00
+date: 2010-10-21 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency