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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
|
autoload :IPAddr, 'ipaddr'
require 'fileutils'
module LeapCli; module Commands
desc "Manage local virtual machines."
long_desc "This command provides a convient way to manage Vagrant-based virtual machines. If FILTER argument is missing, the command runs on all local virtual machines. The Vagrantfile is automatically generated in 'test/Vagrantfile'. If you want to run vagrant commands manually, cd to 'test'."
command [:local, :l] do |local|
local.desc 'Starts up the virtual machine(s)'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :start do |start|
start.flag(:basebox,
:desc => "The basebox to use. This value is passed to vagrant as the "+
"`config.vm.box` option. The value here should be the name of an installed box or a "+
"shorthand name of a box in HashiCorp's Atlas.",
:arg_name => 'BASEBOX',
:default_value => 'LEAP/wheezy'
)
start.action do |global_options,options,args|
vagrant_command(["up", "sandbox on"], args, options)
end
end
local.desc 'Shuts down the virtual machine(s)'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :stop do |stop|
stop.action do |global_options,options,args|
if global_options[:yes]
vagrant_command("halt --force", args)
else
vagrant_command("halt", args)
end
end
end
local.desc 'Destroys the virtual machine(s), reclaiming the disk space'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :destroy do |destroy|
destroy.action do |global_options,options,args|
if global_options[:yes]
vagrant_command("destroy --force", args)
else
vagrant_command("destroy", args)
end
end
end
local.desc 'Print the status of local virtual machine(s)'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :status do |status|
status.action do |global_options,options,args|
vagrant_command("status", args)
end
end
local.desc 'Saves the current state of the virtual machine as a new snapshot'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :save do |status|
status.action do |global_options,options,args|
vagrant_command("sandbox commit", args)
end
end
local.desc 'Resets virtual machine(s) to the last saved snapshot'
local.arg_name 'FILTER', :optional => true #, :multiple => false
local.command :reset do |reset|
reset.action do |global_options,options,args|
vagrant_command("sandbox rollback", args)
end
end
end
public
#
# returns the path to a vagrant ssh private key file.
#
# if the vagrant.key file is owned by root or ourselves, then
# we need to make sure that it owned by us and not world readable.
#
def vagrant_ssh_key_file
file_path = Path.vagrant_ssh_priv_key_file
Util.assert_files_exist! file_path
uid = File.new(file_path).stat.uid
if uid == 0 || uid == Process.euid
FileUtils.install file_path, '/tmp/vagrant.key', :mode => 0600
file_path = '/tmp/vagrant.key'
end
return file_path
end
protected
def vagrant_command(cmds, args, options={})
vagrant_setup(options)
cmds = cmds.to_a
if args.empty?
nodes = [""]
else
nodes = manager.filter(args)[:environment => "local"].field(:name)
end
if nodes.any?
vagrant_dir = File.dirname(Path.named_path(:vagrantfile))
exec = ["cd #{vagrant_dir}"]
cmds.each do |cmd|
nodes.each do |node|
exec << "vagrant #{cmd} #{node}"
end
end
execute exec.join('; ')
else
bail! "No nodes found. This command only works on nodes with ip_address in the network #{LeapCli.leapfile.vagrant_network}"
end
end
private
def vagrant_setup(options)
assert_bin! 'vagrant', 'Vagrant is required for running local virtual machines. Run "sudo apt-get install vagrant".'
if vagrant_version <= Gem::Version.new('1.0.0')
gem_path = assert_run!('vagrant gem which sahara')
if gem_path.nil? || gem_path.empty? || gem_path =~ /^ERROR/
log :installing, "vagrant plugin 'sahara'"
assert_run! 'vagrant gem install sahara -v 0.0.13'
end
else
unless assert_run!('vagrant plugin list | grep sahara | cat').chars.any?
log :installing, "vagrant plugin 'sahara'"
assert_run! 'vagrant plugin install sahara'
end
end
create_vagrant_file(options)
end
def vagrant_version
@vagrant_version ||= Gem::Version.new(assert_run!('vagrant --version').split(' ')[1])
end
def execute(cmd)
log 2, :run, cmd
exec cmd
end
def create_vagrant_file(options)
lines = []
netmask = IPAddr.new('255.255.255.255').mask(LeapCli.leapfile.vagrant_network.split('/').last).to_s
basebox = options[:basebox] || 'LEAP/wheezy'
if vagrant_version <= Gem::Version.new('1.1.0')
lines << %[Vagrant::Config.run do |config|]
manager.each_node do |node|
if node.vagrant?
lines << %[ config.vm.define :#{node.name} do |config|]
lines << %[ config.vm.box = "#{basebox}"]
lines << %[ config.vm.network :hostonly, "#{node.ip_address}", :netmask => "#{netmask}"]
lines << %[ config.vm.customize ["modifyvm", :id, "--natdnshostresolver1", "on"]]
lines << %[ config.vm.customize ["modifyvm", :id, "--name", "#{node.name}"]]
lines << %[ #{leapfile.custom_vagrant_vm_line}] if leapfile.custom_vagrant_vm_line
lines << %[ end]
end
end
else
lines << %[Vagrant.configure("2") do |config|]
manager.each_node do |node|
if node.vagrant?
lines << %[ config.vm.define :#{node.name} do |config|]
lines << %[ config.vm.box = "#{basebox}"]
lines << %[ config.vm.network :private_network, ip: "#{node.ip_address}"]
lines << %[ config.vm.provider "virtualbox" do |v|]
lines << %[ v.customize ["modifyvm", :id, "--natdnshostresolver1", "on"]]
lines << %[ v.name = "#{node.name}"]
lines << %[ end]
lines << %[ #{leapfile.custom_vagrant_vm_line}] if leapfile.custom_vagrant_vm_line
lines << %[ end]
end
end
end
lines << %[end]
lines << ""
write_file! :vagrantfile, lines.join("\n")
end
def pick_next_vagrant_ip_address
taken_ips = manager.nodes[:environment => "local"].field(:ip_address)
if taken_ips.any?
highest_ip = taken_ips.map{|ip| IPAddr.new(ip)}.max
new_ip = highest_ip.succ
else
new_ip = IPAddr.new(LeapCli.leapfile.vagrant_network).succ.succ
end
return new_ip.to_s
end
end; end
|
