awsam 0.0.1

data/bin/ascp

@@ -0,0 +1,65 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.join(File.dirname(__FILE__), '../lib')
+
+require 'awsam'
+
+#
+# Will perform an scp to/from an instance ID.
+#
+#
+# Usage: ascp [user@]<instance ID>:remote-file local-file
+#        ascp local-file [user@]<instance ID>:remote-file
+
+if ARGV.length == 0
+  puts 'Usage: ascp [user@]<instance ID>:remote-file local-file'
+  puts '       ascp local-file [user@]<instance ID>:remote-file'
+  exit 1
+end
+
+user = "root"
+
+args = ARGV.clone
+
+Awsam::Accounts::load!
+acct = Awsam::Accounts::active
+unless acct
+  puts "No active account. Use 'aem use <account name>' to select one"
+  exit 1
+end
+
+keyfile = nil
+0.upto(args.length - 1) do |i|
+  m = args[i].match(/([a-zA-Z0-9_]{1,}@){0,1}(i-[0-9a-f]{8}):/)
+  if m
+    user = m[1]
+    instance_id = m[2]
+
+    inst = Awsam::Ec2::find_instance(acct, instance_id)
+    unless inst
+      puts "Unable to locate instance ID #{instance_id}"
+      exit 1
+    end
+
+    if inst[:aws_state] != "running"
+      puts "Instance #{instance_id} is not running, it's #{inst[:aws_state]}"
+      exit 1
+    end
+
+    args[i] = args[i].gsub(instance_id, inst[:dns_name])
+
+    key = acct.find_key(inst[:ssh_key_name])
+    keyfile = key.path if key
+    break
+  end
+end
+
+if keyfile
+  args.unshift('-i', keyfile, '-o', 'IdentitiesOnly=yes')
+end
+
+exec "scp #{args.join(" ")}"
+
+# Local Variables:
+# mode: ruby
+# End:

data/bin/assh

@@ -0,0 +1,73 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.join(File.dirname(__FILE__), '../lib')
+
+require 'awsam'
+require 'trollop'
+
+$opts = Trollop::options do
+  banner <<-EOS
+  
+  NAME
+      assh -- ssh client for EC2
+  
+  SYNOPSIS
+      assh [-f --first-node] [user@]hostname
+      
+  DESCRIPTION
+      assh is a program for logging into an EC2 node.  assh allows you to specify a hostname by either an instance ID or a 
+      tag name.  Specifying a hostname by a substring of multiple tags, by default, will allow you to specify the exact node you 
+      want to use.
+      
+      The options are as follows:
+      
+      -f first-node
+          Use first_node to specify that you want to override assh's default behavior when multiple tag names result from your 
+          requested hostname.  By enabling this flag, you're allowing assh to use any of the resulting hosts.
+  EOS
+  
+  opt :first_node, "Use first node mode", :short => "-f"
+end
+
+user = "root"
+instance_id = ARGV[0]
+
+if instance_id.include?("@")
+  (user, instance_id) = instance_id.split("@")
+elsif ENV['AWS_DEFAULT_USER'].to_s.length > 0
+  user = ENV['AWS_DEFAULT_USER']
+end
+
+Awsam::Accounts::load!
+
+acct = Awsam::Accounts::active
+unless acct
+  puts "No active account. Use 'aem use <account name>' to select one"
+  exit 1
+end
+
+inst = Awsam::Ec2::find_instance(acct, instance_id)
+unless inst
+  puts "Unable to locate instance ID #{instance_id}"
+  exit 1
+end
+
+if inst[:aws_state] != "running"
+  puts "Instance #{instance_id} is not running, it's #{inst[:aws_state]}"
+  exit 1
+end
+
+puts "Logging in as #{user} to #{inst[:dns_name]}"
+puts
+
+key = acct.find_key(inst[:ssh_key_name])
+
+if key
+  exec "ssh -i #{key.path} -o IdentitiesOnly=yes #{user}@#{inst[:dns_name]}"
+else
+  exec "ssh #{user}@#{inst[:dns_name]}"
+end
+
+# Local Variables:
+# mode: ruby
+# End:

data/bin/raem

@@ -0,0 +1,260 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.join(File.dirname(__FILE__), '../lib')
+
+begin
+  require 'awsam'
+rescue LoadError => err
+  $stderr.puts "ERROR: Unable to load AWSAM gem dependencies: #{err.message}"
+  exit 1
+end
+
+require 'optparse'
+
+def err(*args)
+  $stderr.puts *args
+end
+
+def read_val(prompt, default = nil)
+  begin
+    if default
+      print "#{prompt} [#{default}]: "
+    else
+      print "#{prompt}: "
+    end
+
+    val = gets.chomp
+    if val.empty? && default
+      val = default
+    end
+  end while val.empty?
+
+  val
+end
+
+Awsam::Accounts::load!
+
+$options = {}
+$cmd = nil
+
+def selected_account(default_fallback = false)
+  if !$options[:account] || $options[:account].empty?
+    err "Command requires an account"
+    exit 1
+  end
+
+  acct = Awsam::Accounts.find($options[:account])
+  unless acct
+    exit 1
+  end
+  acct
+end
+
+def usage
+  puts "Usage: raem --init"
+  exit
+end
+
+optparse = OptionParser.new do|opts|
+  opts.banner = "Usage: raem [options]\n"
+
+  # help
+  opts.on( '-h', '--help', 'Display this screen' ) do
+    usage
+  end
+
+  opts.on('--add') do
+    $cmd = :add_account
+  end
+
+  opts.on('--remove') do
+    $cmd = :remove_account
+  end
+
+  opts.on('--account <ACCOUNT>') do |acctname|
+    $options[:account] = acctname
+  end
+
+  opts.on('--import-key') do
+    $cmd = :import_key
+  end
+
+  opts.on('--remove-key') do
+    $cmd = :remove_key
+  end
+
+  opts.on('--keyname <KEYNAME>') do |keyname|
+    $options[:keyname] = keyname
+  end
+
+  opts.on('--keyfile <KEYFILE>') do |keyfile|
+    $options[:keyfile] = keyfile
+  end
+
+  opts.on('--list') do
+    $cmd = :list
+  end
+
+  opts.on('--environ') do
+    $cmd = :environ
+  end
+
+  opts.on('--environ-key') do
+    $cmd = :environ_key
+  end
+
+  opts.on('--init') do
+    $cmd = :init
+  end
+
+  opts.on('--print-default-acct') do
+    $cmd = :print_default_acct
+  end
+
+  opts.on('--print-default-key') do
+    $cmd = :print_default_key
+  end
+
+  opts.on('--default') do
+    $options[:set_default] = true
+  end
+end
+
+optparse.parse!(ARGV)
+
+usage unless $cmd
+
+case $cmd
+when :init
+  Awsam::init_awsam
+
+when :list
+  inuse = ENV['AWSAM_ACTIVE_ACCOUNT']
+  inusekey = ENV['AWSAM_ACTIVE_KEY']
+  puts "\nAWS Accounts:\n\n"
+
+  default = Awsam::Accounts::get_default
+  accts = Awsam::Accounts::get
+  accts.each_pair do |name, acct|
+    desc = acct.desc ? " [#{acct.desc}]" : ""
+    if acct.keys.length > 0
+      dfltkey = acct.get_default_key
+      names = acct.keys.keys.collect{ |k|
+        pfx1 = (inuse == name && inusekey == k) ? ">" : ""
+        pfx2 = (dfltkey && dfltkey.name == k) ? "*" : ""
+        "#{pfx1}#{pfx2}#{k}"
+      }.join(",")
+      plural = "key" + (acct.keys.length > 1 ? "s" : "")
+      desc += " [#{acct.keys.length} #{plural}: #{names}]"
+    end
+    pfx = inuse == name ? "=>" : "  "
+    dfltmark = (default && default.name == name) ? "*" : " "
+    puts "#{pfx}#{dfltmark}#{name}#{desc}"
+  end
+
+  puts
+
+when :print_default_acct
+  default = Awsam::Accounts::get_default
+  exit 1 unless default
+  puts default.name
+
+when :print_default_key
+  acct = selected_account
+  default = acct.get_default_key
+  exit 1 unless default
+  puts default.name
+
+when :environ
+  acct = selected_account
+
+  if $options[:set_default]
+    r = Awsam::Accounts::set_default(acct.name)
+    unless r
+      err "Failed to set account #{acct.name} as the default"
+      exit 1
+    end
+  end
+
+  acct.print_environ
+
+when :environ_key
+  unless $options[:keyname]
+    err "Option requires a keyname"
+    exit 1
+  end
+
+  acct = selected_account
+
+  k = acct.find_key($options[:keyname])
+  unless k
+    err "Could not find key #{$options[:keyname]} in active account"
+    exit 1
+  end
+
+  if $options[:set_default]
+    r = acct.set_default_key(k.name)
+    unless r
+      err "Failed to set key #{k.name} as the default"
+      exit 1
+    end
+  end
+
+  k.print_environ
+
+when :add_account
+  puts "Creating a new AWS account...\n"
+  short_name = read_val("Short name")
+  desc = read_val("Description")
+  aws_region = read_val("AWS Region", ENV['AWS_DEFAULT_REGION'])
+  access_key = read_val("Access key", (ENV['AMAZON_ACCESS_KEY_ID'] || ENV['AWS_ACCESS_KEY']))
+  secret_key = read_val("Secret key", (ENV['AMAZON_SECRET_ACCESS_KEY'] || ENV['AWS_SECRET_KEY']))
+  aws_id = read_val("AWS ID", ENV['AMAZON_AWS_ID'])
+
+  acct = Awsam::Account.new(short_name,
+                            { :description => desc,
+                              :access_key => access_key,
+                              :secret_key => secret_key,
+                              :aws_id => aws_id,
+                              :aws_region => aws_region
+                            })
+  acct.save
+
+when :remove_account
+  selected_account.remove
+
+when :import_key
+  unless $options[:keyname] && $options[:keyfile]
+    err "Requires keyname and keyfile!"
+    exit 1
+  end
+
+  acct = selected_account
+
+  display = acct.name + (acct.desc ? " [#{acct.desc}]" : "")
+
+  unless File.exist?($options[:keyfile])
+    err "Unable to locate key file: #{$options[:keyfile]}"
+    exit 1
+  end
+
+  ret = acct.import_key($options[:keyname], $options[:keyfile])
+  err "Imported key pair #{$options[:keyname]} for account #{display}"
+
+when :remove_key
+  unless $options[:keyname]
+    err "Requires keyname!"
+    exit 1
+  end
+
+  acct = selected_account
+  unless acct.remove_key($options[:keyname])
+    err "Failed to remove key #{$options[:keyname]}"
+    exit 1
+  end
+end
+
+
+# Local Variables:
+# mode: ruby
+# End:

data/lib/awsam.rb

@@ -0,0 +1,46 @@
+$:.unshift File.join(File.dirname(__FILE__), 'awsam')
+
+require 'fileutils'
+
+$:.unshift File.join(File.dirname(__FILE__), '../vendor/right_aws/lib')
+$:.unshift File.join(File.dirname(__FILE__), '../vendor/right_http_connection/lib')
+
+require 'right_aws'
+
+require 'accounts'
+require 'ec2'
+
+module Awsam
+  CONF_BASE_DIR = ".awsam"
+  CONF_DIR = File.join(ENV['HOME'], CONF_BASE_DIR)
+  DEFAULT_LINK_NAME = ".default"
+
+  def self.get_conf_dir
+    FileUtils.mkdir(CONF_DIR) unless File.exist?(CONF_DIR)
+    CONF_DIR
+  end
+
+  def self.get_accts_dir
+    base = get_conf_dir()
+    acctsdir = File.join(base, 'accts')
+    FileUtils.mkdir(acctsdir) unless File.exist?(acctsdir)
+    acctsdir
+  end
+
+  def self.init_awsam
+    dir = get_conf_dir
+    File.open(File.join(dir, "bash.rc"), "w") do |f|
+      f << File.read(File.join(File.dirname(__FILE__), '../bashrc/rc.scr'))
+    end
+
+    puts
+    puts "Initialized AWS Account Manager"
+    puts
+    puts "Add the following to your $HOME/.bashrc:"
+    puts
+    puts "  if [ -s $HOME/#{CONF_BASE_DIR}/bash.rc ]; then"
+    puts "      source $HOME/#{CONF_BASE_DIR}/bash.rc"
+    puts "  fi"
+    puts
+  end
+end

data/lib/awsam/account.rb

@@ -0,0 +1,139 @@
+require 'yaml'
+require 'fileutils'
+
+require 'key'
+require 'utils'
+
+module Awsam
+  class Account
+    DEFAULT_REGION = "us-east-1"
+
+    attr_reader :name, :keys
+
+    def initialize(name, params)
+      if name == Awsam::DEFAULT_LINK_NAME
+        # We require this for our default account symlink
+        raise "Can not name an account: #{Awsam::DEFAULT_LINK_NAME}"
+      end
+      @name = name
+      @params = params
+
+      @params[:aws_region] ||= DEFAULT_REGION
+
+      load_keys
+    end
+
+    def self.load_from_disk(dir)
+      name = File.basename(dir)
+      conffile = File.join(dir, 'conf.yml')
+
+      return nil unless File.exist?(conffile)
+
+      File.open(conffile) do |yf|
+        @conf = YAML::load(yf)
+      end
+
+      Account.new(name, @conf)
+    end
+
+    def load_keys
+      @keys = Hash.new
+      base = conf_file('keys')
+      return unless File.exist?(base)
+      Utils::confdir_scan(base) do |name|
+        @keys[name] = Key.new(File.join(base, name))
+      end
+    end
+
+    def print_environ
+      envs = {
+        "AMAZON_ACCESS_KEY_ID"     => @params[:access_key],
+        "AWS_ACCESS_KEY_ID"        => @params[:access_key],
+        "AWS_ACCESS_KEY"           => @params[:access_key],
+
+        "AMAZON_SECRET_ACCESS_KEY" => @params[:secret_key],
+        "AWS_SECRET_ACCESS_KEY"    => @params[:secret_key],
+        "AWS_SECRET_KEY"           => @params[:secret_key],
+
+        "AMAZON_AWS_ID"            => @params[:aws_id],
+        "AWS_DEFAULT_REGION"       => @params[:aws_region]
+      }
+
+      Utils::bash_environ(envs)
+    end
+
+    def find_key(name)
+      @keys[name]
+    end
+
+    def import_key(name, path)
+      @keys[name] = Key.import(conf_file, name, path)
+    end
+
+    def remove_key(name)
+      return false unless @keys.has_key?(name)
+
+      dflt = get_default_key
+      Utils::remove_default(conf_file('keys')) if dflt && dflt.name == name
+      @keys[name].remove
+      @keys.delete(name)
+      true
+    end
+
+    def set_default_key(keyname)
+      key = @keys[keyname]
+      unless key
+        $stderr.puts "No key named #{keyname}"
+        return false
+      end
+
+      Utils::set_default(conf_file('keys'), keyname)
+    end
+
+    def get_default_key
+      dflt = Utils::get_default(conf_file('keys'))
+      @keys[dflt]
+    end
+
+    def remove
+      dir = conf_file
+      acct = Awsam::Accounts::get_default
+      if acct && acct.name == @name
+        # Need to remove default link if we're the default account
+        Awsam::Accounts::remove_default
+      end
+
+      FileUtils.rm_rf(dir)
+    end
+
+    def save
+      dir = File.join(Awsam::get_accts_dir, @name)
+      FileUtils.mkdir(dir) unless File.exist?(dir)
+
+      File.open(File.join(dir, 'conf.yml'), "w", 0600) do |out|
+        YAML.dump(@params, out )
+      end
+    end
+
+    # Export params...need better way to do this
+    def desc
+      @params[:description]
+    end
+
+    def access_key
+      @params[:access_key]
+    end
+
+    def secret_key
+      @params[:secret_key]
+    end
+
+private
+
+    def conf_file(file = nil)
+      dir = File.join(Awsam::get_accts_dir(), @name)
+
+      return file.nil? ? dir : File.join(dir, file)
+    end
+  end
+end