capistrano-hivequeen 0.1.0

data/README.rdoc

@@ -0,0 +1,16 @@
+== HiveQueen client
+
+Makes your capistrano scripts aware of HiveQueen environments.
+
+=== Installation
+
+In your Gemfile:
+
+    group :development do
+      gem 'capistrano/hivequeen', :require => nil
+    end
+
+And in your config/deploy.rb:
+
+    set :hivequeen_endpoint, "http://your.hivequeen.com"
+    require 'hivequeen_client'

data/capistrano-hivequeen.gemspec

@@ -0,0 +1,48 @@
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+
+  ## Leave these as is they will be modified for you by the rake gemspec task.
+  ## 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.1.0'
+  s.date              = '2011-10-03'
+
+  ## Make sure your summary is short. The description may be as long
+  ## as you like.
+  s.summary     = "Capistrano extensions for interacting with HiveQueen"
+  s.description = "Capistrano extensions for interacting with HiveQueen"
+
+  ## List the primary authors. If there are a bunch of authors, it's probably
+  ## better to set the email to an email list or something. If you don't have
+  ## a custom homepage, consider using your GitHub URL or the like.
+  s.authors  = ["Aaron Suggs"]
+  s.email    = 'aaron@kickstarter.com'
+  s.homepage = 'http://github.com/kickstarter/capistrano-hivequeen'
+
+  ## This sections is only necessary if you have C extensions.
+  # s.require_paths << 'ext'
+  # s.extensions = %w[ext/extconf.rb]
+
+  ## If your gem includes any executables, list them here.
+  s.executables = []
+
+  ## Specify any RDoc options here. You'll want to add your README and
+  ## LICENSE files to the extra_rdoc_files list.
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.extra_rdoc_files = %w[README.rdoc]
+
+  ## List your runtime dependencies here. Runtime dependencies are those
+  ## that are needed for an end user to actually USE your code.
+  s.add_dependency('capistrano-ext')
+  s.add_dependency('capistrano')
+  s.add_dependency('json')
+
+  ## List your development dependencies here. Development dependencies are
+  ## those that are only needed during development
+  #s.add_development_dependency('rake')
+
+  s.files = `git ls-files`.split("\n")
+  s.test_files = `git ls-files -- {spec,tests}/*`.split("\n")
+end

data/lib/capistrano/hivequeen.rb

@@ -0,0 +1,53 @@
+# Load environment configuration from HiveQueen
+require "capistrano/hivequeen/server"
+
+Capistrano::Configuration.instance(:must_exist).load do
+  HiveQueen.endpoint = hivequeen_endpoint
+  HiveQueen.logger = logger
+
+  # Redefine stage tasks from multistage extension
+  # Set the list of available stages
+  set :stages, HiveQueen.environment_names
+  # When no stage is specified, use staging
+  set :default_stage, :staging
+
+  set :repository, HiveQueen.repository
+
+  # Load capistrano multi-stage extension
+  require 'fileutils' # required until https://github.com/capistrano/capistrano-ext/commit/930ca840a0b4adad0ec53546790b3f5ffe726538 is released
+  require 'capistrano/ext/multistage'
+
+  # Redefine stage tasks from multistage extension
+  HiveQueen.environments.each do |env|
+    name = env['name']
+    hive_queen_id = env['id']
+
+    desc "Use environment #{name}"
+    task name do
+      set :stage, name.to_sym
+      set :environment_id, env['id']
+      environment = HiveQueen.roles(hive_queen_id)
+      # Check if environment is ready
+      unless environment['state'] == 'running'
+        abort "Environment #{name} is not ready. State: #{environment['state']}"
+      end
+
+      # Set servers for each role
+      environment['roles'].each do |role_name, role_config|
+        role(role_name.to_sym) { role_config['servers'] }
+      end
+
+      # Ensure some server designated as db server
+      unless roles.key?(:db)
+        # Prefer the bg server
+        db_server = roles[:bg].servers.first if roles.key?(:bg)
+
+        # Otherwise, use any server
+        db_server ||= roles.values.map{|x| x.servers}.flatten.compact.first
+        logger.trace "Using #{db_server} as primary db server"
+        role :db, db_server.to_s, :primary => true
+      end
+
+    end
+  end
+end

data/lib/capistrano/hivequeen/server.rb

@@ -0,0 +1,43 @@
+# HTTP Client for Hive Queen environment configuration
+require 'open-uri'
+require 'json'
+require 'fileutils'
+
+# Special cases:
+# - environment not found
+# - environment not ready: fail
+class HiveQueen
+  class << self
+    attr_accessor :endpoint, :logger
+
+    def project_data
+      @project_data ||= fetch('/projects/kickstarter')
+    end
+
+    def environments
+      project_data['environments']
+    end
+
+    def environment_names
+      environments.map{|e| e['name'].to_sym }
+    end
+
+    def repository
+      project_data['repo']
+    end
+
+    def roles(env_id)
+      env_id = env_id.to_sym
+      @roles ||= {}
+      @roles[env_id] ||= fetch("/environments/#{env_id}")
+    end
+
+    protected
+    def fetch(path)
+      url = endpoint + path + '.json'
+      logger.trace "Fetching #{url}"
+      JSON.parse(open(url).read)
+    end
+
+  end
+end

metadata

@@ -0,0 +1,111 @@
+--- !ruby/object:Gem::Specification 
+name: capistrano-hivequeen
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Aaron Suggs
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-10-03 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: capistrano-ext
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: capistrano
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+description: Capistrano extensions for interacting with HiveQueen
+email: aaron@kickstarter.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- README.rdoc
+- capistrano-hivequeen.gemspec
+- lib/capistrano/hivequeen.rb
+- lib/capistrano/hivequeen/server.rb
+has_rdoc: true
+homepage: http://github.com/kickstarter/capistrano-hivequeen
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 2
+summary: Capistrano extensions for interacting with HiveQueen
+test_files: []
+