@machina.ai/cell-cli 1.0.13-rc9 → 1.0.21-rc1

package/dist/src/ui/hooks/useAtCompletion.js

@@ -0,0 +1,178 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useEffect, useReducer, useRef } from 'react';
+import { FileSearchFactory, escapePath, } from '@machina.ai/cell-cli-core';
+import { MAX_SUGGESTIONS_TO_SHOW, } from '../components/SuggestionsDisplay.js';
+export var AtCompletionStatus;
+(function (AtCompletionStatus) {
+    AtCompletionStatus["IDLE"] = "idle";
+    AtCompletionStatus["INITIALIZING"] = "initializing";
+    AtCompletionStatus["READY"] = "ready";
+    AtCompletionStatus["SEARCHING"] = "searching";
+    AtCompletionStatus["ERROR"] = "error";
+})(AtCompletionStatus || (AtCompletionStatus = {}));
+const initialState = {
+    status: AtCompletionStatus.IDLE,
+    suggestions: [],
+    isLoading: false,
+    pattern: null,
+};
+function atCompletionReducer(state, action) {
+    switch (action.type) {
+        case 'INITIALIZE':
+            return {
+                ...state,
+                status: AtCompletionStatus.INITIALIZING,
+                isLoading: true,
+            };
+        case 'INITIALIZE_SUCCESS':
+            return { ...state, status: AtCompletionStatus.READY, isLoading: false };
+        case 'SEARCH':
+            // Keep old suggestions, don't set loading immediately
+            return {
+                ...state,
+                status: AtCompletionStatus.SEARCHING,
+                pattern: action.payload,
+            };
+        case 'SEARCH_SUCCESS':
+            return {
+                ...state,
+                status: AtCompletionStatus.READY,
+                suggestions: action.payload,
+                isLoading: false,
+            };
+        case 'SET_LOADING':
+            // Only show loading if we are still in a searching state
+            if (state.status === AtCompletionStatus.SEARCHING) {
+                return { ...state, isLoading: action.payload, suggestions: [] };
+            }
+            return state;
+        case 'ERROR':
+            return {
+                ...state,
+                status: AtCompletionStatus.ERROR,
+                isLoading: false,
+                suggestions: [],
+            };
+        case 'RESET':
+            return initialState;
+        default:
+            return state;
+    }
+}
+export function useAtCompletion(props) {
+    const { enabled, pattern, config, cwd, setSuggestions, setIsLoadingSuggestions, } = props;
+    const [state, dispatch] = useReducer(atCompletionReducer, initialState);
+    const fileSearch = useRef(null);
+    const searchAbortController = useRef(null);
+    const slowSearchTimer = useRef(null);
+    useEffect(() => {
+        setSuggestions(state.suggestions);
+    }, [state.suggestions, setSuggestions]);
+    useEffect(() => {
+        setIsLoadingSuggestions(state.isLoading);
+    }, [state.isLoading, setIsLoadingSuggestions]);
+    useEffect(() => {
+        dispatch({ type: 'RESET' });
+    }, [cwd, config]);
+    // Reacts to user input (`pattern`) ONLY.
+    useEffect(() => {
+        if (!enabled) {
+            // reset when first getting out of completion suggestions
+            if (state.status === AtCompletionStatus.READY ||
+                state.status === AtCompletionStatus.ERROR) {
+                dispatch({ type: 'RESET' });
+            }
+            return;
+        }
+        if (pattern === null) {
+            dispatch({ type: 'RESET' });
+            return;
+        }
+        if (state.status === AtCompletionStatus.IDLE) {
+            dispatch({ type: 'INITIALIZE' });
+        }
+        else if ((state.status === AtCompletionStatus.READY ||
+            state.status === AtCompletionStatus.SEARCHING) &&
+            pattern !== state.pattern // Only search if the pattern has changed
+        ) {
+            dispatch({ type: 'SEARCH', payload: pattern });
+        }
+    }, [enabled, pattern, state.status, state.pattern]);
+    // The "Worker" that performs async operations based on status.
+    useEffect(() => {
+        const initialize = async () => {
+            try {
+                const searcher = FileSearchFactory.create({
+                    projectRoot: cwd,
+                    ignoreDirs: [],
+                    useGitignore: config?.getFileFilteringOptions()?.respectGitIgnore ?? true,
+                    useGeminiignore: config?.getFileFilteringOptions()?.respectGeminiIgnore ?? true,
+                    cache: true,
+                    cacheTtl: 30, // 30 seconds
+                    enableRecursiveFileSearch: config?.getEnableRecursiveFileSearch() ?? true,
+                });
+                await searcher.initialize();
+                fileSearch.current = searcher;
+                dispatch({ type: 'INITIALIZE_SUCCESS' });
+                if (state.pattern !== null) {
+                    dispatch({ type: 'SEARCH', payload: state.pattern });
+                }
+            }
+            catch (_) {
+                dispatch({ type: 'ERROR' });
+            }
+        };
+        const search = async () => {
+            if (!fileSearch.current || state.pattern === null) {
+                return;
+            }
+            if (slowSearchTimer.current) {
+                clearTimeout(slowSearchTimer.current);
+            }
+            const controller = new AbortController();
+            searchAbortController.current = controller;
+            slowSearchTimer.current = setTimeout(() => {
+                dispatch({ type: 'SET_LOADING', payload: true });
+            }, 200);
+            try {
+                const results = await fileSearch.current.search(state.pattern, {
+                    signal: controller.signal,
+                    maxResults: MAX_SUGGESTIONS_TO_SHOW * 3,
+                });
+                if (slowSearchTimer.current) {
+                    clearTimeout(slowSearchTimer.current);
+                }
+                if (controller.signal.aborted) {
+                    return;
+                }
+                const suggestions = results.map((p) => ({
+                    label: p,
+                    value: escapePath(p),
+                }));
+                dispatch({ type: 'SEARCH_SUCCESS', payload: suggestions });
+            }
+            catch (error) {
+                if (!(error instanceof Error && error.name === 'AbortError')) {
+                    dispatch({ type: 'ERROR' });
+                }
+            }
+        };
+        if (state.status === AtCompletionStatus.INITIALIZING) {
+            initialize();
+        }
+        else if (state.status === AtCompletionStatus.SEARCHING) {
+            search();
+        }
+        return () => {
+            searchAbortController.current?.abort();
+            if (slowSearchTimer.current) {
+                clearTimeout(slowSearchTimer.current);
+            }
+        };
+    }, [state.status, state.pattern, config, cwd]);
+}
+//# sourceMappingURL=useAtCompletion.js.map

package/dist/src/ui/hooks/useAtCompletion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useAtCompletion.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useAtCompletion.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AACtD,OAAO,EAGL,iBAAiB,EACjB,UAAU,GACX,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,uBAAuB,GACxB,MAAM,qCAAqC,CAAC;AAE7C,MAAM,CAAN,IAAY,kBAMX;AAND,WAAY,kBAAkB;IAC5B,mCAAa,CAAA;IACb,mDAA6B,CAAA;IAC7B,qCAAe,CAAA;IACf,6CAAuB,CAAA;IACvB,qCAAe,CAAA;AACjB,CAAC,EANW,kBAAkB,KAAlB,kBAAkB,QAM7B;AAkBD,MAAM,YAAY,GAAsB;IACtC,MAAM,EAAE,kBAAkB,CAAC,IAAI;IAC/B,WAAW,EAAE,EAAE;IACf,SAAS,EAAE,KAAK;IAChB,OAAO,EAAE,IAAI;CACd,CAAC;AAEF,SAAS,mBAAmB,CAC1B,KAAwB,EACxB,MAA0B;IAE1B,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,YAAY;YACf,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM,EAAE,kBAAkB,CAAC,YAAY;gBACvC,SAAS,EAAE,IAAI;aAChB,CAAC;QACJ,KAAK,oBAAoB;YACvB,OAAO,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,kBAAkB,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC1E,KAAK,QAAQ;YACX,sDAAsD;YACtD,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM,EAAE,kBAAkB,CAAC,SAAS;gBACpC,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC;QACJ,KAAK,gBAAgB;YACnB,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM,EAAE,kBAAkB,CAAC,KAAK;gBAChC,WAAW,EAAE,MAAM,CAAC,OAAO;gBAC3B,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,KAAK,aAAa;YAChB,yDAAyD;YACzD,IAAI,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,SAAS,EAAE,CAAC;gBAClD,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;YAClE,CAAC;YACD,OAAO,KAAK,CAAC;QACf,KAAK,OAAO;YACV,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM,EAAE,kBAAkB,CAAC,KAAK;gBAChC,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,EAAE;aAChB,CAAC;QACJ,KAAK,OAAO;YACV,OAAO,YAAY,CAAC;QACtB;YACE,OAAO,KAAK,CAAC;IACjB,CAAC;AACH,CAAC;AAWD,MAAM,UAAU,eAAe,CAAC,KAA2B;IACzD,MAAM,EACJ,OAAO,EACP,OAAO,EACP,MAAM,EACN,GAAG,EACH,cAAc,EACd,uBAAuB,GACxB,GAAG,KAAK,CAAC;IACV,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,UAAU,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;IACxE,MAAM,UAAU,GAAG,MAAM,CAAoB,IAAI,CAAC,CAAC;IACnD,MAAM,qBAAqB,GAAG,MAAM,CAAyB,IAAI,CAAC,CAAC;IACnE,MAAM,eAAe,GAAG,MAAM,CAAwB,IAAI,CAAC,CAAC;IAE5D,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IACpC,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,SAAS,CAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC,CAAC;IAE/C,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9B,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;IAElB,yCAAyC;IACzC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,yDAAyD;YACzD,IACE,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,KAAK;gBACzC,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,KAAK,EACzC,CAAC;gBACD,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9B,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACrB,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,IAAI,EAAE,CAAC;YAC7C,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;QACnC,CAAC;aAAM,IACL,CAAC,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,KAAK;YACxC,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,SAAS,CAAC;YAChD,OAAO,KAAK,KAAK,CAAC,OAAO,CAAC,yCAAyC;UACnE,CAAC;YACD,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;IAEpD,+DAA+D;IAC/D,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;YAC5B,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC,MAAM,CAAC;oBACxC,WAAW,EAAE,GAAG;oBAChB,UAAU,EAAE,EAAE;oBACd,YAAY,EACV,MAAM,EAAE,uBAAuB,EAAE,EAAE,gBAAgB,IAAI,IAAI;oBAC7D,eAAe,EACb,MAAM,EAAE,uBAAuB,EAAE,EAAE,mBAAmB,IAAI,IAAI;oBAChE,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE,EAAE,EAAE,aAAa;oBAC3B,yBAAyB,EACvB,MAAM,EAAE,4BAA4B,EAAE,IAAI,IAAI;iBACjD,CAAC,CAAC;gBACH,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;gBAC5B,UAAU,CAAC,OAAO,GAAG,QAAQ,CAAC;gBAC9B,QAAQ,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAAC;gBACzC,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;oBAC3B,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,MAAM,GAAG,KAAK,IAAI,EAAE;YACxB,IAAI,CAAC,UAAU,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;gBAClD,OAAO;YACT,CAAC;YAED,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;gBAC5B,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,qBAAqB,CAAC,OAAO,GAAG,UAAU,CAAC;YAE3C,eAAe,CAAC,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBACxC,QAAQ,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACnD,CAAC,EAAE,GAAG,CAAC,CAAC;YAER,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE;oBAC7D,MAAM,EAAE,UAAU,CAAC,MAAM;oBACzB,UAAU,EAAE,uBAAuB,GAAG,CAAC;iBACxC,CAAC,CAAC;gBAEH,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;oBAC5B,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACxC,CAAC;gBAED,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAC9B,OAAO;gBACT,CAAC;gBAED,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACtC,KAAK,EAAE,CAAC;oBACR,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;iBACrB,CAAC,CAAC,CAAC;gBACJ,QAAQ,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,EAAE,CAAC;oBAC7D,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,YAAY,EAAE,CAAC;YACrD,UAAU,EAAE,CAAC;QACf,CAAC;aAAM,IAAI,KAAK,CAAC,MAAM,KAAK,kBAAkB,CAAC,SAAS,EAAE,CAAC;YACzD,MAAM,EAAE,CAAC;QACX,CAAC;QAED,OAAO,GAAG,EAAE;YACV,qBAAqB,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YACvC,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;gBAC5B,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;AACjD,CAAC"}

package/dist/src/ui/hooks/useAutoAcceptIndicator.js

@@ -4,23 +4,23 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { useState, useEffect } from 'react';
-import { useInput } from 'ink';
 import { ApprovalMode } from '@machina.ai/cell-cli-core';
+import { useKeypress } from './useKeypress.js';
 export function useAutoAcceptIndicator({ config, }) {
     const currentConfigValue = config.getApprovalMode();
     const [showAutoAcceptIndicator, setShowAutoAcceptIndicator] = useState(currentConfigValue);
     useEffect(() => {
         setShowAutoAcceptIndicator(currentConfigValue);
     }, [currentConfigValue]);
-    useInput((input, key) => {
+    useKeypress((key) => {
         let nextApprovalMode;
-        if (key.ctrl && input === 'y') {
+        if (key.ctrl && key.name === 'y') {
             nextApprovalMode =
                 config.getApprovalMode() === ApprovalMode.YOLO
                     ? ApprovalMode.DEFAULT
                     : ApprovalMode.YOLO;
         }
-        else if (key.tab && key.shift) {
+        else if (key.shift && key.name === 'tab') {
             nextApprovalMode =
                 config.getApprovalMode() === ApprovalMode.AUTO_EDIT
                     ? ApprovalMode.DEFAULT
@@ -31,7 +31,7 @@ export function useAutoAcceptIndicator({ config, }) {
             // Update local state immediately for responsiveness
             setShowAutoAcceptIndicator(nextApprovalMode);
         }
-    });
+    }, { isActive: true });
     return showAutoAcceptIndicator;
 }
 //# sourceMappingURL=useAutoAcceptIndicator.js.map

package/dist/src/ui/hooks/useAutoAcceptIndicator.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAutoAcceptIndicator.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useAutoAcceptIndicator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAe,MAAM,2BAA2B,CAAC;AAMtE,MAAM,UAAU,sBAAsB,CAAC,EACrC,MAAM,GACqB;IAC3B,MAAM,kBAAkB,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;IACpD,MAAM,CAAC,uBAAuB,EAAE,0BAA0B,CAAC,GACzD,QAAQ,CAAC,kBAAkB,CAAC,CAAC;IAE/B,SAAS,CAAC,GAAG,EAAE;QACb,0BAA0B,CAAC,kBAAkB,CAAC,CAAC;IACjD,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAEzB,QAAQ,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACtB,IAAI,gBAA0C,CAAC;QAE/C,IAAI,GAAG,CAAC,IAAI,IAAI,KAAK,KAAK,GAAG,EAAE,CAAC;YAC9B,gBAAgB;gBACd,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,IAAI;oBAC5C,CAAC,CAAC,YAAY,CAAC,OAAO;oBACtB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC;QAC1B,CAAC;aAAM,IAAI,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YAChC,gBAAgB;gBACd,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,SAAS;oBACjD,CAAC,CAAC,YAAY,CAAC,OAAO;oBACtB,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC;QAC/B,CAAC;QAED,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YACzC,oDAAoD;YACpD,0BAA0B,CAAC,gBAAgB,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,uBAAuB,CAAC;AACjC,CAAC"}
+{"version":3,"file":"useAutoAcceptIndicator.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useAutoAcceptIndicator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAe,MAAM,2BAA2B,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAM/C,MAAM,UAAU,sBAAsB,CAAC,EACrC,MAAM,GACqB;IAC3B,MAAM,kBAAkB,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;IACpD,MAAM,CAAC,uBAAuB,EAAE,0BAA0B,CAAC,GACzD,QAAQ,CAAC,kBAAkB,CAAC,CAAC;IAE/B,SAAS,CAAC,GAAG,EAAE;QACb,0BAA0B,CAAC,kBAAkB,CAAC,CAAC;IACjD,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAEzB,WAAW,CACT,CAAC,GAAG,EAAE,EAAE;QACN,IAAI,gBAA0C,CAAC;QAE/C,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;YACjC,gBAAgB;gBACd,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,IAAI;oBAC5C,CAAC,CAAC,YAAY,CAAC,OAAO;oBACtB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC;QAC1B,CAAC;aAAM,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAC3C,gBAAgB;gBACd,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,SAAS;oBACjD,CAAC,CAAC,YAAY,CAAC,OAAO;oBACtB,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC;QAC/B,CAAC;QAED,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YACzC,oDAAoD;YACpD,0BAA0B,CAAC,gBAAgB,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC,EACD,EAAE,QAAQ,EAAE,IAAI,EAAE,CACnB,CAAC;IAEF,OAAO,uBAAuB,CAAC;AACjC,CAAC"}

package/dist/src/ui/hooks/useAutoAcceptIndicator.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/ui/hooks/useAutoAcceptIndicator.test.js

@@ -0,0 +1,191 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, } from 'vitest';
+import { renderHook, act } from '@testing-library/react';
+import { useAutoAcceptIndicator } from './useAutoAcceptIndicator.js';
+import { Config, ApprovalMode, } from '@machina.ai/cell-cli-core';
+import { useKeypress } from './useKeypress.js';
+vi.mock('./useKeypress.js');
+vi.mock('@machina.ai/cell-cli-core', async () => {
+    const actualServerModule = (await vi.importActual('@machina.ai/cell-cli-core'));
+    return {
+        ...actualServerModule,
+        Config: vi.fn(),
+    };
+});
+describe('useAutoAcceptIndicator', () => {
+    let mockConfigInstance;
+    let capturedUseKeypressHandler;
+    let mockedUseKeypress;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        Config.mockImplementation(() => {
+            const instanceGetApprovalModeMock = vi.fn();
+            const instanceSetApprovalModeMock = vi.fn();
+            const instance = {
+                getApprovalMode: instanceGetApprovalModeMock,
+                setApprovalMode: instanceSetApprovalModeMock,
+                getCoreTools: vi.fn().mockReturnValue([]),
+                getToolDiscoveryCommand: vi.fn().mockReturnValue(undefined),
+                getTargetDir: vi.fn().mockReturnValue('.'),
+                getApiKey: vi.fn().mockReturnValue('test-api-key'),
+                getModel: vi.fn().mockReturnValue('test-model'),
+                getSandbox: vi.fn().mockReturnValue(false),
+                getDebugMode: vi.fn().mockReturnValue(false),
+                getQuestion: vi.fn().mockReturnValue(undefined),
+                getFullContext: vi.fn().mockReturnValue(false),
+                getUserAgent: vi.fn().mockReturnValue('test-user-agent'),
+                getUserMemory: vi.fn().mockReturnValue(''),
+                getGeminiMdFileCount: vi.fn().mockReturnValue(0),
+                getToolRegistry: vi
+                    .fn()
+                    .mockReturnValue({ discoverTools: vi.fn() }),
+            };
+            instanceSetApprovalModeMock.mockImplementation((value) => {
+                instanceGetApprovalModeMock.mockReturnValue(value);
+            });
+            return instance;
+        });
+        mockedUseKeypress = useKeypress;
+        mockedUseKeypress.mockImplementation((handler, _options) => {
+            capturedUseKeypressHandler = handler;
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        mockConfigInstance = new Config();
+    });
+    it('should initialize with ApprovalMode.AUTO_EDIT if config.getApprovalMode returns ApprovalMode.AUTO_EDIT', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.AUTO_EDIT);
+        const { result } = renderHook(() => useAutoAcceptIndicator({
+            config: mockConfigInstance,
+        }));
+        expect(result.current).toBe(ApprovalMode.AUTO_EDIT);
+        expect(mockConfigInstance.getApprovalMode).toHaveBeenCalledTimes(1);
+    });
+    it('should initialize with ApprovalMode.DEFAULT if config.getApprovalMode returns ApprovalMode.DEFAULT', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.DEFAULT);
+        const { result } = renderHook(() => useAutoAcceptIndicator({
+            config: mockConfigInstance,
+        }));
+        expect(result.current).toBe(ApprovalMode.DEFAULT);
+        expect(mockConfigInstance.getApprovalMode).toHaveBeenCalledTimes(1);
+    });
+    it('should initialize with ApprovalMode.YOLO if config.getApprovalMode returns ApprovalMode.YOLO', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.YOLO);
+        const { result } = renderHook(() => useAutoAcceptIndicator({
+            config: mockConfigInstance,
+        }));
+        expect(result.current).toBe(ApprovalMode.YOLO);
+        expect(mockConfigInstance.getApprovalMode).toHaveBeenCalledTimes(1);
+    });
+    it('should toggle the indicator and update config when Shift+Tab or Ctrl+Y is pressed', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.DEFAULT);
+        const { result } = renderHook(() => useAutoAcceptIndicator({
+            config: mockConfigInstance,
+        }));
+        expect(result.current).toBe(ApprovalMode.DEFAULT);
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'tab',
+                shift: true,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.AUTO_EDIT);
+        expect(result.current).toBe(ApprovalMode.AUTO_EDIT);
+        act(() => {
+            capturedUseKeypressHandler({ name: 'y', ctrl: true });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.YOLO);
+        expect(result.current).toBe(ApprovalMode.YOLO);
+        act(() => {
+            capturedUseKeypressHandler({ name: 'y', ctrl: true });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.DEFAULT);
+        expect(result.current).toBe(ApprovalMode.DEFAULT);
+        act(() => {
+            capturedUseKeypressHandler({ name: 'y', ctrl: true });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.YOLO);
+        expect(result.current).toBe(ApprovalMode.YOLO);
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'tab',
+                shift: true,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.AUTO_EDIT);
+        expect(result.current).toBe(ApprovalMode.AUTO_EDIT);
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'tab',
+                shift: true,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.DEFAULT);
+        expect(result.current).toBe(ApprovalMode.DEFAULT);
+    });
+    it('should not toggle if only one key or other keys combinations are pressed', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.DEFAULT);
+        renderHook(() => useAutoAcceptIndicator({
+            config: mockConfigInstance,
+        }));
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'tab',
+                shift: false,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'unknown',
+                shift: true,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'a',
+                shift: false,
+                ctrl: false,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({ name: 'y', ctrl: false });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({ name: 'a', ctrl: true });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({ name: 'y', shift: true });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+        act(() => {
+            capturedUseKeypressHandler({
+                name: 'a',
+                ctrl: true,
+                shift: true,
+            });
+        });
+        expect(mockConfigInstance.setApprovalMode).not.toHaveBeenCalled();
+    });
+    it('should update indicator when config value changes externally (useEffect dependency)', () => {
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.DEFAULT);
+        const { result, rerender } = renderHook((props) => useAutoAcceptIndicator(props), {
+            initialProps: {
+                config: mockConfigInstance,
+            },
+        });
+        expect(result.current).toBe(ApprovalMode.DEFAULT);
+        mockConfigInstance.getApprovalMode.mockReturnValue(ApprovalMode.AUTO_EDIT);
+        rerender({ config: mockConfigInstance });
+        expect(result.current).toBe(ApprovalMode.AUTO_EDIT);
+        expect(mockConfigInstance.getApprovalMode).toHaveBeenCalledTimes(3);
+    });
+});
+//# sourceMappingURL=useAutoAcceptIndicator.test.js.map

package/dist/src/ui/hooks/useAutoAcceptIndicator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useAutoAcceptIndicator.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useAutoAcceptIndicator.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,GAGX,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAErE,OAAO,EACL,MAAM,EAEN,YAAY,GACb,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,WAAW,EAAO,MAAM,kBAAkB,CAAC;AAEpD,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE5B,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;IAC9C,MAAM,kBAAkB,GAAG,CAAC,MAAM,EAAE,CAAC,YAAY,CAC/C,2BAA2B,CAC5B,CAA4B,CAAC;IAC9B,OAAO;QACL,GAAG,kBAAkB;QACrB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;KAChB,CAAC;AACJ,CAAC,CAAC,CAAC;AAsBH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,kBAA2C,CAAC;IAChD,IAAI,0BAA8C,CAAC;IACnD,IAAI,iBAAqD,CAAC;IAE1D,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAGjB,MACD,CAAC,kBAAkB,CAAC,GAAG,EAAE;YACxB,MAAM,2BAA2B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,2BAA2B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAE5C,MAAM,QAAQ,GAA4B;gBACxC,eAAe,EAAE,2BAEhB;gBACD,eAAe,EAAE,2BAEhB;gBACD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAyB;gBACjE,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAEzD;gBACD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAuB;gBAChE,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,cAAc,CAEhD;gBACD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAuB;gBACrE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAExC;gBACD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAwB;gBACnE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAE7C;gBACD,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAwB;gBACrE,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAEtD;gBACD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAuB;gBAChE,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAuB;gBACtE,eAAe,EAAE,EAAE;qBAChB,EAAE,EAAE;qBACJ,eAAe,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAE5C;aACF,CAAC;YACF,2BAA2B,CAAC,kBAAkB,CAAC,CAAC,KAAmB,EAAE,EAAE;gBACrE,2BAA2B,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,iBAAiB,GAAG,WAAiD,CAAC;QACtE,iBAAiB,CAAC,kBAAkB,CAClC,CAAC,OAA2B,EAAE,QAAQ,EAAE,EAAE;YACxC,0BAA0B,GAAG,OAAO,CAAC;QACvC,CAAC,CACF,CAAC;QAEF,8DAA8D;QAC9D,kBAAkB,GAAG,IAAK,MAAc,EAA6B,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wGAAwG,EAAE,GAAG,EAAE;QAChH,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,sBAAsB,CAAC;YACrB,MAAM,EAAE,kBAAiD;SAC1D,CAAC,CACH,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oGAAoG,EAAE,GAAG,EAAE;QAC5G,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACzE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,sBAAsB,CAAC;YACrB,MAAM,EAAE,kBAAiD;SAC1D,CAAC,CACH,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8FAA8F,EAAE,GAAG,EAAE;QACtG,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACtE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,sBAAsB,CAAC;YACrB,MAAM,EAAE,kBAAiD;SAC1D,CAAC,CACH,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;QAC3F,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACzE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,sBAAsB,CAAC;YACrB,MAAM,EAAE,kBAAiD;SAC1D,CAAC,CACH,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAElD,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,IAAI;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,SAAS,CACvB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAEpD,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAS,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,IAAI,CAClB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAE/C,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAS,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,OAAO,CACrB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAElD,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAS,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,IAAI,CAClB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAE/C,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,IAAI;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,SAAS,CACvB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAEpD,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,IAAI;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,YAAY,CAAC,OAAO,CACrB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QAClF,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACzE,UAAU,CAAC,GAAG,EAAE,CACd,sBAAsB,CAAC;YACrB,MAAM,EAAE,kBAAiD;SAC1D,CAAC,CACH,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,KAAK;aACN,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,IAAI;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,GAAG;gBACT,KAAK,EAAE,KAAK;gBACZ,IAAI,EAAE,KAAK;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAS,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAS,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAS,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,0BAA0B,CAAC;gBACzB,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,IAAI;gBACV,KAAK,EAAE,IAAI;aACL,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;QAC7F,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACzE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CACrC,CAAC,KAAmC,EAAE,EAAE,CAAC,sBAAsB,CAAC,KAAK,CAAC,EACtE;YACE,YAAY,EAAE;gBACZ,MAAM,EAAE,kBAAiD;aAC1D;SACF,CACF,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAElD,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAE3E,QAAQ,CAAC,EAAE,MAAM,EAAE,kBAAiD,EAAE,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useCommandCompletion.d.ts

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Suggestion } from '../components/SuggestionsDisplay.js';
+import { CommandContext, SlashCommand } from '../commands/types.js';
+import { TextBuffer } from '../components/shared/text-buffer.js';
+import { Config } from '@machina.ai/cell-cli-core';
+export declare enum CompletionMode {
+    IDLE = "IDLE",
+    AT = "AT",
+    SLASH = "SLASH"
+}
+export interface UseCommandCompletionReturn {
+    suggestions: Suggestion[];
+    activeSuggestionIndex: number;
+    visibleStartIndex: number;
+    showSuggestions: boolean;
+    isLoadingSuggestions: boolean;
+    isPerfectMatch: boolean;
+    setActiveSuggestionIndex: React.Dispatch<React.SetStateAction<number>>;
+    setShowSuggestions: React.Dispatch<React.SetStateAction<boolean>>;
+    resetCompletionState: () => void;
+    navigateUp: () => void;
+    navigateDown: () => void;
+    handleAutocomplete: (indexToUse: number) => void;
+}
+export declare function useCommandCompletion(buffer: TextBuffer, dirs: readonly string[], cwd: string, slashCommands: readonly SlashCommand[], commandContext: CommandContext, reverseSearchActive?: boolean, config?: Config): UseCommandCompletionReturn;

package/dist/src/ui/hooks/useCommandCompletion.js

@@ -0,0 +1,165 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useCallback, useMemo, useEffect } from 'react';
+import { logicalPosToOffset, } from '../components/shared/text-buffer.js';
+import { isSlashCommand } from '../utils/commandUtils.js';
+import { toCodePoints } from '../utils/textUtils.js';
+import { useAtCompletion } from './useAtCompletion.js';
+import { useSlashCompletion } from './useSlashCompletion.js';
+import { useCompletion } from './useCompletion.js';
+export var CompletionMode;
+(function (CompletionMode) {
+    CompletionMode["IDLE"] = "IDLE";
+    CompletionMode["AT"] = "AT";
+    CompletionMode["SLASH"] = "SLASH";
+})(CompletionMode || (CompletionMode = {}));
+export function useCommandCompletion(buffer, dirs, cwd, slashCommands, commandContext, reverseSearchActive = false, config) {
+    const { suggestions, activeSuggestionIndex, visibleStartIndex, showSuggestions, isLoadingSuggestions, isPerfectMatch, setSuggestions, setShowSuggestions, setActiveSuggestionIndex, setIsLoadingSuggestions, setIsPerfectMatch, setVisibleStartIndex, resetCompletionState, navigateUp, navigateDown, } = useCompletion();
+    const cursorRow = buffer.cursor[0];
+    const cursorCol = buffer.cursor[1];
+    const { completionMode, query, completionStart, completionEnd } = useMemo(() => {
+        const currentLine = buffer.lines[cursorRow] || '';
+        if (cursorRow === 0 && isSlashCommand(currentLine.trim())) {
+            return {
+                completionMode: CompletionMode.SLASH,
+                query: currentLine,
+                completionStart: 0,
+                completionEnd: currentLine.length,
+            };
+        }
+        const codePoints = toCodePoints(currentLine);
+        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) {
+                    return {
+                        completionMode: CompletionMode.IDLE,
+                        query: null,
+                        completionStart: -1,
+                        completionEnd: -1,
+                    };
+                }
+            }
+            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,
+                };
+            }
+        }
+        return {
+            completionMode: CompletionMode.IDLE,
+            query: null,
+            completionStart: -1,
+            completionEnd: -1,
+        };
+    }, [cursorRow, cursorCol, buffer.lines]);
+    useAtCompletion({
+        enabled: completionMode === CompletionMode.AT,
+        pattern: query || '',
+        config,
+        cwd,
+        setSuggestions,
+        setIsLoadingSuggestions,
+    });
+    const slashCompletionRange = useSlashCompletion({
+        enabled: completionMode === CompletionMode.SLASH,
+        query,
+        slashCommands,
+        commandContext,
+        setSuggestions,
+        setIsLoadingSuggestions,
+        setIsPerfectMatch,
+    });
+    useEffect(() => {
+        setActiveSuggestionIndex(suggestions.length > 0 ? 0 : -1);
+        setVisibleStartIndex(0);
+    }, [suggestions, setActiveSuggestionIndex, setVisibleStartIndex]);
+    useEffect(() => {
+        if (completionMode === CompletionMode.IDLE || reverseSearchActive) {
+            resetCompletionState();
+            return;
+        }
+        // Show suggestions if we are loading OR if there are results to display.
+        setShowSuggestions(isLoadingSuggestions || suggestions.length > 0);
+    }, [
+        completionMode,
+        suggestions.length,
+        isLoadingSuggestions,
+        reverseSearchActive,
+        resetCompletionState,
+        setShowSuggestions,
+    ]);
+    const handleAutocomplete = useCallback((indexToUse) => {
+        if (indexToUse < 0 || indexToUse >= suggestions.length) {
+            return;
+        }
+        const suggestion = suggestions[indexToUse].value;
+        let start = completionStart;
+        let end = completionEnd;
+        if (completionMode === CompletionMode.SLASH) {
+            start = slashCompletionRange.completionStart;
+            end = slashCompletionRange.completionEnd;
+        }
+        if (start === -1 || end === -1) {
+            return;
+        }
+        let suggestionText = suggestion;
+        if (completionMode === CompletionMode.SLASH) {
+            if (start === end &&
+                start > 1 &&
+                (buffer.lines[cursorRow] || '')[start - 1] !== ' ') {
+                suggestionText = ' ' + suggestionText;
+            }
+        }
+        suggestionText += ' ';
+        buffer.replaceRangeByOffset(logicalPosToOffset(buffer.lines, cursorRow, start), logicalPosToOffset(buffer.lines, cursorRow, end), suggestionText);
+    }, [
+        cursorRow,
+        buffer,
+        suggestions,
+        completionMode,
+        completionStart,
+        completionEnd,
+        slashCompletionRange,
+    ]);
+    return {
+        suggestions,
+        activeSuggestionIndex,
+        visibleStartIndex,
+        showSuggestions,
+        isLoadingSuggestions,
+        isPerfectMatch,
+        setActiveSuggestionIndex,
+        setShowSuggestions,
+        resetCompletionState,
+        navigateUp,
+        navigateDown,
+        handleAutocomplete,
+    };
+}
+//# sourceMappingURL=useCommandCompletion.js.map

package/dist/src/ui/hooks/useCommandCompletion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useCommandCompletion.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useCommandCompletion.tsx"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAGxD,OAAO,EACL,kBAAkB,GAEnB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,MAAM,CAAN,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,+BAAa,CAAA;IACb,2BAAS,CAAA;IACT,iCAAe,CAAA;AACjB,CAAC,EAJW,cAAc,KAAd,cAAc,QAIzB;AAiBD,MAAM,UAAU,oBAAoB,CAClC,MAAkB,EAClB,IAAuB,EACvB,GAAW,EACX,aAAsC,EACtC,cAA8B,EAC9B,sBAA+B,KAAK,EACpC,MAAe;IAEf,MAAM,EACJ,WAAW,EACX,qBAAqB,EACrB,iBAAiB,EACjB,eAAe,EACf,oBAAoB,EACpB,cAAc,EAEd,cAAc,EACd,kBAAkB,EAClB,wBAAwB,EACxB,uBAAuB,EACvB,iBAAiB,EACjB,oBAAoB,EAEpB,oBAAoB,EACpB,UAAU,EACV,YAAY,GACb,GAAG,aAAa,EAAE,CAAC;IAEpB,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAEnC,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,GAC7D,OAAO,CAAC,GAAG,EAAE;QACX,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAClD,IAAI,SAAS,KAAK,CAAC,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1D,OAAO;gBACL,cAAc,EAAE,cAAc,CAAC,KAAK;gBACpC,KAAK,EAAE,WAAW;gBAClB,eAAe,EAAE,CAAC;gBAClB,aAAa,EAAE,WAAW,CAAC,MAAM;aAClC,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;QAC7C,KAAK,IAAI,CAAC,GAAG,SAAS,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAE3B,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,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;oBAC1D,cAAc,EAAE,CAAC;gBACnB,CAAC;gBACD,IAAI,cAAc,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC7B,OAAO;wBACL,cAAc,EAAE,cAAc,CAAC,IAAI;wBACnC,KAAK,EAAE,IAAI;wBACX,eAAe,EAAE,CAAC,CAAC;wBACnB,aAAa,EAAE,CAAC,CAAC;qBAClB,CAAC;gBACJ,CAAC;YACH,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxB,IAAI,GAAG,GAAG,UAAU,CAAC,MAAM,CAAC;gBAC5B,KAAK,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACnD,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;wBAC1B,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;4BAC1D,cAAc,EAAE,CAAC;wBACnB,CAAC;wBAED,IAAI,cAAc,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC7B,GAAG,GAAG,CAAC,CAAC;4BACR,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBAC1D,OAAO;oBACL,cAAc,EAAE,cAAc,CAAC,EAAE;oBACjC,KAAK,EAAE,WAAW;oBAClB,eAAe,EAAE,SAAS;oBAC1B,aAAa,EAAE,GAAG;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO;YACL,cAAc,EAAE,cAAc,CAAC,IAAI;YACnC,KAAK,EAAE,IAAI;YACX,eAAe,EAAE,CAAC,CAAC;YACnB,aAAa,EAAE,CAAC,CAAC;SAClB,CAAC;IACJ,CAAC,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAE3C,eAAe,CAAC;QACd,OAAO,EAAE,cAAc,KAAK,cAAc,CAAC,EAAE;QAC7C,OAAO,EAAE,KAAK,IAAI,EAAE;QACpB,MAAM;QACN,GAAG;QACH,cAAc;QACd,uBAAuB;KACxB,CAAC,CAAC;IAEH,MAAM,oBAAoB,GAAG,kBAAkB,CAAC;QAC9C,OAAO,EAAE,cAAc,KAAK,cAAc,CAAC,KAAK;QAChD,KAAK;QACL,aAAa;QACb,cAAc;QACd,cAAc;QACd,uBAAuB;QACvB,iBAAiB;KAClB,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,wBAAwB,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,oBAAoB,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC,EAAE,CAAC,WAAW,EAAE,wBAAwB,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAElE,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,KAAK,cAAc,CAAC,IAAI,IAAI,mBAAmB,EAAE,CAAC;YAClE,oBAAoB,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QACD,yEAAyE;QACzE,kBAAkB,CAAC,oBAAoB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrE,CAAC,EAAE;QACD,cAAc;QACd,WAAW,CAAC,MAAM;QAClB,oBAAoB;QACpB,mBAAmB;QACnB,oBAAoB;QACpB,kBAAkB;KACnB,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,UAAkB,EAAE,EAAE;QACrB,IAAI,UAAU,GAAG,CAAC,IAAI,UAAU,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACvD,OAAO;QACT,CAAC;QACD,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC;QAEjD,IAAI,KAAK,GAAG,eAAe,CAAC;QAC5B,IAAI,GAAG,GAAG,aAAa,CAAC;QACxB,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC5C,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC;YAC7C,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC3C,CAAC;QAED,IAAI,KAAK,KAAK,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,IAAI,cAAc,GAAG,UAAU,CAAC;QAChC,IAAI,cAAc,KAAK,cAAc,CAAC,KAAK,EAAE,CAAC;YAC5C,IACE,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,EAClD,CAAC;gBACD,cAAc,GAAG,GAAG,GAAG,cAAc,CAAC;YACxC,CAAC;QACH,CAAC;QAED,cAAc,IAAI,GAAG,CAAC;QAEtB,MAAM,CAAC,oBAAoB,CACzB,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,EAClD,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,EAChD,cAAc,CACf,CAAC;IACJ,CAAC,EACD;QACE,SAAS;QACT,MAAM;QACN,WAAW;QACX,cAAc;QACd,eAAe;QACf,aAAa;QACb,oBAAoB;KACrB,CACF,CAAC;IAEF,OAAO;QACL,WAAW;QACX,qBAAqB;QACrB,iBAAiB;QACjB,eAAe;QACf,oBAAoB;QACpB,cAAc;QACd,wBAAwB;QACxB,kBAAkB;QAClB,oBAAoB;QACpB,UAAU;QACV,YAAY;QACZ,kBAAkB;KACnB,CAAC;AACJ,CAAC"}