@kenkaiiii/ggcoder 4.3.193 → 4.3.194

package/dist/ui/App.js

@@ -52,6 +52,8 @@ import { loadCustomCommands } from "../core/custom-commands.js";
 import { buildSystemPrompt } from "../system-prompt.js";
 import { detectLanguages, LANGUAGE_DISPLAY_NAMES, } from "../core/language-detector.js";
 import { detectVerifyCommands } from "../core/verify-commands.js";
+import { FOCUSED_REPO_MAP_MAX_CHARS, FIRST_TURN_REPO_MAP_MAX_CHARS, buildRepoMap, createRepoMapCache, } from "../core/repomap.js";
+import { getLatestUserText, injectRepoMapContextMessages, stripRepoMapContextMessages, } from "../core/repomap-context.js";
 import { extractPlanSteps, findCompletedMarkers, markStepsCompleted, segmentDisplayText, stripDoneMarkers, } from "../utils/plan-steps.js";
 import { getMCPServers } from "../core/mcp/index.js";
 import { trimFlushedItems, flushOnTurnText, flushOnTurnEnd, flushOverflow, } from "./live-item-flush.js";
@@ -337,6 +339,12 @@ export function App(props) {
     const currentToolsRef = useRef(props.tools);
     const [thinkingEnabled, setThinkingEnabled] = useState(!!props.thinking);
     const messagesRef = useRef(props.sessionStore?.messages ?? props.messages);
+    const repoMapInjectionEnabledRef = useRef(true);
+    const repoMapDirtyRef = useRef(true);
+    const repoMapMarkdownRef = useRef("");
+    const repoMapSnapshotRef = useRef(undefined);
+    const repoMapChangedCountRef = useRef(0);
+    const repoMapCacheRef = useRef(createRepoMapCache());
     const [planAutoExpand, setPlanAutoExpand] = useState(props.sessionStore?.planAutoExpand ?? false);
     const approvedPlanPathRef = useRef(props.sessionStore?.approvedPlanPath);
     const planStepsRef = useRef(props.sessionStore?.planSteps ?? []);
@@ -740,59 +748,92 @@ export function App(props) {
             return messages; // Return unchanged on failure
         }
     }, [currentModel, currentProvider, activeApiKey]);
+    const getRepoMapSignalCount = useCallback(() => {
+        return ((props.repoMapChangedFilesRef?.current.size ?? 0) +
+            (props.repoMapReadFilesRef?.current.size ?? 0));
+    }, [props.repoMapChangedFilesRef, props.repoMapReadFilesRef]);
+    const getRepoMapBudget = useCallback(() => {
+        const userTurns = messagesRef.current.filter((message) => message.role === "user").length;
+        const readCount = props.repoMapReadFilesRef?.current.size ?? 0;
+        if (userTurns <= 1 && readCount === 0)
+            return FIRST_TURN_REPO_MAP_MAX_CHARS;
+        if (readCount > 0)
+            return FOCUSED_REPO_MAP_MAX_CHARS;
+        return FOCUSED_REPO_MAP_MAX_CHARS + 1000;
+    }, [props.repoMapReadFilesRef]);
+    const refreshRepoMap = useCallback(async (latestUserPrompt) => {
+        const rendered = await buildRepoMap({
+            cwd: cwdRef.current,
+            maxChars: getRepoMapBudget(),
+            changedFiles: [...(props.repoMapChangedFilesRef?.current ?? new Set())],
+            readFiles: [...(props.repoMapReadFilesRef?.current ?? new Set())],
+            focusTerms: latestUserPrompt ? [latestUserPrompt] : [],
+            cache: repoMapCacheRef.current,
+        });
+        repoMapMarkdownRef.current = rendered.markdown;
+        repoMapSnapshotRef.current = rendered.snapshot;
+        repoMapChangedCountRef.current = getRepoMapSignalCount();
+        repoMapDirtyRef.current = false;
+        return rendered.markdown;
+    }, [
+        getRepoMapBudget,
+        getRepoMapSignalCount,
+        props.repoMapChangedFilesRef,
+        props.repoMapReadFilesRef,
+    ]);
+    const stripRepoMapMessages = useCallback((messages) => {
+        return stripRepoMapContextMessages(messages);
+    }, []);
+    const injectRepoMapContext = useCallback(async (messages) => {
+        if (!repoMapInjectionEnabledRef.current)
+            return stripRepoMapMessages(messages);
+        const stripped = stripRepoMapMessages(messages);
+        const latestUserPrompt = getLatestUserText(stripped);
+        const signalCount = getRepoMapSignalCount();
+        if (signalCount !== repoMapChangedCountRef.current)
+            repoMapDirtyRef.current = true;
+        if (repoMapDirtyRef.current || !repoMapMarkdownRef.current) {
+            await refreshRepoMap(latestUserPrompt);
+        }
+        if (!repoMapMarkdownRef.current)
+            return stripped;
+        return injectRepoMapContextMessages(stripped, repoMapMarkdownRef.current);
+    }, [props.repoMapChangedFilesRef, props.repoMapReadFilesRef, refreshRepoMap, stripRepoMapMessages]);
     /**
      * transformContext callback for the agent loop.
      * Called before each LLM call and on context overflow.
-     * Checks if auto-compaction is needed and runs it.
-     *
-     * Uses actual API-reported token counts (from previous turn_end) when
-     * available, falling back to the character-based estimate. A 30-second
-     * cooldown prevents repeated compaction — matching the pattern used by
-     * Mysti, openclaw, and other real-world agent frameworks.
+     * Compacts persistent chat only, then injects the dynamic repo map transiently.
      */
     const transformContext = useCallback(async (messages, options) => {
+        const stripped = stripRepoMapMessages(messages);
         const settings = settingsRef.current;
         const autoCompact = settings?.get("autoCompact") ?? true;
         const threshold = settings?.get("compactThreshold") ?? 0.8;
         // Force-compact on context overflow regardless of settings
         if (options?.force) {
-            const result = await compactConversation(messages);
+            const result = await compactConversation(stripped);
             lastCompactionTimeRef.current = Date.now();
-            return result;
+            return injectRepoMapContext(result);
         }
         if (!autoCompact)
-            return messages;
+            return injectRepoMapContext(stripped);
         // Time-based cooldown: skip if compaction ran within the last 30 seconds
         if (Date.now() - lastCompactionTimeRef.current < 30_000) {
             log("INFO", "compaction", `Skipping compaction — cooldown active`);
-            return messages;
+            return injectRepoMapContext(stripped);
         }
         const contextWindow = getContextWindow(currentModel);
-        // Reserve = max output budget + ~5K headroom for system prompt + tool
-        // schemas. Otherwise the API rejects requests where the prompt fits the
-        // window but leaves no room for the response (e.g. Codex Mini at 200K
-        // ctx / 100K out — pre-turn estimate may say 160K but a 100K reasoning
-        // response then overflows).
         const modelInfo = getModel(currentModel);
         const reserveTokens = (modelInfo?.maxOutputTokens ?? 0) + 5_000;
-        // Prefer actual API-reported tokens over char-based estimate, but only
-        // when the token count was recorded AFTER the most recent compaction.
-        // A count from before compaction is stale — it reflects the old context
-        // size and would trigger compaction again immediately for no reason.
         const tokensFresh = lastActualTokensTimestampRef.current > lastCompactionTimeRef.current;
         const actualTokens = lastActualTokensRef.current > 0 && tokensFresh ? lastActualTokensRef.current : undefined;
-        if (shouldCompact(messages, contextWindow, threshold, actualTokens, reserveTokens)) {
-            const before = messages.length;
-            const result = await compactConversation(messages);
+        if (shouldCompact(stripped, contextWindow, threshold, actualTokens, reserveTokens)) {
+            const result = await compactConversation(stripped);
             lastCompactionTimeRef.current = Date.now();
-            // If compaction was a no-op (e.g. too few middle messages to summarize),
-            // return the original reference so the agent loop doesn't replace messages.
-            if (result.length === before)
-                return messages;
-            return result;
+            return injectRepoMapContext(result);
         }
-        return messages;
-    }, [currentModel, compactConversation]);
+        return injectRepoMapContext(stripped);
+    }, [currentModel, compactConversation, injectRepoMapContext, stripRepoMapMessages]);
     // ── Background task bar state (external store) ──────────
     const { bgTasks, focused: taskBarFocused, expanded: taskBarExpanded, selectedIndex: selectedTaskIndex, } = useTaskBarStore();
     useTaskBarPolling(props.processManager);
@@ -828,6 +869,7 @@ export function App(props) {
         transformContext,
     }, {
         onComplete: useCallback(() => {
+            messagesRef.current = stripRepoMapMessages(messagesRef.current);
             persistNewMessages();
             // Auto-clear plan progress and approved plan when all steps are completed
             const steps = planStepsRef.current;
@@ -880,6 +922,7 @@ export function App(props) {
             }
         }, [
             persistNewMessages,
+            stripRepoMapMessages,
             planMode,
             props.cwd,
             props.skills,
@@ -1706,6 +1749,47 @@ export function App(props) {
             setLiveItems([{ kind: "plan_event", event: "dismissed", id: getId() }]);
             return;
         }
+        // Handle /map — show, refresh, or toggle dynamic repo map injection
+        if (trimmed === "/map" ||
+            trimmed === "/map refresh" ||
+            trimmed === "/map on" ||
+            trimmed === "/map off") {
+            const action = trimmed.slice("/map".length).trim();
+            if (action === "on") {
+                repoMapInjectionEnabledRef.current = true;
+                repoMapDirtyRef.current = true;
+                setLiveItems((prev) => [
+                    ...prev,
+                    { kind: "info", text: "Dynamic repo map injection is on.", id: getId() },
+                ]);
+                return;
+            }
+            if (action === "off") {
+                repoMapInjectionEnabledRef.current = false;
+                messagesRef.current = stripRepoMapMessages(messagesRef.current);
+                setLiveItems((prev) => [
+                    ...prev,
+                    {
+                        kind: "info",
+                        text: "Dynamic repo map injection is off for this session.",
+                        id: getId(),
+                    },
+                ]);
+                return;
+            }
+            if (action === "refresh")
+                repoMapDirtyRef.current = true;
+            const markdown = await refreshRepoMap(getLatestUserText(messagesRef.current));
+            setLiveItems((prev) => [
+                ...prev,
+                {
+                    kind: "info",
+                    text: formatRepoMapCommandOutput(repoMapInjectionEnabledRef.current, markdown, action === "refresh"),
+                    id: getId(),
+                },
+            ]);
+            return;
+        }
         // Handle /plans — open plan pane
         if (trimmed === "/plans") {
             if (props.resetUI && props.sessionStore && !agentLoop.isRunning) {
@@ -1895,6 +1979,8 @@ export function App(props) {
         compactConversation,
         rebuildSystemPrompt,
         replaceSystemPrompt,
+        refreshRepoMap,
+        stripRepoMapMessages,
     ]);
     const handleDoubleExit = useDoublePress(setExitPending, () => process.exit(0));
     const handleAbort = useCallback(() => {
@@ -2280,6 +2366,13 @@ export function App(props) {
                     log("WARN", "pixel", `chdir failed: ${err.message}`);
                 }
                 cwdRef.current = prep.projectPath;
+                repoMapDirtyRef.current = true;
+                repoMapMarkdownRef.current = "";
+                repoMapSnapshotRef.current = undefined;
+                repoMapChangedCountRef.current = 0;
+                repoMapCacheRef.current = createRepoMapCache();
+                props.repoMapChangedFilesRef?.current.clear();
+                props.repoMapReadFilesRef?.current.clear();
                 setDisplayedCwd(prep.projectPath);
                 let toolsForPixelFix = currentToolsRef.current;
                 if (props.rebuildToolsForCwd) {
@@ -2618,4 +2711,11 @@ export function App(props) {
                             ]);
                         }, cwd: props.cwd, commands: allCommands, eyesCount: eyesCount }), overlay === "model" ? (_jsx(ModelSelector, { onSelect: handleModelSelect, onCancel: () => setOverlay(null), loggedInProviders: props.loggedInProviders ?? [currentProvider], currentModel: currentModel, currentProvider: currentProvider })) : overlay === "theme" ? (_jsx(ThemeSelector, { onSelect: handleThemeSelect, onCancel: () => setOverlay(null), currentTheme: theme.name })) : (_jsx(Footer, { model: currentModel, tokensIn: agentLoop.contextUsed, cwd: displayedCwd, gitBranch: gitBranch, thinkingLevel: thinkingEnabled ? getMaxThinkingLevel(currentModel) : undefined, planMode: planMode, exitPending: exitPending })), (bgTasks.length > 0 || (eyesCount !== undefined && eyesCount > 0) || updatePending) && (_jsxs(Box, { children: [bgTasks.length > 0 && (_jsx(BackgroundTasksBar, { tasks: bgTasks, focused: taskBarFocused, expanded: taskBarExpanded, selectedIndex: selectedTaskIndex, onExpand: handleTaskBarExpand, onCollapse: handleTaskBarCollapse, onKill: handleTaskKill, onExit: handleTaskBarExit, onNavigate: handleTaskNavigate })), eyesCount !== undefined && eyesCount > 0 && (_jsx(Box, { paddingLeft: bgTasks.length > 0 ? 2 : 1, paddingRight: 1, children: _jsx(Text, { color: theme.accent, bold: true, children: `${eyesCount} eyes signal${eyesCount === 1 ? "" : "s"} · Run /eyes-improve to enhance GG Coder` }) })), updatePending && (_jsx(Box, { paddingLeft: bgTasks.length > 0 || (eyesCount !== undefined && eyesCount > 0) ? 2 : 1, paddingRight: 1, children: _jsx(Text, { color: theme.success, bold: true, children: "\u2728 Update ready \u00B7 restart to apply" }) }))] }))] }))] }));
 }
+function formatRepoMapCommandOutput(enabled, markdown, refreshed) {
+    const status = enabled ? "on" : "off";
+    const prefix = refreshed
+        ? `Dynamic repo map refreshed · injection: ${status}`
+        : `Dynamic repo map · injection: ${status}`;
+    return `${prefix}\n\n${markdown}`;
+}
 //# sourceMappingURL=App.js.map