|
| 1 | +import { detectUltraworkKeyword, extractPromptText } from "./detector" |
| 2 | +import { ULTRAWORK_CONTEXT } from "./constants" |
| 3 | +import type { UltraworkModeState } from "./types" |
| 4 | +import { log } from "../../shared" |
| 5 | + |
| 6 | +export * from "./detector" |
| 7 | +export * from "./constants" |
| 8 | +export * from "./types" |
| 9 | + |
| 10 | +const ultraworkModeState = new Map<string, UltraworkModeState>() |
| 11 | + |
| 12 | +export function clearUltraworkModeState(sessionID: string): void { |
| 13 | + ultraworkModeState.delete(sessionID) |
| 14 | +} |
| 15 | + |
| 16 | +export function createUltraworkModeHook() { |
| 17 | + return { |
| 18 | + /** |
| 19 | + * chat.message hook - detect ultrawork/ulw keywords, inject context |
| 20 | + * |
| 21 | + * Execution timing: AFTER claudeCodeHooks["chat.message"] |
| 22 | + * Behavior: |
| 23 | + * 1. Extract text from user prompt |
| 24 | + * 2. Detect ultrawork/ulw keywords (excluding code blocks) |
| 25 | + * 3. If detected, prepend ULTRAWORK_CONTEXT to first text part |
| 26 | + */ |
| 27 | + "chat.message": async ( |
| 28 | + input: { |
| 29 | + sessionID: string |
| 30 | + agent?: string |
| 31 | + model?: { providerID: string; modelID: string } |
| 32 | + messageID?: string |
| 33 | + }, |
| 34 | + output: { |
| 35 | + message: Record<string, unknown> |
| 36 | + parts: Array<{ type: string; text?: string; [key: string]: unknown }> |
| 37 | + } |
| 38 | + ): Promise<void> => { |
| 39 | + const state: UltraworkModeState = { |
| 40 | + detected: false, |
| 41 | + injected: false, |
| 42 | + } |
| 43 | + |
| 44 | + const promptText = extractPromptText(output.parts) |
| 45 | + |
| 46 | + if (!detectUltraworkKeyword(promptText)) { |
| 47 | + ultraworkModeState.set(input.sessionID, state) |
| 48 | + return |
| 49 | + } |
| 50 | + |
| 51 | + state.detected = true |
| 52 | + log("Ultrawork keyword detected", { sessionID: input.sessionID }) |
| 53 | + |
| 54 | + const parts = output.parts as Array<{ type: string; text?: string }> |
| 55 | + const idx = parts.findIndex((p) => p.type === "text" && p.text) |
| 56 | + |
| 57 | + if (idx >= 0) { |
| 58 | + parts[idx].text = `${ULTRAWORK_CONTEXT}${parts[idx].text ?? ""}` |
| 59 | + state.injected = true |
| 60 | + log("Ultrawork context injected", { sessionID: input.sessionID }) |
| 61 | + } |
| 62 | + |
| 63 | + ultraworkModeState.set(input.sessionID, state) |
| 64 | + }, |
| 65 | + |
| 66 | + /** |
| 67 | + * event hook - cleanup session state on deletion |
| 68 | + */ |
| 69 | + event: async ({ |
| 70 | + event, |
| 71 | + }: { |
| 72 | + event: { type: string; properties?: unknown } |
| 73 | + }) => { |
| 74 | + if (event.type === "session.deleted") { |
| 75 | + const props = event.properties as |
| 76 | + | { info?: { id?: string } } |
| 77 | + | undefined |
| 78 | + if (props?.info?.id) { |
| 79 | + ultraworkModeState.delete(props.info.id) |
| 80 | + } |
| 81 | + } |
| 82 | + }, |
| 83 | + } |
| 84 | +} |
0 commit comments