befog 0.1.0

data/bin/befog

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+require "rubygems"
+require "bundler/setup"
+
+# resolve bin path, ignoring symlinks
+require "pathname"
+bin_file = Pathname.new(__FILE__).realpath
+
+# add self to libpath
+$:.unshift File.expand_path("../../lib", bin_file)
+
+require "befog"
+
+Befog::CLI.run(*ARGV)

data/lib/befog.rb

@@ -0,0 +1 @@
+require "befog/cli"

data/lib/befog/cli.rb

@@ -0,0 +1,72 @@
+require "befog/commands/add"
+require "befog/commands/remove"
+require "befog/commands/start"
+require "befog/commands/stop"
+require "befog/commands/run"
+require "befog/commands/list"
+require "befog/commands/configure"
+
+module Befog
+  module CLI
+
+    COMMANDS = {
+      "add" => Befog::Commands::Add,
+      "remove" => Befog::Commands::Remove,
+      "start" => Befog::Commands::Start,
+      "stop" => Befog::Commands::Stop,
+      "run" => Befog::Commands::Run,
+      "list" => Befog::Commands::List,
+      "ls" => Befog::Commands::List,
+      "configure" => Befog::Commands::Configure,
+      "config" => Befog::Commands::Configure
+    }
+    
+    def self.run(subcommand=nil,*args)
+      if command = COMMANDS[subcommand]
+        begin
+          command.run(args)
+          
+        # TODO: use a befog-specific error class to 
+        # differentiate between expected exceptions
+        # (just display the error message) and un-
+        # expected (display the backtrace)
+        rescue => e
+          $stderr.puts "befog: #{e.message}"
+          $stderr.puts e.backtrace
+          exit(-1)
+        end
+      else
+        if subcommand
+          usage "'#{subcommand}' is not a supported command"
+        else
+          usage "No subcommand provided."
+        end
+      end
+    end
+    
+    def self.usage(message)
+      $stderr.puts "befog: #{message}"
+      $stderr.puts <<-eos
+        
+Usage: befog <subcommand> [<bank>] [<options>]
+
+The befog command allows you to manage your cloud servers from the command line. 
+
+Valid commands:
+
+    configure,  Configure a bank of servers
+    config
+    add         Provision new servers for a bank of servers
+    remove      De-provision servers for a bank of servers
+    start       Start a bank of servers
+    stop        Stop (suspend) a bank of servers
+    run         Run a command on each of a bank of servers
+    list,ls     List all servers with optional bank, region, or provider
+    
+You can get more options for any command with --help or -h.
+eos
+      exit(-1)
+      
+    end
+  end
+end

data/lib/befog/commands.rb

@@ -0,0 +1 @@
+require "befog/commands/cli"

data/lib/befog/commands/add.rb

@@ -0,0 +1,55 @@
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/help"
+
+module Befog
+  module Commands
+    
+    class Add
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Help
+  
+
+      command "befog add <bank>",
+        :default_to_help => true
+
+      option :count, 
+        :short => "-c COUNT",
+        :long  => "--count COUNT",
+        :required => true,
+        :description => "The number of machines to provision"
+        
+      # TODO: Add support for machine type
+    
+      def run        
+        count = options[:count].to_i
+        if count <= 0
+          $stderr.puts "Number must be an integer greater than 0."
+          return
+        end
+        servers = []
+        count.times do |i|
+          $stdout.puts "Provisioning server #{i+1} ..."
+          # TODO: Figure out how to give the server a name
+          servers << compute.servers.create(:image_id => bank["image"], 
+              :key_name => bank["keypair"], :region => bank["region"])
+        end
+        $stdout.puts "This may take a few minutes ..."
+        servers.each do |server|
+          server.wait_for { $stdout.puts "Still working ..." ; ready? }
+          self.servers << server.id
+        end
+        servers.each do |server|
+          $stdout.puts "Server #{server.id} is ready at #{server.dns_name}."
+        end
+        save
+      end
+    end
+  end
+end

data/lib/befog/commands/configure.rb

@@ -0,0 +1,80 @@
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/help"
+
+module Befog
+  
+  module Commands
+    
+    class Configure
+    
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Provider
+      include Mixins::Bank
+      include Mixins::Help
+      
+      command "befog configure [<bank>]",
+        :default_to_help => true
+        
+      option :key, 
+        :short => "-k KEY",
+        :long  => "--key KEY",
+        :description => "Your account key"
+
+      option :secret, 
+        :short => "-s SECRET",
+        :long  => "--secret SECRET",
+        :description => "Your account secret"
+      
+      option :provider,
+        :short => "-q PROVIDER",
+        :long => "--provider PROVIDER",
+        :description => "The provider provisioning a bank of servers"
+
+      option :region,
+        :short => "-r REGION",
+        :long => "--region REGION",
+        :description => "The region (datacenter) where a bank is provisioned"
+
+      option :image, 
+        :short => "-i IMAGE",
+        :long  => "--image IMAGE",
+        :description => "The image for provisioning pods"
+    
+      option :keypair, 
+        :short => "-x KEYPAIR",
+        :long  => "--keypair KEYPAIR",
+        :description => "The keypair name to use with SSH"
+
+      option :group, 
+        :short => "-g GROUP",
+        :long  => "--group GROUP",
+        :description => "The security group to use for new instances"
+
+      def self.run(args)
+        self.new(args).run
+      end
+      
+      def initialize(arguments)
+        process_arguments(arguments)
+      end
+      
+      def run
+        %w( key secret ).each do |key|
+          _key = key.to_sym
+          provider[key] = options[_key] if options[_key]
+        end
+        if options[:bank]
+          %w( provider region image keypair group ).each do |key|
+            _key = key.to_sym
+            bank[key] = options[_key] if options[_key]
+          end
+        end
+        save
+      end
+    end
+  end
+end

data/lib/befog/commands/list.rb

@@ -0,0 +1,76 @@
+require "fog"
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/server"
+require "befog/commands/mixins/help"
+
+module Befog
+  module Commands
+    
+    class List
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Server
+      include Mixins::Help
+
+      command "befog list [<bank>]",
+        :default_to_help => false
+
+      option :provider,
+        :short => "-q PROVIDER",
+        :long => "--provider PROVIDER",
+        :description => "The provider provisioning a bank of servers"
+
+
+      def run
+        if options[:bank]
+          list_bank(options[:bank])
+        elsif options[:provider]
+          list_provider(options[:provider])
+        else
+          list_all
+        end
+      end
+      
+      def list_all
+        $stdout.puts "All Servers"
+        banks.keys.select do |name|
+          list_bank(name)
+        end
+      end
+      
+      def list_provider(provider,indent="")
+        $stdout.puts "#{indent}- Provider: #{provider}"
+        indent += "  "
+        banks.select do |name,b|
+          if b["configuration"] and b["configuration"]["provider"] == provider
+            list_bank(name,indent)
+          end
+        end
+      end
+      
+      def list_bank(name, indent="")
+        out = []
+        out << "#{indent}- Bank: #{name}"
+        indent += "  "
+        configuration = banks[name]["configuration"]
+        out += %w( provider region image keypair type ).reduce([]) do |rval,key|        
+          rval << "#{key}: #{configuration[key]}" unless configuration[key].nil?
+          rval
+        end
+        out << "instances:"
+        banks[name]["servers"].each do |id|
+          c = compute(configuration["provider"])
+          out << "- #{c.servers.get(id).public_ip_address}"
+        end
+        $stdout.puts out.join("\n#{indent}")
+      end
+
+    end
+  end
+end

data/lib/befog/commands/mixins/bank.rb

@@ -0,0 +1,50 @@
+module Befog
+  module Commands
+    module Mixins
+      module Bank
+        
+        def banks
+          configuration["banks"] ||= {}
+        end
+        
+        def _bank
+          raise "You must specify a bank to do this" unless options[:bank]
+          banks[options[:bank]] ||= {}
+        end
+        
+        def bank
+          _bank["configuration"] ||= {}
+        end
+        
+        def servers
+          _bank["servers"] ||= []
+        end
+        
+        def process_arguments(arguments)
+          _bank,*rest = arguments
+          if _bank =~ /^\-/ or rest.empty?
+            super
+          else
+            super(rest)
+            options[:bank] = _bank
+            bank.each do |key,value|
+              options[key.to_sym] = value
+            end
+          end
+        end
+        
+        def run_for_each_server
+          if servers.empty?
+            $stderr.puts "No servers are in bank '#{options[:bank]}'."
+            return
+          end
+          servers.each do |id|
+            run_for_server(id)
+          end
+        end
+      
+      end
+    end
+  end
+end
+    

data/lib/befog/commands/mixins/command.rb

@@ -0,0 +1,80 @@
+require "optparse"
+
+module Befog
+  module Commands
+    module Mixins
+      module Command
+        
+        def self.included(target)
+          target.module_eval do
+            
+            class << self 
+              
+              def command(name,descriptor)
+                @command = (Struct.new(:name,:descriptor)).new(name,descriptor)
+              end
+              
+              def option(name,descriptor)
+                options << [name,descriptor]
+              end
+              
+              def options ; @options||=[] ; end
+
+              # TODO: Support multi-line descriptions?
+              def process_arguments(arguments)
+                results = {}; required = [] 
+                parser = OptionParser.new do |parser|
+                  parser.banner = "Usage: #{@command.name} [options]"
+                  options.each do |name,descriptor|
+                    if descriptor[:help]
+                      parser.on_tail(*descriptor.values_at(:short,:long,:description)) do |value|
+                        $stdout.puts parser
+                        exit(0)
+                      end
+                    else
+                      results[name] = descriptor[:default] if descriptor[:default]
+                      required << name if descriptor[:required]
+                      parser.on(*descriptor.values_at(:short,:long,:description)) do |value|
+                        results[name] = value
+                      end
+                    end
+                  end
+                end
+                if @command.descriptor[:default_to_help] and arguments.empty?
+                  $stdout.puts parser
+                  exit(0)
+                end
+                parser.parse!(arguments)
+                required.each do |name|
+                  unless results[name]
+                    $stderr.puts "Missing required option '#{name}'"
+                    $stderr.puts parser
+                    exit(-1)
+                  end
+                end
+                results
+              end
+              
+              def run(arguments)
+                new(arguments).run
+              end
+
+            end
+            
+          end
+        end
+        
+        attr_reader :options
+        
+        def initialize(arguments)
+          process_arguments(arguments)
+        end
+
+        def process_arguments(arguments)
+          @options = self.class.process_arguments(arguments)
+        end
+      end
+    end
+  end
+end
+            

data/lib/befog/commands/mixins/configurable.rb

@@ -0,0 +1,45 @@
+module Befog
+  module Commands
+    module Mixins
+      module Configurable
+            
+        def self.included(target)
+
+          target.module_eval do
+
+            option :path, 
+              :short => "-p PATH",
+              :long => "--path PATH",
+              :default => "~/.befog",
+              :description => "Path to the configuration file you want to use (defaults to '~/.befog')"
+
+            option :name,
+              :short => "-n NAME",
+              :long => "--name NAME",
+              :default => "default",
+              :description => "The name of this configuration (defaults to 'default')"
+
+          end
+        end
+
+        def configuration_path 
+          @configuration_path = File.expand_path(options[:path])
+        end
+          
+        def _configuration
+          @configuration ||= (YAML.load_file(configuration_path) rescue {})
+        end
+        
+        def configuration
+          _configuration[options[:name]] ||= {}
+        end
+
+        def save
+          File.open(File.expand_path(configuration_path),"w") { |f| YAML.dump(_configuration,f) }
+        end
+              
+
+      end
+    end
+  end
+end

data/lib/befog/commands/mixins/help.rb

@@ -0,0 +1,19 @@
+module Befog
+  module Commands
+    module Mixins
+      module Help
+        def self.included(target)
+          target.module_eval do
+            option :help,
+              :short => "-h",
+              :long => "--help",
+              :description => "Show this message",
+              :help => true
+          end
+        end
+        
+      end
+    end
+  end
+end
+            

data/lib/befog/commands/mixins/provider.rb

@@ -0,0 +1,40 @@
+require "befog/providers/aws"
+
+module Befog
+  module Commands
+    module Mixins
+      module Provider
+        
+        PROVIDERS = {
+          "aws" => Befog::Providers::AWS
+        }
+
+        def providers
+          configuration["providers"] ||= {}
+        end
+        
+        def provider
+          raise "You must specify a provider (with --provider or -q) to do this" unless options[:provider]
+          providers[options[:provider]] ||= {}
+        end
+      
+        def regions
+          provider["regions"] ||= {}
+        end
+
+        def region
+          raise "Please set the region using -r or --region." unless options[:region]
+          regions[options[:region]] ||= {}
+        end
+
+        def compute(provider=nil)
+          provider ||= options[:provider]
+          @compute ||= PROVIDERS[provider].compute(providers[provider])
+        end
+      
+      end
+    end
+  end
+end
+
+  

data/lib/befog/commands/mixins/server.rb

@@ -0,0 +1,13 @@
+module Befog
+  module Commands
+    module Mixins
+      module Server
+
+        def get_server(id)
+          compute.servers.get(id)
+        end
+
+      end
+    end
+  end
+end

data/lib/befog/commands/remove.rb

@@ -0,0 +1,57 @@
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/server"
+require "befog/commands/mixins/help"
+
+module Befog
+  module Commands
+    
+    class Remove
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Server
+      include Mixins::Help
+
+      command "befog remove <bank>",
+        :default_to_help => true
+
+      option :count, 
+        :short => "-c COUNT",
+        :long  => "--count COUNT",
+        :required => true,
+        :description => "The number of machines to de-provision"
+
+      def run
+        if servers.empty?
+          $stderr.puts "No servers are in bank '#{name}'."
+          return
+        end
+        count = options[:count].to_i
+        if count <= 0
+          $stderr.puts "Number must be an integer greater than 0."
+          return
+        end
+        if count > servers.size
+          $stderr.puts "Number must be less than or equal to the bank size of #{bank.size}."
+          return
+        end
+        count.times do |i|
+          id = servers.pop
+          $stdout.puts "Deprovisioning server #{id} ..."
+          begin
+            get_server(id).destroy
+          rescue => e
+            $stderr.puts "Error deprovisioning server #{id}"
+            servers.push id
+          end
+          save
+        end
+      end
+    end
+  end
+end

data/lib/befog/commands/run.rb

@@ -0,0 +1,45 @@
+require "fog"
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/server"
+require "befog/commands/mixins/help"
+
+
+module Befog
+  module Commands
+    
+    class Run
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Server
+      include Mixins::Help
+
+      command "befog run <bank>",
+        :default_to_help => true
+
+      option :command, 
+        :short => "-c COMMAND",
+        :long  => "--command COMMAND",
+        :required => true,
+        :description => "Command to run (required)"
+        
+      def run
+        run_for_each_server
+      end
+
+      def run_for_server(id)
+        address = get_server(id).public_ip_address
+        $stdout.puts "Running command '#{options[:command]}' for #{address} ..."
+        result = Fog::SSH.new(address, "root").run(options[:command]).first
+        $stdout.puts "[#{address}: STDOUT] #{result.stdout}"
+        $stderr.puts "[#{address}: STDERR] #{result.stderr}"
+      end
+
+    end
+  end
+end

data/lib/befog/commands/start.rb

@@ -0,0 +1,33 @@
+require "fog"
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/server"
+require "befog/commands/mixins/help"
+
+module Befog
+  module Commands
+    
+    class Start
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Server
+      include Mixins::Help
+
+      command "befog start <bank>",
+        :default_to_help => true
+
+      def run
+        run_for_each_server
+      end
+
+      def run_for_server(id)
+      end
+
+    end
+  end
+end

data/lib/befog/commands/stop.rb

@@ -0,0 +1,34 @@
+require "fog"
+require "befog/commands/mixins/command"
+require "befog/commands/mixins/configurable"
+require "befog/commands/mixins/bank"
+require "befog/commands/mixins/provider"
+require "befog/commands/mixins/server"
+require "befog/commands/mixins/help"
+
+module Befog
+  module Commands
+    
+  
+    class Stop
+  
+      include Mixins::Command
+      include Mixins::Configurable
+      include Mixins::Bank
+      include Mixins::Provider
+      include Mixins::Server
+      include Mixins::Help
+
+      command "befog stop <bank>",
+        :default_to_help => true
+
+      def run
+        run_for_each_server
+      end
+
+      def run_for_server(id)
+      end
+
+    end
+  end
+end

data/lib/befog/providers/aws.rb

@@ -0,0 +1,20 @@
+require "fog"
+Excon.defaults[:ssl_verify_peer] = false
+
+module Befog
+  
+  module Providers
+    
+    class AWS
+      
+      def self.compute(configuration)
+        Fog::Compute.new(:provider => "AWS", 
+          :aws_access_key_id => configuration["key"], 
+          :aws_secret_access_key => configuration["secret"])
+      end
+      
+    end
+    
+  end
+  
+end

metadata

@@ -0,0 +1,132 @@
+--- !ruby/object:Gem::Specification 
+name: befog
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Carlo Flores  carlo@spire.io
+- Dan Yoder dan@spire.io
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-04-03 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: fog
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 1
+        - 2
+        version: 1.1.2
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 2
+        - 7
+        version: "2.7"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 0
+        - 7
+        version: "0.7"
+  type: :development
+  version_requirements: *id003
+description: "\t\tThe befog gem allows you to manage your cloud servers\n\
+  \t\tdirectly from the command-line. \n"
+email: 
+- 
+- 
+executables: 
+- befog
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/befog/cli.rb
+- lib/befog/commands/add.rb
+- lib/befog/commands/configure.rb
+- lib/befog/commands/list.rb
+- lib/befog/commands/mixins/bank.rb
+- lib/befog/commands/mixins/command.rb
+- lib/befog/commands/mixins/configurable.rb
+- lib/befog/commands/mixins/help.rb
+- lib/befog/commands/mixins/provider.rb
+- lib/befog/commands/mixins/server.rb
+- lib/befog/commands/remove.rb
+- lib/befog/commands/run.rb
+- lib/befog/commands/start.rb
+- lib/befog/commands/stop.rb
+- lib/befog/commands.rb
+- lib/befog/providers/aws.rb
+- lib/befog.rb
+- bin/befog
+homepage: https://github.com/spire-io/befog
+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: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Cloud provisioning CLI
+test_files: []
+
+has_rdoc: