require "log4r" require "vagrant-lxc/action" require "vagrant-lxc/driver" require "vagrant-lxc/sudo_wrapper" module Vagrant module LXC class Provider < Vagrant.plugin("2", :provider) attr_reader :driver def initialize(machine) @logger = Log4r::Logger.new("vagrant::provider::lxc") @machine = machine ensure_lxc_installed! machine_id_changed end def sudo_wrapper @shell ||= begin wrapper = @machine.provider_config.sudo_wrapper wrapper = Pathname(wrapper).expand_path(@machine.env.root_path).to_s if wrapper SudoWrapper.new(wrapper) end end def ensure_lxc_installed! unless system("which lxc-version > /dev/null 2>&1") begin sudo_wrapper.run("which", "lxc-version") rescue Vagrant::LXC::Errors::ExecuteError raise Errors::LxcNotInstalled end end end # If the machine ID changed, then we need to rebuild our underlying # container. def machine_id_changed id = @machine.id begin @logger.debug("Instantiating the container for: #{id.inspect}") @driver = Driver.new(id, self.sudo_wrapper) @driver.validate! rescue Driver::ContainerNotFound # The container doesn't exist, so we probably have a stale # ID. Just clear the id out of the machine and reload it. @logger.debug("Container not found! Clearing saved machine ID and reloading.") id = nil retry end end # @see Vagrant::Plugin::V2::Provider#action def action(name) # Attempt to get the action method from the Action class if it # exists, otherwise return nil to show that we don't support the # given action. action_method = "action_#{name}" return LXC::Action.send(action_method) if LXC::Action.respond_to?(action_method) nil end # Returns the SSH info for accessing the Container. def ssh_info # If the Container is not created then we cannot possibly SSH into it, so # we return nil. return nil if state == :not_created # Run a custom action called "fetch_ip" which does what it says and puts # the IP found into the `:machine_ip` key in the environment. env = @machine.action("fetch_ip") # If we were not able to identify the container's IP, we return nil # here and we let Vagrant core deal with it ;) return nil unless env[:machine_ip] { :host => env[:machine_ip], :port => @machine.config.ssh.guest_port } end def state state_id = nil state_id = :not_created if !@driver.container_name state_id = @driver.state if !state_id state_id = :unknown if !state_id short = state_id.to_s.gsub("_", " ") long = I18n.t("vagrant.commands.status.#{state_id}") Vagrant::MachineState.new(state_id, short, long) end def to_s id = @machine.id ? @machine.id : "new VM" "LXC (#{id})" end end end end