94 lines
3.1 KiB
Ruby
94 lines
3.1 KiB
Ruby
require "rubygems"
|
|
require "rubygems/dependency_installer"
|
|
|
|
begin
|
|
require "rubygems/format"
|
|
rescue LoadError
|
|
# rubygems 2.x
|
|
end
|
|
|
|
require "log4r"
|
|
require "vagrant/plugin/manager"
|
|
|
|
module VagrantPlugins
|
|
module CommandPlugin
|
|
module Action
|
|
# This action takes the `:plugin_name` variable in the environment
|
|
# and installs it using the RubyGems API.
|
|
class InstallGem
|
|
def initialize(app, env)
|
|
@app = app
|
|
@logger = Log4r::Logger.new("vagrant::plugins::plugincommand::installgem")
|
|
end
|
|
|
|
def call(env)
|
|
entrypoint = env[:plugin_entry_point]
|
|
plugin_name = env[:plugin_name]
|
|
version = env[:plugin_version]
|
|
|
|
# Determine the plugin name we'll look for in the installed set
|
|
# in order to determine the version and all that.
|
|
find_plugin_name = plugin_name
|
|
if plugin_name =~ /\.gem$/
|
|
# If we're installing from a gem file, determine the name
|
|
# based on the spec in the file.
|
|
pkg = if defined?(Gem::Format)
|
|
# RubyGems 1.x
|
|
Gem::Format.from_file_by_path(plugin_name)
|
|
else
|
|
# RubyGems 2.x
|
|
Gem::Package.new(plugin_name)
|
|
end
|
|
|
|
find_plugin_name = pkg.spec.name
|
|
version = pkg.spec.version
|
|
end
|
|
|
|
# Install the gem
|
|
plugin_name_label = plugin_name
|
|
plugin_name_label += " --version '#{version}'" if version
|
|
env[:ui].info(I18n.t("vagrant.commands.plugin.installing",
|
|
:name => plugin_name_label))
|
|
|
|
# TODO: support version, pre-release, custom sources
|
|
manager = Vagrant::Plugin::Manager.instance
|
|
plugin_spec = manager.install_plugin(
|
|
plugin_name, version: version, require: entrypoint)
|
|
|
|
# Tell the user
|
|
env[:ui].success(I18n.t("vagrant.commands.plugin.installed",
|
|
:name => plugin_spec.name,
|
|
:version => plugin_spec.version.to_s))
|
|
|
|
# If the plugin's spec includes a post-install message display it
|
|
post_install_message = plugin_spec.post_install_message
|
|
if post_install_message
|
|
if post_install_message.is_a?(Array)
|
|
post_install_message = post_install_message.join(" ")
|
|
end
|
|
|
|
env[:ui].info(I18n.t("vagrant.commands.plugin.post_install",
|
|
:name => plugin_spec.name,
|
|
:message => post_install_message.to_s))
|
|
end
|
|
|
|
# Continue
|
|
@app.call(env)
|
|
end
|
|
|
|
def recover(env)
|
|
# If any error happens, we uninstall it and remove it from
|
|
# the state file. We can only do this if we successfully installed
|
|
# the gem in the first place.
|
|
if @installed_plugin_name
|
|
new_env = env.dup
|
|
new_env.delete(:interrupted)
|
|
new_env[:plugin_name] = @installed_plugin_name
|
|
new_env[:action_runner].run(Action.action_uninstall, new_env)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|