kumo 0.0.1

data/bin/kumo

@@ -0,0 +1,135 @@
+#!/usr/bin/env ruby
+require 'gli'
+require 'kumo'
+require 'kumo/commands'
+
+include GLI::App
+version Kumo::VERSION
+
+if ARGV[0] == '--version'
+  puts Kumo::VERSION
+  exit
+end
+
+program_desc 'Manage workspaces in EC2'
+
+desc 'config file path'
+arg_name 'PATH'
+default_value '~/.kumorc'
+flag [:c, :'config-file']
+
+desc 'List instances in a set'
+command :list do |c|
+  c.desc 'keypair'
+  c.flag [:keypair]
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'type id'
+  c.flag [:'type-id']
+  c.desc 'AMI'
+  c.flag [:'image-id']
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.action do |global_options, options, args|
+    Kumo::Commands.list(global_options, options, args)
+  end
+end
+
+desc 'Launch an instance'
+command :launch do |c|
+  c.desc 'keypair'
+  c.flag [:keypair]
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'type id'
+  c.flag [:'type-id']
+  c.desc 'AMI'
+  c.flag [:'image-id']
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.desc 'security groups'
+  c.arg_name 'GROUP1,GROUP2,...'
+  c.flag [:g, :groups]
+  c.action do |global_options, options, args|
+    Kumo::Commands.launch(global_options, options, args)
+  end
+end
+
+desc 'Terminate an instance'
+arg_name '[INSTANCE_ID]'
+command :terminate do |c|
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.action do |global_options,options,args|
+    Kumo::Commands.terminate(global_options, options, args)
+  end
+end
+
+desc 'Start an instance'
+arg_name '[INSTANCE_ID]'
+command :start do |c|
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.action do |global_options, options, args|
+    Kumo::Commands.start(global_options, options, args)
+  end
+end
+
+desc 'Stop an instance'
+arg_name '[INSTANCE_ID]'
+command :stop do |c|
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.action do |global_options, options, args|
+    Kumo::Commands.stop(global_options, options, args)
+  end
+end
+
+desc 'DNS of an instance'
+arg_name '[INSTANCE_ID]'
+command :dns do |c|
+  c.desc 'region'
+  c.flag [:region]
+  c.desc 'name tag'
+  c.flag [:t, :tag]
+  c.desc 'configuration sets'
+  c.arg_name 'SET1,SET2,...'
+  c.flag [:s, :sets]
+  c.action do |global_options, options, args|
+    Kumo::Commands.dns(global_options, options, args)
+  end
+end
+
+pre do |global, command, options, args|
+  true
+end
+
+post do |global, command, options, args|
+end
+
+on_error do |exception|
+  true
+end
+
+exit run(ARGV)

data/lib/kumo/commands.rb

@@ -0,0 +1,125 @@
+module Kumo
+  # Commands for managing instances
+  class Commands
+    class << self
+      # List all instances with the same tag
+      #
+      # Example:
+      #
+      #     kumo list
+      def list(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options), parse_sets(options))
+        Kumo::Server.find_all.each { |server| puts "%-15s%-15s%-15s%-30s%-45s" % [server.id, server.tag, server.state, server.launch_time.localtime, server.public_dns] }
+      end
+
+      # Launch an instance
+      #
+      # Examples:
+      #
+      #     kumo launch
+      def launch(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options), parse_sets(options))
+        puts Kumo::Server.launch.id
+      end
+
+      # Terminate an instance
+      #
+      # Examples:
+      #
+      #     kumo terminate
+      def terminate(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options))
+        if args[0]
+          Kumo::Server.find_by_id(args[0]).terminate
+        else
+          last_server.terminate
+        end
+      end
+
+      # Start an instance
+      #
+      # Examples:
+      #
+      #     kumo start
+      def start(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options))
+        if args[0]
+          Kumo::Server.find_by_id(args[0]).start
+        else
+          last_server.start
+        end
+      end
+
+      # Stop an instance
+      #
+      # Examples:
+      #
+      #     kumo stop
+      def stop(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options))
+        if args[0]
+          Kumo::Server.find_by_id(args[0]).stop
+        else
+          last_server.stop
+        end
+      end
+
+      # Print the public DNS of an instance
+      #
+      # Examples:
+      #
+      #     kumo dns
+      def dns(global_options, options, args)
+        Kumo.initialize_config(File.expand_path(global_options[:'config-file']), sanitize_options(options))
+        if args[0]
+          puts Kumo::Server.find_by_id(args[0]).public_dns
+        else
+          puts last_server.public_dns
+        end
+      end
+
+      private
+
+      # Parse the sets
+      def parse_sets(options)
+        if options[:sets]
+          options[:sets].split(',').map { |s| s.to_sym }
+        else
+          []
+        end
+      end
+
+      # Parse the groups
+      #
+      # +groups_string+:: a string containing a list of comma separated groups
+      #
+      # Returns an +array+ of +symbols+
+      def parse_groups(groups_string)
+        groups_string.split(',').map { |g| g.to_sym }
+      end
+
+      # Sanitize the options, keeping only those that are defined
+      #
+      # +options+:: configuration options
+      #
+      # Returns a +hash+ suitable to initialize the main configuration
+      def sanitize_options(options)
+        sanitized = {}
+        [:keypair, :region, :'type-id', :tag].each do |o|
+          sanitized[o] = options[o] unless options[o].nil?
+        end
+
+        unless options[:groups].nil?
+          sanitized[:groups] = parse_groups(options[:groups])
+        end
+
+        sanitized
+      end
+
+      # Returns the last server that was launched/started
+      def last_server
+        Kumo::Server.find_all.last
+      end
+    end
+  end
+end

data/lib/kumo/config.rb

@@ -0,0 +1,67 @@
+require 'yaml'
+
+module Kumo
+  # Configuration for EC2
+  class Config
+    attr_reader :config
+
+    # New configuration
+    #
+    # +config_file_path+:: configuration file path (YAML)
+    # +options+:: instance options
+    # +sets+:: configuration sets (applied in order)
+    def initialize(config_file_path, options = {}, sets = [])
+      set_default_config
+      merge_config_file(config_file_path, sets)
+      merge_extra_config(options)
+      validate_config
+    end
+
+    private
+
+    # Validate the configuration
+    def validate_config
+      if @config[:'access-key-id'].nil? || @config[:'secret-access-key'].nil?
+        raise Exception, 'AWS "access key id" and "secret access key" must be included in the configuration file'
+      elsif @config[:keypair].nil?
+        raise Exception, 'AWS "keypair" is missing'
+      end
+    end
+
+    # Sets the default configuration. This is the base configuration,
+    # which will be merged with other configurations later.
+    def set_default_config
+      @config = {
+        :region => 'us-east-1',
+        :'type-id' => 't1.micro',
+        :'image-id' => 'ami-a29943cb',  # Ubuntu 12.04
+        :groups => ['kumo'],
+        :tag => 'kumo'
+      }
+    end
+
+    # Merge the configuration file, overriding the previous configuration
+    def merge_config_file(config_file_path, sets)
+      begin
+        config_file = YAML.load_file(config_file_path)
+      rescue
+        raise Exception, "Could not load config file: " + config_file_path
+      end
+
+      sets.unshift :default
+      sets.each do |set|
+        begin
+          @config.merge!(config_file[:sets][set])
+        rescue
+          raise Exception, "The set '" + set.to_s + "' was not found"
+        end
+      end
+    end
+
+    # Merge the extra configuration, overriding the previous configuration
+    def merge_extra_config(options)
+      options[:groups] = @config[:groups] if options[:groups] && options[:groups].empty?
+      @config.merge!(options)
+    end
+  end
+end

data/lib/kumo/server.rb

@@ -0,0 +1,112 @@
+require 'fog'
+
+module Kumo
+  # The server is an EC2 instance
+  class Server
+    include Comparable
+
+    # Find an instance by id
+    #
+    # +id+:: the id of the instance
+    def self.find_by_id(id)
+      Server.new(infra.servers.get(id))
+    end
+
+    # Find all instances
+    def self.find_all
+      infra.servers.select do |server|
+        (server.tags["Name"] == Kumo.config[:tag]) && (server.state != "terminated")
+      end.map do |server|
+        Server.new(server)
+      end.sort
+    end
+
+    # Launch an instance
+    def self.launch
+      Server.new(infra.servers.create(
+        :image_id => Kumo.config[:'image-id'],
+        :flavor_id => Kumo.config[:'type-id'],
+        :key_name => Kumo.config[:keypair],
+        :groups => Kumo.config[:groups],
+        :tags => { 'Name' => Kumo.config[:tag] }
+      ))
+    end
+
+    # Create a new instance based on a fog server
+    def initialize(fog_server)
+      @server = fog_server
+    end
+
+    # Get the ID of the instance
+    def id
+      @server.id
+    end
+
+    # Get the launch time of the instance
+    def launch_time
+      @server.created_at
+    end
+
+    # Get the running state of the instance
+    def state
+      @server.state
+    end
+
+    # Get the tag of the instance
+    def tag
+      @server.tags['Name']
+    end
+
+    # Get the public DNS of the instance
+    def public_dns
+      @server.dns_name
+    end
+
+    # Get the private DNS of the instance
+    def private_dns
+      @server.private_dns_name
+    end
+
+    # Terminate the instance
+    def terminate
+      @server.destroy
+    end
+
+    # Start the instance
+    def start
+      if @server.state == "stopped"
+        @server.start
+      end
+    end
+
+    # Stop the instance
+    def stop
+      if @server.state == "running"
+        @server.stop
+      end
+    end
+
+    # :nodoc:
+    def <=>(other_server)
+      if self.launch_time < other_server.launch_time
+        -1
+      elsif self.launch_time > other_server.launch_time
+        1
+      else
+        0
+      end
+    end
+
+    private
+
+    # The connection to EC2
+    def self.infra
+      @fog ||= Fog::Compute.new(
+        :provider => 'AWS',
+        :region => Kumo.config[:region],
+        :aws_access_key_id => Kumo.config[:'access-key-id'],
+        :aws_secret_access_key => Kumo.config[:'secret-access-key']
+      )
+    end
+  end
+end

data/lib/kumo/version.rb

@@ -0,0 +1,3 @@
+module Kumo
+  VERSION = '0.0.1'
+end

data/lib/kumo.rb

@@ -0,0 +1,13 @@
+require 'kumo/version'
+
+require 'kumo/config'
+require 'kumo/server'
+
+module Kumo
+  def self.config
+    @config
+  end
+  def self.initialize_config(file_path, options = {}, sets = [])
+    @config = Config.new(file_path, options, sets).config
+  end
+end

metadata

@@ -0,0 +1,135 @@
+--- !ruby/object:Gem::Specification
+name: kumo
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Michael V. O'Brien
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: gli
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: michael@michaelvobrien.com
+executables:
+- kumo
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/kumo
+- lib/kumo/commands.rb
+- lib/kumo/config.rb
+- lib/kumo/server.rb
+- lib/kumo/version.rb
+- lib/kumo.rb
+homepage: https://github.com/notbrien/kumo
+licenses: []
+post_install_message: 
+rdoc_options:
+- --title
+- kumo
+- -ri
+require_paths:
+- lib
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Manage workspaces in EC2
+test_files: []