rudy 0.4.0 → 0.6.0

data/bin/ird

@@ -0,0 +1,175 @@
+#!/usr/bin/ruby
+
+# = Interactive Rudy (experimental)
+# 
+# === Not your granparent's deployment tool
+#
+#   See ird -h for usage
+#
+
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib') # Put our local lib in first place
+$:.unshift File.join(File.dirname(__FILE__), '..', 'vendor', 'highline-1.5.1', 'lib')
+%w{drydock caesars rye}.each { |dir| $:.unshift File.join(File.dirname(__FILE__), '..', '..', dir, 'lib') }
+require 'rubygems'
+
+# SEE: http://github.com/rubyspec/matzruby/blob/a34130eb7c4ecc164115a59aa1b76c643bd98202/lib/irb/xmp.rb
+# SEE: http://github.com/blackwinter/wirble/tree/master
+require "irb"
+require 'irb/completion'
+
+require 'rudy'
+require 'drydock'
+require 'rudy/cli'
+
+
+module RudyCLI_Interactive # :nodoc:all
+  extend self
+  extend Drydock
+  include Readline
+  
+  global :A, :accesskey, String, "AWS Access Key"
+  global :S, :secretkey, String, "AWS Secret Access Key"
+  #global :R, :region, String, "Connect to a specific EC2 region (ie: #{Rudy::DEFAULT_REGION})"
+  #global :n, :nocolor, "Disable output colors"
+  global :f, :config, String, "Specify another configuration file to read (ie: #{Rudy::CONFIG_FILE})"
+  global :z, :zone, String, "Connect to a specific EC2 zone (ie: #{Rudy::DEFAULT_ZONE})"
+  global :e, :environment, String, "Connect to the specified environment (ie: #{Rudy::DEFAULT_ENVIRONMENT})"
+  global :r, :role, String, "Connect to a machine with the specified role (ie: #{Rudy::DEFAULT_ROLE})"
+  global :p, :position, String, "Position in the machine in its group (ie: #{Rudy::DEFAULT_POSITION})"
+  global :u, :user, String, "Provide a username (ie: #{Rudy::DEFAULT_USER})"
+  #global :q, :quiet, "Run with less output"
+  #global :v, :verbose, "Increase verbosity of output (i.e. -v or -vv or -vvv)" do
+  #  @verbose ||= 0
+  #  @verbose += 1
+  #end
+  global :V, :version, "Display version number" do
+    puts "Rudy version: #{Rudy::VERSION}"
+    exit 0
+  end
+
+
+  default :interactive
+  debug :off
+  desc "An interactive REPL for Rudy."
+  usage "#{$0} [global options]"
+  command :interactive do |obj|
+  
+    @@rmach = Rudy::Machines.new(:global => obj.global)
+    @@rkey = Rudy::KeyPairs.new(:global => obj.global)
+    @@rgrp = Rudy::Groups.new(:global => obj.global)
+    @@rvol = Rudy::Volumes.new(:global => obj.global)
+    @@radd = Rudy::Addresses.new(:global => obj.global)
+  
+    puts Rudy::CLI.generate_header(@@rmach.global, @@rmach.config)
+    puts Rudy::Utils.without_indent(%Q`
+    ------------------------------------------------------------
+    Welcome Interactive Rudy! (EXPERIMENTAL)
+    ------------------------------------------------------------
+  
+    `)
+    h
+  
+    # What's the deal with this error message when defining a help method?
+    # => rb: warn: can't alias help from irb_help
+    if __FILE__ == $0
+      IRB.start(__FILE__)
+    else
+      # check -e option
+      if /^-e$/ =~ $0
+        IRB.start(__FILE__)
+      else
+        IRB.start(__FILE__)
+      end
+    end
+  
+  end
+  
+  
+  
+  
+  def h;
+    puts (%Q`
+  HELP
+
+    h             Help (this screen)
+    m             Rudy::Machines instance
+    g             Rudy::Groups instance
+    k             Rudy::KeyPairs instance
+    v             Rudy::Volumes instance
+    a             Rudy::Addresses instance
+    ssh [GROUP]   Connect to a machine
+    example       Start an EC2 instance from scratch.
+    `)
+  end
+  def example
+    puts (%Q`
+  EXAMPLE
+      Start an EC2 instance from scratch.
+
+      # Create a security group
+      >> g.create
+      => #<Rudy::AWS::EC2::Group @name="stage-app" ...>
+
+      # Create a keypair (used to login as root)
+      >> k.create
+      => #<Rudy::AWS::EC2::KeyPair @name="key-stage-app" ...>
+
+      # Start an EC2 machine instance:    
+      >> m.create
+      Instance: i-228def51
+      Waiting for the instance to startup
+      ........It's up!
+
+      # Login to the new instance
+      >> ssh 
+
+      All commands in Rudy refer to the global defaults unless
+      otherwise specified. "stage" is the default environment and 
+      "app" is the default role. You can change these with command 
+      line arguments or by modifying the defaults in your Rudy 
+      config file: #{@@rmach.config_dirname}
+
+      You can also specify your own names by providing
+      one as the first argument: g.create 'group5000'.
+
+      # List instances in stage-app
+      >> m.list :running
+      => [#<Rudy::AWS::EC2::Instance:0x18c5888 ...>]
+
+      # Print the instance in a human readable way
+      >> puts m.list.first.to_s
+
+      # Print instance system console output
+      >> puts m.console
+
+      # Destroy ALL INSTANCES in stage-app
+      >> m.destroy
+    `)
+  end
+
+  def m; @@rmach; end
+  def k; @@rkey; end
+  def g; @@rgrp; end
+  def v; @@rvol; end
+  def a; @@radd; end
+
+  # Change Group
+  #def cg(name); end
+
+  def ssh(*args)
+    @@rmach.switch_user :root
+    @@rmach.connect(*args)
+    @@rmach.switch_user # back to previous user
+    nil
+  end
+  
+  
+end
+
+include CLI  # So we don't have to say CLI.example in the REPL. 
+
+
+__END__
+#@rbox = Rye::Box.new('ec2-174-129-173-3.compute-1.amazonaws.com', 'root')
+#@rbox2 = Rye::Box.new('ec2-174-129-173-3.compute-1.amazonaws.com', 'root')
+#@rbox.add_keys('/Users/delano/Projects/git/rudy/.rudy/key-test-app.private')

data/bin/rudy

@@ -1,174 +1,277 @@
 #!/usr/bin/ruby
 
-# Rudy -- Your friend in staging and deploying to EC2
+
+# = Rudy
+# 
+# === Not your granparent's deployment tool
 #
 #   See rudy -h for usage
 # 
 
-RUDY_HOME = File.join(File.dirname(__FILE__), '..')
-RUDY_LIB = File.join(RUDY_HOME, 'lib')
-$:.unshift RUDY_LIB # Put our local lib in first place
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib') # Put our local lib in first place
+$:.unshift File.join(File.dirname(__FILE__), '..', 'vendor', 'highline-1.5.1', 'lib')
+#%w{amazon-ec2 rye}.each { |dir| $:.unshift File.join(File.dirname(__FILE__), '..', '..', dir, 'lib') }
+#require 'rubygems'
 
-require 'rubygems' if RUBY_VERSION < "1.9"
-require 'date'
-require 'drydock'
-extend Drydock
-
-project "Rudy"  # This also runs require 'ruby'
-
-global :A, :accesskey, String, "AWS Access Key"
-global :S, :secretkey, String, "AWS Secret Access Key"
-#global :R, :region, String, "Connect to a specific EC2 region (ie: #{Rudy::DEFAULT_REGION})"
-global :f, :config, String, "Specify another configuration file to read (ie: #{Rudy::RUDY_CONFIG_FILE})"
-global :z, :zone, String, "Connect to a specific EC2 zone (ie: #{Rudy::DEFAULT_ZONE})"
-global :e, :environment, String, "Connect to the specified environment (ie: #{Rudy::DEFAULT_ENVIRONMENT})"
-global :r, :role, String, "Connect to a machine with the specified role (ie: #{Rudy::DEFAULT_ROLE})"
-global :p, :position, String, "Position in the machine in its group (ie: #{Rudy::DEFAULT_POSITION})"
-global :u, :user, String, "Provide a username (ie: #{Rudy::DEFAULT_USER})"
-global :q, :quiet, "Run with less output"
-global :v, :verbose, "Increase verbosity of output (i.e. -v or -vv or -vvv)" do
-  @verbose ||= 0
-  @verbose += 1
-end
-global :V, :version, "Display version number" do
-  puts "Rudy version: #{Rudy::VERSION}"
-  exit 0
-end
+#$SAFE = 1   # require is unsafe in Ruby 1.9??
 
-#desc "Run this the first time you use Rudy (it's immutable so running it again does no harm)."
-#command :setup => Rudy::Command::Metadata
-
-
-# ------------------------------------  RUDY INFO COMMANDS  --------
-# ------------------------------------------------------------------
-
-#usage "rudy info"
-#desc "Displays info about the current Rudy configuration"
-#command :info => Rudy::Command::Metadata
-
-usage "rudy [-f config-file] config [param-name]"
-desc "Check Rudy configuration."
-option :l, :all, "Display config settings for all machines"
-option :d, :defaults, "Display the default value for the supplied parameter"
-argv :name
-command :config => Rudy::Command::Config
-
-usage "rudy myaddress [-i] [-e]"
-desc "Displays you current internal and external IP addresses"
-option :e, :external, "Display only external IP address"
-option :i, :internal, "Display only internal IP address"
-command :myaddress do |obj|
-  ea = Rudy::Utils::external_ip_address || '' 
-  ia = Rudy::Utils::internal_ip_address || ''
-  if obj.global.quiet
-    puts ia unless obj.option.external && !obj.option.internal
-    puts ea unless obj.option.internal && !obj.option.external
-  else
-    puts "%10s: %s" % ['Internal', ia] unless obj.option.external && !obj.option.internal
-    puts "%10s: %s" % ['External', ea] unless obj.option.internal && !obj.option.external
+require 'drydock'
+require 'rudy'
+require 'rudy/cli'
+
+
+# Command-line interface for /bin/rudy
+class RudyCLI < Rudy::CLI::Base
+  
+  # ------------------------------------------  RUDY GLOBALS  --------
+  # ------------------------------------------------------------------
+  
+  global :e, :environment, String, "Connect to the specified environment (ie: #{Rudy::DEFAULT_ENVIRONMENT})"
+  global :r, :role, String, "Connect to a machine with the specified role (ie: #{Rudy::DEFAULT_ROLE})"
+  global :p, :position, String, "Position in the machine in its group (ie: #{Rudy::DEFAULT_POSITION})"
+  
+  
+
+  # ---------------------------------  RUDY MACHINE COMMANDS  --------
+  # ------------------------------------------------------------------
+
+
+  #desc "Machine Status"
+  #usage "rudy [global options] status [-g group-name] [-s state] [--all] [instance-ID]"
+  #option :g, :group, String, "A security group name"
+  #option :s, :state, String, "Machine state. One of: running (default), pending, terminated"
+  #option :l, :all, "Show all machines in this group, regardless of state."
+  #argv :awsid
+  #command :status => Rudy::CLI::Machines
+  #
+  #usage "rudy [global options] connect [-g group-name] [-i instance-ID] [cmd]"
+  #desc "Open an SSH connection"
+  #option :print, "Only print the SSH command, don't connect"
+  #option :g, :group, String, "A security group name"
+  #option :i, :awsid, String, "An instance ID"
+  #argv :cmd
+  #command :connect => Rudy::CLI::Machines 
+  #command_alias :connect, :ssh
+  #
+  #usage "rudy [global options] copy [-p] [-r] source target"
+  #desc "Copy files to or from machines. NOTE: You must use quotes when using a tilda for your remote dir ('~/')."
+  #option :r, :recursive, "Recursively copy entire directories"
+  #option :p, :preserve, "Preserve atimes and ctimes."
+  #option :d, :download, "Download FROM the remote machine to the local machine"
+  #option :print, "Only print the SSH command, don't connect"
+  #option :g, :group, String, "A security group name"
+  #option :i, :awsid, String, "An instance ID"
+  #command :copy => Rudy::CLI::Machines
+  #command_alias :copy, :scp
+  #command_alias :copy, :upload
+  #command_alias :copy, :download
+
+
+
+
+  # --------------------------------  MISCELLANEOUS COMMANDS  --------
+  # ------------------------------------------------------------------
+  
+  desc "Display time (in UTC)"
+  option :l, :local, "Display local time"
+  command :time do |obj|
+    t = obj.option.local ? Time.now : Time.now.utc
+    puts '%s' % t.strftime("%Y-%m-%d %T %Z (%z)")
+  end
+  
+  usage "rudy [-f config-file] config [param-name]"
+  desc "Check Rudy configuration."
+  option :l, :all, "Display configs for all machines"
+  option :s, :script, "Output configuration identical to what is provided to scripts called in routines"
+  option :r, :rudy, "Output a skeleton Rudyfile"
+  option :d, :defaults, "Display the default value for the supplied parameter"
+  option :g, :group, String, "Display configuration for a specific group"
+  argv :name
+  command :config => Rudy::CLI::Config
+
+  usage "rudy myaddress [-i] [-e]"
+  desc "Displays you current internal and external IP addresses"
+  option :e, :external, "Display only external IP address"
+  option :i, :internal, "Display only internal IP address"
+  command :myaddress do |obj|
+    ea = Rudy::Utils::external_ip_address || '' 
+    ia = Rudy::Utils::internal_ip_address || ''
+    if obj.global.quiet
+      puts ia unless obj.option.external && !obj.option.internal
+      puts ea unless obj.option.internal && !obj.option.external
+    else
+      puts "%10s: %s" % ['Internal', ia] unless obj.option.external && !obj.option.internal
+      puts "%10s: %s" % ['External', ea] unless obj.option.internal && !obj.option.external
+    end
+    obj.global.quiet = true  # don't print elapsed time
   end
-end
-
-
-
-# -----------------------------  RUDY MAINTENANCE COMMANDS  --------
-# ------------------------------------------------------------------
-
-usage "#{$/} [global options] disks [-C -p path -d device -s size] [-A] [-D] [disk name]"
-desc "Manage Disks"
-option :l, :all, "Display all disk definitions"
-option :p, :path, String, "The filesystem path to use as the mount point"
-option :d, :device, String, "The device id (default: /dev/sdh)"
-option :s, :size, Integer, "The size of disk (in GB)"
-action :C, :create, "Create a disk definition"
-action :D, :destroy, "Destroy a disk definition"
-action :A, :attach, "Attach a disk"
-action :N, :unattach, "Unattach a disk"
-argv :diskname
-command :disk => Rudy::Command::Disks
-
-
-usage "rudy [global options] backups [-C] [disk name]"
-desc "Manage Backups"
-option :s, :snapshot, String, "Create a backup entry from an existing snapshot"
-action :Z, :sync, "Check for and delete backup metadata with no snapshot. DOES NOT delete snapshots."
-#action :T, :tidy, "Tidy existing backups"
-action :D, :destroy, "Destroy a backup and DELETE its snapshots."
-action :C, :create, "Create a backup"
-argv :disk
-command :'backup' => Rudy::Command::Backups
-command_alias :backup, :bu
-
-
-usage "rudy [global options] metadata instance-ID"
-desc "Display Rudy metadata."
-command :metadata => Rudy::Command::Metadata
-command_alias :metadata, :md
-
-
-desc "Machine Group Status"
-command :status => Rudy::Command::Machines
-
-
-#desc "Update a Machine Group with the current version of Rudy"
-#command :update => Rudy::Command::Machines
-
-
-usage "rudy [-e env] [-u user] connect [-p] [cmd]"
-desc "Open an SSH connection"
-option :p, :print, "Only print the SSH command, don't connect"
-argv :cmd
-command :connect => Rudy::Command::Environment 
-command_alias :connect, :ssh
-
-usage "rudy [-e env] [-u user] copy [-p] -r [from path] [to path]"
-desc "Copy files to or from machines. NOTE: You must use quotes when using a tilda for your remote dir ('~/')."
-option :r, :remote, "Copy FROM the remote machine to the local machine"
-option :p, :print, "Only print the SSH command, don't connect"
-argv :from, :to
-command :copy => Rudy::Command::Environment
-command_alias :copy, :scp
-command_alias :copy, :upload
-command_alias :copy, :download
-
-
-# --------------------------------  RUDY ROUTINES COMMANDS  --------
-# ------------------------------------------------------------------
-
-desc "Shutdown a Machine Group"
-command :shutdown => Rudy::Command::Machines
-
-desc "Start a Machine Group"
-command :startup => Rudy::Command::Machines
-command_alias :startup, :start
-
-desc "Restart a Machine Group"
-command :restart => Rudy::Command::Machines
 
-desc "Release to a machine group"
-option :s, :switch, "Switch to the release branch/tag"
-option :m, :msg, String, "A short release note"
-command :release => Rudy::Command::Release
+  usage "rudy [global options] annoy [-h -m -l] [-e]"
+  desc "Play around with Rudy's annoying challenges"
+  option :s, :string, "A numeric challenge"
+  option :n, :numeric, "A numeric challenge"
+  option :i, :insane, "Insane annoyance factor"
+  option :h, :high, "High annoyance factor"
+  option :m, :medium, "Medium annoyance factor"
+  option :l, :low, "Low annoyance factor"
+  option :r, :rand, "Random challenge type"
+  command :annoy do |obj|
+    srand(Time.now.to_f)
+    flavor = [:numeric, :string, :rand].detect { |v| obj.option.send(v) } || :string
+    factor = [:insane, :high, :medium, :low].detect { |v| obj.option.send(v) } || :medium
+    success = Annoy.challenge?("Is this annoying?", factor, flavor)
+    puts (success ? "Correct!" : "WRONG!").bright
+    obj.global.quiet = true  # don't print elapsed time
+  end
 
-desc "Update the release currently running in a machine group"
-command :rerelease => Rudy::Command::Release
-command_alias :rerelease, :rere
+  desc "Display the current Rudy slogan"
+  command :slogan do |obj|
+    puts "Rudy: Not your grandparent's deployment tool!"
+    obj.global.quiet = true  # don't print elapsed time
+  end
 
-#desc "Deploy disk snapshots from one machine to another"
-#command :deploy => Rudy::Command::Deploy
+  desc "Generates a configuration template to #{Rudy::CONFIG_FILE}"
+  command :generate_config do |obj|
+    unless File.exists?(Rudy::CONFIG_FILE)
+      Rudy::Config.init_config_dir
+      puts "Add your AWS credentials to #{Rudy::CONFIG_FILE}"
+    else
+      puts "#{Rudy::CONFIG_FILE} already exists"
+    end
+  end
 
 
+  desc "Initialize Rudy configuration"
+  command :init do |obj|
+  
+    unless File.exists?(Rudy::CONFIG_FILE)
+      Rudy::Config.init_config_dir
+    end
+  
+    begin
+
+      unless Rudy::Huxtable.domain_exists?
+        puts "Creating SimpleDB domain #{Rudy::Huxtable.domain}"
+        Rudy::Huxtable.create_domain
+        puts "Initialized"
+      else
+        puts "Already Initialized"
+      end
+    
+    rescue Rudy::AWS::SDB::NoSecretKey, 
+           Rudy::AWS::SDB::NoAccessKey, 
+           Rudy::NoConfig => ex
+      puts "AWS credentials must be configured to continue."
+      puts "You can modify these in #{Rudy::CONFIG_FILE}"
+      exit 1
+    end
+  
+    obj.global.quiet = true  # don't print elapsed time
+  end
 
 
-# -------------------------------------------  UGLY STUFFS  --------
-# ------------------------------------------------------------------
-debug :on
-capture :stderr       
-before do
-  @start = Time.now
-end
-after do
-  @elapsed = Time.now - @start
-  puts $/, "Elapsed: %.2f seconds" % @elapsed.to_f if @elapsed > 0.1
+  # ---------------------------------  RUDY MANAGER COMMANDS  --------
+  # ------------------------------------------------------------------
+
+  #usage "rudy [global options] backups [-C] [disk name]"
+  #desc "Manage Backups"
+  #option :s, :snapshot, String, "Create a backup entry from an existing snapshot"
+  #action :Z, :sync, "Check for and delete backup metadata with no snapshot. DOES NOT delete snapshots."
+  ##action :T, :tidy, "Tidy existing backups"
+  #action :D, :destroy, "Destroy a backup and DELETE its snapshots."
+  #action :C, :create, "Create a backup"
+  #argv :disk
+  #command :'backup' => Rudy::CLI::Backups
+  #command_alias :backup, :bu
+  
+  desc "Startup a machine group"
+  usage "rudy startup"
+  command :startup => Rudy::CLI::Routines
+  command_alias :startup, :s
+
+  desc "Shutdown a machine group"
+  usage "rudy shutdown"
+  command :shutdown => Rudy::CLI::Routines
+  
+  desc "Create a release"
+  usage "rudy release"
+  command :release => Rudy::CLI::Routines
+  command_alias :release, :r
+
+  desc "Manage Machines"
+  action :W, :wash, "Wash machine metadata."
+  command :machines => Rudy::CLI::Machines
+  command_alias :machines, :m
+  
+  desc "Log in to a machine"
+  command :ssh => Rudy::CLI::Machines
+  
+  desc "Manage Disks"
+  action :W, :wash, "Wash disk metadata."
+  command :disks => Rudy::CLI::Disks
+  command_alias :disks, :d
+  
+  # --------------------------  RUDY RELEASE/DEPLOY COMMANDS  --------
+  # ------------------------------------------------------------------
+
+  #desc "Release to a Machine Group"
+  #option :g, :group, String, "A security group name"
+  #option :s, :switch, "Switch to the release branch/tag"
+  #option :m, :msg, String, "A short release note"
+  #command :release => Rudy::CLI::Release
+
+  #desc "Update the release currently running in a machine group"
+  #command :rerelease => Rudy::CLI::Release
+  #command_alias :rerelease, :rere
+
+  #desc "Deploy disk snapshots from one machine to another"
+  #command :deploy => Rudy::CLI::Deploy
+
+
+  #desc "Shutdown a Machine Group"
+  #usage "rudy [global options] shutdown [-g group-name] [instance-ID]"
+  #option :g, :group, String, "A security group name"
+  #argv :awsid
+  #command :shutdown => Rudy::CLI::Routines
+  #
+  #
+  #desc "Start a Machine Group"
+  #usage "rudy [global options] startup [-g group-name] [-i image-ID]"
+  #option :ami, String, "EC2 image ID (AMI)"
+  #option :g, :group, String, "A security group name"
+  #command :startup => Rudy::CLI::Routines
+  #command_alias :startup, :start
+
+  #desc "Restart a Machine Group"
+  #option :g, :group, String, "A security group name"
+  #argv :awsid
+  #command :restart => Rudy::CLI::Routines
+  #
+
+  
+  # -------------------------------------------  UGLY STUFFS  --------
+  # ------------------------------------------------------------------
+  debug :off
+  default :machines
+  
+
+  
 end
 
+# We call Drydock specifically otherwise it will run at_exit. Rye also
+# uses at_exit for shutting down the ssh-agent. Ruby executes at_exit
+# blocks in reverse order so if Drydock is required first, it's block
+# will run after Rye shuts down the ssh-agent. 
+begin
+  Drydock.run!(ARGV, STDIN) if Drydock.run? && !Drydock.has_run?
+rescue Drydock::ArgError, Drydock::OptError=> ex
+  STDERR.puts ex.message
+  STDERR.puts ex.usage
+rescue Rudy::Error => ex
+  STDERR.puts ex.message
+rescue => ex
+  STDERR.puts "ERROR (#{ex.class.to_s}): #{ex.message}"
+  STDERR.puts ex.backtrace if Drydock.debug?
+rescue SystemExit
+  # Don't balk
+end