Merge pull request #128 from Opencode-DCP/fix/nudge-counter-after-restart

fix: nudge counter incorrect after session restart

Author: Tarquinen

lib/hooks.ts

@@ -14,11 +14,12 @@ export function createChatMessageTransformHandler(
     logger: Logger,
     config: PluginConfig
 ) {
-    return async(
+    return async (
         input: {},
         output: { messages: WithParts[] }
     ) => {
-        checkSession(client, state, logger, output.messages);
+        await checkSession(client, state, logger, output.messages)
+
         if (state.isSubAgent) {
             return
         }

lib/state/state.ts

@@ -4,12 +4,12 @@ import { loadSessionState } from "./persistence"
 import { getLastUserMessage } from "../messages/utils"
 import { isSubAgentSession } from "../utils"
 
-export const checkSession = (
+export const checkSession = async (
     client: any,
     state: SessionState,
     logger: Logger,
     messages: WithParts[]
-) => {
+): Promise<void> => {
 
     const lastUserMessage = getLastUserMessage(messages)
     if (!lastUserMessage) {
@@ -20,14 +20,11 @@ export const checkSession = (
 
     if (state.sessionId === null || state.sessionId !== lastSessionId) {
         logger.info(`Session changed: ${state.sessionId} -> ${lastSessionId}`)
-        ensureSessionInitialized(
-            client,
-            state,
-            lastSessionId,
-            logger
-        ).catch((err) => {
+        try {
+            await ensureSessionInitialized(client, state, lastSessionId, logger)
+        } catch (err: any) {
             logger.error("Failed to initialize session state", { error: err.message })
-        } )
+        }
     }
 }
 

lib/state/tool-cache.ts

@@ -6,8 +6,6 @@ const MAX_TOOL_CACHE_SIZE = 500
 
 /**
  * Sync tool parameters from OpenCode's session.messages() API.
- * This is the single source of truth for tool parameters, replacing
- * format-specific parsing from LLM API requests.
  */
 export async function syncToolCache(
     state: SessionState,
@@ -24,6 +22,8 @@ export async function syncToolCache(
                     continue
                 }
 
+                const alreadyPruned = state.prune.toolIds.includes(part.callID)
+
                 state.toolParameters.set(
                     part.callID,
                     {
@@ -35,7 +35,7 @@ export async function syncToolCache(
                     }
                 )
 
-                if (!config.strategies.pruneTool.protectedTools.includes(part.tool)) {
+                if (!alreadyPruned && !config.strategies.pruneTool.protectedTools.includes(part.tool)) {
                     state.nudgeCounter++
                 }