1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
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
|