oussamadouhou/oh-my-opencode-free-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(hook): block grep tool and guide to safe-grep

Browse Source
This commit is contained in:
YeonGyu-Kim
2025-12-04 18:06:29 +09:00
parent 8c6dc77d53
commit b397dd6655
2 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/hooks/context-window-monitor.ts
View File

@@ -1,7 +1,8 @@
import type { PluginInput } from "@opencode-ai/plugin" import type { PluginInput } from "@opencode-ai/plugin"
const ANTHROPIC_CONTEXT_LIMIT = 1_000_000 const ANTHROPIC_DISPLAY_LIMIT = 1_000_000
const CONTEXT_WARNING_THRESHOLD = 0.75 const ANTHROPIC_ACTUAL_LIMIT = 200_000
const CONTEXT_WARNING_THRESHOLD = 0.70
const CONTEXT_REMINDER = `[SYSTEM REMINDER - 1M Context Window] const CONTEXT_REMINDER = `[SYSTEM REMINDER - 1M Context Window]
@@ -65,16 +66,17 @@ export function createContextWindowMonitorHook(ctx: PluginInput) {
return sum + inputTokens + cacheReadTokens return sum + inputTokens + cacheReadTokens
}, 0) }, 0)
const usagePercentage = totalInputTokens / ANTHROPIC_CONTEXT_LIMIT const actualUsagePercentage = totalInputTokens / ANTHROPIC_ACTUAL_LIMIT
if (usagePercentage < CONTEXT_WARNING_THRESHOLD) return if (actualUsagePercentage < CONTEXT_WARNING_THRESHOLD) return
remindedSessions.add(sessionID) remindedSessions.add(sessionID)
const usedPct = (usagePercentage * 100).toFixed(1) const displayUsagePercentage = totalInputTokens / ANTHROPIC_DISPLAY_LIMIT
const remainingPct = ((1 - usagePercentage) * 100).toFixed(1) const usedPct = (displayUsagePercentage * 100).toFixed(1)
const remainingPct = ((1 - displayUsagePercentage) * 100).toFixed(1)
const usedTokens = totalInputTokens.toLocaleString() const usedTokens = totalInputTokens.toLocaleString()
const limitTokens = ANTHROPIC_CONTEXT_LIMIT.toLocaleString() const limitTokens = ANTHROPIC_DISPLAY_LIMIT.toLocaleString()
output.output += `\n\n${CONTEXT_REMINDER} output.output += `\n\n${CONTEXT_REMINDER}
[Context Status: ${usedPct}% used (${usedTokens}/${limitTokens} tokens), ${remainingPct}% remaining]` [Context Status: ${usedPct}% used (${usedTokens}/${limitTokens} tokens), ${remainingPct}% remaining]`

4
src/index.ts
View File

@@ -96,6 +96,10 @@ const OhMyOpenCodePlugin: Plugin = async (ctx) => {
}, },
"tool.execute.before": async (input, _output) => { "tool.execute.before": async (input, _output) => {
if (input.tool === "grep") {
throw new Error(`[BLOCKED] grep has no timeout and can freeze the system. It is permanently disabled. Use 'safe-grep' instead.`)
}
if (input.sessionID === mainSessionID) { if (input.sessionID === mainSessionID) {
updateTerminalTitle({ updateTerminalTitle({
sessionId: input.sessionID, sessionId: input.sessionID,