buckknife 0.0.4

data/lib/buckknife/commands/rackspace_create.rb

@@ -0,0 +1,58 @@
+module BuckKnife
+  module Commands
+    class RackspaceCreate < Create
+      include Base
+
+      #     --bootstrap-version VERSION  The version of Chef to install
+      # -N, --node-name NAME             The Chef node name for your new node
+      # -s, --server-url URL             Chef Server URL
+      # -k, --key KEY                    API Client Key
+      #     --color                      Use colored output
+      # -c, --config CONFIG              The configuration file to use
+      #     --defaults                   Accept default values for all questions
+      # -d, --distro DISTRO              Bootstrap a distro using a template; default is 'ubuntu10.04-gems'
+      # -e, --editor EDITOR              Set the editor to use for interactive commands
+      # -E, --environment ENVIRONMENT    Set the Chef environment
+      # -f, --flavor FLAVOR              The flavor of server; default is 2 (512 MB)
+      # -F, --format FORMAT              Which format to use for output
+      # -I, --image IMAGE                The image of the server
+      #     --no-color                   Don't use colors in the output
+      # -n, --no-editor                  Do not open EDITOR, just accept the data as is
+      # -u, --user USER                  API Client Username
+      #     --prerelease                 Install the pre-release chef gems
+      #     --print-after                Show the data after a destructive operation
+      #     --rackspace-api-auth-url URL Your rackspace API auth url; default is 'auth.api.rackspacecloud.com'
+      # -K, --rackspace-api-key KEY      Your rackspace API key
+      # -A USERNAME                      Your rackspace API username
+      #     --rackspace-api-username
+      # -r, --run-list RUN_LIST          Comma separated list of roles/recipes to apply
+      # -S, --server-name NAME           The server name
+      # -P, --ssh-password PASSWORD      The ssh password
+      # -x, --ssh-user USERNAME          The ssh username; default is 'root'
+      #     --template-file TEMPLATE     Full path to location of template to use
+      #     --sudo                       Execute the bootstrap via sudo
+      # -V, --verbose                    More verbose output. Use twice for max verbosity
+      # -v, --version                    Show chef version
+      # -y, --yes                        Say yes to all prompts for confirmation
+      # -h, --help                       Show this message
+
+      def command
+        super.tap do |c|
+          c.opt '-S', node.name
+          c.opt '-K', project.rs_api_key
+          c.opt '-A', project.rs_api_username
+        end
+      end
+
+      protected
+
+      def image_id
+        node.image_id || '49'
+      end
+
+      def flavor
+        BuckKnife::Helper.rackspace_flavors[ node.size ].to_s
+      end
+    end
+  end
+end

data/lib/buckknife/commands/run_client.rb

@@ -0,0 +1,49 @@
+module BuckKnife
+  module Commands
+    # options is a hash with one or more of the following keys specified:
+    # :project      => Project Name eg. myapp
+    # :environment  => Environment Name eg. staging
+    # :node         => Node Name eg. myapp-staging-app1
+    # :ssh_user     => (optional) ssh user to connect with
+    # :ssh_password => (optional) ssh user's password
+    class RunClient
+      include Base
+
+      def initialize(options)
+        @project = BuckKnife::Project.find( options[:project] )
+        @options = options
+      end
+
+      def command
+        super.tap do |c|
+          c.arg 'ssh', query, 'sudo chef-client'
+          c.opt '--attribute', 'ipaddress'
+          c.opt '-x', ssh_username
+          c.opt '-P', ssh_password
+        end
+      end
+
+      protected
+
+      def query
+        result = []
+        result << "project:#{@options[:project]}"         if @options[:project]
+        result << "environment:#{@options[:environment]}" if @options[:environment]
+        result << "name:#{@options[:node]}"               if @options[:node]
+        result.join(' AND ')
+      end
+
+      def node
+        @project.node(@options[:node]) || @project.nodes.first
+      end
+
+      def ssh_username
+        node.ssh_username
+      end
+
+      def ssh_password
+        node.ssh_password
+      end
+    end
+  end
+end

data/lib/buckknife/data_accessor.rb

@@ -0,0 +1,29 @@
+module BuckKnife
+  module DataAccessor
+    def self.included(base)
+      base.extend(ClassMethods)
+
+      def _read_data(key)
+        @data[key.to_s]
+      end
+
+      def _write_data(key, val)
+        @data[key.to_s] = val
+      end
+    end
+
+    module ClassMethods
+      def data_accessor(*names)
+        names.each do |name|
+          define_method(name) do
+            _read_data(name)
+          end
+
+          define_method("#{name}=") do |val|
+            _write_data(name, val)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/buckknife/environment.rb

@@ -0,0 +1,16 @@
+module BuckKnife
+  class Environment
+    attr_reader :name, :nodes
+
+    def initialize(name, data, nodes)
+      @data  = data
+      @name  = name
+      @nodes = nodes
+    end
+
+    def domain_name
+      @data['domain_name']
+    end
+
+  end
+end

data/lib/buckknife/helper.rb

@@ -0,0 +1,39 @@
+module BuckKnife::Helper
+  extend self
+
+  def parse_node_ip(node, interface)
+    network    = node['network']
+    interfaces = network['interfaces']
+    iface      = interfaces[interface]
+    addresses  = iface['addresses']
+
+    return addresses.keys.select {|key| addresses[key]['family'] == 'inet'}.first
+  end
+
+  def read_project_data_bag(project)
+    project_data_bag_file = File.expand_path File.join( BuckKnife.project_root, "#{project}.json" )
+    project_data          = JSON.parse File.read(project_data_bag_file)
+  end
+
+  def combine_roles_and_recipes(roles=[], recipes=[], separator=' ')
+    role_array        = roles.map { |role| "role[#{role}]" }
+    recipe_array      = recipes.map { |recipe| "recipe[#{recipe}]" }
+    roles_and_recipes = role_array.concat(recipe_array).join(separator)
+
+    roles_and_recipes
+  end
+
+  def rackspace_flavors
+    {
+      '256M' => 1,
+      '512M' => 2,
+      '1G'   => 3,
+      '2G'   => 4,
+      '4G'   => 5,
+      '8G'   => 6,
+      '16G'  => 7
+    }
+  end
+
+end
+

data/lib/buckknife/knife_helper.rb

@@ -0,0 +1,22 @@
+module BuckKnife
+  module KnifeHelper
+    def project_from_arg_or_ask
+      project_name = @name_args[0]
+
+      if project_name.nil?
+        show_usage
+        ui.fatal("You must specify a project name")
+        exit 1
+      end
+
+      begin
+        project = Project.find(project_name)
+      rescue BuckKnife::UnknownProjectError
+        ui.fatal("A project data file '#{project_name}.json' was not found in #{Project.root}")
+        exit 1
+      end
+
+      return project
+    end
+  end
+end

data/lib/buckknife/node.rb

@@ -0,0 +1,30 @@
+module BuckKnife
+  class Node
+    include DataAccessor
+    attr_accessor :name
+
+    def initialize(name, data)
+      @name = name
+      @data = data
+    end
+
+    data_accessor :image_id, :size, :flavor, :environment, :address, :group
+    data_accessor :ssh_username, :ssh_password
+
+    def roles
+      @data["roles"].split(' ')
+    end
+
+    def recipes
+      @data["recipes"].split(' ')
+    end
+
+    def run_list
+      BuckKnife::Helper.combine_roles_and_recipes(roles, recipes)
+    end
+
+    def run_list_items
+      run_list.split(' ')
+    end
+  end
+end

data/lib/buckknife/project.rb

@@ -0,0 +1,80 @@
+require 'pathname'
+require 'active_model'
+
+module BuckKnife
+  class Project
+    include DataAccessor
+    include ActiveModel::Validations
+
+    data_accessor :repository, :distro, :base_run_list, :provider
+    data_accessor :ssh_username, :identity_file
+    data_accessor :rs_api_key, :rs_api_username
+
+    validates_presence_of :provider, :distro, :base_run_list
+
+    class << self
+      def all_names
+        Dir["#{root}/*.json"].map { |f| File.basename(f).split(".").first }
+      end
+
+      def all
+        Dir["#{root}/*.json"].map { |f| from_file(f) }
+      end
+
+      def find(name)
+        from_file root.join("#{name}.json")
+      end
+
+      def from_file(path)
+        raise UnknownProjectError unless File.exist?(path)
+        new( JSON.parse( File.read(path) ) )
+      end
+
+      def root
+        @root ||= begin
+          unless path = Chef::Config[:knife][:project_dir]
+            raise UnknownProjectRootError, "You must define knife[:project_dir] in your knife.rb"
+          end
+
+          Pathname.new path
+        end
+      end
+
+      def root=(path)
+        @root = Pathname.new(path)
+      end
+    end
+
+    def initialize(data)
+      @data = data
+    end
+
+    def name
+      @data["id"]
+    end
+
+    def node_names
+      @data["nodes"].keys.sort
+    end
+
+    def nodes
+      @data["nodes"].map { |name, data| Node.new(name, data) }
+    end
+
+    def node(name)
+      nodes.find { |n| n.name == name }
+    end
+
+    def environment_names
+      @data["environments"].keys.sort
+    end
+
+    def environments
+      @data["environments"].map do |name, data|
+        Environment.new( name, data, nodes.select { |n| n.environment == name } )
+      end
+    end
+
+  end
+end
+

data/lib/buckknife/templates/deploy_environments.rb.erb

@@ -0,0 +1,85 @@
+# This file is automatically generated by BuckKnife.
+#
+# DO NOT MODIFY THIS FILE
+#
+# Place your custom variables, tasks and overrides in the deploy.rb 
+# file.  To add this file to your Rails application copy it the the 
+# config directory as config/deploy_environments.rb
+#
+# Then load it at the top of your deploy.rb file with the following
+# statement:
+#
+# load 'config/deploy_environments'
+
+
+require "bundler/capistrano"
+require 'highline/import'
+
+
+default_run_options[:pty] = true
+
+set :repository,       "<%= project.repository %>"
+set :scm,              :git
+set :app_user,         "deploy"
+set :user,             "<%= project.nodes.first.ssh_username %>"
+set :password,         "<%= project.nodes.first.ssh_password %>"
+set :use_sudo,         true
+
+set :checkout,         "export"
+set :deploy_via,       "copy"
+set :copy_strategy,    :export
+set :copy_exclude,     ".git/*"
+set :copy_compression, :zip
+
+
+
+<% environments.each do |environment| %>
+
+desc "Deploy to <%= environment[:name] %> (usage: cap <%= environment[:name] %> deploy)"
+task :<%= environment[:name] %> do
+  set :domain,      "<%= environment[:domain_name] %>"
+  set :application, "<%= project.name %>"
+  set :rails_env,   "<%= environment[:name] %>"
+  set :deploy_to,   "/var/www/<%= environment[:domain_name] %>"
+
+  <% environment[:app_servers].each do |app_server| %>
+
+  role :app,        "<%= app_server.address %>"                      # <%= app_server.name %>
+  <% end %>
+  <% environment[:job_servers].each do |job_server| %>
+
+  role :job,        "<%= job_server.address %>"                      # <%= job_server.name %>
+  <% end %>
+
+  role :db,         "<%= environment[:primary_db].address %>", :primary => true    # <%= environment[:primary_db].name %>
+
+end
+
+<% end %>
+
+namespace :deploy do
+  task :start do ; end
+  task :stop  do ; end
+
+  task :restart, :roles => :app, :except => { :no_release => true } do
+    run "#{try_sudo} touch #{File.join(current_path,'tmp','restart.txt')}"
+  end
+
+  task :chown_project, :roles => [:app, :job] do
+    sudo "chmod 777 #{deploy_to}/releases"
+    sudo "chown -R #{app_user}:sysadmin #{deploy_to}"
+  end
+
+  task :symlink_database_config, :roles => [:app, :job] do
+    run "rm #{release_path}/config/database.yml"
+    run "ln -s #{deploy_to}/shared/database.yml #{release_path}/config/database.yml"
+  end
+  
+end
+
+
+after 'deploy:setup',           'deploy:chown_project'
+after 'deploy',                 'deploy:chown_project'
+after 'deploy:finalize_update', 'deploy:symlink_database_config'
+
+

data/lib/buckknife/templates/project.json.erb

@@ -0,0 +1,75 @@
+{
+  "id": "<%= project_name %>",
+  "domain": "<%= project_name %>.com",
+  "repository": "FIXME",
+  "provider": "rackspace",
+  "public_interface": "eth0",
+  "internal_interface": "eth1",
+  "distro": "FIXME",
+  "base_run_list": "monitor_client",
+
+  "environments": {
+    "production": {
+      "domain_name": "origin.<%= project_name %>.com",
+      "www_virtual_ip": "FIXME",
+      "database_user": "FIXME",
+      "database_password": "FIXME"
+    },
+    "staging": {
+      "domain_name": "staging.<%= project_name %>.com",
+      "www_virtual_ip": "127.0.0.1",
+      "database_user": "FIXME",
+      "database_password": "FIXME"
+    },
+    "dev": {
+      "domain_name": "dev.<%= project_name %>.com",
+      "www_virtual_ip": "127.0.0.1",
+      "database_user": "FIXME",
+      "database_password": "FIXME"
+    }
+  },
+
+  "nodes": {
+    "myapp-monitor1": {
+      "group": "monitor",
+      "size": "2G",
+      "environment": "production",
+      "roles": "",
+      "recipes": "",
+      "address": "127.0.0.1",
+      "ssh_username": "FIXME",
+      "ssh_password": "FIXME"
+    },
+
+    "myapp-dev-app1": {
+      "group": "app",
+      "size": "2G",
+      "environment": "dev",
+      "roles": "app_heartbeat_pair rails_app rails_nginx_app",
+      "recipes": "",
+      "address": "127.0.0.1",
+      "ssh_username": "FIXME",
+      "ssh_password": "FIXME"
+    },
+    "myapp-dev-app2": {
+      "group": "app",
+      "size": "2G",
+      "environment": "dev",
+      "roles": "app_heartbeat_pair rails_app rails_nginx_app",
+      "recipes": "",
+      "address": "127.0.0.1",
+      "ssh_username": "username",
+      "ssh_password": "password"
+    },
+    "myapp-dev-job1": {
+      "group": "job",
+      "size": "2G",
+      "environment": "dev",
+      "roles": "rails_app rails_job",
+      "recipes": "",
+      "address": "127.0.0.1",
+      "ssh_username": "username",
+      "ssh_password": "password"
+    }
+  }
+}