forked from jedi4ever/veewee
/
basebox.rb
84 lines (70 loc) · 3.14 KB
/
basebox.rb
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
require 'veewee'
require 'optparse'
require 'veewee/command/vagrant/ostypes'
require 'veewee/command/vagrant/templates'
require 'veewee/command/vagrant/list'
require 'veewee/command/vagrant/build'
require 'veewee/command/vagrant/destroy'
require 'veewee/command/vagrant/up'
require 'veewee/command/vagrant/halt'
require 'veewee/command/vagrant/ssh'
require 'veewee/command/vagrant/winrm'
require 'veewee/command/vagrant/define'
require 'veewee/command/vagrant/undefine'
require 'veewee/command/vagrant/export'
module Veewee
module Command
module Vagrant
class Basebox < ::Vagrant::Command::Base
def initialize(argv,env)
super
@main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
@subcommands = ::Vagrant::Registry.new
@subcommands.register(:ostypes) { Veewee::Command::Vagrant::Ostypes }
@subcommands.register(:templates) { Veewee::Command::Vagrant::Templates }
@subcommands.register(:list) { Veewee::Command::Vagrant::List }
@subcommands.register(:build) { Veewee::Command::Vagrant::Build }
@subcommands.register(:destroy) { Veewee::Command::Vagrant::Destroy }
@subcommands.register(:up) { Veewee::Command::Vagrant::Up }
@subcommands.register(:halt) { Veewee::Command::Vagrant::Halt }
@subcommands.register(:ssh) { Veewee::Command::Vagrant::Ssh }
@subcommands.register(:winrm) { Veewee::Command::Vagrant::Winrm }
@subcommands.register(:define) { Veewee::Command::Vagrant::Define }
@subcommands.register(:undefine) { Veewee::Command::Vagrant::Undefine }
@subcommands.register(:export) { Veewee::Command::Vagrant::Export }
end
def execute
if @main_args.include?("-h") || @main_args.include?("--help")
# Print the help for all the box commands.
return help
end
# If we reached this far then we must have a subcommand. If not,
# then we also just print the help and exit.
command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
return help if !command_class || !@sub_command
@logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")
# Initialize and execute the command class
command_class.new(@sub_args, @env).execute
end
# Prints the help out for this command
def help
opts = OptionParser.new do |opts|
opts.banner = "Usage: vagrant basebox <command> [<args>]"
opts.separator ""
opts.separator "Available subcommands:"
# Add the available subcommands as separators in order to print them
# out as well.
keys = []
@subcommands.each { |key, value| keys << key.to_s }
keys.sort.each do |key|
opts.separator " #{key}"
end
opts.separator ""
opts.separator "For help on any individual command run `vagrant basebox COMMAND -h`"
end
@env.ui.info(opts.help, :prefix => false)
end
end
end
end
end