rightscale-autocomplete 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Joshua Bates
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= rightscale-autocomplete
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Joshua Bates. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,19 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "rightscale-autocomplete"
+    gem.summary = %Q{SSH completion for rightscale servers}
+    gem.description = %Q{SSH completion for rightscale servers}
+    gem.email = "joshuabates@gmail.com"
+    gem.homepage = "http://github.com/joshuabates/rightscale-autocomplete"
+    gem.authors = ["Joshua Bates"]
+    gem.add_development_dependency "rightscaler"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/rs-autocomplete

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+require 'rightscale-autocomplete'
+if ENV['COMP_LINE'].nil?
+  RightscaleAutocomplete.send(ARGV.first)
+else
+  RightscaleAutocomplete.complete(ENV['COMP_LINE'])
+  exit 0
+end

data/bin/sshrs

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require 'rightscale-autocomplete'
+RightscaleAutocomplete.ssh(ARGV.first)

data/lib/rightscale-autocomplete.rb

@@ -0,0 +1,121 @@
+module RightscaleAutocomplete
+  BASE_DIR = File.join(File.expand_path('~'), '.rightscale')
+  COOKIE_CACHE = File.join(BASE_DIR, "cookie")
+  SERVER_CACHE = File.join(BASE_DIR, "servers")
+  CREDENTIALS_FILE = File.join(BASE_DIR, "credentials")
+  SSH_CONFIG = File.join(BASE_DIR, "ssh")
+  
+  def self.ssh(server)
+    authenticate
+    
+    ssh_config = Marshal.load(File.read(SSH_CONFIG))
+    servers = File.readlines(SERVER_CACHE)
+    server_id = servers.grep(/#{server.strip}/).first.split("::").last.strip
+    dns = Rightscaler::Manage::Server.find(server_id).get(:settings)['dns_name']
+    
+    cmd = "ssh #{ssh_config[:user]}@#{dns} -i #{ssh_config[:key]}"
+    puts cmd
+    exec cmd
+  end
+  
+  def self.complete(partial_server_name)
+    after_match = partial_server_name.split[1..-1].join(' ')
+    server_match = (after_match.empty? || after_match =~ /\s$/) ? nil : after_match.strip
+    servers = server_list.split("\n").map { |line| line.split('::').first }
+    servers = servers.select { |t| /^#{Regexp.escape server_match}/i =~ t } if server_match
+
+    if server_match =~ /^([-\w:]+:)/
+      upto_last_colon = $1
+      after_match = $'
+      servers = servers.map { |t| (t =~ /^#{Regexp.escape upto_last_colon}([-\w:]+)$/i) ? "#{$1}" : t }
+    end
+
+    puts servers
+  end
+  
+  def self.reset
+    File.delete(COOKIE_CACHE) rescue nil
+    File.delete(SERVER_CACHE) rescue nil
+    
+    authenticate
+    get_server_list
+  end
+  
+  def self.setup
+    Dir.mkdir(BASE_DIR) unless File.exists?(BASE_DIR)
+    
+    credentials = {}
+    print "Enter your Rightscale account_id: "
+    credentials[:account_id] = STDIN.gets.chomp
+    print "Enter your Rightscale username: "
+    credentials[:user] = STDIN.gets.chomp
+    print "Enter your Rightscale password: "
+    credentials[:password] = STDIN.gets.chomp
+    
+    File.open(CREDENTIALS_FILE, 'w') { |f| f << Marshal.dump(credentials) }
+    
+    ssh_config = {}
+    print "Enter the username to login as (eg root): "
+    ssh_config[:user] = gets.chomp
+    
+    print "Enter the location of your ssh key (eg ~/.ssh/ec2_prod_key): "
+    ssh_config[:key] = gets.chomp
+    
+    File.open(SSH_CONFIG, 'w') { |f| f << Marshal.dump(ssh_config) }
+    
+    puts "\nAdd this line to your profile"
+    puts "complete -C rs-autocomplete -o default sshrs"
+  end
+  
+  private
+  
+  def self.authenticate
+    require "rightscaler"
+    
+    credentials = load_credentials
+    
+    Rightscaler::Resource.account_id = credentials[:account_id]
+    Rightscaler::Resource.user = credentials[:user]
+    Rightscaler::Resource.password = credentials[:password]
+
+    if cookie = valid_cookie
+      Rightscaler::Account.set_auth_cookie(cookie)
+    else
+      Rightscaler::Account.authenticate
+      File.open(COOKIE_CACHE, 'w') { |f| f.puts Rightscaler::Resource.headers['cookie'] }
+    end
+  end
+  
+  def self.load_credentials
+    setup unless File.exists? CREDENTIALS_FILE
+    Marshal.load(File.read(CREDENTIALS_FILE))
+  end
+
+  def self.valid_cookie
+    return unless File.exists?(COOKIE_CACHE) && File.mtime(COOKIE_CACHE) > (Time.now - 360)
+    cookie = File.read(COOKIE_CACHE)
+    !cookie.blank? && cookie
+  end
+
+  def self.server_list
+    if File.exists?(SERVER_CACHE)
+      File.read(SERVER_CACHE)
+    else
+      get_server_list
+    end
+  end
+  
+  def self.get_server_list
+    authenticate
+    deployments = Rightscaler::Manage::Deployment.find(:all)
+    servers = deployments.inject([]) do |mem, deployment|
+      mem << deployment.servers.map { |s| "#{underscore(deployment.nickname)}:#{underscore(s.nickname)}::#{s.id}"}
+    end.join("\n")
+    File.open(SERVER_CACHE, 'w') { |f| f.puts servers }
+    servers
+  end
+
+  def self.underscore(string)
+    string.strip.gsub(/\W+/, '_')
+  end
+end

data/rightscale-autocomplete.gemspec

@@ -0,0 +1,57 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{rightscale-autocomplete}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Joshua Bates"]
+  s.date = %q{2010-03-24}
+  s.description = %q{SSH completion for rightscale servers}
+  s.email = %q{joshuabates@gmail.com}
+  s.executables = ["rs-autocomplete", "sshrs"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/rs-autocomplete",
+     "bin/sshrs",
+     "lib/rightscale-autocomplete.rb",
+     "rightscale-autocomplete.gemspec",
+     "test/helper.rb",
+     "test/test_rightscale-autocomplete.rb"
+  ]
+  s.homepage = %q{http://github.com/joshuabates/rightscale-autocomplete}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{SSH completion for rightscale servers}
+  s.test_files = [
+    "test/helper.rb",
+     "test/test_rightscale-autocomplete.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rightscaler>, [">= 0"])
+    else
+      s.add_dependency(%q<rightscaler>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<rightscaler>, [">= 0"])
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'rightscale-autocomplete'
+
+class Test::Unit::TestCase
+end

data/test/test_rightscale-autocomplete.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestRightscaleAutocomplete < 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,87 @@
+--- !ruby/object:Gem::Specification 
+name: rightscale-autocomplete
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Joshua Bates
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-24 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rightscaler
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: SSH completion for rightscale servers
+email: joshuabates@gmail.com
+executables: 
+- rs-autocomplete
+- sshrs
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/rs-autocomplete
+- bin/sshrs
+- lib/rightscale-autocomplete.rb
+- rightscale-autocomplete.gemspec
+- test/helper.rb
+- test/test_rightscale-autocomplete.rb
+has_rdoc: true
+homepage: http://github.com/joshuabates/rightscale-autocomplete
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: SSH completion for rightscale servers
+test_files: 
+- test/helper.rb
+- test/test_rightscale-autocomplete.rb