62 lines
1.5 KiB
Ruby
62 lines
1.5 KiB
Ruby
|
module Guard
|
||
|
|
module Hook
|
||
|
|
def self.included(base)
|
||
|
|
base.send :include, InstanceMethods
|
||
|
|
end
|
||
|
|
|
||
|
|
module InstanceMethods
|
||
|
|
|
||
|
|
# When passed a sybmol, #hook will generate a hook name
|
||
|
|
# from the symbol and calling method name. When passed
|
||
|
|
# a string, #hook will turn the string into a symbol
|
||
|
|
# directly.
|
||
|
|
def hook(event)
|
||
|
|
if event.class == Symbol
|
||
|
|
calling_method = caller[0][/`([^']*)'/, 1]
|
||
|
|
hook_name = "#{calling_method}_#{event}".to_sym
|
||
|
|
else
|
||
|
|
hook_name = event.to_sym
|
||
|
|
end
|
||
|
|
|
||
|
|
UI.info "\nHook :#{hook_name} executed for #{self.class}"
|
||
|
|
Hook.notify(self.class, hook_name)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
class << self
|
||
|
|
def callbacks
|
||
|
|
@callbacks ||= Hash.new { |hash, key| hash[key] = [] }
|
||
|
|
end
|
||
|
|
|
||
|
|
def add(listener, guard_class, events)
|
||
|
|
_events = events.class == Array ? events : [events]
|
||
|
|
_events.each do |event|
|
||
|
|
callbacks[[guard_class, event]] << listener
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def has_callback?(listener, guard_class, event)
|
||
|
|
listeners(guard_class, event).include? listener
|
||
|
|
end
|
||
|
|
|
||
|
|
def notify(guard_class, event)
|
||
|
|
callbacks[[guard_class, event]].each do |listener|
|
||
|
|
listener.call(guard_class, event)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def reset!
|
||
|
|
@callbacks = nil
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def listeners(guard_class, event)
|
||
|
|
callbacks[[guard_class, event]].inject([]) do |all, arr|
|
||
|
|
all << arr
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|