engineyard-local 0.2.1 → 1.0.1

data/lib/engineyard-local/command.rb

@@ -1,6 +1,7 @@
 module Engineyard
   module Local
     module Command
+
       def self.all
         Engineyard::Local.locales[:commands].keys.map { |cmd| cmd.to_s }
       end
@@ -53,6 +54,14 @@ module Engineyard
   end
 end
 
+module Vagrant
+  class CLI
+    def help
+      Engineyard::Local::Command.dispatch(['help'])      
+    end
+  end
+end
+
 require "thor"
 # TODO autoload
 require "engineyard-local/command/helpers"
@@ -63,3 +72,17 @@ require "engineyard-local/command/exec"
 require "engineyard-local/command/list"
 require "engineyard-local/command/rails"
 require "engineyard-local/command/up"
+require "engineyard-local/command/status"
+require "engineyard-local/command/help"
+require "engineyard-local/command/update"
+require "engineyard-local/command/start"
+require "engineyard-local/command/stop"
+require "engineyard-local/command/terminate"
+
+module Engineyard
+  module Local
+    module Command
+      COMMANDS_WITH_NO_COOKBOOK_CHECK = [Engineyard::Local::Command::Terminate, Engineyard::Local::Command::Status, Engineyard::Local::Command::Update, Vagrant::Command::Suspend, Vagrant::Command::Halt]
+    end
+  end
+end

data/lib/engineyard-local/command/base.rb

@@ -1,13 +1,90 @@
+module Vagrant
+  module Command
+    class Base
+      include Engineyard::Local::Command::Helpers
+
+      def cookbook_status(scope = :partial, get_installed_version = true)
+        installed_cookbook_version = []
+        la_brea_data = []
+        cb_status = ''
+
+        begin
+          run( Vagrant::Action::Builder.new do
+            use( Engineyard::Local::Middleware::Cookbooks,:get_installed_version, installed_cookbook_version ) if get_installed_version
+            use( Engineyard::Local::Middleware::Cookbooks,:get_la_brea_data, la_brea_data )
+          end )
+        rescue Exception
+          cookbooks_status = "ERROR: Failure while determining cookbook status."
+        end
+
+        installed_cookbook_version = installed_cookbook_version.first
+        la_brea_data = la_brea_data.first || {}
+
+        if ( installed_cookbook_version == la_brea_data[:cookbooks_version] ) && ( scope == :full )
+          cb_status << I18n.t("vagrant.commands.status.cookbooks_current",
+            :version => installed_cookbook_version)
+        elsif ( installed_cookbook_version != la_brea_data[:cookbooks_version] )
+          cb_status << I18n.t("vagrant.commands.status.cookbooks_outdated",
+            :version => installed_cookbook_version,
+            :new_version => la_brea_data[:cookbooks_version],
+            :changelog => Nokogiri::HTML(la_brea_data[:changelog]).text)
+        end
+
+        cb_status
+      end
+
+      def initialize(argv, env)
+        @argv = argv
+        @env  = env
+        @logger = Log4r::Logger.new("vagrant::command::#{self.class.to_s.downcase}")
+        @cookbook_status_already_checked ||= :not_checked
+
+        begin
+          with_target_vms() do |vm|
+            if vm.created? && vm.state == :running
+              @env.ui.info(cookbook_status, :prefix => false) unless Engineyard::Local.cookbook_version_verified?
+              Engineyard::Local.cookbook_version_verified
+            else
+              cookbook_status(:partial, false)
+            end
+          end
+        rescue Errors::NoEnvironmentError
+          # Ignore this error. It will happen any time an ey-local/vagrant command is invoked
+          # in a directory that is not yet setup with a Vagrant environment.
+        end
+      end
+
+    end
+  end
+end
+
 module Engineyard
   module Local
     module Command
       class Base < Vagrant::Command::Base
-        include Command::Helpers
 
         attr_reader :env, :options
 
         def initialize(env, options)
           @env, @options = env, options
+          @cookbook_status_already_checked ||= :not_checked
+
+          unless Command::COMMANDS_WITH_NO_COOKBOOK_CHECK.include? self.class
+            begin
+              with_target_vms() do |vm|
+                if vm.created? && vm.state == :running
+                  @env.ui.info(cookbook_status, :prefix => false) unless Engineyard::Local.cookbook_version_verified?
+                  Engineyard::Local.cookbook_version_verified
+                else
+                  cookbook_status(:partial, false)
+                end
+              end
+            rescue Vagrant::Errors::NoEnvironmentError
+              # Ignore this error. It will happen any time an ey-local/vagrant command is invoked
+              # in a directory that is not yet setup with a Vagrant environment.
+            end
+          end
+	
         end
       end
     end

data/lib/engineyard-local/command/group.rb

@@ -1,8 +1,21 @@
+require 'engineyard-local/command'
+
 module Engineyard
   module Local
     module Command
       # TODO move to purely vagrant command based setup
       class Group < Thor
+
+        def self.printable_tasks(all = true, subcommand = false)
+          (all ? all_tasks : tasks).map do |_, task|
+            next if task.hidden?
+            item = []
+            item << banner(task, false, subcommand)
+            item << (task.description ? "# #{task.description.gsub(/ +/m,' ')}" : "")
+            item
+          end.compact
+        end
+
         # TODO this class level env crap will go away once
         #      these commands are moved out of thor to Vagrant commands
         def self.env=(env)
@@ -13,6 +26,25 @@ module Engineyard
           def env
             @@env
           end
+
+          def externally_register_vm(e)
+            eylocal_up_path = File.join(e.home_path,'eylocal-up')
+            lines = FileTest.exists?(eylocal_up_path) &&
+                    FileTest.file?(eylocal_up_path) &&
+                    FileTest.readable?(eylocal_up_path) ? File.read(eylocal_up_path).split : []
+            lines << e.cwd
+            File.open(eylocal_up_path, 'w+') {|fh| fh.write lines.join("\n")}
+          end
+
+          def externally_unregister_vm(e)
+            eylocal_up_path = File.join(e.home_path,'eylocal-up')
+            lines = FileTest.exists?(eylocal_up_path) &&
+                    FileTest.file?(eylocal_up_path) &&
+                    FileTest.readable?(eylocal_up_path) ? File.read(eylocal_up_path).split : []
+            lines.reject! {|l| l =~ /#{e.cwd}/}
+            File.open(eylocal_up_path, 'w+') {|fh| fh.write lines.join("\n")}
+          end
+
         end
 
         # NOTE this is generally a bad idea as it might break between Vagrant versions
@@ -32,6 +64,14 @@ module Engineyard
         # register the command group eylocal
         # register(*command_info[:eylocal])
 
+	map "-h" => :help
+	default_task :help
+
+        desc(*command_info[:help])
+	def help(*args)
+          Command::Help.new(env, options).exec(self, *args)
+	end
+
         # make sure rails is installed, run rails commands in the guest project share
         method_option "version", :type => :string, :default => nil
         desc(*command_info[:rails])
@@ -45,12 +85,20 @@ module Engineyard
         end
 
         # streamlining of Vagrant up
-        method_option "no-bundle", :type => :boolean, :default => nil
-        method_option "no-db-setup", :type => :boolean, :default => nil
-        method_option "silent", :type => :boolean, :default => nil
+
+        method_option "no-chef",      :type => :boolean, :default => nil
+        method_option "no-dna",       :type => :boolean, :default => nil
+        method_option "no-bundle",    :type => :boolean, :default => nil
+        method_option "no-db-setup",  :type => :boolean, :default => nil
+        method_option "silent",       :type => :boolean, :default => nil
         desc(*command_info[:up])
         def up(*args)
-          Command::Up.new(env, options).exec(*args)
+          Command::Up.new(env, options).exec(*args) && externally_register_vm(env)
+        end
+
+        desc(*command_info[:down])
+        def down(*args)
+          Command::VagrantAction.new(env, options).exec(:halt) && externally_unregister_vm(env)
         end
 
         method_option "environment", :type => :string, :default => "development"
@@ -73,13 +121,29 @@ module Engineyard
 
         desc(*command_info[:start])
         def start
-          Command::VagrantAction.new(env, options).exec(:resume)
+          Command::Start.new(env, options).exec()
         end
 
         desc(*command_info[:stop])
         def stop
-          Command::VagrantAction.new(env, options).exec(:suspend)
+          Command::Stop.new(env, options).exec()
+        end
+
+        desc(*command_info[:status])
+        def status(*args)
+          Command::Status.new(env, options).exec(env)
         end
+
+        desc(*command_info[:update])
+        def update(*args)
+          Command::Update.new(env, options).exec()
+        end
+
+        desc(*command_info[:terminate])
+        def terminate(*args)
+          Command::Terminate.new(env, options).exec()
+        end
+
       end
     end
   end

data/lib/engineyard-local/command/help.rb

@@ -0,0 +1,36 @@
+module Engineyard
+  module Local
+    module Command
+      class Help < Base
+
+        def exec(thor, *args)
+          list = Engineyard::Local::Command::Group.printable_tasks
+          Thor::Util.thor_classes_in(thor.class).each do |klass|
+            list += klass.printable_tasks(false)
+          end
+
+          truncate_output = true
+
+          if !args.empty?
+            list = list.select { |i| args.any? { |a| i.first =~ /^[\w\-]+\s+#{a}\b/ } }
+            truncate_output = false
+            list.sort!{ |a,b| a[0] <=> b[0] }.each do |h|
+              thor.shell.say(h[0], nil, true)
+              insert_linebreaks(h[1]).each do |line|
+                thor.shell.say("#{' ' * 4}#{line}", nil, true)
+              end
+            end
+          else
+            list.sort!{ |a,b| a[0] <=> b[0] }
+
+            thor.shell.say "Commands:"
+            thor.shell.print_table(list, :indent => 2, :truncate => truncate_output)
+            thor.shell.say
+          end
+
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/command/helpers.rb

@@ -2,6 +2,18 @@ module Engineyard
   module Local
     module Command
       module Helpers
+
+        # Stupid, simple formatter for text. Returns an array of lines.
+        def insert_linebreaks(txt, maxlength = 76, ensure_linebreaks = true)
+          "#{txt} ".scan(/.{0,#{maxlength}}\s+/).collect do |line|
+            if ensure_linebreaks
+              line =~ /\n$/ ? line : line + "\n"
+            else
+              line
+            end
+          end
+        end
+
         def run(middleware, opts={}, &block)
           with_target_vms do |vm|
             opts = merge_run_options(vm, opts)

data/lib/engineyard-local/command/start.rb

@@ -0,0 +1,26 @@
+module Engineyard
+  module Local
+    module Command
+      class Start < Base
+
+        # Wrap Vagrant's :resume so that it checks to see if the instance is suspended first, and gives helpful output if it is not.
+        def exec(*args)
+          with_target_vms do |vm|
+            if vm.created?
+              if ( vm.state == :saved )
+                vm.resume
+              elsif ( vm.state == :running )
+                env.ui.info( insert_linebreaks( I18n.t( "eylocal.start.running" ) ) )
+              else
+                env.ui.info( insert_linebreaks( I18n.t( "eylocal.start.not_startable" , :state => vm.state ) ) )
+              end
+            else
+              env.ui.info( I18n.t( "eylocal.start.not_created" ) )
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/command/status.rb

@@ -0,0 +1,33 @@
+require 'optparse'
+require 'nokogiri'
+
+module Engineyard
+  module Local
+    module Command
+      class Status < Base
+
+        def exec(*args)
+          state = nil
+          results = []
+
+          cb_status = nil
+
+          with_target_vms do |vm|
+            cb_status = cookbook_status(:full) if vm.state == :running
+
+            state = vm.state.to_s if !state
+            results << "#{vm.name.to_s.ljust(25)}#{vm.state.to_s.gsub("_", " ")}"
+          end
+
+          state = results.length == 1 ? state : "listing"
+
+          @env.ui.info(I18n.t("vagrant.commands.status.output",
+                              :states => results.join("\n"),
+                              :message => I18n.t("vagrant.commands.status.#{state}"),
+                              :alerts => insert_linebreaks( cb_status ).join ),
+                       :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/command/stop.rb

@@ -0,0 +1,26 @@
+module Engineyard
+  module Local
+    module Command
+      class Stop < Base
+
+        # Wrap Vagrant's :resume so that it checks to see if the instance is suspended first, and gives helpful output if it is not.
+        def exec(*args)
+          with_target_vms do |vm|
+            if vm.created?
+              if ( vm.state == :running )
+                vm.suspend
+              elsif ( vm.state == :saved )
+                env.ui.info( insert_linebreaks( I18n.t( "eylocal.stop.saved" ) ) )
+              else
+                env.ui.info( insert_linebreaks( I18n.t( "eylocal.stop.not_stoppable" , :state => vm.state ) ) )
+              end
+            else
+              env.ui.info( I18n.t( "eylocal.stop.not_created" ) )
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/command/terminate.rb

@@ -0,0 +1,26 @@
+module Engineyard
+  module Local
+    module Command
+      class Terminate < Base
+
+        # Wrap Vagrant's :destroy 
+        def exec(*args)
+          with_target_vms do |vm|
+            if vm.created?
+              env.ui.info( I18n.t( "eylocal.terminate.destroying" ) )
+              vm.destroy
+              File.unlink(cached_data_path) if FileTest.exist? cached_data_path
+            else
+              env.ui.info( I18n.t( "eylocal.terminate.not_created" ) )
+            end
+          end
+        end
+
+        def cached_data_path
+          "#{@env.root_path}/.ey_local_data"
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/command/up.rb

@@ -2,6 +2,19 @@ module Engineyard
   module Local
     module Command
       class Up < Base
+
+        # install the application bundle and create/migrate the database
+        # TODO check for the database.yml and fail gracefully
+        def self.setup_stack(options)
+          Vagrant::Action::Builder.new do
+            use Middleware::DNA unless options["no-dna"]
+            use Middleware::Chef unless options["no-chef"]
+            use Middleware::Bundle unless options["no-bundle"]
+            use Middleware::Rails::DB unless options["no-db-setup"]
+            use Middleware::Rails::Assets
+          end
+        end
+
         def box_defaults
           Local.config[:box_defaults]
         end
@@ -31,17 +44,20 @@ module Engineyard
           env.boxes.reload!
 
           env.ui.info(I18n.t("eylocal.up.root_path", :root_path => env.root_path))
-          if File.expand_path(env.root_path) != File.expand_path(Dir.pwd)
-            env.ui.info(I18n.t("eylocal.up.root_path_instructions", :root_path => env.root_path, :pwd => Dir.pwd))
+          if !pwd_matches_root_path
+            env.ui.info(I18n.t("eylocal.up.root_path_instructions",
+                               :root_path => env.root_path, :pwd => Dir.pwd))
           end
 
           # don't build a box if its already built
           with_target_vms do |vm|
             if vm.created?
               env.ui.info I18n.t("vagrant.commands.up.vm_created")
+              if ! ( vm.state == 'running' )
+                vm.start
+              end
             else
               up = Vagrant.actions.get(:up)
-
               # insert the tag before the box is in a state where it can't be saved eg running
               up.insert_before(Vagrant::Action::VM::CheckGuestAdditions, Middleware::Tag)
               up.insert_before(Vagrant::Action::VM::CheckGuestAdditions, Middleware::Network, @options)
@@ -52,11 +68,15 @@ module Engineyard
           end
 
           # finish the box setup
-          run setup_stack(options)
+          run self.class.setup_stack(options)
         end
 
         private
 
+        def pwd_matches_root_path
+          File.expand_path(env.root_path) == File.expand_path(Dir.pwd)
+        end
+
         # prefer the the base box included with the installer if it exists
         # otherwise use the default downloadable box
         def box_uri
@@ -70,17 +90,6 @@ module Engineyard
             box_defaults[:uri]
           end
         end
-
-        # install the application bundle and create/migrate the database
-        # TODO check for the database.yml and fail gracefully
-        def setup_stack(options)
-          Vagrant::Action::Builder.new do
-            use Middleware::DNA unless options["no-dna"]
-            use Middleware::Chef unless options["no-chef"]
-            use Middleware::Bundle unless options["no-bundle"]
-            use Middleware::Rails::DB unless options["no-db-setup"]
-          end
-        end
       end
     end
   end