blob: d75697afafac2eec00a89d8738a235dfff21d167 (
plain)
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
|
require 'puppet/interface/action_builder'
module Puppet::Interface::ActionManager
# Declare that this app can take a specific action, and provide
# the code to do so.
def action(name, &block)
@actions ||= {}
raise "Action #{name} already defined for #{self}" if action?(name)
action = Puppet::Interface::ActionBuilder.build(self, name, &block)
@actions[action.name] = action
end
# This is the short-form of an action definition; it doesn't use the
# builder, just creates the action directly from the block.
def script(name, &block)
@actions ||= {}
raise "Action #{name} already defined for #{self}" if action?(name)
@actions[name] = Puppet::Interface::Action.new(self, name, :when_invoked => block)
end
def actions
@actions ||= {}
result = @actions.keys
if self.is_a?(Class) and superclass.respond_to?(:actions)
result += superclass.actions
elsif self.class.respond_to?(:actions)
result += self.class.actions
end
result.sort
end
def get_action(name)
@actions ||= {}
result = @actions[name.to_sym]
if result.nil?
if self.is_a?(Class) and superclass.respond_to?(:get_action)
found = superclass.get_action(name)
elsif self.class.respond_to?(:get_action)
found = self.class.get_action(name)
end
if found then
# This is not the nicest way to make action equivalent to the Ruby
# Method object, rather than UnboundMethod, but it will do for now,
# and we only have to make this change in *one* place. --daniel 2011-04-12
result = @actions[name.to_sym] = found.__dup_and_rebind_to(self)
end
end
return result
end
def action?(name)
actions.include?(name.to_sym)
end
end
|
