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
|
require '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 node-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 do |local|
local.desc 'Starts up the virtual machine(s)'
local.arg_name 'node-filter', :optional => true #, :multiple => false
local.command :start do |start|
start.action do |global_options,options,args|
vagrant_command(["up", "sandbox on"], args)
end
end
local.desc 'Shuts down the virtual machine(s)'
local.arg_name 'node-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 'node-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 'node-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 'node-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 'node-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
def vagrant_ssh_key_file
file_path = File.expand_path('../../../vendor/vagrant_ssh_keys/vagrant.key', File.dirname(__FILE__))
Util.assert_files_exist! file_path
if File.new(file_path).stat.uid == Process.euid
# if the vagrant.key file is owned by ourselves, we need to make sure that it is not world readable
FileUtils.cp file_path, '/tmp/vagrant.key'
FileUtils.chmod 0600, '/tmp/vagrant.key'
file_path = '/tmp/vagrant.key'
end
return file_path
end
protected
def vagrant_command(cmds, args)
vagrant_setup
cmds = cmds.to_a
assert_config! 'provider.vagrant.network'
if args.empty?
nodes = [""]
else
nodes = manager.filter(args)[:local => true].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 #{provider.vagrant.network}"
end
end
private
def vagrant_setup
assert_bin! 'vagrant', 'run "sudo gem install vagrant"'
unless `vagrant gem which sahara`.chars.any?
log :installing, "vagrant plugin 'sahara'"
assert_run! 'vagrant gem install sahara'
end
create_vagrant_file
end
def execute(cmd)
log 2, :run, cmd
exec cmd
end
def create_vagrant_file
lines = []
netmask = IPAddr.new('255.255.255.255').mask(provider.vagrant.network.split('/').last).to_s
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 = "leap-wheezy"]
lines << %[ config.vm.box_url = "http://download.leap.se/leap-debian.box"]
lines << %[ config.vm.network :hostonly, "#{node.ip_address}", :netmask => "#{netmask}"]
lines << %[ config.vm.customize ["modifyvm", :id, "--natdnshostresolver1", "on"]]
lines << %[ #{leapfile.custom_vagrant_vm_line}] if leapfile.custom_vagrant_vm_line
lines << %[ end]
end
end
lines << %[end]
lines << ""
write_file! :vagrantfile, lines.join("\n")
end
def pick_next_vagrant_ip_address
taken_ips = manager.nodes[:local => true].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(provider.vagrant.network).succ.succ
end
return new_ip.to_s
end
end; end
|
