3d2390fc94
This commit basically grepped the code base for all uses of Dir.mktmpdir and Tempfile.new/open and ensures the value is unique within the code base and also prefixed with `vagrant-`. Previously, most invocations of these commands simply used "vagrant", thus making them indistinguishable when trying to identify leaks.
39 lines
982 B
Ruby
39 lines
982 B
Ruby
require "fileutils"
|
|
|
|
require_relative "../../../../lib/vagrant/action/general/package"
|
|
|
|
module VagrantPlugins
|
|
module ProviderVirtualBox
|
|
module Action
|
|
class PackageSetupFolders
|
|
include Vagrant::Util::Presence
|
|
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
env["package.output"] ||= "package.box"
|
|
env["package.directory"] ||= Dir.mktmpdir("vagrant-package-", env[:tmp_path])
|
|
|
|
# Match up a couple environmental variables so that the other parts of
|
|
# Vagrant will do the right thing.
|
|
env["export.temp_dir"] = env["package.directory"]
|
|
|
|
Vagrant::Action::General::Package.validate!(
|
|
env["package.output"], env["package.directory"])
|
|
|
|
@app.call(env)
|
|
end
|
|
|
|
def recover(env)
|
|
dir = env["package.directory"]
|
|
if File.exist?(dir)
|
|
FileUtils.rm_rf(dir)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|