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
|
require 'puppet/application'
require 'puppet/interface'
class Puppet::Application::InterfaceBase < Puppet::Application
should_parse_config
run_mode :agent
def preinit
super
trap(:INT) do
$stderr.puts "Cancelling Interface"
exit(0)
end
end
option("--debug", "-d") do |arg|
Puppet::Util::Log.level = :debug
end
option("--verbose", "-v") do
Puppet::Util::Log.level = :info
end
option("--format FORMAT") do |arg|
@format = arg.to_sym
end
option("--mode RUNMODE", "-r") do |arg|
raise "Invalid run mode #{arg}; supported modes are user, agent, master" unless %w{user agent master}.include?(arg)
self.class.run_mode(arg.to_sym)
set_run_mode self.class.run_mode
end
attr_accessor :interface, :type, :verb, :name, :arguments, :format
attr_writer :exit_code
# This allows you to set the exit code if you don't want to just exit
# immediately but you need to indicate a failure.
def exit_code
@exit_code || 0
end
def main
# Call the method associated with the provided action (e.g., 'find').
if result = interface.send(verb, name, *arguments)
puts render(result)
end
exit(exit_code)
end
# Override this if you need custom rendering.
def render(result)
render_method = Puppet::Network::FormatHandler.format(format).render_method
result.send(render_method)
end
def setup
Puppet::Util::Log.newdestination :console
@verb, @name, @arguments = command_line.args
@arguments ||= []
@type = self.class.name.to_s.sub(/.+:/, '').downcase.to_sym
unless @interface = Puppet::Interface.interface(@type)
raise "Could not find interface '#{@type}'"
end
@format ||= @interface.default_format
validate
end
def validate
unless verb
raise "You must specify #{interface.actions.join(", ")} as a verb; 'save' probably does not work right now"
end
unless interface.action?(verb)
raise "Command '#{verb}' not found for #{type}"
end
end
end
|
