134cbdb007
2) Pass args from hooks to callbacks 3) Suppress UI message from hooks unless in 'development'
56 lines
1.4 KiB
Ruby
56 lines
1.4 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, *args)
|
|
hook_name = if event.is_a? Symbol
|
|
calling_method = caller[0][/`([^']*)'/, 1]
|
|
"#{calling_method}_#{event}".to_sym
|
|
else
|
|
event.to_sym
|
|
end
|
|
|
|
if ENV["GUARD_ENV"] == "development"
|
|
UI.info "\nHook :#{hook_name} executed for #{self.class}"
|
|
end
|
|
|
|
Hook.notify(self.class, hook_name, *args)
|
|
end
|
|
end
|
|
|
|
class << self
|
|
def callbacks
|
|
@callbacks ||= Hash.new { |hash, key| hash[key] = [] }
|
|
end
|
|
|
|
def add_callback(listener, guard_class, events)
|
|
_events = events.is_a?(Array) ? events : [events]
|
|
_events.each do |event|
|
|
callbacks[[guard_class, event]] << listener
|
|
end
|
|
end
|
|
|
|
def has_callback?(listener, guard_class, event)
|
|
callbacks[[guard_class, event]].include?(listener)
|
|
end
|
|
|
|
def notify(guard_class, event, *args)
|
|
callbacks[[guard_class, event]].each do |listener|
|
|
listener.call(guard_class, event, *args)
|
|
end
|
|
end
|
|
|
|
def reset_callbacks!
|
|
@callbacks = nil
|
|
end
|
|
end
|
|
end
|
|
end
|