guignol 0.1.2.1 → 0.3.0

data/lib/guignol/models/instance.rb

@@ -0,0 +1,244 @@
+
+require 'yaml'
+require 'fog'
+require 'active_support/core_ext/hash/slice'
+require 'guignol'
+require 'guignol/models/base'
+require 'guignol/models/volume'
+
+require 'pry'
+
+module Guignol::Models
+  class Instance < Base
+    class Error < Exception; end
+
+    def initialize(name, options)
+      super
+      subject.username = options[:username] if options[:username] && exists?
+    end
+
+    def fqdn
+      name and domain and "#{name}.#{domain}"
+    end
+
+
+    def create
+      log "server already exists" and return self if exist?
+        
+      create_options = Guignol::DefaultServerOptions.merge options.slice(:image_id, :flavor_id, :key_name, :security_group_ids, :user_data, :username)
+
+      # check for pre-existing volume(s). if any exist, add their AZ to the server's options
+      zones = create_options[:volumes].map { |name,volume_options| Volume.new(name, volume_options).availability_zone }.compact.uniq
+      if zones.size > 1
+        raise "pre-existing volumes volumes are in different availability zones"
+      elsif zones.size == 1
+        log "using AZ '#{zones.first}' since volumes already exist"
+        create_options[:availability_zone] = zones.first
+      end
+
+      log "building server..."
+      set_subject connection.servers.create(create_options)
+      setup
+      log "created as #{subject.dns_name}"
+
+      return self
+    rescue Exception => e
+      log "error while creating", :error => e
+      destroy
+      raise
+    end
+
+
+    def start
+      log "server doesn't exist (ignoring)" and return unless exist?
+      wait_for_state_transitions
+
+      if subject.state != "stopped"
+        log "server #{subject.state}."
+      else
+        log "starting server..."
+        subject.start
+        setup
+        log "server started"
+      end
+      return self
+    rescue Exception => e
+      log "error while starting", :error => e
+      stop
+      raise
+    end
+
+    def stop
+      wait_for_state_transitions
+      reload
+      if !exist?
+        log "server doesn't exist (ignoring)."
+      elsif subject.state != "running"
+        log "server #{subject.state}."
+      else
+        log "stopping server..."
+        remove_dns
+        subject.stop
+        wait_for_state 'stopped', 'terminated'
+      end
+      return self
+    end
+
+
+    def destroy
+      log "server doesn't exist (ignoring)." and return self unless exist?
+
+      log "tearing server down..."
+      remove_dns
+      subject.destroy
+      wait_for_state 'stopped', 'terminated', 'nonexistent'
+      # FIXME: remove tags here
+      set_subject nil
+      return self
+    end
+
+
+    def update_dns
+      return unless options[:domain]
+      if subject.nil? || subject.state != 'running'
+        log "server not running, not updating DNS"
+        return
+      end
+      log "updating DNS"
+      
+      unless dns_zone
+        log "DNS zone does not exist"
+        return self
+      end
+
+      if record = dns_record
+        if dns_record_matches?(record)
+          log "DNS record already exists"
+          return self
+        else
+          log "warning, while creating, DNS record exists but points to wrong server (fixing)"
+          record.destroy
+        end
+      end
+
+      dns_zone.records.create(:name => fqdn, :type => 'CNAME', :value => subject.dns_name, :ttl => 5)
+      log "#{fqdn} -> #{subject.dns_name}"
+      return self
+    end
+
+
+  private
+
+
+    def default_options
+      { :volumes => {} }
+    end
+
+
+    def domain
+      options[:domain]
+    end
+
+
+    def update_tags
+      log "updating server tags"
+      tags = { 'Name' => name, 'Domain' => domain, 'UUID' => uuid }
+      response = connection.create_tags(subject.id, tags)
+      raise Error.new("updating server tags") unless response.status == 200
+      return self
+    end
+
+
+    def update_root_volume_tags
+      log "updating root volume tags"
+      tags = { 'Name' => "#{name}-root", 'UUID' => uuid }
+
+      # we assume the root volume is the first in the block device map
+      if blockdev = subject.block_device_mapping.first
+        root_volume_id = blockdev['volumeId']
+        response = connection.create_tags(root_volume_id, tags)
+        raise Error.new("updating root volume tags") unless response.status == 200
+      end
+      return self
+    end
+
+
+    def update_volumes
+      options[:volumes].each_pair do |name,options|
+        options[:availability_zone] = subject.availability_zone
+        Volume.new(name, options).attach(subject.id)
+      end
+    end
+
+
+    # shared between create and start
+    def setup
+      update_tags
+      log "waiting for public dns to be set up..."
+      wait_for { subject.dns_name }
+      update_dns
+      update_volumes
+      update_root_volume_tags
+      wait_for_state 'running'
+      return self
+    end
+
+
+    def remove_dns
+      return unless domain
+      log "removing dns record"
+
+      unless dns_zone
+        log "dns zone does not exist"
+        return self
+      end
+
+      if record = dns_record
+        unless dns_record_matches?(record)
+          log "warning, while removing, DNS record exist but does not point to the current server"
+        end
+        record.destroy
+      end
+
+      return self
+    end
+
+
+    def wait_for_state_transitions
+      return unless exists?
+      return unless %w(stopping pending).include? subject.state
+      log "waiting for state transition from '#{subject.state}' to complete"
+      wait_for { subject.state != 'pending' }  if subject.state != 'pending'
+      wait_for { subject.state != 'stopping' } if subject.state != 'stopping'
+    end
+
+
+    def dns_connection
+      @@dns_connection ||= Fog::DNS.new(:provider => :aws)
+    end
+
+
+    def dns_zone
+      @dns_zone ||= dns_connection.zones.find { |zone| zone.domain == domain }
+    end
+
+    def dns_record
+      dns_zone.records.find { |record| record.name == fqdn }
+    end      
+
+    def dns_record_matches?(record)
+      !!record and record.value.any? { |dns_name| dns_name == subject.dns_name }
+    end
+
+
+
+
+    # walks the connection for matching servers, return 
+    # either the found server of nil
+    def find_subject
+      connection.servers.
+        select { |s| s.state != 'terminated' }.
+        find { |s| s.tags['UUID'] == uuid }
+    end
+  end
+end

data/lib/guignol/models/volume.rb

@@ -0,0 +1,91 @@
+
+require 'fog'
+require 'active_support/core_ext/hash/slice'
+require 'guignol'
+require 'guignol/models/base'
+
+
+module Guignol::Models
+  class Volume < Base
+    class Error < Exception; end
+
+
+    def availability_zone
+      subject && subject.availability_zone
+    end
+
+
+    def create
+      log "volume already exists" and return self if exist?
+
+      log "creating volume"
+      create_options = Guignol::DefaultVolumeOptions.merge options.slice(:availability_zone, :size, :snapshot, :delete_on_termination)
+      set_subject connection.volumes.create(create_options)
+      update_tags
+
+      wait_for_state 'available'
+      return self
+    rescue Exception => e
+      log "error while creating (#{e.class.name})"
+      destroy
+      raise
+    end
+
+
+    def destroy
+      return self unless exist?
+
+      log "destroying volume"
+      subject.destroy
+      wait_for_state 'deleted'
+      # FIXME: remove tags here
+
+      set_subject nil
+      return self
+    end
+
+
+    def attach(server_id)
+      exist? or create
+      subject.reload
+
+      if subject.server_id == server_id
+        if subject.device == options[:dev]
+          log "volume already attached"
+          return
+        else
+          log "error: volume attached to device #{subject.device} instead of options[:dev]"
+          raise Error.new('already attached')
+        end
+      end
+
+      response = connection.attach_volume(server_id, subject.id, options[:dev])
+      response.status == 200 or raise Error.new('failed to attach volume')
+      update_tags
+      return self
+    end
+
+
+    private
+
+
+    def find_subject
+      connection.volumes.
+        select { |s| %w(in-use available).include?(s.state) }.
+        find { |s| s.tags['UUID'] == uuid }
+    end
+
+
+    def update_tags
+      log "updating tags"
+      tags = { 'Name' => name, 'UUID' => uuid }
+      response = connection.create_tags(subject.id, tags)
+      unless response.status == 200
+        log "updating tags failed"
+        destroy and raise Error.new('updating tags failed')
+      end
+    end
+
+
+  end
+end

data/lib/guignol/shell.rb

@@ -1,50 +1,35 @@
-# Copyright (c) 2012, HouseTrip SA.
-# All rights reserved.
-# 
-# Redistribution and use in source and binary forms, with or without
-# modification, are permitted provided that the following conditions are met: 
-# 
-# 1. Redistributions of source code must retain the above copyright notice, this
-#    list of conditions and the following disclaimer. 
-# 2. Redistributions in binary form must reproduce the above copyright notice,
-#    this list of conditions and the following disclaimer in the documentation
-#    and/or other materials provided with the distribution. 
-# 
-# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
-# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
-# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
-# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
-# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
-# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
-# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
-# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
-# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
-# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-# 
-# The views and conclusions contained in the software and documentation are those
-# of the authors and should not be interpreted as representing official policies, 
-# either expressed or implied, of the authors.
-
-require 'term/ansicolor'
-require 'guignol/commands'
+require 'thor'
 
 module Guignol
-  class Shell
-    include Singleton
+  class Shell < Thor
+    def help(*args)
+      shell.say
+      shell.say "Guignol -- manipulate EC2 instances from your command line.", :cyan
+      shell.say
+      super
+    end
+
+    def self.start
+      super(ARGV, :shell => shared_shell)
+    end
 
-    def execute(command_name, *argv)
-      command_name ||= 'help'
-      unless command = Commands::Map[command_name]
-        Commands::Help.new.run
-        die "no such command '#{command_name}'."
-      end
-      command.new(*argv).run
-      exit 0
+    def self.shared_shell
+      @shared_shell ||= Thor::Base.shell.new
     end
 
-    def die(message)
-      puts Term::ANSIColor.red("fatal: #{message}")
-      exit 1
+    def self.exit_on_failure?
+      true
     end
   end
 end
+
+
+require 'guignol/commands/create'
+require 'guignol/commands/kill'
+require 'guignol/commands/start'
+require 'guignol/commands/stop'
+require 'guignol/commands/list'
+require 'guignol/commands/uuid'
+require 'guignol/commands/fix_dns'
+require 'guignol/commands/clone'
+require 'guignol/commands/execute'

data/lib/guignol/tty_spinner.rb

@@ -1,31 +1,6 @@
-# Copyright (c) 2012, HouseTrip SA.
-# All rights reserved.
-# 
-# Redistribution and use in source and binary forms, with or without
-# modification, are permitted provided that the following conditions are met: 
-# 
-# 1. Redistributions of source code must retain the above copyright notice, this
-#    list of conditions and the following disclaimer. 
-# 2. Redistributions in binary form must reproduce the above copyright notice,
-#    this list of conditions and the following disclaimer in the documentation
-#    and/or other materials provided with the distribution. 
-# 
-# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
-# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
-# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
-# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
-# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
-# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
-# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
-# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
-# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
-# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-# 
-# The views and conclusions contained in the software and documentation are those
-# of the authors and should not be interpreted as representing official policies, 
-# either expressed or implied, of the authors.
 
 require 'singleton'
+require 'guignol'
 
 module Guignol
   class TtySpinner
@@ -38,9 +13,11 @@ module Guignol
     end
 
     def spin!
-      $stderr.write(Chars[@state % Chars.size] + "\r")
-      $stderr.flush
-      @state += 1
+      if $stderr.tty? && !Guignol.env.test?
+        $stderr.write(Chars[@state % Chars.size] + "\r")
+        $stderr.flush
+        @state += 1
+      end
       Thread.pass
     end