Contents

module VagrantShellCommander
  # Main plugin command
  class Command < Vagrant.plugin("2", "command")
    attr_accessor :env

    # Main entry point of this command
    #
    def execute
      cli_options = OptionManager.new.execute
      argv = parse_options(cli_options[:parser])
      
      return unless argv
      
      if [nil, ''].include? cli_options[:values][:cmd]
        env.ui.info cli_options[:parser]
      else
        with_target_vms(argv) do |machine|
          manage_machine(machine, cli_options) 
        end
      end
      0
    end

    private

    # Executes actions for a given machine
    #
    # @param [Vagrant::Machine] subject vm 
    # @param [Hash] Parser (:parser key) and parsed options (:values key)
    # @return nil
    #
    def manage_machine(machine, cli_options)
      if machine.state.id != :running
        env.ui.warn("Machine #{machine.name} is not running.")
        return
      end
      
      env.ui.success("#{machine.name}::")
      machine.action(:ssh_run, 
                     ssh_run_command: add_options_to_command(cli_options[:values][:cmd], 
                                                             cli_options[:values][:dir],
                                                             cli_options[:values][:user]))
    end

    # Adds the options to the given command
    #
    # @param cmd [String] Shell command
    # @param cwd [String] Optional working directory
    # @param user [String] Optional executing user
    # @return [String] Command with directory change if cwd is present and optional executing user
    #
    def add_options_to_command(cmd, cwd=nil, user=nil)
      cmd = "cd #{cwd} && #{cmd}" if cwd
      cmd = "sudo su - #{user} -c \"#{cmd}\"" if user
      cmd
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
vagrant-shell-commander-0.2.0	lib/vagrant-shell-commander/command.rb