File size: 2,059 Bytes
064bfd6 | 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 | import { HOOK_EVENTS } from 'src/entrypoints/agentSdkTypes.js'
import type { AppState } from 'src/state/AppState.js'
import { logForDebugging } from '../debug.js'
import type { HooksSettings } from '../settings/types.js'
import { addSessionHook, removeSessionHook } from './sessionHooks.js'
/**
* Registers hooks from a skill's frontmatter as session hooks.
*
* Hooks are registered as session-scoped hooks that persist for the duration
* of the session. If a hook has `once: true`, it will be automatically removed
* after its first successful execution.
*
* @param setAppState - Function to update the app state
* @param sessionId - The current session ID
* @param hooks - The hooks settings from the skill's frontmatter
* @param skillName - The name of the skill (for logging)
* @param skillRoot - The base directory of the skill (for CLAUDE_PLUGIN_ROOT env var)
*/
export function registerSkillHooks(
setAppState: (updater: (prev: AppState) => AppState) => void,
sessionId: string,
hooks: HooksSettings,
skillName: string,
skillRoot?: string,
): void {
let registeredCount = 0
for (const eventName of HOOK_EVENTS) {
const matchers = hooks[eventName]
if (!matchers) continue
for (const matcher of matchers) {
for (const hook of matcher.hooks) {
// For once: true hooks, use onHookSuccess callback to remove after execution
const onHookSuccess = hook.once
? () => {
logForDebugging(
`Removing one-shot hook for event ${eventName} in skill '${skillName}'`,
)
removeSessionHook(setAppState, sessionId, eventName, hook)
}
: undefined
addSessionHook(
setAppState,
sessionId,
eventName,
matcher.matcher || '',
hook,
onHookSuccess,
skillRoot,
)
registeredCount++
}
}
}
if (registeredCount > 0) {
logForDebugging(
`Registered ${registeredCount} hooks from skill '${skillName}'`,
)
}
}
|