aids

Author: jorgenwh

index.ts

@@ -3,7 +3,7 @@ import { getConfig } from "./lib/config"
 import { Logger } from "./lib/logger"
 import { createSessionState } from "./lib/state"
 import { createPruneTool } from "./lib/strategies/prune-tool"
-import { createChatMessageTransformHandler } from "./lib/hooks"
+import { createChatMessageTransformHandler, createEventHandler } from "./lib/hooks"
 
 const plugin: Plugin = (async (ctx) => {
     const config = getConfig(ctx)
@@ -54,6 +54,7 @@ const plugin: Plugin = (async (ctx) => {
                 logger.info("Added 'prune' to experimental.primary_tools via config mutation")
             }
         },
+        event: createEventHandler(ctx.client, config, state, logger),
     }
 }) satisfies Plugin
 

lib/hooks.ts

@@ -31,3 +31,24 @@ export function createChatMessageTransformHandler(
         insertPruneToolContext(state, config, logger, output.messages)
     }
 }
+
+export function createEventHandler(
+    client: any,
+    config: PluginConfig,
+    state: SessionState,
+    logger: Logger
+) {
+    return async (
+        { event }: { event: any }
+    ) => {
+        if (state.sessionId === null || state.isSubAgent) {
+            return
+        }
+
+        if (event.type === "session.status" && event.properties.status.type === "idle") {
+            if (!config.strategies.onIdle.enabled) {
+                return
+            }
+        }
+    }
+}

lib/state/state.ts

@@ -43,7 +43,8 @@ export function createSessionState(): SessionState {
             totalPruneTokens: 0,
         },
         toolParameters: new Map<string, ToolParameterEntry>(),
-        nudgeCounter: 0
+        nudgeCounter: 0,
+        lastToolPrune: false
     }
 }
 
@@ -59,6 +60,7 @@ export function resetSessionState(state: SessionState): void {
     }
     state.toolParameters.clear()
     state.nudgeCounter = 0
+    state.lastToolPrune = false
 }
 
 export async function ensureSessionInitialized(

lib/state/tool-cache.ts

@@ -17,6 +17,7 @@ export async function syncToolCache(
 ): Promise<void> {
     try {
         logger.info("Syncing tool parameters from OpenCode messages")
+
         for (const msg of messages) {
             for (const part of msg.parts) {
                 if (part.type !== "tool" || !part.callID || state.toolParameters.has(part.callID)) {
@@ -36,6 +37,9 @@ export async function syncToolCache(
                 if (!config.strategies.pruneTool.protectedTools.includes(part.tool)) {
                     state.nudgeCounter++
                 }
+
+                state.lastToolPrune = part.tool === "prune"
+                logger.info("lastToolPrune=" + String(state.lastToolPrune))
             }
         }
 

lib/state/types.ts

@@ -30,4 +30,5 @@ export interface SessionState {
     stats: SessionStats
     toolParameters: Map<string, ToolParameterEntry>
     nudgeCounter: number
+    lastToolPrune: boolean
 }