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
|
require 'puppet/interface'
require 'puppet/interface/action'
class Puppet::Interface::ActionBuilder
attr_reader :action
def self.build(face, name, &block)
raise "Action #{name.inspect} must specify a block" unless block
new(face, name, &block).action
end
# Ideally the method we're defining here would be added to the action, and a
# method on the face would defer to it, but we can't get scope correct, so
# we stick with this. --daniel 2011-03-24
def when_invoked(&block)
@action.when_invoked = block
end
def when_rendering(type = nil, &block)
if type.nil? then # the default error message sucks --daniel 2011-04-18
raise ArgumentError, 'You must give a rendering format to when_rendering'
end
if block.nil? then
raise ArgumentError, 'You must give a block to when_rendering'
end
@action.set_rendering_method_for(type, block)
end
def option(*declaration, &block)
option = Puppet::Interface::OptionBuilder.build(@action, *declaration, &block)
@action.add_option(option)
end
def default(value = true)
@action.default = !!value
end
def render_as(value = nil)
value.nil? and raise ArgumentError, "You must give a rendering format to render_as"
formats = Puppet::Network::FormatHandler.formats
unless formats.include? value
raise ArgumentError, "#{value.inspect} is not a valid rendering format: #{formats.sort.join(", ")}"
end
@action.render_as = value
end
# Metaprogram the simple DSL from the target class.
Puppet::Interface::Action.instance_methods.grep(/=$/).each do |setter|
next if setter =~ /^=/
property = setter.to_s.chomp('=')
unless method_defined? property
# Using eval because the argument handling semantics are less awful than
# when we use the define_method/block version. The later warns on older
# Ruby versions if you pass the wrong number of arguments, but carries
# on, which is totally not what we want. --daniel 2011-04-18
eval <<-METHOD
def #{property}(value)
@action.#{property} = value
end
METHOD
end
end
private
def initialize(face, name, &block)
@face = face
@action = Puppet::Interface::Action.new(face, name)
instance_eval(&block)
@action.when_invoked or raise ArgumentError, "actions need to know what to do when_invoked; please add the block"
end
end
|
