ec2cli 0.1.0

data/README

@@ -0,0 +1,71 @@
+= ec2cli
+
+== Description
+
+ec2cli is an interactive command-line client of Amazon EC2.
+
+== Source Code
+
+https://bitbucket.org/winebarrel/ec2cli
+
+== Install
+
+  shell> gem install ec2cli
+  shell> export AWS_ACCESS_KEY_ID='...'
+  shell> export AWS_SECRET_ACCESS_KEY='...'
+  shell> export EC2_REGION=ap-northeast-1
+  shell> ec2cli -h
+  Usage: ec2cli [options]
+      -k, --access-key=ACCESS_KEY
+      -s, --secret-key=SECRET_KEY
+      -r, --region=REGION_OR_ENDPOINT
+          --debug
+      -h, --help
+      ...
+  shell> ec2cli # show prompt
+  ap-northeast-1> desc instances;
+  [
+    ["cthulhu","i-01bd4903","stopped","10.0.114.178","t1.micro",["default"],{"Name":"cthulhu"},"ap-northeast-1a"],
+    ["hastur","i-e78d79e5","stopped","10.0.130.129","m1.large",["default"],{"Name":"hastur","aaa":"1100"},"ap-northeast-1b"],
+    ["nyar","i-a5e119a7","stopped","10.0.254.210","c1.xlarge",["default"],{"aaa":"1100","Name":"nyar"},"ap-northeast-1b"],
+  ]
+  // 3 rows in set (0.23 sec)
+
+== Help
+
+  ##### Query #####
+  
+  DESC[RIBE] INSTANCES [WHERE name = '...' AND ...]
+    Describes one or more of your instances.
+  
+  DESC[RIBE] IMAGES [WHERE name = '...' AND ...]
+  DESC[RIBE] ALL IMAGES [WHERE [Owner = {amazon|aws-marketplace|self} AND] name = '...' AND ...]
+    Describes the images.
+  
+  RUN INSTANCES image_id
+    Launches instances
+  
+  START INSTANCES id_or_name
+  START INSTANCES (id_or_name, ...)
+    Starts an Amazon EBS-backed AMI that you've previously stopped.
+  
+  STOP INSTANCES id_or_name
+  STOP INSTANCES (id_or_name, ...)
+    Stops an Amazon EBS-backed instance
+  
+  SET TAGS name = value, ...  [WHERE name = '...' AND ...]
+    Adds or overwrites one or more tags for the specified EC2 resource or resources.
+  
+  USE region_or_endpoint
+    changes an endpoint
+  
+  SHOW REGIONS
+    displays a region list
+  
+  ##### Command #####
+  
+  .help                           displays this message
+  .quit | .exit                   exits sdbcli
+  .debug           (true|false)?  displays a debug status or changes it
+  .version                        displays a version
+

data/bin/ec2cli

@@ -0,0 +1,67 @@
+#!/usr/bin/env ruby
+$LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'lib')
+
+Version = '0.1.0'
+
+HISTORY_FILE = File.join((ENV['HOME'] || ENV['USERPROFILE'] || '.'), '.ec2cli_history')
+HISTSIZE = 500
+
+require 'rubygems'
+require 'ec2cli'
+require 'readline'
+
+options = parse_options
+
+driver = EC2::Driver.new(
+  options.access_key_id,
+  options.secret_access_key,
+  options.ec2_endpoint_or_region)
+
+driver.debug = options.debug
+
+# load history file
+if File.exist?(HISTORY_FILE)
+  open(HISTORY_FILE) do |f|
+    f.each_line do |line|
+      line = line.strip
+      Readline::HISTORY.push(line) unless line.empty?
+    end
+  end
+end
+
+# interactive mode
+src = ''
+prompt1 = lambda { "#{driver.region || 'unknown'}> " }
+prompt2 = lambda { "#{' ' * (prompt1.call.length - 3)}-> " }
+
+while buf = Readline.readline((src.empty? ? prompt1.call : prompt2.call), true)
+  # ignore blank lines
+  if /\A\s*\Z/ =~ buf
+    Readline::HISTORY.pop
+    next
+  end
+
+  if src.empty? and buf =~ /\A\.(.+)/
+    evaluate_command(driver, $1)
+  else
+    begin
+      src << (src.empty? ? buf : ("\n" + buf))
+      evaluate_query(driver, src, :show_rows => true)
+    rescue => e
+      print_error(e.message)
+      print_error(e.backtrace) if driver.debug
+    end
+
+    prompt = src.empty? ? prompt1.call : prompt2.call
+  end
+end # of while
+
+# save history file
+unless Readline::HISTORY.empty?
+  open(HISTORY_FILE, 'wb') do |f|
+    (Readline::HISTORY.to_a.slice(-(Readline::HISTORY.length < HISTSIZE ? Readline::HISTORY.length : HISTSIZE)..-1) || []).each do |line|
+      next if /\A\s*\Z/ =~ line
+      f.puts line
+    end
+  end
+end

data/lib/ec2cli/cli/evaluate.rb

@@ -0,0 +1,53 @@
+require 'ec2cli/ec2-driver'
+
+def evaluate_query(driver, src, opts = {})
+  ss = StringScanner.new(src.dup)
+  buf = ''
+
+  until ss.eos?
+    if (tok = ss.scan %r{[^`'";\\/#]+}) #'
+      buf << tok
+    elsif (tok = ss.scan /`(?:[^`]|``)*`/)
+      buf << tok
+    elsif (tok = ss.scan /'(?:[^']|'')*'/) #'
+      buf << tok
+    elsif (tok = ss.scan /"(?:[^"]|"")*"/) #"
+      buf << tok
+    elsif (tok = ss.scan %r{/\*/?(?:\n|[^/]|[^*]/)*\*/})
+      # nothing to do
+    elsif (tok = ss.scan /--[^\r\n]*(?:\r\n|\r|\n|\Z)/)
+      # nothing to do
+    elsif (tok = ss.scan /#[^\r\n]*(?:\r\n|\r|\n|\Z)/)
+      # nothing to do
+    elsif (tok = ss.scan /(?:\\;)/)
+      buf << ';' # escape of ';'
+    elsif (tok = ss.scan /(?:;|\\G)/)
+      src.replace(ss.rest)
+      query = buf
+      buf = ''
+
+      if query.strip.empty?
+        print_error('No query specified')
+        next
+      end
+
+      start_time = Time.new
+      out = driver.execute(query)
+      elapsed = Time.now - start_time
+
+      if out.kind_of?(EC2::Driver::Rownum)
+        print_rownum(out, :time => elapsed)
+      elsif out.kind_of?(String)
+        puts out
+      elsif out
+        opts = opts.merge(:inline => (tok != '\G'), :time => elapsed)
+        print_json(out, opts)
+      end
+    elsif (tok = ss.scan /./)
+      buf << tok # 落ち穂拾い
+    end
+  end
+
+  src.replace(buf.strip)
+  buf
+end

data/lib/ec2cli/cli/functions.rb

@@ -0,0 +1,104 @@
+require 'json'
+
+def print_error(errmsg, opts = {})
+  errmsg = errmsg.join("\n") if errmsg.kind_of?(Array)
+  errmsg = errmsg.strip.split("\n").map {|i| "// #{i.strip}" }.join("\n")
+  errmsg += "\n\n" unless opts[:strip]
+  $stderr.puts errmsg
+end
+
+def print_rownum(data, opts = {})
+  rownum = data.to_i
+  msg = "// #{rownum} #{rownum > 1 ? 'rows' : 'row'} changed"
+  msg << " (%.2f sec)" % opts[:time] if opts[:time]
+  msg << "\n\n"
+  puts msg
+end
+
+def print_json(data, opts = {})
+  str = nil
+
+  if data.kind_of?(Array) and opts[:inline]
+    str = "[\n"
+
+    data.each_with_index do |item, i|
+      str << "  #{item.to_json}"
+      str << ',' if i < (data.length - 1)
+      str << "\n"
+    end
+
+    str << "]"
+  else
+    if data.kind_of?(Array) or data.kind_of?(Hash)
+      str = JSON.pretty_generate(data)
+    else
+      str = data.to_json
+    end
+  end
+
+  str.sub!(/(?:\r\n|\r|\n)*\Z/, "\n")
+
+  if opts[:show_rows] and data.kind_of?(Array)
+    str << "// #{data.length} #{data.length > 1 ? 'rows' : 'row'} in set"
+    str << " (%.2f sec)" % opts[:time] if opts[:time]
+    str << "\n"
+  end
+
+  str << "\n"
+  puts str
+end
+
+def print_version
+  puts "ec2cli #{Version}"
+end
+
+def evaluate_command(driver, cmd_arg)
+  cmd, arg = cmd_arg.split(/\s+/, 2).map {|i| i.strip }
+  arg = nil if (arg || '').strip.empty?
+
+  r = /\A#{Regexp.compile(cmd)}/i
+
+  commands = {
+    'help' => lambda {
+      print_help
+    },
+
+    ['exit', 'quit'] => lambda {
+      exit 0
+    },
+
+    'debug' => lambda {
+      if arg
+        r_arg = /\A#{Regexp.compile(arg)}/i
+
+        if r_arg =~ 'true'
+          driver.debug = true
+        elsif r_arg =~ 'false'
+          driver.debug = false
+        else
+          print_error('Invalid argument')
+        end
+      else
+        puts driver.debug
+      end
+    },
+
+    'version' => lambda {
+      print_version
+    }
+  }
+
+  cmd_name, cmd_proc = commands.find do |name, proc|
+    if name.kind_of?(Array)
+      name.any? {|i| r =~ i }
+    else
+      r =~ name
+    end
+  end
+
+  if cmd_proc
+    cmd_proc.call
+  else
+    print_error('Unknown command')
+  end
+end

data/lib/ec2cli/cli/help.rb

@@ -0,0 +1,40 @@
+def print_help
+  puts <<EOS
+##### Query #####
+
+DESC[RIBE] INSTANCES [WHERE name = '...' AND ...]
+  Describes one or more of your instances.
+
+DESC[RIBE] IMAGES [WHERE name = '...' AND ...]
+DESC[RIBE] ALL IMAGES [WHERE [Owner = {amazon|aws-marketplace|self} AND] name = '...' AND ...]
+  Describes the images.
+
+RUN INSTANCES image_id
+  Launches instances
+
+START INSTANCES id_or_name
+START INSTANCES (id_or_name, ...)
+  Starts an Amazon EBS-backed AMI that you've previously stopped.
+
+STOP INSTANCES id_or_name
+STOP INSTANCES (id_or_name, ...)
+  Stops an Amazon EBS-backed instance
+
+SET TAGS name = value, ...  [WHERE name = '...' AND ...]
+  Adds or overwrites one or more tags for the specified EC2 resource or resources.
+
+USE region_or_endpoint
+  changes an endpoint
+
+SHOW REGIONS
+  displays a region list
+
+##### Command #####
+
+.help                           displays this message
+.quit | .exit                   exits sdbcli
+.debug           (true|false)?  displays a debug status or changes it
+.version                        displays a version
+
+EOS
+end

data/lib/ec2cli/cli/options.rb

@@ -0,0 +1,35 @@
+require 'optparse'
+require 'ostruct'
+
+def parse_options
+  options = OpenStruct.new
+  options.access_key_id     = ENV['AWS_ACCESS_KEY_ID']
+  options.secret_access_key = ENV['AWS_SECRET_ACCESS_KEY']
+  options.ec2_endpoint_or_region = ENV['EC2_ENDPOINT'] || ENV['EC2_REGION'] || 'ec2.us-east-1.amazonaws.com'
+
+  # default value
+  options.debug       = false
+
+  ARGV.options do |opt|
+    opt.on('-k', '--access-key=ACCESS_KEY')          {|v| options.access_key_id          = v      }
+    opt.on('-s', '--secret-key=SECRET_KEY')          {|v| options.secret_access_key      = v      }
+    opt.on('-r', '--region=REGION_OR_ENDPOINT')      {|v| options.ddb_endpoint_or_region = v      }
+    opt.on('',   '--debug')                          {    options.debug                  = true   }
+
+    opt.on('-h', '--help') {
+      puts opt.help
+      puts
+      print_help
+      exit
+    }
+
+    opt.parse!
+
+    unless options.access_key_id and options.secret_access_key and options.ec2_endpoint_or_region
+      puts opt.help
+      exit 1
+    end
+  end
+
+  options
+end

data/lib/ec2cli/ec2-client.rb

@@ -0,0 +1,108 @@
+require 'cgi'
+require 'base64'
+require 'net/https'
+require 'openssl'
+require 'optparse'
+require 'pp'
+require 'rexml/document'
+
+require 'ec2cli/ec2-error'
+require 'ec2cli/ec2-endpoint'
+
+module EC2
+  class Client
+
+    API_VERSION = '2013-02-01'
+    SIGNATURE_VERSION = 2
+    SIGNATURE_ALGORITHM = :SHA256
+
+    attr_reader :endpoint
+    attr_reader :region
+    attr_accessor :debug
+
+    def initialize(accessKeyId, secretAccessKey, endpoint_or_region)
+      @accessKeyId = accessKeyId
+      @secretAccessKey = secretAccessKey
+      set_endpoint_and_region(endpoint_or_region)
+      @debug = false
+    end
+
+    def set_endpoint_and_region(endpoint_or_region)
+      @endpoint, @region = EC2::Endpoint.endpoint_and_region(endpoint_or_region)
+    end
+
+    def query(action, params = {})
+      params = {
+        :Action           => action,
+        :Version          => API_VERSION,
+        :Timestamp        => Time.now.getutc.strftime('%Y-%m-%dT%H:%M:%SZ'),
+        :SignatureVersion => SIGNATURE_VERSION,
+        :SignatureMethod  => "Hmac#{SIGNATURE_ALGORITHM}",
+        :AWSAccessKeyId   => @accessKeyId,
+      }.merge(params)
+
+      signature = aws_sign(params)
+      params[:Signature] = signature
+
+     if @debug
+        $stderr.puts(<<EOS)
+---request begin---
+#{params.pretty_inspect}
+---request end---
+EOS
+      end
+
+      https = Net::HTTP.new(@endpoint, 443)
+      https.use_ssl = true
+      https.verify_mode = OpenSSL::SSL::VERIFY_NONE
+
+      source = nil
+
+      https.start do |w|
+        req = Net::HTTP::Post.new('/',
+          'Host' => @endpoint,
+          'Content-Type' => 'application/x-www-form-urlencoded'
+        )
+
+        req.set_form_data(params)
+        res = w.request(req)
+
+        unless res.kind_of?(Net::HTTPOK)
+          raise EC2::Error, "#{res.code} #{res.message}"
+        end
+
+        source = res.body
+      end
+
+      if /<Response>\s*<Errors>\s*<Error>/ =~ source
+        errors = []
+
+        REXML::Document.new(source).each_element('//Errors/Error') do |element|
+          code = element.text('Code')
+          message = element.text('Message')
+          errors << "#{code}:#{message}"
+        end
+
+        rraise EC2::Error, errors.join(', ') unless errors.empty?
+      end
+
+      if @debug
+        $stderr.puts(<<EOS)
+---response begin---
+#{source}
+---response end---
+EOS
+      end
+
+      return source
+    end
+
+    def aws_sign(params)
+      params = params.sort_by {|a, b| a.to_s }.map {|k, v| "#{CGI.escape(k.to_s)}=#{CGI.escape(v.to_s)}" }.join('&')
+      string_to_sign = "POST\n#{@endpoint}\n/\n#{params}"
+      digest = OpenSSL::HMAC.digest(OpenSSL::Digest.const_get(SIGNATURE_ALGORITHM).new, @secretAccessKey, string_to_sign)
+      Base64.encode64(digest).gsub("\n", '')
+    end
+
+  end # Client
+end # EC2