ec2ssh 1.0.1

data/.gitignore

@@ -0,0 +1,5 @@
+*.swp
+vendor
+.bundle
+tmp
+*.gem

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+gemspec
+

data/Gemfile.lock

@@ -0,0 +1,25 @@
+PATH
+  remote: .
+  specs:
+    schop (1.0.0)
+      highline (~> 1.6)
+      ktheory-right_aws (~> 2.0.3)
+      thor (~> 0.14.6)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    highline (1.6.1)
+    ktheory-right_aws (2.0.3)
+      right_http_connection (>= 1.2.1)
+    right_http_connection (1.2.4)
+    thor (0.14.6)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  highline (~> 1.6)
+  ktheory-right_aws (~> 2.0.3)
+  schop!
+  thor (~> 0.14.6)

data/README.rdoc

@@ -0,0 +1,48 @@
+== Introduction
+ec2ssh is a ssh_config manager for amazonaws ec2.
+<tt>ec2ssh</tt> command adds <tt>Host</tt> descriptions to ssh_config (~/.ssh/config default). 'Name' tag of instances are used as <tt>Host</tt> descriptions.
+== How to use
+1. Set 'Name' tag to your instances
+eg. Tag 'app-server-1' as 'Name' to an instance i-xxxxx in us-west-1 region.
+2. install ec2ssh
+    gem install ec2ssh
+3. Set environment variables "AMAZON_ACCESS_KEY_ID" and "AMAZON_SECRET_ACCESS_KEY"
+    export AMAZON_ACCESS_KEY_ID="..."
+    export AMAZON_SECRET_ACCESS_KEY="..."
+4. execute <tt>ec2ssh</tt> init and <tt>update</tt>
+    ec2ssh init
+    ec2ssh update
+Then host-names of your instances are generated and wrote to .ssh/config
+5. And you can ssh to your instances with your tagged name.
+    ssh user@app-server-1.us-west-1
+== Commands
+    ec2ssh help [TASK]  # Describe available tasks or one specific task
+    ec2ssh init         # Add ec2ssh mark to ssh_config
+    ec2ssh update       # Update ec2 hosts list in ssh_config
+    ec2ssh remove       # Remove ec2ssh mark from ssh_config
+Each command can be used <tt>--path</tt> option to set ssh_config path. The default is "~/.ssh/config"
+    ec2ssh init --path=/path/to/ssh_config
+== ssh_config and mark lines
+<tt>ec2ssh init</tt> command inserts mark lines your <tt>.ssh/config</tt> such as:
+    ### EC2SSH BEGIN ###
+    # Generated by ec2ssh http://github.com/mirakui/ec2ssh
+    # DO NOT edit this block!
+    # Updated Sun Dec 05 00:00:14 +0900 2010
+    ### EC2SSH END ###
+<tt>ec2ssh update</tt> command inserts 'Host' descriptions between 'BEGIN' line and 'END' line.
+    ### EC2SSH BEGIN ###
+    # Generated by ec2ssh http://github.com/mirakui/ec2ssh
+    # DO NOT edit this block!
+    # Updated Sun Dec 05 00:00:14 +0900 2010
+    Host app-server-1.us-west-1
+      HostName ec2-xxx-xxx-xxx-xxx.us-west-1.compute.amazonaws.com
+    Host db-server-1.ap-southeast-1
+      HostName ec2-xxx-xxx-xxx-xxx.ap-southeast-1.compute.amazonaws.com
+        :
+        :
+    ### EC2SSH END ###
+<tt>ec2ssh remove</tt> command removes the mark lines.
+== Notice
+<tt>ec2ssh</tt> command updates your <tt>.ssh/config</tt> file default. You should make a backup of it.
+== License
+ec2ssh is released under the MIT license.

data/bin/ec2ssh

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require 'ec2ssh'
+Ec2ssh::CLI.start

data/ec2ssh.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "ec2ssh/version"
+
+Gem::Specification.new do |s|
+  s.name        = "ec2ssh"
+  s.version     = Ec2ssh::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["mirakui"]
+  s.email       = ["mirakui@tsuyabu.in"]
+  s.homepage    = "http://github.com/mirakui/ec2ssh"
+  s.summary     = %q{A ssh_config manager for AWS EC2}
+  s.description = %q{A ssh_config manager for AWS EC2}
+
+  s.rubyforge_project = "ec2ssh"
+  s.add_dependency "thor", "~> 0.14.6"
+  s.add_dependency "highline", "~> 1.6"
+  s.add_dependency "ktheory-right_aws", "~> 2.0.3"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/lib/ec2ssh/cli.rb

@@ -0,0 +1,69 @@
+require 'thor'
+require 'right_aws'
+require 'highline'
+
+module Ec2ssh
+  class CLI < Thor
+    path_option = [:path, {:banner => "/path/to/ssh_config", :default=>"#{ENV['HOME']}/.ssh/config"}]
+
+    desc "init", "Add ec2ssh mark to ssh_config"
+    method_option *path_option
+    def init
+      config = Config.new(options.path)
+      if config.mark_exist?
+        red "Marker already exists on #{options.path}"
+        return
+      end
+      config.append_mark!
+      green "Added mark to #{options.path}"
+    end
+
+    desc "update", "Update ec2 hosts list in ssh_config"
+    method_option *path_option
+    def update
+      config = Config.new(options.path)
+      unless config.mark_exist?
+        red "Marker not found on #{options.path}"
+        red "Execute '#{$0} init --path=#{options.path}' first!"
+        return
+      end
+      hosts = Hosts.new.all
+      config_str = config.wrap(hosts.map{|h|<<-END}.join)
+Host #{h[:host]}
+  HostName #{h[:dns_name]}
+      END
+      config.replace! config_str
+      yellow config_str
+      green "Updated #{hosts.size} hosts on #{options.path}"
+    rescue AwsEnvNotDefined
+      red <<-END 
+Set environment variables to access AWS such as:
+  export AMAZON_ACCESS_KEY_ID="..."
+  export AMAZON_SECRET_ACCESS_KEY="..."
+      END
+    end
+
+    desc "remove", "Remove ec2ssh mark from ssh_config"
+    method_option *path_option
+    def remove
+      config = Config.new(options.path)
+      unless config.mark_exist?
+        red "Marker not found on #{options.path}"
+        return
+      end
+      config.replace! ""
+      green "Removed mark from #{options.path}"
+    end
+
+    private
+    def hl
+      @hl ||= HighLine.new
+    end
+
+    [:red,:green,:yellow].each do |col|
+      define_method(col) do |str|
+        puts hl.color(str, col)
+      end
+    end
+  end
+end

data/lib/ec2ssh/config.rb

@@ -0,0 +1,52 @@
+require 'pathname'
+
+module Ec2ssh
+  class Config
+    HEADER = "### EC2SSH BEGIN ###"
+    FOOTER = "### EC2SSH END ###"
+
+    attr_reader :path
+
+    def initialize(path=nil)
+      @path = Pathname(path || "#{ENV['HOME']}/.ssh/config")
+    end
+
+    def append_mark!
+      replace! ""
+      open(@path, "a") do |f|
+        f.puts wrap("")
+      end
+    end
+
+    def mark_exist?
+      config_src =~ /#{HEADER}\n.*#{FOOTER}\n/m
+    end
+
+    def replace!(str)
+      save! config_src.gsub(/#{HEADER}\n.*#{FOOTER}\n/m, str)
+    end
+
+    def config_src
+      @config_src ||= open(@path, "r") do |f|
+        f.read
+      end
+    end
+
+    def save!(str)
+      open(@path, "w") do |f|
+        f.puts str
+      end
+    end
+
+    def wrap(text)
+      return <<-END
+#{HEADER}
+# Generated by ec2ssh http://github.com/mirakui/ec2ssh
+# DO NOT edit this block!
+# Updated #{Time.now}
+#{text}
+#{FOOTER}
+      END
+    end
+  end
+end

data/lib/ec2ssh/hosts.rb

@@ -0,0 +1,34 @@
+module Ec2ssh
+  class AwsEnvNotDefined < StandardError; end
+  class Hosts
+    def initialize
+      @access_key_id = ENV["AMAZON_ACCESS_KEY_ID"].to_s
+      @secret_access_key = ENV["AMAZON_SECRET_ACCESS_KEY"].to_s
+      if @access_key_id.length == 0 || @secret_access_key.length == 0
+        raise AwsEnvNotDefined
+      end
+    end
+
+    def all
+      ec2 = RightAws::Ec2.new(@access_key_id, @secret_access_key)
+      regions = ec2.describe_regions
+      regions.map do |region|
+        process_region region
+      end.flatten
+    end
+
+    private
+    def process_region(region)
+      ec2 = RightAws::Ec2.new(@access_key_id, @secret_access_key, :region => region)
+      instance_names = {}
+      ec2.describe_tags.each do |tag|
+        next unless tag[:key]=='Name' && tag[:resource_type]=='instance'
+        instance_names[tag[:resource_id]] = tag[:value]
+      end
+      ec2.describe_instances.map do |instance|
+        instance_name = instance_names[instance[:aws_instance_id]]
+        instance_name ? {:host => "#{instance_name}.#{region}", :dns_name => instance[:dns_name]} : nil
+      end.compact
+    end
+  end
+end

data/lib/ec2ssh/version.rb

@@ -0,0 +1,3 @@
+module Ec2ssh
+  VERSION = '1.0.1'
+end

data/lib/ec2ssh.rb

@@ -0,0 +1,5 @@
+module Ec2ssh
+  autoload :CLI, "ec2ssh/cli"
+  autoload :Config, "ec2ssh/config"
+  autoload :Hosts, "ec2ssh/hosts"
+end

metadata

@@ -0,0 +1,124 @@
+--- !ruby/object:Gem::Specification 
+name: ec2ssh
+version: !ruby/object:Gem::Version 
+  hash: 21
+  prerelease: false
+  segments: 
+  - 1
+  - 0
+  - 1
+  version: 1.0.1
+platform: ruby
+authors: 
+- mirakui
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-05 00:00:00 +09:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 43
+        segments: 
+        - 0
+        - 14
+        - 6
+        version: 0.14.6
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: highline
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 1
+        - 6
+        version: "1.6"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: ktheory-right_aws
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 2
+        - 0
+        - 3
+        version: 2.0.3
+  type: :runtime
+  version_requirements: *id003
+description: A ssh_config manager for AWS EC2
+email: 
+- mirakui@tsuyabu.in
+executables: 
+- ec2ssh
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- README.rdoc
+- bin/ec2ssh
+- ec2ssh.gemspec
+- lib/ec2ssh.rb
+- lib/ec2ssh/cli.rb
+- lib/ec2ssh/config.rb
+- lib/ec2ssh/hosts.rb
+- lib/ec2ssh/version.rb
+has_rdoc: true
+homepage: http://github.com/mirakui/ec2ssh
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: ec2ssh
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A ssh_config manager for AWS EC2
+test_files: []
+