Class: Clacky::HookManager

Inherits:
Object
  • Object
show all
Defined in:
lib/clacky/agent/hook_manager.rb

Constant Summary collapse

HOOK_EVENTS = 
[
  :before_tool_use,
  :after_tool_use,
  :on_tool_error,
  :on_start,
  :on_complete,
  :on_iteration,
  :session_rollback
].freeze

Instance Method Summary collapse

Constructor Details

#initialize ⇒ HookManager

Returns a new instance of HookManager.



15
16
17
 
# File 'lib/clacky/agent/hook_manager.rb', line 15

def initialize
  @hooks = Hash.new { |h, k| h[k] = [] }
end

Instance Method Details

#add(event, &block) ⇒ Object 



19
20
21
22
 
# File 'lib/clacky/agent/hook_manager.rb', line 19

def add(event, &block)
  validate_event!(event)
  @hooks[event] << block
end

#clear(event = nil) ⇒ Object 



45
46
47
48
49
50
51
52
 
# File 'lib/clacky/agent/hook_manager.rb', line 45

def clear(event = nil)
  if event
    validate_event!(event)
    @hooks[event].clear
  else
    @hooks.clear
  end
end

#has_hooks?(event) ⇒ Boolean

Returns:

  • (Boolean) 


41
42
43
 
# File 'lib/clacky/agent/hook_manager.rb', line 41

def has_hooks?(event)
  @hooks[event].any?
end

#trigger(event, *args) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/clacky/agent/hook_manager.rb', line 24

def trigger(event, *args)
  validate_event!(event)
  result = { action: :allow }

  @hooks[event].each do |hook|
    begin
      hook_result = hook.call(*args)
      result.merge!(hook_result) if hook_result.is_a?(Hash)
    rescue StandardError => e
      # Log error but don't fail
      Clacky::Logger.error("Hook error", event: event, error: e)
    end
  end

  result
end

#validate_event!(event) ⇒ Object

Raises:

  • (ArgumentError) 


55
56
57
58
59
 
# File 'lib/clacky/agent/hook_manager.rb', line 55

def validate_event!(event)
  return if HOOK_EVENTS.include?(event)

  raise ArgumentError, "Invalid hook event: #{event}. Must be one of #{HOOK_EVENTS.join(', ')}"
end