capistrano-hivequeen 0.5.0 → 0.6.0

data/capistrano-hivequeen.gemspec

@@ -6,7 +6,7 @@ Gem::Specification.new do |s|
   ## If your rubyforge_project name is different, then edit it and comment out
   ## the sub! line in the Rakefile
   s.name              = 'capistrano-hivequeen'
-  s.version           = '0.5.0'
+  s.version           = '0.6.0'
   s.date              = Time.now.strftime("%Y-%m-%d")
 
   ## Make sure your summary is short. The description may be as long

data/lib/capistrano/hivequeen.rb

@@ -16,6 +16,10 @@ Capistrano::Configuration.instance(:must_exist).load do
   set :repository, HiveQueen.repository
   set :scm, :git
 
+  # By default, don't override deployments if there's another deployment in progress.
+  # From the command line, use -s override=true to force a deployment
+  set :override, false
+
   # Load capistrano multi-stage extension
   require 'fileutils' # required until https://github.com/capistrano/capistrano-ext/commit/930ca840a0b4adad0ec53546790b3f5ffe726538 is released
   require 'capistrano/ext/multistage'

data/lib/capistrano/hivequeen/deploy.rb

@@ -8,6 +8,28 @@ Capistrano::Configuration.instance.load do
     end
   end
 
+  before "deploy:update", "hivequeen:start"
+  before "setup",         "hivequeen:start"
+  namespace :hivequeen do
+    desc "[internal] Start a deployment in hivequeen"
+    task :start do
+      # TODO: is there a better way to determine what cap tasks are running?
+      tasks = ARGV.reject{|task_name| stage.to_s == task_name}
+      params = {
+        :task => tasks.join(' '),
+        :commit => real_revision,
+        :override => override
+      }
+      begin
+        deployment = HiveQueen.start_deployment(environment_id, params)
+        set :deployment_id, deployment['id']
+        at_exit { HiveQueen.finish_deployment(environment_id, deployment['id']) }
+      rescue HiveQueen::DeploymentError
+        abort "Cannot start deployment. Errors: #{$!.message}"
+      end
+    end
+  end
+
   desc "Deploy without migrations"
   task(:hotfix) { deploy.default }
 

data/lib/capistrano/hivequeen/server.rb

@@ -8,11 +8,14 @@ require 'base64'
 # - environment not found
 # - environment not ready: fail
 class HiveQueen
+  class DeploymentError < Exception; end
+  class InsecureCredentials < Exception; end
+
   class << self
     attr_accessor :endpoint, :logger, :project, :username, :password
 
     def project_data
-      @project_data ||= fetch("/projects/#{project}.json")
+      @project_data ||= get("/projects/#{project}.json")
     end
 
     def environments
@@ -30,18 +33,40 @@ class HiveQueen
     def roles(env_id)
       env_id = env_id.to_sym
       @roles ||= {}
-      @roles[env_id] ||= fetch("/environments/#{env_id}.json")
+      @roles[env_id] ||= get("/environments/#{env_id}.json")
+    end
+
+    def start_deployment(environment_id, params)
+      required_params = [:task, :commit]
+      required_params.each do |key|
+        raise ArgumentError.new("#{key} is a required param") unless params.key?(key)
+      end
+      put_or_post('POST', "/environments/#{environment_id}/deployments.json", :deployment => params)
+    end
+
+    def finish_deployment(environment_id, deployment_id)
+      state = $! ? 'failed' : 'succeeded'
+      puts "Finishing deployment in Hivequeen. State: #{state}"
+      params = {:deployment => {:state => state}}
+      put_or_post('PUT', "/environments/#{environment_id}/deployments/#{deployment_id}.json", params)
     end
 
     # Load credentials from ~/.hivequeen
     def get_credentials!
       @username, @password = File.read(credential_path).chomp.split(':')
       raise unless username && password
+      # Check that credentials are not accessible to world or group
+      mode = File.stat(credential_path).mode
+      raise InsecureCredentials unless (mode % 64 == 0)
+    rescue InsecureCredentials
+      puts "#{credential_path} is insecure. Please change you password and run"
+      puts "chmod 600 #{credential_path}"
+      exit 1
     rescue Errno::ENOENT, RuntimeError
       puts "Could not read HiveQueen credentials from #{credential_path}."
       puts "#{credential_path} should contain your username and password seperated by a colon"
       puts "Run this command with your credentials:"
-      puts " $ echo username:password > #{credential_path}"
+      puts " $ echo username:password > #{credential_path}; chmod 600 #{credential_path}"
       exit 1
     end
 
@@ -59,7 +84,7 @@ class HiveQueen
       { 'Authorization' => "Basic #{value}" }
     end
 
-    def fetch(path)
+    def get(path)
       url = "#{endpoint}#{path}"
       logger.trace "Fetching #{url}"
       response = connection.request(:method => 'GET', :path => path, :headers => auth_header)
@@ -69,5 +94,19 @@ class HiveQueen
       JSON.parse(response.body)
     end
 
+    def put_or_post(method, path, data)
+      headers = auth_header.merge("Content-type"=>"application/json", "Accept"=>"application/json")
+      response = connection.request(:path => path, :method => method, :headers => headers, :body => data.to_json)
+      case response.status
+      when (200..299)
+        JSON.parse(response.body)
+      when (400..499)
+        errors = JSON.parse(response.body)
+        raise DeploymentError.new(errors.inspect)
+      else
+        raise "Request to #{path} returned #{response.status} status"
+      end
+    end
+
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: capistrano-hivequeen
 version: !ruby/object:Gem::Version 
-  hash: 11
+  hash: 7
   prerelease: 
   segments: 
   - 0
-  - 5
+  - 6
   - 0
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors: 
 - Aaron Suggs
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-10-18 00:00:00 -04:00
+date: 2011-10-20 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency