Adding rackspace/fog support

Author: adamhjk

chef/lib/chef/config.rb

@@ -197,5 +197,7 @@ def self.manage_secret_key
 cache_type"BasicFile"
 cache_options({:path=>"/var/chef/cache/checksums",:skip_expires=>true})
 
+# Arbitrary knife configuration data
+knifeHash.new
 end
 end

chef/lib/chef/knife/rackspace_server_create.rb

@@ -0,0 +1,156 @@
+#
+# Author:: Adam Jacob (<[email protected]>)
+# 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'
+
+classChef
+classKnife
+classRackspaceServerCreate < Knife
+
+banner"Sub-Command: rackspace server create [RUN LIST...] (options)"
+
+option:flavor,
+:short=>"-f FLAVOR",
+:long=>"--flavor FLAVOR",
+:description=>"The flavor of server",
+:proc=>Proc.new{ |f| f.to_i},
+:default=>1
+
+option:image,
+:short=>"-i IMAGE",
+:long=>"--image IMAGE",
+:description=>"The image of the server",
+:proc=>Proc.new{ |i| i.to_i},
+:default=>14362
+
+option:server_name,
+:short=>"-N NAME",
+:long=>"--server-name NAME",
+:description=>"The server name",
+:default=>"wtf"
+
+option:api_key,
+:short=>"-K KEY",
+:long=>"--rackspace-api-key KEY",
+:description=>"Your rackspace API key",
+:proc=>Proc.new{ |key| Chef::Config[:knife][:rackspace_api_key]=key}
+
+option:api_username,
+:short=>"-A USERNAME",
+:long=>"--rackspace-api-username USERNAME",
+:description=>"Your rackspace API username",
+:proc=>Proc.new{ |username| Chef::Config[:knife][:rackspace_api_username]=username}
+
+defh
+@highline ||= HighLine.new
+end
+
+defrun
+require'fog'
+require'highline'
+require'net/ssh/multi'
+require'readline'
+
+connection=Fog::Rackspace::Servers.new(
+:rackspace_api_key=>Chef::Config[:knife][:rackspace_api_key],
+:rackspace_username=>Chef::Config[:knife][:rackspace_api_username]
+)
+
+server=connection.servers.new
+
+server.flavor_id=config[:flavor]
+server.image_id=config[:image]
+server.name=config[:server_name]
+server.personality=[
+{
+'path'=>'/etc/install-chef',
+'contents'=><<-EOH
+#!/bin/bash
+# Customized rc.local for chef installation
+
+if [ ! -f /usr/bin/chef-client ]; then
+ apt-get update
+ apt-get install -y ruby ruby1.8-dev build-essential wget libruby-extras libruby1.8-extras
+ cd /tmp
+ wget http://rubyforge.org/frs/download.php/69365/rubygems-1.3.6.tgz
+ tar xvf rubygems-1.3.6.tgz
+ cd rubygems-1.3.6
+ ruby setup.rb
+ cp /usr/bin/gem1.8 /usr/bin/gem
+ gem install chef ohai --no-rdoc --no-ri --verbose
+fi
+
+exit 0
+EOH
+},
+{
+'path'=>"/etc/chef/validation.pem",
+'contents'=>IO.read(Chef::Config[:validation_key])
+},
+{
+'path'=>"/etc/chef/client.rb",
+'contents'=><<-EOH
+log_level :info
+log_location STDOUT
+chef_server_url "#{Chef::Config[:chef_server_url]}" 
+validation_client_name "#{Chef::Config[:validation_client_name]}"
+EOH
+},
+{
+'path'=>"/etc/chef/first-boot.json",
+'contents'=>{"run_list"=>@name_args}.to_json
+},
+]
+
+server.save
+
+ $stdout.sync=true
+
+puts"#{h.color("Name",:cyan)}: #{server.name}"
+puts"#{h.color("Flavor",:cyan)}: #{server.flavor_id}"
+puts"#{h.color("Image",:cyan)}: #{server.image_id}"
+puts"#{h.color("Public Address",:cyan)}: #{server.addresses["public"]}"
+puts"#{h.color("Private Address",:cyan)}: #{server.addresses["private"]}"
+puts"#{h.color("Password",:cyan)}: #{server.password}"
+
+print"\n#{h.color("Requesting server",:magenta)}"
+saved_password=server.password
+
+# wait for it to be ready to do stuff
+server.wait_for{print".";ready?}
+
+puts"\nServer ready, waiting 15 seconds to bootstrap."
+sleep15
+
+puts"\nBootstrapping #{h.color(server.name,:bold)}..."
+
+ssh=Chef::Knife::Ssh.new
+ssh.name_args=[server.addresses["public"][0],"/bin/bash /etc/install-chef && /usr/bin/chef-client -j /etc/chef/first-boot.json"]
+ssh.config[:ssh_user]="root"
+ssh.config[:manual]=true
+ssh.config[:password]=saved_password
+ssh.password=saved_password
+ssh.run
+
+end
+end
+end
+end
+
+

chef/lib/chef/knife/rackspace_server_delete.rb

@@ -0,0 +1,57 @@
+#
+# Author:: Adam Jacob (<[email protected]>)
+# 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'
+
+classChef
+classKnife
+classRackspaceServerDelete < Knife
+
+banner"Sub-Command: rackspace server delete SERVER (options)"
+
+defh
+@highline ||= HighLine.new
+end
+
+defrun
+require'fog'
+require'highline'
+require'net/ssh/multi'
+require'readline'
+
+connection=Fog::Rackspace::Servers.new(
+:rackspace_api_key=>Chef::Config[:knife][:rackspace_api_key],
+:rackspace_username=>Chef::Config[:knife][:rackspace_api_username]
+)
+
+server=connection.servers.get(@name_args[0])
+
+confirm("Do you really want to delete server ID #{server.id} named #{server.name}")
+
+server.destroy
+
+Chef::Log.warn("Deleted server #{server.id} named #{server.name}")
+end
+end
+end
+end
+
+
+
+

chef/lib/chef/knife/rackspace_server_list.rb

@@ -0,0 +1,59 @@
+#
+# Author:: Adam Jacob (<[email protected]>)
+# 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'
+
+classChef
+classKnife
+classRackspaceServerList < Knife
+
+banner"Sub-Command: rackspace server list (options)"
+
+defh
+@highline ||= HighLine.new
+end
+
+defrun
+require'fog'
+require'highline'
+require'net/ssh/multi'
+require'readline'
+
+connection=Fog::Rackspace::Servers.new(
+:rackspace_api_key=>Chef::Config[:knife][:rackspace_api_key],
+:rackspace_username=>Chef::Config[:knife][:rackspace_api_username]
+)
+
+server_list=[h.color('ID',:bold),h.color('Name',:bold),h.color('Public IP',:bold),h.color('Private IP',:bold),h.color('Flavor ID',:bold)]
+connection.servers.all.eachdo |server|
+server_list << server.id.to_s
+server_list << server.name
+server_list << server.addresses["public"][0]
+server_list << server.addresses["private"][0]
+server_list << server.flavor_id.to_s
+end
+putsh.list(server_list,:columns_across,5)
+
+end
+end
+end
+end
+
+
+

chef/lib/chef/knife/ssh.rb

@@ -23,6 +23,8 @@ class Chef
 classKnife
 classSsh < Knife
 
+attr_writer:password
+
 banner"Sub-Command: ssh QUERY COMMAND (options)"
 
 option:concurrency,
@@ -49,6 +51,11 @@ class Ssh < Knife
 :long=>"--ssh-user USERNAME",
 :description=>"The ssh username"
 
+option:ssh_password,
+:short=>"-P PASSWORD",
+:long=>"--ssh-password PASSWORD",
+:description=>"The ssh password"
+
 defsession
 @session ||= Net::SSH::Multi.start(:concurrent_connections=>config[:concurrency])
 end
@@ -75,7 +82,12 @@ def configure_session
 defsession_from_list(list)
 list.eachdo |item|
 Chef::Log.debug("Adding #{item}")
-session.useconfig[:ssh_user] ? "#{config[:ssh_user]}@#{item}" : item
+
+ifconfig[:password]
+session.useconfig[:ssh_user] ? "#{config[:ssh_user]}@#{item}" : item,:password=>config[:password]
+else
+session.useconfig[:ssh_user] ? "#{config[:ssh_user]}@#{item}" : item
+end
 @longest=item.lengthifitem.length > @longest
 end
 session