Class: Sumo

Inherits:
Object
  • Object
show all
Defined in:
lib/sumo.rb

Instance Method Summary collapse

Instance Method Details

#attach(volume, instance, device) ⇒ Object 



57
58
59
60
61
62
63
64
 
# File 'lib/sumo.rb', line 57

def attach(volume, instance, device)
	result = ec2.attach_volume(
		:volume_id => volume,
		:instance_id => instance,
		:device => device
	)
	"done"
end

#attached_volumes ⇒ Object 



49
50
51
 
# File 'lib/sumo.rb', line 49

def attached_volumes
	volumes.select { |vol| vol[:status] == 'in-use' }
end

#available_volumes ⇒ Object 



45
46
47
 
# File 'lib/sumo.rb', line 45

def available_volumes
	volumes.select { |vol| vol[:status] == 'available' }
end

#bootstrap_chef(hostname) ⇒ Object 



190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
 
# File 'lib/sumo.rb', line 190

def bootstrap_chef(hostname)
	commands = [
		'sudo apt-get update',
		'sudo apt-get autoremove -y',
		'if [ ! -f /usr/lib/ruby/1.8/net/https.rb ]; then sudo apt-get install -y xfsprogs xfsdump xfslibs-dev ruby ruby-dev rubygems libopenssl-ruby1.8 git-core; fi',
		'if [ ! -d /etc/chef ]; then sudo mkdir /etc/chef; fi',
		'if [ ! -d /var/lib/gems/1.8/gems/chef-* ]; then sudo gem install chef ohai --no-rdoc --no-ri; fi',
		config['cookbooks_url'] ? "if [ -d chef-cookbooks ]; then cd chef-cookbooks; git pull; else git clone #{config['cookbooks_url']} chef-cookbooks; fi" : "echo done"
	]
	ssh(hostname, commands)
	if !config['cookbooks_url'] && config['cookbooks_dir']
	  scp(hostname, config['cookbooks_dir'], "chef-cookbooks")
  end
  if config['chef-validation']
    scp(hostname, config['chef-validation'], "validation.pem")
    ssh(hostname, ["sudo mv validation.pem /etc/chef/"])
   end
end

#config ⇒ Object 



307
308
309
 
# File 'lib/sumo.rb', line 307

def config
	@config ||= default_config.merge read_config
end

#console_output(instance_id) ⇒ Object 



303
304
305
 
# File 'lib/sumo.rb', line 303

def console_output(instance_id)
	ec2.get_console_output(:instance_id => instance_id)["output"]
end

#create_keypair ⇒ Object 



338
339
340
341
342
 
# File 'lib/sumo.rb', line 338

def create_keypair
	keypair = ec2.create_keypair(:key_name => "sumo").keyMaterial
	File.open(keypair_file, 'w') { |f| f.write keypair }
	File.chmod 0600, keypair_file
end

#create_security_group ⇒ Object 



344
345
346
347
 
# File 'lib/sumo.rb', line 344

def create_security_group
	ec2.create_security_group(:group_name => 'sumo', :group_description => 'Sumo')
rescue AWS::InvalidGroupDuplicate
end

#create_volume(size) ⇒ Object 



71
72
73
74
75
76
77
 
# File 'lib/sumo.rb', line 71

def create_volume(size)
	result = ec2.create_volume(
		:availability_zone => config['availability_zone'],
		:size => size.to_s
	)
	result["volumeId"]
end

#default_config ⇒ Object 



315
316
317
318
319
320
321
322
 
# File 'lib/sumo.rb', line 315

def default_config
	{
		'user' => 'ubuntu',
		'ami' => 'ami-1234de7b', # Ubuntu 10.04 LTS (Lucid Lynx)
		'availability_zone' => 'us-east-1d',
		'instance_size' => 't1.micro'
	}
end

#destroy_volume(volume) ⇒ Object 



87
88
89
90
 
# File 'lib/sumo.rb', line 87

def destroy_volume(volume)
	ec2.delete_volume(:volume_id => volume)
	"done"
end

#detach(volume) ⇒ Object 



66
67
68
69
 
# File 'lib/sumo.rb', line 66

def detach(volume)
	result = ec2.detach_volume(:volume_id => volume, :force => "true")
	"done"
end

#ec2 ⇒ Object 



360
361
362
363
364
365
366
 
# File 'lib/sumo.rb', line 360

def ec2
   @ec2 ||= AWS::EC2::Base.new(
     :access_key_id => config['access_id'], 
     :secret_access_key => config['access_secret'], 
     :server => server
   )
end

#fetch_list ⇒ Object 



92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/sumo.rb', line 92

def fetch_list
	result = ec2.describe_instances
	return [] unless result.reservationSet

	instances = []
	result.reservationSet.item.each do |r|
		r.instancesSet.item.each do |item|
			instances << {
				:instance_id => item.instanceId,
				:status => item.instanceState.name,
				:hostname => item.dnsName,
				:local_dns => item.privateDnsName,
				:private_ip => item.privateIpAddress
			}
		end
	end
	instances
end

#fetch_resources(hostname) ⇒ Object 



286
287
288
289
290
291
 
# File 'lib/sumo.rb', line 286

def fetch_resources(hostname)
	cmd = "ssh -i #{keypair_file} #{config['user']}@#{hostname} 'sudo cat /root/resources' 2>&1"
	out = IO.popen(cmd, 'r') { |pipe| pipe.read }
	abort "failed to read resources, output:\n#{out}" unless $?.success?
	parse_resources(out, hostname)
end

#find(id_or_hostname) ⇒ Object 



111
112
113
114
115
116
117
118
119
 
# File 'lib/sumo.rb', line 111

def find(id_or_hostname)
	return unless id_or_hostname
	id_or_hostname = id_or_hostname.strip.downcase
	list.detect do |inst|
		inst[:hostname] == id_or_hostname or
		inst[:instance_id] == id_or_hostname or
		inst[:instance_id].gsub(/^i-/, '') == id_or_hostname
	end
end

#find_volume(volume_id) ⇒ Object 



121
122
123
124
125
126
127
128
 
# File 'lib/sumo.rb', line 121

def find_volume(volume_id)
	return unless volume_id
	volume_id = volume_id.strip.downcase
	volumes.detect do |volume|
		volume[:volume_id] == volume_id or
		volume[:volume_id].gsub(/^vol-/, '') == volume_id
	end
end

#format_volume(volume, instance, device, mountpoint) ⇒ Object 



79
80
81
82
83
84
85
 
# File 'lib/sumo.rb', line 79

def format_volume(volume, instance, device, mountpoint)
commands = [
	"if [ ! -d #{mountpoint} ]; then sudo mkdir #{mountpoint}; fi",
	"if [ -b /dev/#{device}1 ]; then sudo mount /dev/#{device}1 #{mountpoint}; else echo ',,L' | sudo sfdisk /dev/#{device} && sudo mkfs.xfs /dev/#{device}1 && sudo mount /dev/#{device}1 #{mountpoint}; fi"
]
ssh(instance, commands)
end

#instance_info(instance_id) ⇒ Object 



142
143
144
145
146
 
# File 'lib/sumo.rb', line 142

def instance_info(instance_id)
	fetch_list.detect do |inst|
		inst[:instance_id] == instance_id
	end
end

#keypair_file ⇒ Object 



334
335
336
 
# File 'lib/sumo.rb', line 334

def keypair_file
	"#{sumo_dir}/keypair.pem"
end

#launch ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/sumo.rb', line 7

def launch
	ami = config['ami']
	raise "No AMI selected" unless ami

	create_keypair unless File.exists? keypair_file

	create_security_group
	open_firewall(22)

	result = ec2.run_instances(
		:image_id => ami,
		:instance_type => config['instance_size'],
		:key_name => 'sumo',
		:security_group => [ 'sumo' ],
		:availability_zone => config['availability_zone']
	)
	result.instancesSet.item[0].instanceId
end

#list ⇒ Object 



26
27
28
 
# File 'lib/sumo.rb', line 26

def list
	@list ||= fetch_list
end

#list_by_status(status) ⇒ Object 



138
139
140
 
# File 'lib/sumo.rb', line 138

def list_by_status(status)
	list.select { |i| i[:status] == status }
end

#new_ssh(hostname, cmds) ⇒ Object 



247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
 
# File 'lib/sumo.rb', line 247

def new_ssh(hostname, cmds)
   Net::SSH.start(hostname, config['user'], :keys => [keypair_file], :compression => "none") do |ssh|
     # capture all stderr and stdout output from a remote process
           
     File.open(File.expand_path("~/.sumo/ssh.log"), "w") do |log|
       ssh.open_channel do |channel|          
         cmds.each do |cmd|
           channel.exec cmd do |ch, success|
       			abort "failed on #{cmd}\nCheck ~/.sumo/ssh.log for the output" unless success

             channel.on_data do |ch, data|
               puts "Got data #{data.inspect}"
               log << data
               log.flush
             end

             channel.on_extended_data do |ch, type, data|
               puts "Got data #{data.inspect}"
               log << data
               log.flush
             end
             
             channel.on_close do |ch|
               puts "channel is closing!"
             end
           end
         end
       end

       ssh.loop
     end
   end
end

#nondestroyed_volumes ⇒ Object 



53
54
55
 
# File 'lib/sumo.rb', line 53

def nondestroyed_volumes
	volumes.select { |vol| vol[:status] != 'deleting' }
end

#open_firewall(port) ⇒ Object 



349
350
351
352
353
354
355
356
357
358
 
# File 'lib/sumo.rb', line 349

def open_firewall(port)
	ec2.authorize_security_group_ingress(
		:group_name => 'sumo',
		:ip_protocol => 'tcp',
		:from_port => port,
		:to_port => port,
		:cidr_ip => '0.0.0.0/0'
	)
rescue AWS::InvalidPermissionDuplicate
end

#parse_resources(raw, hostname) ⇒ Object 



293
294
295
296
297
 
# File 'lib/sumo.rb', line 293

def parse_resources(raw, hostname)
	raw.split("\n").map do |line|
		line.gsub(/localhost/, hostname)
	end
end

#pending ⇒ Object 



134
135
136
 
# File 'lib/sumo.rb', line 134

def pending
	list_by_status('pending')
end

#read_config ⇒ Object 



328
329
330
331
332
 
# File 'lib/sumo.rb', line 328

def read_config
	YAML.load File.read("#{sumo_dir}/config.yml")
rescue Errno::ENOENT
	raise "Sumo is not configured, please fill in ~/.sumo/config.yml"
end

#resources(hostname) ⇒ Object 



281
282
283
284
 
# File 'lib/sumo.rb', line 281

def resources(hostname)
	@resources ||= {}
	@resources[hostname] ||= fetch_resources(hostname)
end

#running ⇒ Object 



130
131
132
 
# File 'lib/sumo.rb', line 130

def running
	list_by_status('running')
end

#scp(hostname, directory, endpoint = ".") ⇒ Object 



240
241
242
243
244
245
 
# File 'lib/sumo.rb', line 240

def scp(hostname, directory, endpoint=".")
	`scp -i #{keypair_file} -r #{directory} #{config['user']}@#{hostname}:#{endpoint}`
	unless $?.success?
		abort "failed to transfer #{directory}"
	end
end

#server ⇒ Object 



368
369
370
371
372
 
# File 'lib/sumo.rb', line 368

def server
 zone = config['availability_zone']
 host = zone.slice(0, zone.length - 1)
 "#{host}.ec2.amazonaws.com"
end

#set(key, value) ⇒ Object 



311
312
313
 
# File 'lib/sumo.rb', line 311

def set(key, value)
	config[key] = value
end

#setup_role(hostname, instance_id, role) ⇒ Object 



209
210
211
212
213
214
215
216
217
218
219
220
 
# File 'lib/sumo.rb', line 209

def setup_role(hostname, instance_id, role)
	commands = [
	  "if [ ! -f /etc/chef/client.pem ]; then cd chef-cookbooks",
		"sudo /var/lib/gems/1.8/bin/chef-solo -c config/solo.rb -j roles/bootstrap.json -r http://s3.amazonaws.com/chef-solo/bootstrap-latest.tar.gz",
		"if [ -f config/client.rb ]; then sudo cp config/client.rb /etc/chef/client.rb; fi",
		"sudo /var/lib/gems/1.8/bin/chef-client",
		"sudo rm /etc/chef/validation.pem; fi"
	]
	ssh(hostname, commands)
	`knife node run_list add #{instance_id} "role[#{role}]"`
	ssh(hostname, ["sudo /var/lib/gems/1.8/bin/chef-client"])
end

#ssh(hostname, cmds) ⇒ Object 



222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
 
# File 'lib/sumo.rb', line 222

def ssh(hostname, cmds)
  unless IO.read(File.expand_path("~/.ssh/known_hosts")).include?(hostname)
    `ssh-keyscan -t rsa #{hostname} >> $HOME/.ssh/known_hosts`
    if config['deploy_key']
      scp(hostname, config['deploy_key'], ".ssh/id_rsa")
     end
     if config['known_hosts']
       scp(hostname, config['known_hosts'], ".ssh/known_hosts")
     end
   end
	IO.popen("ssh -i #{keypair_file} #{config['user']}@#{hostname} > #{config['logfile'] || "~/.sumo/ssh.log"} 2>&1", "w") do |pipe|
		pipe.puts cmds.join(' && ')
	end
	unless $?.success?
		abort "failed\nCheck #{config['logfile'] || "~/.sumo/ssh.log"} for the output"
	end
end

#sumo_dir ⇒ Object 



324
325
326
 
# File 'lib/sumo.rb', line 324

def sumo_dir
	"#{ENV['HOME']}/.sumo"
end

#terminate(instance_id) ⇒ Object 



299
300
301
 
# File 'lib/sumo.rb', line 299

def terminate(instance_id)
	ec2.terminate_instances(:instance_id => [ instance_id ])
end

#volumes ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
 
# File 'lib/sumo.rb', line 30

def volumes
	result = ec2.describe_volumes
	return [] unless result.volumeSet

	result.volumeSet.item.map do |row|
		{
			:volume_id => row["volumeId"],
			:size => row["size"],
			:status => row["status"],
			:device => (row["attachmentSet"]["item"].first["device"] rescue ""),
			:instance_id => (row["attachmentSet"]["item"].first["instanceId"] rescue ""),
		}
	end
end

#wait_for_hostname(instance_id) ⇒ Object

Raises:

  • (ArgumentError) 


148
149
150
151
152
153
154
155
156
157
158
 
# File 'lib/sumo.rb', line 148

def wait_for_hostname(instance_id)
	raise ArgumentError unless instance_id and instance_id.match(/^i-/)
	loop do
		if inst = instance_info(instance_id)
			if hostname = inst[:hostname]
				return hostname
			end
		end
		sleep 1
	end
end

#wait_for_private_ip(instance_id) ⇒ Object

Raises:

  • (ArgumentError) 


160
161
162
163
164
165
166
167
168
169
170
 
# File 'lib/sumo.rb', line 160

def wait_for_private_ip(instance_id)
	raise ArgumentError unless instance_id and instance_id.match(/^i-/)
	loop do
		if inst = instance_info(instance_id)
			if private_ip = inst[:private_ip]
				return private_ip
			end
		end
		sleep 1
	end
end

#wait_for_ssh(hostname) ⇒ Object

Raises:

  • (ArgumentError) 


172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
 
# File 'lib/sumo.rb', line 172

def wait_for_ssh(hostname)
	raise ArgumentError unless hostname
	loop do
		begin
			Timeout::timeout(4) do
				TCPSocket.new(hostname, 22)
				IO.popen("ssh -i #{keypair_file} #{config['user']}@#{hostname} > #{config['logfile'] || "~/.sumo/ssh.log"} 2>&1", "w") do |pipe|
     			pipe.puts "ls"
     		end
     		if $?.success?
     			return
     		end
			end
		rescue SocketError, Timeout::Error, Errno::ECONNREFUSED, Errno::EHOSTUNREACH
		end
	end
end