buckknife 0.0.4

data/lib/buckknife/version.rb

@@ -0,0 +1,3 @@
+module BuckKnife
+  VERSION = "0.0.4"
+end

data/lib/chef/knife/project_add_role.rb

@@ -0,0 +1,24 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectAddRole < Chef::Knife
+    banner "knife project add role PROJECT [NODE]"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    def run
+      project   = project_from_arg_or_ask
+      node_name = @name_args[1]
+      nodes     = Array(project.node(node_name) || project.nodes)
+
+      nodes.each do |node|
+        ui.output "# Run these command to add roles one at a time to #{node.name}:"
+        ui.output BuckKnife::Commands::AddRole.new( node )
+      end
+    end
+
+  end
+end

data/lib/chef/knife/project_bootstrap.rb

@@ -0,0 +1,24 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectBootstrap < Chef::Knife
+    banner "knife project bootstrap PROJECT [NODE]"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    def run
+      project   = project_from_arg_or_ask
+      node_name = @name_args[1]
+      nodes     = Array(project.node(node_name) || project.nodes)
+
+      nodes.each do |node|
+        ui.output "# Run this command to bootstrap #{node.name}:"
+        ui.output BuckKnife::Commands::Bootstrap.new( project, node )
+      end
+    end
+
+  end
+end

data/lib/chef/knife/project_capistrano.rb

@@ -0,0 +1,19 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectCapistrano < Chef::Knife
+    banner "knife project capistrano PROJECT"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    def run
+      project = project_from_arg_or_ask
+      ui.output "# Copy and commit this file to the project's repository:"
+      ui.output BuckKnife::Commands::Capistrano.new(project).call
+    end
+
+  end
+end

data/lib/chef/knife/project_create.rb

@@ -0,0 +1,53 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectCreate < Chef::Knife
+    banner "knife project create PROJECT [NODE]"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    option :identity_file,
+      :short => "-i IDENTITY_FILE",
+      :long => "--identity-file",
+      :description => "The SSH identity file used for authentication"
+
+    option :provider,
+      :short => "-p PROVIDER",
+      :long => "--provider",
+      :description => "The cloud provider to use (euca, rackspace, ...)"
+
+    option :ssh_username,
+      :short => "-x USERNAME",
+      :long => "--ssh-user",
+      :description => "The ssh username"
+
+    def run
+      project   = project_from_arg_or_ask
+      node_name = @name_args[1]
+      nodes     = Array(project.node(node_name) || project.nodes)
+
+      project.identity_file = Chef::Config[:knife][:euca_identity_file] if Chef::Config[:knife][:euca_identity_file]
+      project.provider      = config[:provider] if config[:provider]
+      project.ssh_username  = config[:ssh_username] if config[:ssh_username]
+
+      # TODO: Move this into project, ActiveModel validation?
+      unless project.valid?
+        ui.fatal("The #{project.name}.json file has some errors that need to be fixed:")
+        project.errors.each do |att, err|
+          ui.fatal("'#{att}' #{err}")
+        end
+
+        exit 1
+      end
+
+      nodes.each do |node|
+        ui.output "# Run this command to create #{node.name}:"
+        ui.output BuckKnife::Commands::Create[project.provider].new(project, node)
+      end
+    end
+
+  end
+end

data/lib/chef/knife/project_info.rb

@@ -0,0 +1,13 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectInfo < Chef::Knife
+    banner "knife project info"
+
+    def run
+      ui.output "Config file: %s" % Chef::Config[:config_file] || config[:config_file]
+      ui.output "Chef Server: %s" % Chef::Config[:chef_server_url]
+      ui.output "Project Dir: %s" % Chef::Config[:knife][:project_dir]
+    end
+  end
+end

data/lib/chef/knife/project_list.rb

@@ -0,0 +1,17 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectList < Chef::Knife
+    banner "knife project list"
+
+    deps do
+      require 'buckknife'
+    end
+
+    def run
+      BuckKnife::Project.all.each do |project|
+        ui.output "  #{project.name}"
+      end
+    end
+  end
+end

data/lib/chef/knife/project_new.rb

@@ -0,0 +1,28 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectNew < Chef::Knife
+    banner "knife project new PROJECT"
+
+    deps do
+      require 'buckknife'
+    end
+
+    def run
+      project_name = @name_args[0]
+
+      if project_name.nil?
+        show_usage
+        ui.fatal("You must specify a project name")
+        exit 1
+      end
+
+      template = BuckKnife.template_root.join("project.json.erb").read
+      compiled = ERB.new(template, nil, '%>').result(binding)
+      project_path = Project.root.join("#{project_name}.json")
+      project_path.open("w") { |f| f.write compiled }
+
+      ui.output ui.color("#{project_name}.json written to #{project_path}", :green)
+    end
+  end
+end

data/lib/chef/knife/project_run_client.rb

@@ -0,0 +1,30 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectRunClient < Chef::Knife
+    banner "knife project run client PROJECT [NODE] [options]"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    option :env,
+      :short => "-e ENVIRONMENT",
+      :long => "--env ENVIRONMENT",
+      :description => "Project environment to scope to (dev,staging,production)"
+
+    def run
+      project   = project_from_arg_or_ask
+      node_name = @name_args[1]
+
+      ui.output "# Run this command to run client:"
+      ui.output BuckKnife::Commands::RunClient.new(
+        :project     => project.name,
+        :node        => node_name,
+        :environment => config[:env]
+      )
+    end
+
+  end
+end

data/lib/chef/knife/project_show.rb

@@ -0,0 +1,34 @@
+require 'chef/knife'
+
+module BuckKnife
+  class ProjectShow < Chef::Knife
+    banner "knife project show PROJECT"
+
+    deps do
+      require 'buckknife'
+      include BuckKnife::KnifeHelper
+    end
+
+    def run
+      ui.use_presenter BuckKnife::ProjectPresenter
+      project = project_from_arg_or_ask
+      output(project)
+    end
+  end
+
+  class ProjectPresenter < Chef::Knife::Core::GenericPresenter
+
+    def summarize(project)
+      summarized = <<-SUMMARY
+#{ui.color('Project Name:', :bold)} #{ui.color(project.name, :bold)}
+#{key('Environments:')} #{project.environment_names.join(', ')}
+#{key('Nodes:')}        #{project.node_names.join("\n" + ' ' * 14)}
+      SUMMARY
+    end
+
+    def key(key_text)
+      ui.color(key_text, :cyan)
+    end
+  end
+
+end

data/spec/buckknife/commands/add_role_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+
+describe BuckKnife::Commands::AddRole do
+  it "should output a knife command to add roles" do
+    project = BuckKnife::Project.find('myapp')
+    node    = project.node('myapp-monitor1')
+    cmd     = BuckKnife::Commands::AddRole.new(node)
+
+    cmd.to_s.should == [
+      "knife node run_list add myapp-monitor1 'role[monitor_client]'",
+      "knife node run_list add myapp-monitor1 'role[rails_app]'"
+    ].join("\n")
+  end
+end

data/spec/buckknife/commands/bootstrap_spec.rb

@@ -0,0 +1,22 @@
+# cli = BuckKnife::CLI.new(args, options, config)
+# cli.invoke(:task, [first_arg, second_arg, third_arg], {options hash})
+require 'spec_helper'
+
+describe BuckKnife::Commands::Bootstrap do
+  it "should output a knife command to bootstrap the server on rackspace" do
+    project = BuckKnife::Project.find('myapp')
+    node    = project.node('myapp-monitor1')
+    cmd     = BuckKnife::Commands::Bootstrap.new(project, node)
+
+    cmd.to_s.should == [
+      "knife bootstrap 127.0.0.1",
+      "-N myapp-monitor1",
+      "-x username",
+      "-P password",
+      "-r 'role[base],recipe[base_projects::myapp],role[base_server]'",
+      "-d ubuntu10.04.deb",
+      "--sudo"
+    ].join(' ')
+
+  end
+end

data/spec/buckknife/commands/capistrano_spec.rb

@@ -0,0 +1,13 @@
+# cli = BuckKnife::CLI.new(args, options, config)
+# cli.invoke(:task, [first_arg, second_arg, third_arg], {options hash})
+require 'spec_helper'
+
+describe BuckKnife::Commands::Capistrano do
+  it "should output a capistrano deploy script for a project" do
+    BuckKnife::Commands::Capistrano.any_instance.stub(:fetch_ip) { "127.0.0.1" }
+    project = BuckKnife::Project.find('myapp')
+    cmd     = BuckKnife::Commands::Capistrano.new(project)
+
+    cmd.call.should =~ /DO NOT MODIFY THIS FILE/
+  end
+end

data/spec/buckknife/commands/create_spec.rb

@@ -0,0 +1,48 @@
+# cli = BuckKnife::CLI.new(args, options, config)
+# cli.invoke(:task, [first_arg, second_arg, third_arg], {options hash})
+require 'spec_helper'
+
+describe BuckKnife::Commands::Create do
+  it "should output a knife command to create the server on rackspace" do
+    project = BuckKnife::Project.find('myapp')
+    node    = project.node('myapp-monitor1')
+    cmd     = BuckKnife::Commands::Create[project.provider].new(project, node)
+
+    cmd.to_s.should == [
+      "knife rackspace server create",
+      "-r 'role[base],recipe[base_projects::myapp],role[base_server]'",
+      "-f 4",
+      "-I 49",
+      "-d ubuntu10.04.deb",
+      "-s http://localhost:4000",
+      "-N myapp-monitor1",
+      "-S myapp-monitor1",
+      "-K ''",
+      "-A ''"
+    ].join(' ')
+
+  end
+
+  it "should output a knife command to create the server on eucalyptus" do
+    project = BuckKnife::Project.find('myapp')
+    project.should_receive(:provider) { "euca" }.twice
+    node    = project.node('myapp-monitor1')
+
+    node.flavor.should be_nil
+
+    cmd = BuckKnife::Commands::Create[project.provider].new(project, node)
+    cmd.should_receive(:availability_zone) { "us-west-1" }.twice
+
+    cmd.to_s.should == [
+      "knife euca server create",
+      "-r 'role[base],recipe[base_projects::myapp],role[base_server]'",
+      "-f m1.large",
+      "-I emi-DFB9107F",
+      "-d ubuntu10.04.deb",
+      "-s http://localhost:4000",
+      "-N myapp-monitor1",
+      "-Z us-west-1"
+    ].join(' ')
+
+  end
+end

data/spec/buckknife/commands/run_client_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe BuckKnife::Commands::RunClient do
+  it "should output a knife command to add roles" do
+    cmd = BuckKnife::Commands::RunClient.new(:project => 'myapp', :node => 'myapp-monitor1')
+
+    cmd.to_s.should == [
+      "knife ssh 'project:myapp AND name:myapp-monitor1'",
+      "'sudo chef-client'",
+      "--attribute ipaddress",
+      "-x username",
+      "-P password"
+    ].join(' ')
+  end
+end

data/spec/buckknife/environment_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+
+describe BuckKnife::Environment do
+  context "initialized with a name and hash of data" do
+    subject { BuckKnife::Environment.new("dev", {"domain_name" => "dev.myapp.com"}, []) }
+    it "should have a name" do
+      subject.name.should == "dev"
+    end
+
+    it "should have a domain name" do
+      subject.domain_name.should == "dev.myapp.com"
+    end
+  end
+end

data/spec/buckknife/helper_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe BuckKnife::Helper do
+  it "should create a run list from roles and recipes" do
+    roles   = ['base', 'base_server']
+    recipes = ["base_projects::projectname"]
+    sep     = ','
+    result = BuckKnife::Helper.combine_roles_and_recipes(roles, recipes, sep)
+    result.should == "role[base],role[base_server],recipe[base_projects::projectname]"
+  end
+end

data/spec/buckknife/node_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+describe BuckKnife::Node do
+  context "initialized with a hash of data" do
+    subject { BuckKnife::Project.find("myapp").node("myapp-monitor1") }
+    it "should have a name" do
+      subject.name.should == "myapp-monitor1"
+    end
+
+    it "should have a size" do
+      subject.size.should == "2G"
+    end
+
+    it "should have a run list" do
+      subject.run_list.should == "role[monitor_client] role[rails_app]"
+    end
+  end
+end

data/spec/buckknife/project_spec.rb

@@ -0,0 +1,49 @@
+require 'spec_helper'
+
+describe BuckKnife::Project do
+  context "initialized with a hash of data" do
+    subject { BuckKnife::Project.find("myapp") }
+    it "should have a name" do
+      subject.name.should == "myapp"
+    end
+
+    it "should have node names" do
+      subject.node_names.should == %w[
+        myapp-dev-app1
+        myapp-dev-app2
+        myapp-dev-job1
+        myapp-monitor1
+        myapp-prod-app1
+        myapp-prod-app2
+        myapp-prod-job1
+        myapp-staging-app1
+        myapp-staging-app2
+        myapp-staging-job1
+      ]
+    end
+
+    it "should have nodes" do
+      subject.nodes.first.should be_instance_of BuckKnife::Node
+    end
+
+    it "can look up a node based on name" do
+      subject.node("myapp-monitor1").should be_instance_of BuckKnife::Node
+    end
+
+    it "should have a repository" do
+      subject.repository.should == "http://svn.domain.com/myapp/trunk"
+    end
+
+    it "should have environment names" do
+      subject.environment_names.should == ["dev", "production", "staging"]
+    end
+
+    it "should have environments" do
+      subject.environments.first.should be_instance_of BuckKnife::Environment
+    end
+
+    it "should have a provider" do
+      subject.provider.should == "rackspace"
+    end
+  end
+end