mbailey-chef 0.9.12.1

data/lib/chef/knife/ec2_instance_data.rb

@@ -0,0 +1,46 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class Ec2InstanceData < Knife
+
+      banner "knife ec2 instance data [RUN LIST...] (options)"
+
+      option :edit,
+        :short => "-e",
+        :long => "--edit",
+        :description => "Edit the instance data"
+
+      def run 
+        data = {
+          "chef_server" => Chef::Config[:chef_server_url],
+          "validation_client_name" => Chef::Config[:validation_client_name],
+          "validation_key" => IO.read(Chef::Config[:validation_key]),
+          "attributes" => { "run_list" => @name_args } 
+        }
+        data = edit_data(data) if config[:edit]
+        output(data)
+      end
+    end
+  end
+end
+

data/lib/chef/knife/ec2_server_create.rb

@@ -0,0 +1,216 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'socket'
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class Ec2ServerCreate < Knife
+
+      banner "knife ec2 server create [RUN LIST...] (options)"
+
+      attr_accessor :initial_sleep_delay
+
+      option :flavor,
+        :short => "-f FLAVOR",
+        :long => "--flavor FLAVOR",
+        :description => "The flavor of server (m1.small, m1.medium, etc)",
+        :default => "m1.small"
+
+      option :image,
+        :short => "-i IMAGE",
+        :long => "--image IMAGE",
+        :description => "The AMI for the server"
+
+      option :security_groups,
+        :short => "-G X,Y,Z",
+        :long => "--groups X,Y,Z",
+        :description => "The security groups for this server",
+        :default => ["default"],
+        :proc => Proc.new { |groups| groups.split(',') }
+
+      option :availability_zone,
+        :short => "-Z ZONE",
+        :long => "--availability-zone ZONE",
+        :description => "The Availability Zone",
+        :default => "us-east-1b"
+
+      option :chef_node_name,
+        :short => "-N NAME",
+        :long => "--node-name NAME",
+        :description => "The Chef node name for your new node"
+
+      option :ssh_key_name,
+        :short => "-S KEY",
+        :long => "--ssh-key KEY",
+        :description => "The AWS SSH key id",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_ssh_key_id] = key }
+
+      option :ssh_user,
+        :short => "-x USERNAME",
+        :long => "--ssh-user USERNAME",
+        :description => "The ssh username",
+        :default => "root" 
+ 
+      option :ssh_password,
+        :short => "-P PASSWORD",
+        :long => "--ssh-password PASSWORD",
+        :description => "The ssh password"
+
+      option :identity_file,
+        :short => "-I IDENTITY_FILE",
+        :long => "--identity-file IDENTITY_FILE",
+        :description => "The SSH identity file used for authentication"
+ 
+      option :aws_access_key_id,
+        :short => "-A ID",
+        :long => "--aws-access-key-id KEY",
+        :description => "Your AWS Access Key ID",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_access_key_id] = key } 
+
+      option :aws_secret_access_key,
+        :short => "-K SECRET",
+        :long => "--aws-secret-access-key SECRET",
+        :description => "Your AWS API Secret Access Key",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_secret_access_key] = key } 
+
+      option :prerelease,
+        :long => "--prerelease",
+        :description => "Install the pre-release chef gems"
+
+      option :region,
+        :long => "--region REGION",
+        :description => "Your AWS region",
+        :default => "us-east-1"
+
+      option :distro,
+        :short => "-d DISTRO",
+        :long => "--distro DISTRO",
+        :description => "Bootstrap a distro using a template",
+        :default => "ubuntu10.04-gems"
+
+      option :template_file,
+        :long => "--template-file TEMPLATE",
+        :description => "Full path to location of template to use",
+        :default => false
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def tcp_test_ssh(hostname)
+        tcp_socket = TCPSocket.new(hostname, 22)
+        readable = IO.select([tcp_socket], nil, nil, 5)
+        if readable
+          Chef::Log.debug("sshd accepting connections on #{hostname}, banner is #{tcp_socket.gets}")
+          yield
+          true
+        else
+          false
+        end
+      rescue Errno::ETIMEDOUT
+        false
+      rescue Errno::ECONNREFUSED
+        sleep 2
+        false
+      ensure
+        tcp_socket && tcp_socket.close
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+
+        $stdout.sync = true
+
+        connection = Fog::AWS::EC2.new(
+          :aws_access_key_id => Chef::Config[:knife][:aws_access_key_id],
+          :aws_secret_access_key => Chef::Config[:knife][:aws_secret_access_key],
+          :region => config[:region]
+        )
+
+        server = connection.servers.create(
+          :image_id => config[:image],
+          :groups => config[:security_groups],
+          :flavor_id => config[:flavor],
+          :key_name => Chef::Config[:knife][:aws_ssh_key_id],
+          :availability_zone => config[:availability_zone]
+        )
+
+        puts "#{h.color("Instance ID", :cyan)}: #{server.id}"
+        puts "#{h.color("Flavor", :cyan)}: #{server.flavor_id}"
+        puts "#{h.color("Image", :cyan)}: #{server.image_id}"
+        puts "#{h.color("Availability Zone", :cyan)}: #{server.availability_zone}"
+        puts "#{h.color("Security Groups", :cyan)}: #{server.groups.join(", ")}"
+        puts "#{h.color("SSH Key", :cyan)}: #{server.key_name}"
+     
+        print "\n#{h.color("Waiting for server", :magenta)}"
+
+        # wait for it to be ready to do stuff
+        server.wait_for { print "."; ready? }
+
+        puts("\n")
+
+
+        puts "#{h.color("Public DNS Name", :cyan)}: #{server.dns_name}"
+        puts "#{h.color("Public IP Address", :cyan)}: #{server.ip_address}"
+        puts "#{h.color("Private DNS Name", :cyan)}: #{server.private_dns_name}"
+        puts "#{h.color("Private IP Address", :cyan)}: #{server.private_ip_address}"
+
+        print "\n#{h.color("Waiting for sshd", :magenta)}"
+
+        print(".") until tcp_test_ssh(server.dns_name) { sleep @initial_sleep_delay ||= 10; puts("done") }
+
+
+        bootstrap_for_node(server).run
+
+        puts "\n"
+        puts "#{h.color("Instance ID", :cyan)}: #{server.id}"
+        puts "#{h.color("Flavor", :cyan)}: #{server.flavor_id}"
+        puts "#{h.color("Image", :cyan)}: #{server.image_id}"
+        puts "#{h.color("Availability Zone", :cyan)}: #{server.availability_zone}"
+        puts "#{h.color("Security Groups", :cyan)}: #{server.groups.join(", ")}"
+        puts "#{h.color("SSH Key", :cyan)}: #{server.key_name}"
+        puts "#{h.color("Public DNS Name", :cyan)}: #{server.dns_name}"
+        puts "#{h.color("Public IP Address", :cyan)}: #{server.ip_address}"
+        puts "#{h.color("Private DNS Name", :cyan)}: #{server.private_dns_name}"
+        puts "#{h.color("Private IP Address", :cyan)}: #{server.private_ip_address}"
+        puts "#{h.color("Run List", :cyan)}: #{@name_args.join(', ')}"
+      end
+
+      def bootstrap_for_node(server)
+        bootstrap = Chef::Knife::Bootstrap.new
+        bootstrap.name_args = [server.dns_name]
+        bootstrap.config[:run_list] = @name_args
+        bootstrap.config[:ssh_user] = config[:ssh_user]
+        bootstrap.config[:identity_file] = config[:identity_file]
+        bootstrap.config[:chef_node_name] = config[:chef_node_name] || server.id
+        bootstrap.config[:prerelease] = config[:prerelease]
+        bootstrap.config[:distro] = config[:distro]
+        bootstrap.config[:use_sudo] = true
+        bootstrap.config[:template_file] = config[:template_file]
+        bootstrap
+      end
+
+    end
+  end
+end

data/lib/chef/knife/ec2_server_delete.rb

@@ -0,0 +1,90 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class Ec2ServerDelete < Knife
+
+      banner "knife ec2 server delete SERVER [SERVER] (options)"
+      
+      option :aws_access_key_id,
+        :short => "-A ID",
+        :long => "--aws-access-key-id KEY",
+        :description => "Your AWS Access Key ID",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_access_key_id] = key } 
+
+      option :aws_secret_access_key,
+        :short => "-K SECRET",
+        :long => "--aws-secret-access-key SECRET",
+        :description => "Your AWS API Secret Access Key",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_secret_access_key] = key } 
+
+      option :region,
+        :long => "--region REGION",
+        :description => "Your AWS region",
+        :default => "us-east-1"
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+
+        connection = Fog::AWS::EC2.new(
+          :aws_access_key_id => Chef::Config[:knife][:aws_access_key_id],
+          :aws_secret_access_key => Chef::Config[:knife][:aws_secret_access_key],
+          :region => config[:region]
+        )
+
+        @name_args.each do |instance_id|
+          server = connection.servers.get(instance_id)
+
+          puts "#{h.color("Instance ID", :cyan)}: #{server.id}"
+          puts "#{h.color("Flavor", :cyan)}: #{server.flavor_id}"
+          puts "#{h.color("Image", :cyan)}: #{server.image_id}"
+          puts "#{h.color("Availability Zone", :cyan)}: #{server.availability_zone}"
+          puts "#{h.color("Security Groups", :cyan)}: #{server.groups.join(", ")}"
+          puts "#{h.color("SSH Key", :cyan)}: #{server.key_name}"
+          puts "#{h.color("Public DNS Name", :cyan)}: #{server.dns_name}"
+          puts "#{h.color("Public IP Address", :cyan)}: #{server.ip_address}"
+          puts "#{h.color("Private DNS Name", :cyan)}: #{server.private_dns_name}"
+          puts "#{h.color("Private IP Address", :cyan)}: #{server.private_ip_address}"
+
+          puts "\n"
+          confirm("Do you really want to delete this server")
+
+          server.destroy
+
+          Chef::Log.warn("Deleted server #{server.id}")
+        end
+      end
+    end
+  end
+end
+
+
+
+
+

data/lib/chef/knife/ec2_server_list.rb

@@ -0,0 +1,88 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class Ec2ServerList < Knife
+
+      banner "knife ec2 server list [RUN LIST...] (options)"
+
+      option :aws_access_key_id,
+        :short => "-A ID",
+        :long => "--aws-access-key-id KEY",
+        :description => "Your AWS Access Key ID",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_access_key_id] = key } 
+
+      option :aws_secret_access_key,
+        :short => "-K SECRET",
+        :long => "--aws-secret-access-key SECRET",
+        :description => "Your AWS API Secret Access Key",
+        :proc => Proc.new { |key| Chef::Config[:knife][:aws_secret_access_key] = key } 
+
+      option :region,
+        :long => "--region REGION",
+        :description => "Your AWS region",
+        :default => "us-east-1"
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+
+        $stdout.sync = true
+
+        connection = Fog::AWS::EC2.new(
+          :aws_access_key_id => Chef::Config[:knife][:aws_access_key_id],
+          :aws_secret_access_key => Chef::Config[:knife][:aws_secret_access_key],
+          :region => config[:region]
+        )
+
+        server_list = [ 
+          h.color('Instance ID', :bold), 
+          h.color('Public IP', :bold), 
+          h.color('Private IP', :bold),
+          h.color('Flavor', :bold),
+          h.color('Image', :bold),
+          h.color('Security Groups', :bold),
+          h.color('State', :bold)
+        ]
+        connection.servers.all.each do |server|
+          server_list << server.id.to_s
+          server_list << (server.ip_address == nil ? "" : server.ip_address)
+          server_list << (server.private_ip_address == nil ? "" : server.private_ip_address) 
+          server_list << (server.flavor_id == nil ? "" : server.flavor_id)
+          server_list << (server.image_id == nil ? "" : server.image_id)
+          server_list << server.groups.join(", ")
+          server_list << server.state
+        end
+        puts h.list(server_list, :columns_across, 7)
+
+      end
+    end
+  end
+end
+
+

data/lib/chef/knife/exec.rb

@@ -0,0 +1,52 @@
+#--
+# Author:: Daniel DeLeo (<dan@opscode.com)
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+
+class Chef::Knife::Exec < Chef::Knife
+
+  banner "knife exec [SCRIPT] (options)"
+
+  option :exec,
+    :short => "-E CODE",
+    :long => "--exec CODE",
+    :description => "a string of Chef code to execute"
+
+  def late_load_deps
+    require 'chef/shef/ext'
+  end
+
+  def run
+    late_load_deps
+    scripts = Array(name_args)
+    context = Object.new
+    Shef::Extensions.extend_context_object(context)
+    if config[:exec]
+      context.instance_eval(config[:exec], "-E Argument", 0)
+    elsif !scripts.empty?
+      scripts.each do |script|
+        file = File.expand_path(script)
+        context.instance_eval(IO.read(file), file, 0)
+      end
+    else
+      script = STDIN.read
+      context.instance_eval(script, "STDIN", 0)
+    end
+  end
+  
+end