@sunub/obsidian-mcp-server 0.3.19 → 0.3.20

package/build/cli/hooks/useCommandCompletion.js

@@ -0,0 +1,344 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useAtCompletion } from "../hooks/useAtCompletion.js";
+import { useCompletion } from "../hooks/useCompletion.js";
+import { PROMPT_COMPLETION_MIN_LENGTH, usePromptCompletion, } from "../hooks/usePromptCompletion.js";
+import { useShellCompletion } from "../hooks/useShellCompletion.js";
+import { useSlashCompletion } from "../hooks/useSlashCompletion.js";
+import { logicalPosToOffset } from "../key/text-buffer.js";
+import { isSlashCommand } from "../utils/commandUtils.js";
+import { toCodePoints } from "../utils/textUtil.js";
+import { useCallback, useEffect, useMemo, useState } from "react";
+export var CompletionMode;
+(function (CompletionMode) {
+    CompletionMode["IDLE"] = "IDLE";
+    CompletionMode["AT"] = "AT";
+    CompletionMode["SLASH"] = "SLASH";
+    CompletionMode["PROMPT"] = "PROMPT";
+    CompletionMode["SHELL"] = "SHELL";
+})(CompletionMode || (CompletionMode = {}));
+export function useCommandCompletion({ buffer, cwd, slashCommands, commandContext, reverseSearchActive = false, shellModeActive, config, active, }) {
+    const [forceShowShellSuggestions, setForceShowShellSuggestions] = useState(false);
+    const { suggestions, activeSuggestionIndex, visibleStartIndex, isLoadingSuggestions, isPerfectMatch, setSuggestions, setActiveSuggestionIndex, setIsLoadingSuggestions, setIsPerfectMatch, setVisibleStartIndex, resetCompletionState: baseResetCompletionState, navigateUp, navigateDown, } = useCompletion();
+    const resetCompletionState = useCallback(() => {
+        baseResetCompletionState();
+        setForceShowShellSuggestions(false);
+    }, [baseResetCompletionState]);
+    const cursorRow = buffer.cursor[0];
+    const cursorCol = buffer.cursor[1];
+    const { completionMode, query: memoQuery, completionStart, completionEnd, } = useMemo(() => {
+        const currentLine = buffer.lines[cursorRow] || "";
+        const codePoints = toCodePoints(currentLine);
+        if (shellModeActive) {
+            return {
+                completionMode: currentLine.trim().length === 0
+                    ? CompletionMode.IDLE
+                    : CompletionMode.SHELL,
+                query: "",
+                completionStart: -1,
+                completionEnd: -1,
+            };
+        }
+        // FIRST: Check for @ completion (scan backwards from cursor)
+        // This must happen before slash command check so that `/cmd @file`
+        // triggers file completion, not just slash command completion.
+        for (let i = cursorCol - 1; i >= 0; i--) {
+            const char = codePoints[i];
+            if (char === " ") {
+                let backslashCount = 0;
+                for (let j = i - 1; j >= 0 && codePoints[j] === "\\"; j--) {
+                    backslashCount++;
+                }
+                if (backslashCount % 2 === 0) {
+                    break;
+                }
+            }
+            else if (char === "@") {
+                let end = codePoints.length;
+                for (let i = cursorCol; i < codePoints.length; i++) {
+                    if (codePoints[i] === " ") {
+                        let backslashCount = 0;
+                        for (let j = i - 1; j >= 0 && codePoints[j] === "\\"; j--) {
+                            backslashCount++;
+                        }
+                        if (backslashCount % 2 === 0) {
+                            end = i;
+                            break;
+                        }
+                    }
+                }
+                const pathStart = i + 1;
+                const partialPath = currentLine.substring(pathStart, end);
+                return {
+                    completionMode: CompletionMode.AT,
+                    query: partialPath,
+                    completionStart: pathStart,
+                    completionEnd: end,
+                };
+            }
+        }
+        // THEN: Check for slash command (only if no @ completion is active)
+        if (cursorRow === 0 && isSlashCommand(currentLine.trim())) {
+            return {
+                completionMode: CompletionMode.SLASH,
+                query: currentLine,
+                completionStart: 0,
+                completionEnd: currentLine.length,
+            };
+        }
+        // Check for prompt completion - only if enabled
+        const trimmedText = buffer.text.trim();
+        const isPromptCompletionEnabled = false;
+        if (isPromptCompletionEnabled &&
+            trimmedText.length >= PROMPT_COMPLETION_MIN_LENGTH &&
+            !isSlashCommand(trimmedText) &&
+            !trimmedText.includes("@")) {
+            return {
+                completionMode: CompletionMode.PROMPT,
+                query: trimmedText,
+                completionStart: 0,
+                completionEnd: trimmedText.length,
+            };
+        }
+        return {
+            completionMode: CompletionMode.IDLE,
+            query: null,
+            completionStart: -1,
+            completionEnd: -1,
+        };
+    }, [cursorRow, cursorCol, buffer.lines, buffer.text, shellModeActive]);
+    useAtCompletion({
+        enabled: active && completionMode === CompletionMode.AT,
+        pattern: memoQuery || "",
+        config,
+        cwd,
+        setSuggestions,
+        setIsLoadingSuggestions,
+    });
+    const slashCompletionRange = useSlashCompletion({
+        enabled: active && completionMode === CompletionMode.SLASH && !shellModeActive,
+        query: memoQuery,
+        slashCommands,
+        commandContext,
+        setSuggestions,
+        setIsLoadingSuggestions,
+        setIsPerfectMatch,
+    });
+    const shellCompletionRange = useShellCompletion({
+        enabled: active && completionMode === CompletionMode.SHELL,
+        line: buffer.lines[cursorRow] || "",
+        cursorCol,
+        cwd,
+        setSuggestions,
+        setIsLoadingSuggestions,
+    });
+    const query = completionMode === CompletionMode.SHELL
+        ? shellCompletionRange.query
+        : memoQuery;
+    const basePromptCompletion = usePromptCompletion({
+        buffer,
+    });
+    const isShellSuggestionsVisible = completionMode !== CompletionMode.SHELL || forceShowShellSuggestions;
+    const promptCompletion = useMemo(() => {
+        if (completionMode === CompletionMode.SHELL &&
+            suggestions.length === 1 &&
+            query != null &&
+            shellCompletionRange.completionStart === shellCompletionRange.activeStart) {
+            const suggestion = suggestions[0];
+            const textToInsertBase = suggestion.value;
+            if (textToInsertBase.startsWith(query) &&
+                textToInsertBase.length > query.length) {
+                const currentLine = buffer.lines[cursorRow] || "";
+                const start = shellCompletionRange.completionStart;
+                const end = shellCompletionRange.completionEnd;
+                let textToInsert = textToInsertBase;
+                const charAfterCompletion = currentLine[end];
+                if (charAfterCompletion !== " " &&
+                    !textToInsert.endsWith("/") &&
+                    !textToInsert.endsWith("\\")) {
+                    textToInsert += " ";
+                }
+                const newText = currentLine.substring(0, start) +
+                    textToInsert +
+                    currentLine.substring(end);
+                return {
+                    text: newText,
+                    isActive: true,
+                    isLoading: false,
+                    accept: () => {
+                        buffer.replaceRangeByOffset(logicalPosToOffset(buffer.lines, cursorRow, start), logicalPosToOffset(buffer.lines, cursorRow, end), textToInsert);
+                    },
+                    clear: () => { },
+                    markSelected: () => { },
+                };
+            }
+        }
+        return basePromptCompletion;
+    }, [
+        completionMode,
+        suggestions,
+        query,
+        basePromptCompletion,
+        buffer,
+        cursorRow,
+        shellCompletionRange,
+    ]);
+    useEffect(() => {
+        setActiveSuggestionIndex(suggestions.length > 0 ? 0 : -1);
+        setVisibleStartIndex(0);
+        // Generic perfect match detection for non-slash modes or as a fallback
+        if (completionMode !== CompletionMode.SLASH) {
+            if (suggestions.length > 0) {
+                const firstSuggestion = suggestions[0];
+                setIsPerfectMatch(firstSuggestion.value === query);
+            }
+            else {
+                setIsPerfectMatch(false);
+            }
+        }
+    }, [
+        suggestions,
+        setActiveSuggestionIndex,
+        setVisibleStartIndex,
+        completionMode,
+        query,
+        setIsPerfectMatch,
+    ]);
+    useEffect(() => {
+        if (!active ||
+            completionMode === CompletionMode.IDLE ||
+            reverseSearchActive) {
+            resetCompletionState();
+        }
+    }, [active, completionMode, reverseSearchActive, resetCompletionState]);
+    const showSuggestions = active &&
+        completionMode !== CompletionMode.IDLE &&
+        !reverseSearchActive &&
+        isShellSuggestionsVisible &&
+        (isLoadingSuggestions || suggestions.length > 0);
+    /**
+     * Gets the completed text by replacing the completion range with the suggestion value.
+     * This is the core string replacement logic used by both autocomplete and auto-execute.
+     *
+     * @param suggestion The suggestion to apply
+     * @returns The completed text with the suggestion applied, or null if invalid
+     */
+    const getCompletedText = useCallback((suggestion) => {
+        const currentLine = buffer.lines[cursorRow] || "";
+        let start = completionStart;
+        let end = completionEnd;
+        if (completionMode === CompletionMode.SLASH) {
+            start = slashCompletionRange.completionStart;
+            end = slashCompletionRange.completionEnd;
+        }
+        else if (completionMode === CompletionMode.SHELL) {
+            start = shellCompletionRange.completionStart;
+            end = shellCompletionRange.completionEnd;
+        }
+        if (start === -1 || end === -1) {
+            return null;
+        }
+        // Apply space padding for slash commands (needed for subcommands like "/chat list")
+        let suggestionText = suggestion.insertValue ?? suggestion.value;
+        if (completionMode === CompletionMode.SLASH) {
+            // Add leading space if completing a subcommand (cursor is after parent command with no space)
+            if (start === end && start > 1 && currentLine[start - 1] !== " ") {
+                suggestionText = ` ${suggestionText}`;
+            }
+        }
+        // Build the completed text with proper spacing
+        return (currentLine.substring(0, start) +
+            suggestionText +
+            currentLine.substring(end));
+    }, [
+        cursorRow,
+        buffer.lines,
+        completionMode,
+        completionStart,
+        completionEnd,
+        slashCompletionRange,
+        shellCompletionRange,
+    ]);
+    const handleAutocomplete = useCallback((indexToUse) => {
+        if (indexToUse < 0 || indexToUse >= suggestions.length) {
+            return;
+        }
+        const suggestion = suggestions[indexToUse];
+        const completedText = getCompletedText(suggestion);
+        if (completedText === null) {
+            return;
+        }
+        let start = completionStart;
+        let end = completionEnd;
+        if (completionMode === CompletionMode.SLASH) {
+            start = slashCompletionRange.completionStart;
+            end = slashCompletionRange.completionEnd;
+        }
+        else if (completionMode === CompletionMode.SHELL) {
+            start = shellCompletionRange.completionStart;
+            end = shellCompletionRange.completionEnd;
+        }
+        // Add space padding for Tab completion (auto-execute gets padding from getCompletedText)
+        let suggestionText = suggestion.insertValue ?? suggestion.value;
+        if (completionMode === CompletionMode.SLASH) {
+            if (start === end &&
+                start > 1 &&
+                (buffer.lines[cursorRow] || "")[start - 1] !== " ") {
+                suggestionText = ` ${suggestionText}`;
+            }
+        }
+        const lineCodePoints = toCodePoints(buffer.lines[cursorRow] || "");
+        const charAfterCompletion = lineCodePoints[end];
+        let shouldAddSpace = true;
+        if (completionMode === CompletionMode.SLASH) {
+            const command = slashCompletionRange.getCommandFromSuggestion(suggestion);
+            // Don't add a space if the command has an action (can be executed)
+            // and doesn't have a completion function (doesn't REQUIRE more arguments)
+            const isExecutableCommand = !!command?.action;
+            const requiresArguments = !!command?.completion;
+            shouldAddSpace = !isExecutableCommand || requiresArguments;
+        }
+        if (charAfterCompletion !== " " &&
+            !suggestionText.endsWith("/") &&
+            !suggestionText.endsWith("\\") &&
+            shouldAddSpace) {
+            suggestionText += " ";
+        }
+        buffer.replaceRangeByOffset(logicalPosToOffset(buffer.lines, cursorRow, start), logicalPosToOffset(buffer.lines, cursorRow, end), suggestionText);
+    }, [
+        cursorRow,
+        buffer,
+        suggestions,
+        completionMode,
+        completionStart,
+        completionEnd,
+        slashCompletionRange,
+        shellCompletionRange,
+        getCompletedText,
+    ]);
+    return {
+        suggestions,
+        activeSuggestionIndex,
+        visibleStartIndex,
+        showSuggestions,
+        isLoadingSuggestions,
+        isPerfectMatch,
+        forceShowShellSuggestions,
+        setForceShowShellSuggestions,
+        isShellSuggestionsVisible,
+        setActiveSuggestionIndex,
+        resetCompletionState,
+        navigateUp,
+        navigateDown,
+        handleAutocomplete,
+        promptCompletion,
+        getCommandFromSuggestion: slashCompletionRange.getCommandFromSuggestion,
+        slashCompletionRange,
+        getCompletedText,
+        completionMode,
+    };
+}
+//# sourceMappingURL=useCommandCompletion.js.map

package/build/cli/hooks/useCommandCompletion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useCommandCompletion.js","sourceRoot":"","sources":["../../../src/cli/hooks/useCommandCompletion.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EACN,4BAA4B,EAE5B,mBAAmB,GACnB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AAEtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAMlE,MAAM,CAAN,IAAY,cAMX;AAND,WAAY,cAAc;IACzB,+BAAa,CAAA;IACb,2BAAS,CAAA;IACT,iCAAe,CAAA;IACf,mCAAiB,CAAA;IACjB,iCAAe,CAAA;AAChB,CAAC,EANW,cAAc,KAAd,cAAc,QAMzB;AA6CD,MAAM,UAAU,oBAAoB,CAAC,EACpC,MAAM,EACN,GAAG,EACH,aAAa,EACb,cAAc,EACd,mBAAmB,GAAG,KAAK,EAC3B,eAAe,EACf,MAAM,EACN,MAAM,GACuB;IAC7B,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAC9D,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEjB,MAAM,EACL,WAAW,EACX,qBAAqB,EACrB,iBAAiB,EACjB,oBAAoB,EACpB,cAAc,EAEd,cAAc,EACd,wBAAwB,EACxB,uBAAuB,EACvB,iBAAiB,EACjB,oBAAoB,EAEpB,oBAAoB,EAAE,wBAAwB,EAC9C,UAAU,EACV,YAAY,GACZ,GAAG,aAAa,EAAE,CAAC;IAEpB,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7C,wBAAwB,EAAE,CAAC;QAC3B,4BAA4B,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC;IAE/B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAEnC,MAAM,EACL,cAAc,EACd,KAAK,EAAE,SAAS,EAChB,eAAe,EACf,aAAa,GACb,GAAG,OAAO,CAAC,GAAG,EAAE;QAChB,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;QAE7C,IAAI,eAAe,EAAE,CAAC;YACrB,OAAO;gBACN,cAAc,EACb,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;oBAC9B,CAAC,CAAC,cAAc,CAAC,IAAI;oBACrB,CAAC,CAAC,cAAc,CAAC,KAAK;gBACxB,KAAK,EAAE,EAAE;gBACT,eAAe,EAAE,CAAC,CAAC;gBACnB,aAAa,EAAE,CAAC,CAAC;aACjB,CAAC;QACH,CAAC;QAED,6DAA6D;QAC7D,mEAAmE;QACnE,+DAA+D;QAC/D,KAAK,IAAI,CAAC,GAAG,SAAS,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAE3B,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBAClB,IAAI,cAAc,GAAG,CAAC,CAAC;gBACvB,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC3D,cAAc,EAAE,CAAC;gBAClB,CAAC;gBACD,IAAI,cAAc,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,MAAM;gBACP,CAAC;YACF,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACzB,IAAI,GAAG,GAAG,UAAU,CAAC,MAAM,CAAC;gBAC5B,KAAK,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACpD,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;wBAC3B,IAAI,cAAc,GAAG,CAAC,CAAC;wBACvB,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;4BAC3D,cAAc,EAAE,CAAC;wBAClB,CAAC;wBAED,IAAI,cAAc,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC9B,GAAG,GAAG,CAAC,CAAC;4BACR,MAAM;wBACP,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBAC1D,OAAO;oBACN,cAAc,EAAE,cAAc,CAAC,EAAE;oBACjC,KAAK,EAAE,WAAW;oBAClB,eAAe,EAAE,SAAS;oBAC1B,aAAa,EAAE,GAAG;iBAClB,CAAC;YACH,CAAC;QACF,CAAC;QAED,oEAAoE;QACpE,IAAI,SAAS,KAAK,CAAC,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YAC3D,OAAO;gBACN,cAAc,EAAE,cAAc,CAAC,KAAK;gBACpC,KAAK,EAAE,WAAW;gBAClB,eAAe,EAAE,CAAC;gBAClB,aAAa,EAAE,WAAW,CAAC,MAAM;aACjC,CAAC;QACH,CAAC;QAED,gDAAgD;QAChD,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,yBAAyB,GAAG,KAAK,CAAC;QACxC,IACC,yBAAyB;YACzB,WAAW,CAAC,MAAM,IAAI,4BAA4B;YAClD,CAAC,cAAc,CAAC,WAAW,CAAC;YAC5B,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,EACzB,CAAC;YACF,OAAO;gBACN,cAAc,EAAE,cAAc,CAAC,MAAM;gBACrC,KAAK,EAAE,WAAW;gBAClB,eAAe,EAAE,CAAC;gBAClB,aAAa,EAAE,WAAW,CAAC,MAAM;aACjC,CAAC;QACH,CAAC;QAED,OAAO;YACN,cAAc,EAAE,cAAc,CAAC,IAAI;YACnC,KAAK,EAAE,IAAI;YACX,eAAe,EAAE,CAAC,CAAC;YACnB,aAAa,EAAE,CAAC,CAAC;SACjB,CAAC;IACH,CAAC,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAEvE,eAAe,CAAC;QACf,OAAO,EAAE,MAAM,IAAI,cAAc,KAAK,cAAc,CAAC,EAAE;QACvD,OAAO,EAAE,SAAS,IAAI,EAAE;QACxB,MAAM;QACN,GAAG;QACH,cAAc;QACd,uBAAuB;KACvB,CAAC,CAAC;IAEH,MAAM,oBAAoB,GAAG,kBAAkB,CAAC;QAC/C,OAAO,EACN,MAAM,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,IAAI,CAAC,eAAe;QACtE,KAAK,EAAE,SAAS;QAChB,aAAa;QACb,cAAc;QACd,cAAc;QACd,uBAAuB;QACvB,iBAAiB;KACjB,CAAC,CAAC;IAEH,MAAM,oBAAoB,GAAG,kBAAkB,CAAC;QAC/C,OAAO,EAAE,MAAM,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK;QAC1D,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE;QACnC,SAAS;QACT,GAAG;QACH,cAAc;QACd,uBAAuB;KACvB,CAAC,CAAC;IAEH,MAAM,KAAK,GACV,cAAc,KAAK,cAAc,CAAC,KAAK;QACtC,CAAC,CAAC,oBAAoB,CAAC,KAAK;QAC5B,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,oBAAoB,GAAG,mBAAmB,CAAC;QAChD,MAAM;KACN,CAAC,CAAC;IAEH,MAAM,yBAAyB,GAC9B,cAAc,KAAK,cAAc,CAAC,KAAK,IAAI,yBAAyB,CAAC;IAEtE,MAAM,gBAAgB,GAAG,OAAO,CAAC,GAAG,EAAE;QACrC,IACC,cAAc,KAAK,cAAc,CAAC,KAAK;YACvC,WAAW,CAAC,MAAM,KAAK,CAAC;YACxB,KAAK,IAAI,IAAI;YACb,oBAAoB,CAAC,eAAe,KAAK,oBAAoB,CAAC,WAAW,EACxE,CAAC;YACF,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,gBAAgB,GAAG,UAAU,CAAC,KAAK,CAAC;YAE1C,IACC,gBAAgB,CAAC,UAAU,CAAC,KAAK,CAAC;gBAClC,gBAAgB,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,EACrC,CAAC;gBACF,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBAClD,MAAM,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;gBACnD,MAAM,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;gBAE/C,IAAI,YAAY,GAAG,gBAAgB,CAAC;gBACpC,MAAM,mBAAmB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;gBAC7C,IACC,mBAAmB,KAAK,GAAG;oBAC3B,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC;oBAC3B,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,EAC3B,CAAC;oBACF,YAAY,IAAI,GAAG,CAAC;gBACrB,CAAC;gBAED,MAAM,OAAO,GACZ,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC;oBAC/B,YAAY;oBACZ,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBAE5B,OAAO;oBACN,IAAI,EAAE,OAAO;oBACb,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,KAAK;oBAChB,MAAM,EAAE,GAAG,EAAE;wBACZ,MAAM,CAAC,oBAAoB,CAC1B,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,EAClD,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,EAChD,YAAY,CACZ,CAAC;oBACH,CAAC;oBACD,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;oBACf,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;iBACtB,CAAC;YACH,CAAC;QACF,CAAC;QACD,OAAO,oBAAoB,CAAC;IAC7B,CAAC,EAAE;QACF,cAAc;QACd,WAAW;QACX,KAAK;QACL,oBAAoB;QACpB,MAAM;QACN,SAAS;QACT,oBAAoB;KACpB,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,wBAAwB,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,oBAAoB,CAAC,CAAC,CAAC,CAAC;QAExB,uEAAuE;QACvE,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,eAAe,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;gBACvC,iBAAiB,CAAC,eAAe,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACP,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;IACF,CAAC,EAAE;QACF,WAAW;QACX,wBAAwB;QACxB,oBAAoB;QACpB,cAAc;QACd,KAAK;QACL,iBAAiB;KACjB,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,IACC,CAAC,MAAM;YACP,cAAc,KAAK,cAAc,CAAC,IAAI;YACtC,mBAAmB,EAClB,CAAC;YACF,oBAAoB,EAAE,CAAC;QACxB,CAAC;IACF,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAExE,MAAM,eAAe,GACpB,MAAM;QACN,cAAc,KAAK,cAAc,CAAC,IAAI;QACtC,CAAC,mBAAmB;QACpB,yBAAyB;QACzB,CAAC,oBAAoB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAElD;;;;;;OAMG;IACH,MAAM,gBAAgB,GAAG,WAAW,CACnC,CAAC,UAAsB,EAAiB,EAAE;QACzC,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAElD,IAAI,KAAK,GAAG,eAAe,CAAC;QAC5B,IAAI,GAAG,GAAG,aAAa,CAAC;QACxB,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;YAC7C,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC1C,CAAC;aAAM,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YACpD,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;YAC7C,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC1C,CAAC;QAED,IAAI,KAAK,KAAK,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC;QACb,CAAC;QAED,oFAAoF;QACpF,IAAI,cAAc,GAAG,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,KAAK,CAAC;QAChE,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,8FAA8F;YAC9F,IAAI,KAAK,KAAK,GAAG,IAAI,KAAK,GAAG,CAAC,IAAI,WAAW,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;gBAClE,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YACvC,CAAC;QACF,CAAC;QAED,+CAA+C;QAC/C,OAAO,CACN,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC;YAC/B,cAAc;YACd,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,CAC1B,CAAC;IACH,CAAC,EACD;QACC,SAAS;QACT,MAAM,CAAC,KAAK;QACZ,cAAc;QACd,eAAe;QACf,aAAa;QACb,oBAAoB;QACpB,oBAAoB;KACpB,CACD,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACrC,CAAC,UAAkB,EAAE,EAAE;QACtB,IAAI,UAAU,GAAG,CAAC,IAAI,UAAU,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACxD,OAAO;QACR,CAAC;QACD,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,aAAa,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAEnD,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;YAC5B,OAAO;QACR,CAAC;QAED,IAAI,KAAK,GAAG,eAAe,CAAC;QAC5B,IAAI,GAAG,GAAG,aAAa,CAAC;QACxB,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;YAC7C,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC1C,CAAC;aAAM,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YACpD,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;YAC7C,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC1C,CAAC;QAED,yFAAyF;QACzF,IAAI,cAAc,GAAG,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,KAAK,CAAC;QAChE,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,IACC,KAAK,KAAK,GAAG;gBACb,KAAK,GAAG,CAAC;gBACT,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,EACjD,CAAC;gBACF,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YACvC,CAAC;QACF,CAAC;QAED,MAAM,cAAc,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACnE,MAAM,mBAAmB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QAEhD,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC7C,MAAM,OAAO,GACZ,oBAAoB,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC;YAC3D,mEAAmE;YACnE,0EAA0E;YAC1E,MAAM,mBAAmB,GAAG,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC;YAC9C,MAAM,iBAAiB,GAAG,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC;YAChD,cAAc,GAAG,CAAC,mBAAmB,IAAI,iBAAiB,CAAC;QAC5D,CAAC;QAED,IACC,mBAAmB,KAAK,GAAG;YAC3B,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC;YAC7B,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC9B,cAAc,EACb,CAAC;YACF,cAAc,IAAI,GAAG,CAAC;QACvB,CAAC;QAED,MAAM,CAAC,oBAAoB,CAC1B,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,EAClD,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,EAChD,cAAc,CACd,CAAC;IACH,CAAC,EACD;QACC,SAAS;QACT,MAAM;QACN,WAAW;QACX,cAAc;QACd,eAAe;QACf,aAAa;QACb,oBAAoB;QACpB,oBAAoB;QACpB,gBAAgB;KAChB,CACD,CAAC;IAEF,OAAO;QACN,WAAW;QACX,qBAAqB;QACrB,iBAAiB;QACjB,eAAe;QACf,oBAAoB;QACpB,cAAc;QACd,yBAAyB;QACzB,4BAA4B;QAC5B,yBAAyB;QACzB,wBAAwB;QACxB,oBAAoB;QACpB,UAAU;QACV,YAAY;QACZ,kBAAkB;QAClB,gBAAgB;QAChB,wBAAwB,EAAE,oBAAoB,CAAC,wBAAwB;QACvE,oBAAoB;QACpB,gBAAgB;QAChB,cAAc;KACd,CAAC;AACH,CAAC"}

package/build/cli/hooks/useCompletion.d.ts

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Suggestion } from "../ui/SuggestionsDisplay.js";
+export interface UseCompletionReturn {
+    suggestions: Suggestion[];
+    activeSuggestionIndex: number;
+    visibleStartIndex: number;
+    isLoadingSuggestions: boolean;
+    isPerfectMatch: boolean;
+    setSuggestions: React.Dispatch<React.SetStateAction<Suggestion[]>>;
+    setActiveSuggestionIndex: React.Dispatch<React.SetStateAction<number>>;
+    setVisibleStartIndex: React.Dispatch<React.SetStateAction<number>>;
+    setIsLoadingSuggestions: React.Dispatch<React.SetStateAction<boolean>>;
+    setIsPerfectMatch: React.Dispatch<React.SetStateAction<boolean>>;
+    resetCompletionState: () => void;
+    navigateUp: () => void;
+    navigateDown: () => void;
+}
+export declare function useCompletion(): UseCompletionReturn;

package/build/cli/hooks/useCompletion.js

@@ -0,0 +1,84 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { MAX_SUGGESTIONS_TO_SHOW, } from "../ui/SuggestionsDisplay.js";
+import { useCallback, useState } from "react";
+export function useCompletion() {
+    const [suggestions, setSuggestions] = useState([]);
+    const [activeSuggestionIndex, setActiveSuggestionIndex] = useState(-1);
+    const [visibleStartIndex, setVisibleStartIndex] = useState(0);
+    const [isLoadingSuggestions, setIsLoadingSuggestions] = useState(false);
+    const [isPerfectMatch, setIsPerfectMatch] = useState(false);
+    const resetCompletionState = useCallback(() => {
+        setSuggestions([]);
+        setActiveSuggestionIndex(-1);
+        setVisibleStartIndex(0);
+        setIsLoadingSuggestions(false);
+        setIsPerfectMatch(false);
+    }, []);
+    const navigateUp = useCallback(() => {
+        if (suggestions.length === 0)
+            return;
+        setActiveSuggestionIndex((prevActiveIndex) => {
+            // Calculate new active index, handling wrap-around
+            const newActiveIndex = prevActiveIndex <= 0 ? suggestions.length - 1 : prevActiveIndex - 1;
+            // Adjust scroll position based on the new active index
+            setVisibleStartIndex((prevVisibleStart) => {
+                // Case 1: Wrapped around to the last item
+                if (newActiveIndex === suggestions.length - 1 &&
+                    suggestions.length > MAX_SUGGESTIONS_TO_SHOW) {
+                    return Math.max(0, suggestions.length - MAX_SUGGESTIONS_TO_SHOW);
+                }
+                // Case 2: Scrolled above the current visible window
+                if (newActiveIndex < prevVisibleStart) {
+                    return newActiveIndex;
+                }
+                // Otherwise, keep the current scroll position
+                return prevVisibleStart;
+            });
+            return newActiveIndex;
+        });
+    }, [suggestions.length]);
+    const navigateDown = useCallback(() => {
+        if (suggestions.length === 0)
+            return;
+        setActiveSuggestionIndex((prevActiveIndex) => {
+            // Calculate new active index, handling wrap-around
+            const newActiveIndex = prevActiveIndex >= suggestions.length - 1 ? 0 : prevActiveIndex + 1;
+            // Adjust scroll position based on the new active index
+            setVisibleStartIndex((prevVisibleStart) => {
+                // Case 1: Wrapped around to the first item
+                if (newActiveIndex === 0 &&
+                    suggestions.length > MAX_SUGGESTIONS_TO_SHOW) {
+                    return 0;
+                }
+                // Case 2: Scrolled below the current visible window
+                const visibleEndIndex = prevVisibleStart + MAX_SUGGESTIONS_TO_SHOW;
+                if (newActiveIndex >= visibleEndIndex) {
+                    return newActiveIndex - MAX_SUGGESTIONS_TO_SHOW + 1;
+                }
+                // Otherwise, keep the current scroll position
+                return prevVisibleStart;
+            });
+            return newActiveIndex;
+        });
+    }, [suggestions.length]);
+    return {
+        suggestions,
+        activeSuggestionIndex,
+        visibleStartIndex,
+        isLoadingSuggestions,
+        isPerfectMatch,
+        setSuggestions,
+        setActiveSuggestionIndex,
+        setVisibleStartIndex,
+        setIsLoadingSuggestions,
+        setIsPerfectMatch,
+        resetCompletionState,
+        navigateUp,
+        navigateDown,
+    };
+}
+//# sourceMappingURL=useCompletion.js.map

package/build/cli/hooks/useCompletion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useCompletion.js","sourceRoot":"","sources":["../../../src/cli/hooks/useCompletion.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACN,uBAAuB,GAEvB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAkB9C,MAAM,UAAU,aAAa;IAC5B,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAe,EAAE,CAAC,CAAC;IACjE,MAAM,CAAC,qBAAqB,EAAE,wBAAwB,CAAC,GACtD,QAAQ,CAAS,CAAC,CAAC,CAAC,CAAC;IACtB,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CAAS,CAAC,CAAC,CAAC;IACtE,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GACpD,QAAQ,CAAU,KAAK,CAAC,CAAC;IAC1B,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,CAAU,KAAK,CAAC,CAAC;IAErE,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7C,cAAc,CAAC,EAAE,CAAC,CAAC;QACnB,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7B,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACxB,uBAAuB,CAAC,KAAK,CAAC,CAAC;QAC/B,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE;QACnC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAErC,wBAAwB,CAAC,CAAC,eAAe,EAAE,EAAE;YAC5C,mDAAmD;YACnD,MAAM,cAAc,GACnB,eAAe,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC;YAErE,uDAAuD;YACvD,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,EAAE;gBACzC,0CAA0C;gBAC1C,IACC,cAAc,KAAK,WAAW,CAAC,MAAM,GAAG,CAAC;oBACzC,WAAW,CAAC,MAAM,GAAG,uBAAuB,EAC3C,CAAC;oBACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,CAAC,MAAM,GAAG,uBAAuB,CAAC,CAAC;gBAClE,CAAC;gBACD,oDAAoD;gBACpD,IAAI,cAAc,GAAG,gBAAgB,EAAE,CAAC;oBACvC,OAAO,cAAc,CAAC;gBACvB,CAAC;gBACD,8CAA8C;gBAC9C,OAAO,gBAAgB,CAAC;YACzB,CAAC,CAAC,CAAC;YAEH,OAAO,cAAc,CAAC;QACvB,CAAC,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzB,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;QACrC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAErC,wBAAwB,CAAC,CAAC,eAAe,EAAE,EAAE;YAC5C,mDAAmD;YACnD,MAAM,cAAc,GACnB,eAAe,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC;YAErE,uDAAuD;YACvD,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,EAAE;gBACzC,2CAA2C;gBAC3C,IACC,cAAc,KAAK,CAAC;oBACpB,WAAW,CAAC,MAAM,GAAG,uBAAuB,EAC3C,CAAC;oBACF,OAAO,CAAC,CAAC;gBACV,CAAC;gBACD,oDAAoD;gBACpD,MAAM,eAAe,GAAG,gBAAgB,GAAG,uBAAuB,CAAC;gBACnE,IAAI,cAAc,IAAI,eAAe,EAAE,CAAC;oBACvC,OAAO,cAAc,GAAG,uBAAuB,GAAG,CAAC,CAAC;gBACrD,CAAC;gBACD,8CAA8C;gBAC9C,OAAO,gBAAgB,CAAC;YACzB,CAAC,CAAC,CAAC;YAEH,OAAO,cAAc,CAAC;QACvB,CAAC,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;IACzB,OAAO;QACN,WAAW;QACX,qBAAqB;QACrB,iBAAiB;QACjB,oBAAoB;QACpB,cAAc;QAEd,cAAc;QACd,wBAAwB;QACxB,oBAAoB;QACpB,uBAAuB;QACvB,iBAAiB;QAEjB,oBAAoB;QACpB,UAAU;QACV,YAAY;KACZ,CAAC;AACH,CAAC"}

package/build/cli/hooks/useDispatcher.d.ts

@@ -0,0 +1,6 @@
+import type { McpToolInfo } from "../services/McpClientService.js";
+import type { CallToolFn, DispatchResult } from "../types.js";
+export interface UseDispatcherReturn {
+    handleDispatch: (text: string, callTool: CallToolFn) => Promise<DispatchResult>;
+}
+export declare function useDispatcher(availableTools?: McpToolInfo[]): UseDispatcherReturn;