crew 0.0.1

data/lib/crew/cli/contexts.rb

@@ -0,0 +1,76 @@
+module Crew
+  class CLI
+    class Contexts
+      include CLI::Processor
+
+      def process(args)
+        cmds = commands
+        global_opts = Trollop::options(args) do
+          stop_on cmds
+        end
+        dispatch(args)
+      end
+
+      def new(args)
+        context_name = args.shift
+        raise "You must specify a context name" if context_name.nil?
+        crew_home.context_new(context_name)
+      end
+
+      def list(args)
+        max_name_length = 0
+        crew_home.contexts do |context_name, context|
+          max_name_length = [max_name_length, context_name.size].max
+        end
+
+        crew_home.contexts do |context_name, context|
+          space_count = max_name_length - context_name.size + 1
+          puts "[#{context_name.color(:cyan)}]#{" " * space_count}Using #{context.adapter_name} adapter from #{context.source}"
+        end
+      end
+
+      def edit(args)
+        context_name = args.shift
+        raise "You must specify a context name" if context_name.nil?
+        crew_home.context_edit(context_name)
+      end
+
+      def info(args)
+        context_name = args.shift
+        raise "You must specify a context name" if context_name.nil?
+        crew_home.context_info(context_name)
+      end
+
+      def remove(args)
+        context_name = args.shift
+        raise "You must specify a context name" if context_name.nil?
+        crew_home.context_remove(context_name)
+      end
+
+      def help(args)
+        puts "# Crew #{Crew::VERSION}".color(:magenta)
+        puts
+        puts help_message
+      end
+
+      def help_message
+        <<-HEREDOCS
+#{"Contexts                       ".bright}
+#{"===============================".bright}
+
+#{"contexts help                  ".color(:blue).bright}provides help
+
+#{"contexts list                  ".color(:blue).bright}lists all the tasks currently installed
+#{"contexts new (context name)    ".color(:blue).bright}lists all the tasks available to be installed
+#{"contexts remove (context name) ".color(:blue).bright}updates the specified task (or all of them)
+#{"contexts edit (context name)   ".color(:blue).bright}lists all the tasks available to be installed
+        HEREDOCS
+      end
+
+      private
+      def commands
+        %w(help list new remove edit)
+      end
+    end
+  end
+end

data/lib/crew/cli/tasks.rb

@@ -0,0 +1,129 @@
+module Crew
+  class CLI
+    class Tasks
+      include CLI::Processor
+
+      def process(args)
+        cmds = commands
+        global_opts = Trollop::options(args) do
+          stop_on cmds
+        end
+        dispatch(args)
+      end
+
+      def new(args)
+        task_name = args.shift
+        raise "You must specify a task name" if task_name.nil?
+        crew_home.new(task_name)
+      end
+
+      def update(args)
+        opts = Trollop::options(args) do
+          opt :all, "Update all", short: "-a"
+        end
+        task_name = args.shift
+        raise "must specify task name or use -a to update all tasks" if task_name.nil? && !opts[:all]
+        crew_home.task_update(task_name)
+      end
+
+      def diff(args)
+        crew_home.task_diff
+      end
+
+      def available(args)
+        sources = Set.new
+        crew_home.available do |task, installed|
+          unless sources.include?(task.from_source)
+            puts "#{task.from_source}\n#{'=' * task.from_source.size}"
+            sources << task.from_source
+          end
+          print ' '
+          print installed ? "i".color(:blue) : " "
+          print task.untested? ? "!".color(:red) : " "
+          puts " %-30s : %s" % [task.name.bright, task.desc.color("#888888")]
+        end
+      end
+
+      def list(args)
+        max_name_length = 0
+        crew_home.task_list do |task|
+          max_name_length = [max_name_length, task.name.size].max
+        end
+        crew_home.task_list do |task|
+          install = if task.installed?
+            "i".color(:green)
+          else
+            ' '
+          end
+          test_status = if task.untested?
+            "x".color(:red).blink
+          elsif task.passing?
+            "√".color(:green)
+          else
+            'f'.color(:red)
+          end
+          print ' ' + install + test_status
+          puts " %-#{max_name_length}s : %s" % [task.name, task.desc]
+        end
+      end
+
+      def edit(args)
+        task_name = args.shift
+        raise "You must specify a task name" if task_name.nil?
+        crew_home.task_edit(task_name)
+      end
+
+      def info(args)
+        task_name = args.shift
+        raise "You must specify a task name" if task_name.nil?
+        crew_home.task_info(task_name)
+      end
+
+      def install(args)
+        task_name = args.shift
+        raise "You must specify a task name" if task_name.nil?
+        crew_home.task_install(task_name)
+      end
+
+      def install_all(args)
+        crew_home.task_install_all
+      end
+
+      def remove(args)
+        task_name = args.shift
+        raise "You must specify a task name" if task_name.nil?
+        crew_home.task_remove(task_name)
+      end
+
+      def help(args)
+        puts "# Crew #{Crew::VERSION}".color(:magenta)
+        puts
+        puts help_message
+      end
+
+      def help_message
+        <<-HEREDOCS
+#{"Tasks                          ".bright}
+#{"===============================".bright}
+
+#{"tasks help                     ".color(:blue).bright}provides help
+
+#{"tasks list                     ".color(:blue).bright}lists all the tasks currently installed
+#{"tasks available                ".color(:blue).bright}lists all the tasks available to be installed
+#{"tasks update (task name)       ".color(:blue).bright}updates the specified task (or all of them)
+#{"tasks diff (task name)         ".color(:blue).bright}shows the differences between the task from source and your local copy
+#{"tasks install [task name]      ".color(:blue).bright}installs a task
+#{"tasks remove [task name]    ".color(:blue).bright}removes a task
+#{"tasks new [task name]          ".color(:blue).bright}edits a task
+#{"tasks edit [task name]         ".color(:blue).bright}edits a task
+#{"tasks info [task name]         ".color(:blue).bright}shows information about a specific task
+        HEREDOCS
+      end
+
+      private
+      def commands
+        %w(help new install uninstall delete list edit info install install-all uninstall available update diff)
+      end
+    end
+  end
+end

data/lib/crew/context.rb

@@ -0,0 +1,137 @@
+require 'timeout'
+require 'set'
+require 'open3'
+require 'io/console'
+
+module Crew
+  class Context
+
+    class FailedCommandError < RuntimeError
+      attr_reader :command, :out, :err, :status
+
+      def initialize(command, out, err, status)
+        @command, @out, @err, @status = command, out, err, status
+        super("`#{@command}' failed with status=#{@status} out=#{@out.inspect} err=#{@err.inspect}")
+      end
+    end
+
+    class TestError < RuntimeError
+      def initialize(original_exception, stdout, stderr)
+        @original_exception, @stdout, @stderr = original_exception, stdout, stderr
+      end
+
+      def to_s
+        %|
+Error: #{@original_exception.message} (#{@original_exception.class})
+#{@original_exception.backtrace.join("\n  ")}
+
+Stderr:
+#{@stderr}
+
+Stdout:
+#{@stdout}
+        |.trim
+      end
+    end
+
+    include Util
+
+    attr_accessor :opts, :adapter_name, :source
+    attr_reader :hints, :home, :name
+
+    def initialize(home, name, file = nil, &blk)
+      @home = home
+      @name = name
+      @hints = Set.new
+      @shell_count = 0
+      load(file, &blk)
+    end
+
+    def load(file = nil, &blk)
+      DSL.new(self).load(file, &blk)
+    end
+
+    def logger
+      @home.logger
+    end
+
+    def enter_sudo_password
+      unless defined?(@sudo_password)
+        puts "Getting your sudo password (just this once):"
+        @sudo_password = ENV['SUDO_PASSWORD'] || STDIN.noecho(&:gets).chomp
+      end
+      nil
+    end
+
+    def run(name, *hints, &blk)
+      with_callbacks do
+        task = Task.new(@home, name) { run &blk }
+        task.default_hints = hints
+        task.run!
+      end
+    end
+
+    def with_callbacks
+      out = nil
+      begin
+        @shell_count += 1
+        if @shell_count == 1
+          logger.context @name do
+            start_shell
+            out = yield
+          end
+        else
+          out = yield
+        end
+      ensure
+        @shell_count -= 1
+        stop_shell if @shell_count == 0
+      end
+      out
+    end
+
+    def run_callbacks(type)
+      home.callbacks[type].each do |before|
+        run before.source_location.join(':'), &before
+      end
+    end
+
+    def start_shell
+      run_callbacks(:before)
+    end
+
+    def stop_shell
+      run_callbacks(:after)
+    end
+
+    def hint(h)
+      @hints << h.to_s
+    end
+
+    def sh(cmd, opts = {})
+      out, err, code = sh_with_code(cmd, opts)
+      assert code.zero?, "Command `#{cmd}' with out: #{out.inspect} err: #{err.inspect} code=#{code}"
+      out
+    end
+
+    def list
+      @home.each_task do |task|
+        yield task
+      end
+    end
+
+    def task(name)
+      @home.find_and_add_task(name)
+    end
+
+    def register
+      @started ||= begin
+        start
+        true
+      end
+    end
+
+    def start
+    end
+  end
+end

data/lib/crew/context/dsl.rb

@@ -0,0 +1,30 @@
+module Crew
+  class Context
+    class DSL
+      def initialize(context, &blk)
+        @context = context
+      end
+
+      def load(file = nil, &blk)
+        if file
+          instance_eval File.read(file), file, 1
+          @context.source = file
+        elsif blk
+          instance_eval(&blk)
+          @context.source = blk.source_location.join(":")
+        end
+      end
+
+      def adapter(name_or_module, opts = {})
+        @context.adapter_name = name_or_module.to_s
+        @context.extend name_or_module.is_a?(Module) ? name_or_module : Context.const_get(name_or_module.to_s)
+        @context.opts = opts
+      end
+
+      def hint(hint)
+        @context.hints << hint
+      end
+    end
+  end
+end
+

data/lib/crew/context/fusion.rb

@@ -0,0 +1,60 @@
+module Crew
+  class Context
+    module Fusion
+      include SSH
+
+      attr_accessor :snapshot_name
+
+      def vm_path
+        opts.fetch(:path)
+      end
+
+      def reset!(name)
+        path = vm_path
+        @home.run_in_context(Local, "Restoring to `#{name}'") do
+          vmrun_snapshots_revert(path, name)
+        end
+      end
+
+      def prepare_snapshot(name, force = false, &initializer)
+        path = vm_path
+        fusion_context = self
+        @home.run_in_context(Local, "Preparing snapshot `#{name}'") do
+          vmrun_snapshots_delete(path, name) if force
+          unless vmrun_snapshots(path).include?(name)
+            vmrun_snapshots_revert(path, "snapshot")
+            @home.run_in_context(fusion_context, "Setting up", &initializer) if initializer
+            vmrun_snapshots_create(path, name)
+          end
+        end
+      end
+
+      def start_shell
+        path = vm_path
+        reset!(@snapshot_name) if @snapshot_name
+        @home.run_in_context(Local, "Starting VM") do
+          vmrun_start path
+        end
+        opts[:host] = @home.run_in_context(Local, "Connecting to VM") do
+          poll("Getting ip address") { vmrun_getguestipaddress(path) }
+        end
+        run_callbacks(:before)
+      end
+
+      def stop_shell
+        if @_shell
+          @_shell.close
+          @_shell = nil
+        end
+        @current_dir = nil
+        run_callbacks(:after)
+        path = vm_path
+        #at_exit do
+        #  @home.run_in_context(Local, "Suspending VM") do
+        #    vmrun_suspend path
+        #  end
+        #end
+      end
+    end
+  end
+end

data/lib/crew/context/local.rb

@@ -0,0 +1,67 @@
+module Crew
+  class Context
+    module Local
+      def start_shell
+        @pwds ||= [Dir.pwd]
+        super
+      end
+
+      def stop_shell
+        super
+      end
+
+      def cd(dir)
+        @pwds.push dir
+        @home.logger.cd dir
+        if block_given?
+          begin
+            yield
+          ensure
+            @pwds.pop
+            @home.logger.cd @pwds.last
+          end
+        end
+      end
+
+      def sh_raw(cmd, stdin = nil)
+        stdout_str, stderr_str, status = Open3.capture3(cmd, opts)
+        [stdout_str, stderr_str, status.exitstatus]
+      end
+
+      def sh_with_code(cmd, opts = {})
+        runner = "bash -l -c %s"
+        stdin = nil
+        opts[:chdir] ||= File.expand_path(@pwds.last, @pwds[-2])
+        if opts[:sudo]
+          enter_sudo_password
+          stdin ||= @sudo_password
+          opts = { stdin_data: @sudo_password }
+          @home.logger.sh "SUDO #{cmd}"
+          cmd = "sudo -S #{runner % escape(cmd)}"
+        else
+          @home.logger.sh cmd
+          cmd = runner % escape(cmd)
+        end
+        sh_raw cmd, stdin
+      end
+
+      def save_data(data, target_path)
+        with_callbacks do
+          @home.logger.sh "Saving #{data.size} bytes to #{target_path}"
+          File.open(target_path, 'w') { |f| f << data }
+        end
+      end
+
+      def save_file(local_path, target_path)
+        with_callbacks do
+          @home.logger.sh "Saving #{local_path} bytes to #{target_path}"
+          system("cp #{escape(local_path)} #{escape(target_path)}")
+        end
+      end
+
+      def reconnect!
+        @context.reconnect!
+      end
+    end
+  end
+end