1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
require 'optparse'
require 'vagrant/util/template_renderer'
module VagrantPlugins
module CommandInit
class Command < Vagrant.plugin("2", :command)
def self.synopsis
"initializes a new Vagrant environment by creating a Vagrantfile"
end
def execute
options = {
force: false,
minimal: false,
output: "Vagrantfile",
template: ENV["VAGRANT_DEFAULT_TEMPLATE"]
}
opts = OptionParser.new do |o|
o.banner = "Usage: vagrant init [options] [name [url]]"
o.separator ""
o.separator "Options:"
o.separator ""
o.on("--box-version VERSION", "Version of the box to add") do |f|
options[:box_version] = f
end
o.on("-f", "--force", "Overwrite existing Vagrantfile") do |f|
options[:force] = f
end
o.on("-m", "--minimal", "Use minimal Vagrantfile template (no help comments). Ignored with --template") do |m|
options[:minimal] = m
end
o.on("--output FILE", String,
"Output path for the box. '-' for stdout") do |output|
options[:output] = output
end
o.on("--template FILE", String, "Path to custom Vagrantfile template") do |template|
options[:template] = template
end
end
# Parse the options
argv = parse_options(opts)
return if !argv
save_path = nil
if options[:output] != "-"
save_path = Pathname.new(options[:output]).expand_path(@env.cwd)
save_path.delete if save_path.exist? && options[:force]
raise Vagrant::Errors::VagrantfileExistsError if save_path.exist?
end
# Determine the template and template root to use
template_root = ""
if options[:template].nil?
options[:template] = "Vagrantfile"
if options[:minimal]
options[:template] = "Vagrantfile.min"
end
template_root = ::Vagrant.source_root.join("templates/commands/init")
end
# Strip the .erb extension off the template if the user passes it in
options[:template] = options[:template].chomp(".erb")
# Make sure the template actually exists
full_template_path = Vagrant::Util::TemplateRenderer.new(options[:template], template_root: template_root).full_template_path
if !File.file?(full_template_path)
raise Vagrant::Errors::VagrantfileTemplateNotFoundError, path: full_template_path
end
contents = Vagrant::Util::TemplateRenderer.render(options[:template],
box_name: argv[0] || "base",
box_url: argv[1],
box_version: options[:box_version],
template_root: template_root
)
if save_path
# Write out the contents
begin
save_path.open("w+") do |f|
f.write(contents)
end
rescue Errno::EACCES
raise Vagrant::Errors::VagrantfileWriteError
end
@env.ui.info(I18n.t("vagrant.commands.init.success"), prefix: false)
else
@env.ui.info(contents, prefix: false)
end
# Success, exit status 0
0
end
end
end
end
|
