vagrant-ec-sandbox 0.1.0.Preview

data/lib/vagrant-ec-sandbox/action/wildfly.rb

@@ -0,0 +1,79 @@
+require_relative "../util/sandbox/get_user_processes"
+require_relative "../dir_structure"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Action
+      module Wildfly
+        def self.logger(class_name)
+          Log4r::Logger.new("vagrantplugins::ecsandbox::action::wildfly::#{class_name}")
+        end
+
+        def self.wildfly_home_path(session)
+          session.dir_structures[:sandbox].internal(:app, :as, "wildfly-8.1.0.Final")
+        end
+
+        def self.wait_host(env)
+          found = false
+          env[:ui].output("Waiting for Host Controller...")
+
+          until found
+            processes = Util::Sandbox::GetUserProcesses.new("wildfly").execute(env)
+            for proc in processes
+              found = (proc[:info][:type] == :wildfly_host_controller)
+              break if found
+            end
+
+            sleep 5
+          end
+          env[:ui].success("Host Controller started.")
+        end
+
+        def self.wait_server(env, server_names, opts=nil)
+          opts = {
+            wait_for_any:false
+          }.merge(opts || {})
+          server_names = server_names || []
+          wait_for_any = opts[:wait_for_any] || false
+
+          if wait_for_any
+            env[:ui].output("Waiting for one of the servers " << server_names.join(", ") << "...")
+          else
+            env[:ui].output("Waiting for servers " << server_names.join(", ") << "...")
+          end
+
+          pending = server_names.clone
+          last_started_server = nil
+
+          while !pending.empty?
+            processes = Util::Sandbox::GetUserProcesses.new("wildfly").execute(env)
+
+            for proc in processes
+              if proc[:info][:type] == :wildfly_server
+                if pending.include?(proc[:info][:server_name])
+                  last_started_server = proc[:info][:server_name]
+                  env[:ui].detail("Server #{last_started_server} started.")
+                  pending.delete(last_started_server)
+                end
+              end
+            end
+
+            if wait_for_any && last_started_server
+              break
+            end
+
+            sleep 5            
+          end
+
+          if wait_for_any
+            env[:ui].output("One or all servers started.")
+          else
+            env[:ui].success("All servers started.")
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/action.rb

@@ -0,0 +1,129 @@
+require "pathname"
+require "vagrant/action/builder"
+require "vagrant/action/builtin/call"
+require_relative "communicator_extender"
+require_relative "action/wildfly"
+require_relative "action/dist"
+require_relative "action/code"
+require_relative "action/init"
+
+
+module VagrantPlugins
+   module ECSandbox
+    module Action
+      autoload :SelectDb, File.expand_path("../action/select_db", __FILE__)
+      autoload :UseAppServer, File.expand_path("../action/use_app_server", __FILE__)
+      Init.autoload :InitVagrant, File.expand_path("../action/init/init_vagrant", __FILE__)
+      Init.autoload :EnvConfigConfirm, File.expand_path("../action/init/env_config_confirm", __FILE__)
+      Init.autoload :InitDir, File.expand_path("../action/init/init_dir", __FILE__)
+      Code.autoload :PrepareDist, File.expand_path("../action/code/prepare_dist", __FILE__)
+      Dist.autoload :GetViaBuild, File.expand_path("../action/dist/get_vbuild", __FILE__)
+      Dist.autoload :CheckArtifact, File.expand_path("../action/dist/check_artifact", __FILE__)
+      Dist.autoload :MakeContents, File.expand_path("../action/dist/make_contents", __FILE__)
+      Dist.autoload :ClearContents, File.expand_path("../action/dist/clear_contents", __FILE__)
+      Dist.autoload :GetViaNexus, File.expand_path("../action/dist/get_vnexus", __FILE__)
+      Dist.autoload :CheckWildflyInstaller, File.expand_path("../action/dist/check_wildfly_installer", __FILE__)
+      Dist.autoload :InstallWildfly, File.expand_path("../action/dist/install_wildfly", __FILE__)
+      Dist.autoload :ConfigEC, File.expand_path("../action/dist/config_ec", __FILE__)
+      Dist.autoload :DeployEcApp, File.expand_path("../action/dist/deploy_ecapp", __FILE__)
+      Wildfly.autoload :Remove, File.expand_path("../action/wildfly/remove", __FILE__)
+      Wildfly.autoload :GetIsInstalled, File.expand_path("../action/wildfly/get_isinstalled", __FILE__)
+      Wildfly.autoload :Start, File.expand_path("../action/wildfly/start", __FILE__)
+      Wildfly.autoload :StartAdmin, File.expand_path("../action/wildfly/start_admin", __FILE__)
+      Wildfly.autoload :Stop, File.expand_path("../action/wildfly/stop", __FILE__)
+      Wildfly.autoload :Status, File.expand_path("../action/wildfly/status", __FILE__)
+      Wildfly.autoload :Reload, File.expand_path("../action/wildfly/reload", __FILE__)
+      Wildfly.autoload :ReloadAdmin, File.expand_path("../action/wildfly/reload_admin", __FILE__)
+      Wildfly.autoload :RedirectDirs, File.expand_path("../action/wildfly/redirect_dirs", __FILE__)
+      Wildfly.autoload :SwitchDb, File.expand_path("../action/wildfly/switch_db", __FILE__)
+
+
+      include Vagrant::Action::Builtin
+
+
+      # This middleware sequence will install wildfly from distribution dir.
+      def self.action_install_wildfly
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use SelectDb
+          b.use UseAppServer
+          b.use Code::PrepareDist
+          b.use Dist::GetViaBuild
+          b.use Dist::CheckArtifact
+          b.use Call, Wildfly::GetIsInstalled do |env1, b2|
+            if env1[:result]
+              b2.use Wildfly::Stop
+              b2.use Wildfly::Remove
+            end
+          end
+          b.use Dist::ClearContents
+          b.use Dist::MakeContents
+          b.use Dist::CheckWildflyInstaller
+          b.use Dist::InstallWildfly
+          b.use Wildfly::StartAdmin
+          b.use Dist::ConfigEC
+          b.use Wildfly::ReloadAdmin
+          b.use Wildfly::RedirectDirs
+          b.use Dist::DeployEcApp
+          b.use Dist::ClearContents
+          b.use Wildfly::Reload
+        end
+      end
+
+      ## TEST ##
+      def self.action_test
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use UseAppServer
+          b.use Code::PrepareDist
+        end
+      end
+
+
+      def self.action_wildfly_status
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Wildfly::Status
+        end
+      end
+
+      def self.action_stop_wildfly
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use UseAppServer
+          b.use Wildfly::Stop
+        end
+      end
+
+      def self.action_start_wildfly
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use UseAppServer
+          b.use Wildfly::Start
+        end
+      end
+
+      def self.action_restart_wildfly
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use UseAppServer
+          b.use Wildfly::Stop
+          b.use Wildfly::Start
+        end
+      end
+
+      def self.action_config_db
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use UseAppServer
+          b.use SelectDb
+          b.use Wildfly::Stop
+          b.use Wildfly::StartAdmin
+          b.use Wildfly::SwitchDb
+          b.use Wildfly::Reload
+        end
+      end
+
+      def self.action_init
+        Vagrant::Action::Builder.new.tap do |b|
+          # b.use Init::EnvConfigConfirm
+          b.use Init::InitDir
+          b.use Init::InitVagrant
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/config_db.rb

@@ -0,0 +1,34 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class ConfigDb < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec config_db [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_config_db, 
+              use_default_db_selector: false,
+              auto_start_machine: true)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/install.rb

@@ -0,0 +1,34 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class Install < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec update [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_install_wildfly,
+              use_default_db_selector: true,
+              auto_start_machine: true)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/restart.rb

@@ -0,0 +1,33 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class Restart < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec restart [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_restart_wildfly, 
+              auto_start_machine: true)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/show_status.rb

@@ -0,0 +1,32 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class ShowStatus < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec status [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_wildfly_status)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/start.rb

@@ -0,0 +1,33 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class Start < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec start [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_start_wildfly, 
+              auto_start_machine: true)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/stop.rb

@@ -0,0 +1,32 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class Stop < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec stop [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_stop_wildfly)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/as/test.rb

@@ -0,0 +1,32 @@
+require 'optparse'
+
+require_relative "../base"
+require_relative "../../action"
+require_relative "../../communicator_extender"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      module AS
+        class Test < Base
+          def execute
+            opts = OptionParser.new do |o|
+              o.banner = "Usage: vagrant ec update [-h]"
+            end
+
+            # Parse the options
+            argv = parse_options(opts)
+            return if !argv
+            raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+            action_machine(Action.action_test)
+
+            # Success, exit status 0
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/base.rb

@@ -0,0 +1,52 @@
+require "vagrant/plugin/state_file"
+require "vagrant/registry"
+require_relative "../session"
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      class Base < Vagrant.plugin("2", :command)
+        # This is a helper for executing an action sequence with the proper
+        # environment hash setup so that the plugin specific helpers are
+        # in.
+        #
+        # @param [Object] callable the Middleware callable
+        # @param [Hash] env Extra environment hash that is merged in.
+        def action(callable, env=nil)
+          merged_env = {
+            sandbox_session: VagrantPlugins::ECSandbox::Session.default
+          }.merge(env || {})
+
+          @env.action_runner.run(callable, merged_env)
+        end
+
+        def action_machine(callable, env=nil)
+          with_target_vms(@argv) do |machine|
+              if machine.state.id != :running
+                if env[:auto_start_machine] == true
+                  @env.cli("up", machine.name.to_s)
+                else
+                  @env.ui.info("#{machine.name} is not running, use command 'vagrant up' to start the virtual machine .")
+                  next
+                end
+              end
+
+              merged_env = {
+                machine: machine,
+                sandbox_session: VagrantPlugins::ECSandbox::Session.create_machine(machine)
+                }.merge(env || {})
+
+              @env.action_runner.run(callable, merged_env)
+          end
+        end
+
+        def initialize(argv, env)
+          super(argv, env)
+
+          # Init session
+          VagrantPlugins::ECSandbox::Session.init_default(env)
+        end        
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/init.rb

@@ -0,0 +1,29 @@
+require 'optparse'
+
+require_relative "base"
+require_relative "../action"
+
+
+module VagrantPlugins
+  module ECSandbox
+    module Command
+      class Init < Base
+        def execute
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant ec init [-h]"
+          end
+
+          # Parse the options
+          argv = parse_options(opts)
+          return if !argv
+          raise Vagrant::Errors::CLIInvalidUsage, help: opts.help.chomp if argv.length > 0
+
+          action(Action.action_init)
+
+          # Success, exit status 0
+          0
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/command/root.rb

@@ -0,0 +1,101 @@
+require 'optparse'
+
+module VagrantPlugins
+   module ECSandbox
+    module Command
+      class Root < Vagrant.plugin("2", :command)
+        def self.synopsis
+          "manages ec sandboxes: init, update, destroy."
+        end
+
+        def initialize(argv, env)
+          super
+
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+          @subcommands.register(:init) do
+            require_relative "init"
+            Init
+          end
+
+          @subcommands.register(:install) do
+            require_relative "as/install"
+            AS::Install
+          end
+
+          @subcommands.register(:status) do
+            require_relative "as/show_status"
+            AS::ShowStatus
+          end
+
+          @subcommands.register(:stop) do
+            require_relative "as/stop"
+            AS::Stop
+          end
+
+          @subcommands.register(:start) do
+            require_relative "as/start"
+            AS::Start
+          end
+
+          @subcommands.register(:restart) do
+            require_relative "as/restart"
+            AS::Restart
+          end
+
+          # @subcommands.register(:test) do
+          #   require_relative "as/test"
+          #   AS::Test
+          # end
+
+          @subcommands.register(:config_db) do
+            require_relative "as/config_db"
+            AS::ConfigDb
+          end
+
+
+        end
+
+        def execute
+          if @main_args.include?("-h") || @main_args.include?("--help")
+            # Print the help for all the sub-commands.
+            return help
+          end
+
+          # If we reached this far then we must have a subcommand. If not,
+          # then we also just print the help and exit.
+          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+          return help if !command_class || !@sub_command
+          @logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")
+
+          # Initialize and execute the command class
+          command_class.new(@sub_args, @env).execute
+        end
+
+        # Prints the help out for this command
+        def help
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant ec <command> [<args>]"
+            o.separator ""
+            o.separator "Available subcommands:"
+
+            # Add the available subcommands as separators in order to print them
+            # out as well.
+            keys = []
+            @subcommands.each { |key, value| keys << key.to_s }
+
+            keys.sort.each do |key|
+              o.separator "     #{key}"
+            end
+
+            o.separator ""
+            o.separator "For help on any individual command run `vagrant ec COMMAND -h`"
+          end
+
+          @env.ui.info(opts.help, prefix: false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-ec-sandbox/communicator_extender.rb

@@ -0,0 +1,45 @@
+require "log4r"
+require "vagrant/plugin/manager"
+require "vagrant/util/platform"
+require_relative "util/name"
+require_relative "dir_structure"
+
+
+module Vagrant
+  module Plugin
+    module V2
+      class Communicator
+        def sh(session, contents, opts=nil, &block)
+          new_opts = {
+            file_name_base:nil,
+            category:nil,
+            sudo:nil
+          }.merge(opts || {})
+
+          session.dir_structures[:sandbox].using(
+            contents, 
+            file_name_base:new_opts[:file_name_base], 
+            category:new_opts[:category], 
+            use_unit_line_ending:true, 
+            communicator:self,
+            permissions:"755",
+            host_dir_structure:session.dir_structures[:host]) do |file|
+
+            sudo = new_opts[:sudo] || false
+            command = "sh #{file}";
+
+            if (sudo)
+              self.sudo(command, opts) do |type, data|
+                block.call(type, data) if block
+              end
+            else
+              self.execute(command, opts) do |type, data|
+                block.call(type, data) if block
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end