~nchohan/+junk/mytools

#!/usr/bin/ruby -w

# Programmer: Chris Bunch

 

$VERBOSE = nil # to supress excessive SSL cert warnings

 

require 'fileutils'

require 'yaml'

 

$:.unshift File.join(File.dirname(__FILE__), "..", "lib")

require 'common_functions'

require 'node_layout'

 

USAGE = <<END_OF_USAGE

#{AS_VERSION}

 

Usage: appscale-add-keypair [OPTIONS]

 

Examples:

  appscale-add-keypair --ips ips.yaml

  

Flags:

  --ips: The YAML file containing the IPs of the machines to use if not using a cloud infrastructure (for example, Xen).

END_OF_USAGE

 

ALL_FLAGS = ["help", "usage", "h", "ips", "keyname", "version"]

 

require 'parse_args'

 

keyname = KEYNAME

ips_yaml = IPS

 

node_layout = NodeLayout.new(IPS, { :database => "cassandra" } )

 

["ssh-keygen", "ssh-copy-id"].each { |cmd|

  abort("You do not have the '#{cmd}' command in your PATH. Please ensure that it is in your path and try again.") unless CommonFunctions.user_has_cmd?(cmd)

}

 

appscale_dir = File.expand_path("~/.appscale")

FileUtils.mkdir(appscale_dir) unless File.exists?(appscale_dir)

 

path = File.expand_path("~/.appscale/#{keyname}")

backup_key = File.expand_path("~/.appscale/#{keyname}.key")

pub_key = File.expand_path("~/.appscale/#{keyname}.pub")

 

#FileUtils.rm_f([path, backup_key, pub_key])

unless File.exists?(path) and File.exists?(pub_key)

  FileUtils.rm_f([path, backup_key, pub_key])

  puts CommonFunctions.shell("ssh-keygen -t rsa -N '' -f #{path}")

end

FileUtils.chmod(0600, [path, pub_key])

 

ips = node_layout.nodes.collect { |node| node.id }

ips.each { |ip|

  puts CommonFunctions.shell("ssh-copy-id -i #{path} root@#{ip}")

#  puts CommonFunctions.shell("scp -i #{path} #{path} root@#{ip}:.ssh/id_rsa")

#  puts CommonFunctions.shell("scp -i #{path} #{path} root@#{ip}:.ssh/id_dsa")

#  puts CommonFunctions.shell("scp -i #{path} #{pub_key} root@#{ip}:.ssh/id_rsa.pub")

}

 

head_ip = node_layout.head_node.id

puts CommonFunctions.shell("scp -i #{path} #{path} root@#{head_ip}:.ssh/id_rsa")

# this is needed for EC2 integration.

puts CommonFunctions.shell("scp -i #{path} #{path} root@#{head_ip}:.ssh/id_dsa")

puts CommonFunctions.shell("scp -i #{path} #{pub_key} root@#{head_ip}:.ssh/id_rsa.pub")

 

FileUtils.cp(path, backup_key)

puts "A new ssh key has been generated for you and placed at #{path}. You can now use this key to log into any of the machines you specified without providing a password via the following command:\n\tssh root@XXX.XXX.XXX.XXX -i #{path}"