awsam 0.0.1

data/lib/awsam/accounts.rb

@@ -0,0 +1,61 @@
+require 'fileutils'
+
+require 'account'
+require 'utils'
+
+module Awsam
+  module Accounts
+
+    @@accounts = {}
+
+    def self.load!
+      accts = Hash.new
+      accts_dir = Awsam::get_accts_dir
+      Utils::confdir_scan(accts_dir) do |name|
+        acct = Account::load_from_disk(File.join(accts_dir, name))
+        accts[name] = acct if acct
+      end
+
+      @@accounts = accts
+    end
+
+    def self.active
+      active = ENV['AWSAM_ACTIVE_ACCOUNT']
+      return nil unless active
+
+      acct = find(active)
+      unless acct
+        puts "No account named '#{active}' found."
+        return nil
+      end
+
+      acct
+    end
+
+    def self.get
+      return @@accounts
+    end
+
+    def self.find(name)
+      @@accounts[name]
+    end
+
+    def self.set_default(name)
+      unless find(name)
+        $stderr.puts "Failed to find account #{name}"
+        return false
+      end
+
+      Utils::set_default(Awsam::get_accts_dir, name)
+    end
+
+    def self.get_default
+      dflt = Utils::get_default(Awsam::get_accts_dir)
+      dflt ? find(dflt) : nil
+    end
+
+    def self.remove_default
+      Utils::remove_default(Awsam::get_accts_dir)
+    end
+  end
+end

data/lib/awsam/ec2.rb

@@ -0,0 +1,95 @@
+module Awsam
+  module Ec2
+
+    LOOKUP_TAGS = ["Name", "aws:autoscaling:groupName"].freeze
+
+    def self.find_instance(acct, instance_id)
+      logger = Logger.new(File.open("/dev/null", "w"))
+      ec2 = RightAws::Ec2.new(acct.access_key, acct.secret_key, :logger => logger)
+      
+      unless ec2
+        puts "Unable to connect to EC2"
+        return nil
+      end
+
+      find(ec2, instance_id)
+    end
+    
+    def self.find(ec2, instance_id)
+      if instance_id =~ /^i-[0-9a-f]{7,9}$/
+        find_by_instance_id(ec2, instance_id)
+      else
+        find_by_tag(ec2, instance_id)
+      end
+    end
+    
+    def self.find_by_instance_id(ec2, instance_id)
+      begin
+        ec2.describe_instances(instance_id).first
+      rescue RightAws::AwsError
+        puts "instance_id does not exist"
+        exit 1
+      end
+    end
+    
+    def self.find_by_tag(ec2, instance_id)
+      results = []
+
+      ec2.describe_tags(:filters => {
+                          "resource-type" => "instance"
+                        }).each do |tag|
+        if LOOKUP_TAGS.include?(tag[:key]) &&
+            tag[:value].downcase.include?(instance_id.downcase)
+          results << tag
+        end
+      end
+
+      if !results || results.length == 0
+        puts "No tags by this name are available in your account"
+        exit 1
+      end
+      
+      results.uniq! { |a| a[:resource_id] }
+      results.sort! { |a,b| a[:value] <=> b[:value] }
+
+      rmap = {}
+      ec2.describe_instances(results.map{|a| a[:resource_id]},
+                             :filters => {
+                               "instance-state-name" => "running"
+                             }).each do |inst|
+        rmap[inst[:aws_instance_id]] = inst
+      end
+
+      results.reject! { |a| rmap[a[:resource_id]].nil? }
+
+      if results.length == 0
+        puts "No running instances by that tag name are available"
+        exit 1
+      end
+
+      if $opts[:first_node] || results.length == 1
+        node = results.first
+      else
+        puts "Please select which node you wish to use:"
+        puts
+
+        results.each_with_index do |elem, i|
+          inst = rmap[elem[:resource_id]]
+          puts "%d) %s (%s, %s, %s)" %
+            [i, elem[:value], inst[:aws_instance_id],
+             inst[:aws_instance_type], inst[:aws_launch_time]]
+        end
+        puts "q) Quit"
+        puts
+
+        print "> "
+        input = $stdin.gets
+        puts
+        exit unless input =~ /^\d+$/
+        node = results[input.to_i]
+      end
+
+      return rmap[node[:resource_id]]
+    end
+  end
+end

data/lib/awsam/key.rb

@@ -0,0 +1,53 @@
+require 'fileutils'
+
+module Awsam
+  class Key
+    KEYFILE = "key.pem"
+
+    attr_reader :name
+
+    def initialize(keydir)
+      @name = File.basename(keydir)
+      @dir = keydir
+      if @name == Awsam::DEFAULT_LINK_NAME
+        # This is required for the default link
+        raise "Can not name a key: #{Awsam::DEFAULT_LINK_NAME}"
+      end
+    end
+
+    def path
+      File.join(@dir, KEYFILE)
+    end
+
+    def self.import(acctdir, key_name, key_file)
+      dir = File.join(Key::keys_dir(acctdir), key_name)
+      FileUtils.mkdir(dir) unless File.exist?(dir)
+
+      File.open(File.join(dir, KEYFILE), "w", 0400) do |f|
+        f << File.read(key_file)
+      end
+
+      Key.new(dir)
+    end
+
+    def self.keys_dir(base)
+      dir = File.join(base, "keys")
+      FileUtils.mkdir(dir) unless File.exist?(dir)
+      dir
+    end
+
+    def print_environ
+      envs = {
+        "AMAZON_SSH_KEY_NAME" => @name,
+        "AMAZON_SSH_KEY_FILE" => self.path
+      }
+
+      Utils::bash_environ(envs)
+    end
+
+    def remove
+      FileUtils.rm(self.path)
+      FileUtils.rmdir(@dir)
+    end
+  end
+end

data/lib/awsam/utils.rb

@@ -0,0 +1,50 @@
+
+module Awsam
+  module Utils
+    # Scan a directory yielding for each file
+    def self.confdir_scan(dir)
+      Dir.entries(dir).each do |name|
+        next if name == '.' || name == '..' || name == Awsam::DEFAULT_LINK_NAME
+        yield(name)
+      end
+    end
+
+    # Print the appropriate environment variables set commands for bash
+    def self::bash_environ(envs)
+      envs.each_pair do |k, v|
+        puts "export #{k}=\"#{v}\""
+      end
+    end
+
+    # Set the default resource with link directory and target
+    def self.set_default(basedir, target)
+      link = File.join(basedir, Awsam::DEFAULT_LINK_NAME)
+      if File.exist?(link)
+        begin
+          FileUtils.rm(link)
+        rescue => err
+          $stderr.puts "Failed to remove link #{link}: #{err.message}"
+          return false
+        end
+      end
+      begin
+        FileUtils.ln_s(target, link)
+      rescue => err
+        $stderr.puts "Failed to create symlink: #{err.message}"
+        return false
+      end
+      true
+    end
+
+    # Get the target of the default link
+    def self.get_default(basedir)
+      link = File.join(basedir, Awsam::DEFAULT_LINK_NAME)
+      File.exist?(link) ? File.readlink(link) : nil
+    end
+
+    # Remove the default link
+    def self.remove_default(basedir)
+      FileUtils.rm File.join(basedir, Awsam::DEFAULT_LINK_NAME)
+    end
+  end
+end

data/lib/awsam/version.rb

@@ -0,0 +1,3 @@
+module Awsam
+  VERSION = "0.0.1"
+end

data/test/helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'awsam'
+
+class Test::Unit::TestCase
+end

data/test/test_awsem.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestAwsam < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

metadata

@@ -0,0 +1,140 @@
+--- !ruby/object:Gem::Specification
+name: awsam
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Mike Heffner
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: right_aws
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Amazon Web Services Account Manager (modeled after 'rvm')
+email:
+- mikeh@fesnel.com
+executables:
+- ascp
+- assh
+- raem
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- awsam.gemspec
+- bashrc/rc.scr
+- bin/ascp
+- bin/assh
+- bin/raem
+- lib/awsam.rb
+- lib/awsam/account.rb
+- lib/awsam/accounts.rb
+- lib/awsam/ec2.rb
+- lib/awsam/key.rb
+- lib/awsam/utils.rb
+- lib/awsam/version.rb
+- test/helper.rb
+- test/test_awsem.rb
+homepage: ''
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 4361486896786609133
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 4361486896786609133
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23.2
+signing_key: 
+specification_version: 3
+summary: Amazon Web Services Account Manager
+test_files:
+- test/helper.rb
+- test/test_awsem.rb