From 55bc8f08dff07bd60ad7792f5201cf0f47ddc640 Mon Sep 17 00:00:00 2001
From: YeonGyu-Kim <code.yeon.gyu@gmail.com>
Date: Sat, 13 Dec 2025 14:05:17 +0900
Subject: [PATCH] refactor(ultrawork-mode): use history injection instead of
 direct message modification
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Replace direct parts[idx].text modification with injectHookMessage
- Context now injected via filesystem (like UserPromptSubmitHook)
- Preserves original user message without modification

🤖 GENERATED WITH ASSISTANCE OF [OhMyOpenCode](https://github.com/code-yeongyu/oh-my-opencode)
---
 src/hooks/ultrawork-mode/index.ts | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/src/hooks/ultrawork-mode/index.ts b/src/hooks/ultrawork-mode/index.ts
index 312ba7f..58c378c 100644
--- a/src/hooks/ultrawork-mode/index.ts
+++ b/src/hooks/ultrawork-mode/index.ts
@@ -2,6 +2,7 @@ import { detectUltraworkKeyword, extractPromptText } from "./detector"
 import { ULTRAWORK_CONTEXT } from "./constants"
 import type { UltraworkModeState } from "./types"
 import { log } from "../../shared"
+import { injectHookMessage } from "../../features/hook-message-injector"
 
 export * from "./detector"
 export * from "./constants"
@@ -16,13 +17,13 @@ export function clearUltraworkModeState(sessionID: string): void {
 export function createUltraworkModeHook() {
   return {
     /**
-     * chat.message hook - detect ultrawork/ulw keywords, inject context
+     * chat.message hook - detect ultrawork/ulw keywords, inject context via history
      *
      * Execution timing: AFTER claudeCodeHooks["chat.message"]
      * Behavior:
      *   1. Extract text from user prompt
      *   2. Detect ultrawork/ulw keywords (excluding code blocks)
-     *   3. If detected, prepend ULTRAWORK_CONTEXT to first text part
+     *   3. If detected, inject ULTRAWORK_CONTEXT via injectHookMessage (history injection)
      */
     "chat.message": async (
       input: {
@@ -51,13 +52,25 @@ export function createUltraworkModeHook() {
       state.detected = true
       log("Ultrawork keyword detected", { sessionID: input.sessionID })
 
-      const parts = output.parts as Array<{ type: string; text?: string }>
-      const idx = parts.findIndex((p) => p.type === "text" && p.text)
+      const message = output.message as {
+        agent?: string
+        model?: { modelID?: string; providerID?: string }
+        path?: { cwd?: string; root?: string }
+        tools?: Record<string, boolean>
+      }
 
-      if (idx >= 0) {
-        parts[idx].text = `${ULTRAWORK_CONTEXT}${parts[idx].text ?? ""}`
+      const success = injectHookMessage(input.sessionID, ULTRAWORK_CONTEXT, {
+        agent: message.agent,
+        model: message.model,
+        path: message.path,
+        tools: message.tools,
+      })
+
+      if (success) {
         state.injected = true
-        log("Ultrawork context injected", { sessionID: input.sessionID })
+        log("Ultrawork context injected via history", { sessionID: input.sessionID })
+      } else {
+        log("Ultrawork context injection failed", { sessionID: input.sessionID })
       }
 
       ultraworkModeState.set(input.sessionID, state)