59 lines
1.6 KiB
Ruby
59 lines
1.6 KiB
Ruby
require "tempfile"
|
|
|
|
module Vagrant
|
|
module Util
|
|
class Tempfile
|
|
# Utility function for creating a temporary file that will persist for
|
|
# the duration of the block and then be removed after the block finishes.
|
|
#
|
|
# @example
|
|
#
|
|
# Tempfile.create("arch-configure-networks") do |f|
|
|
# f.write("network-1")
|
|
# f.fsync
|
|
# f.close
|
|
# do_something_with_file(f.path)
|
|
# end
|
|
#
|
|
# @example
|
|
#
|
|
# Tempfile.create(["runner", "ps1"]) do |f|
|
|
# # f will have a suffix of ".ps1"
|
|
# # ...
|
|
# end
|
|
#
|
|
# @param [String, Array] name the prefix of the tempfile to create
|
|
# @param [Hash] options a list of options
|
|
# @param [Proc] block the block to yield the file object to
|
|
#
|
|
# @yield [File]
|
|
def self.create(name, options = {})
|
|
raise "No block given!" if !block_given?
|
|
|
|
options = {
|
|
binmode: true
|
|
}.merge(options)
|
|
|
|
# The user can specify an array where the first parameter is the prefix
|
|
# and the last parameter is the file suffix. We want to prefix the
|
|
# "prefix" with `vagrant-`, and this does that
|
|
if name.is_a?(Array)
|
|
basename = ["vagrant-#{name[0]}", name[1]]
|
|
else
|
|
basename = "vagrant-#{name}"
|
|
end
|
|
|
|
Dir::Tmpname.create(basename) do |path|
|
|
begin
|
|
f = File.open(path, "w+")
|
|
f.binmode if options[:binmode]
|
|
yield f
|
|
ensure
|
|
File.unlink(f.path) if File.file?(f.path)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|