@xelauvas/xela-cli 0.1.0

package/src/hooks/usePasteHandler.ts

@@ -0,0 +1,285 @@
+import { basename } from 'path'
+import React from 'react'
+import { logError } from 'src/utils/log.js'
+import { useDebounceCallback } from 'usehooks-ts'
+import type { InputEvent, Key } from '../ink.js'
+import {
+  getImageFromClipboard,
+  isImageFilePath,
+  PASTE_THRESHOLD,
+  tryReadImageFromPath,
+} from '../utils/imagePaste.js'
+import type { ImageDimensions } from '../utils/imageResizer.js'
+import { getPlatform } from '../utils/platform.js'
+
+const CLIPBOARD_CHECK_DEBOUNCE_MS = 50
+const PASTE_COMPLETION_TIMEOUT_MS = 100
+
+type PasteHandlerProps = {
+  onPaste?: (text: string) => void
+  onInput: (input: string, key: Key) => void
+  onImagePaste?: (
+    base64Image: string,
+    mediaType?: string,
+    filename?: string,
+    dimensions?: ImageDimensions,
+    sourcePath?: string,
+  ) => void
+}
+
+export function usePasteHandler({
+  onPaste,
+  onInput,
+  onImagePaste,
+}: PasteHandlerProps): {
+  wrappedOnInput: (input: string, key: Key, event: InputEvent) => void
+  pasteState: {
+    chunks: string[]
+    timeoutId: ReturnType<typeof setTimeout> | null
+  }
+  isPasting: boolean
+} {
+  const [pasteState, setPasteState] = React.useState<{
+    chunks: string[]
+    timeoutId: ReturnType<typeof setTimeout> | null
+  }>({ chunks: [], timeoutId: null })
+  const [isPasting, setIsPasting] = React.useState(false)
+  const isMountedRef = React.useRef(true)
+  // Mirrors pasteState.timeoutId but updated synchronously. When paste + a
+  // keystroke arrive in the same stdin chunk, both wrappedOnInput calls run
+  // in the same discreteUpdates batch before React commits — the second call
+  // reads stale pasteState.timeoutId (null) and takes the onInput path. If
+  // that key is Enter, it submits the old input and the paste is lost.
+  const pastePendingRef = React.useRef(false)
+
+  const isMacOS = React.useMemo(() => getPlatform() === 'macos', [])
+
+  React.useEffect(() => {
+    return () => {
+      isMountedRef.current = false
+    }
+  }, [])
+
+  const checkClipboardForImageImpl = React.useCallback(() => {
+    if (!onImagePaste || !isMountedRef.current) return
+
+    void getImageFromClipboard()
+      .then(imageData => {
+        if (imageData && isMountedRef.current) {
+          onImagePaste(
+            imageData.base64,
+            imageData.mediaType,
+            undefined, // no filename for clipboard images
+            imageData.dimensions,
+          )
+        }
+      })
+      .catch(error => {
+        if (isMountedRef.current) {
+          logError(error as Error)
+        }
+      })
+      .finally(() => {
+        if (isMountedRef.current) {
+          setIsPasting(false)
+        }
+      })
+  }, [onImagePaste])
+
+  const checkClipboardForImage = useDebounceCallback(
+    checkClipboardForImageImpl,
+    CLIPBOARD_CHECK_DEBOUNCE_MS,
+  )
+
+  const resetPasteTimeout = React.useCallback(
+    (currentTimeoutId: ReturnType<typeof setTimeout> | null) => {
+      if (currentTimeoutId) {
+        clearTimeout(currentTimeoutId)
+      }
+      return setTimeout(
+        (
+          setPasteState,
+          onImagePaste,
+          onPaste,
+          setIsPasting,
+          checkClipboardForImage,
+          isMacOS,
+          pastePendingRef,
+        ) => {
+          pastePendingRef.current = false
+          setPasteState(({ chunks }) => {
+            // Join chunks and filter out orphaned focus sequences
+            // These can appear when focus events split during paste
+            const pastedText = chunks
+              .join('')
+              .replace(/\[I$/, '')
+              .replace(/\[O$/, '')
+
+            // Check if the pasted text contains image file paths
+            // When dragging multiple images, they may come as:
+            // 1. Newline-separated paths (common in some terminals)
+            // 2. Space-separated paths (common when dragging from Finder)
+            // For space-separated paths, we split on spaces that precede absolute paths:
+            // - Unix: space followed by `/` (e.g., `/Users/...`)
+            // - Windows: space followed by drive letter and `:\` (e.g., `C:\Users\...`)
+            // This works because spaces within paths are escaped (e.g., `file\ name.png`)
+            const lines = pastedText
+              .split(/ (?=\/|[A-Za-z]:\\)/)
+              .flatMap(part => part.split('\n'))
+              .filter(line => line.trim())
+            const imagePaths = lines.filter(line => isImageFilePath(line))
+
+            if (onImagePaste && imagePaths.length > 0) {
+              const isTempScreenshot =
+                /\/TemporaryItems\/.*screencaptureui.*\/Screenshot/i.test(
+                  pastedText,
+                )
+
+              // Process all image paths
+              void Promise.all(
+                imagePaths.map(imagePath => tryReadImageFromPath(imagePath)),
+              ).then(results => {
+                const validImages = results.filter(
+                  (r): r is NonNullable<typeof r> => r !== null,
+                )
+
+                if (validImages.length > 0) {
+                  // Successfully read at least one image
+                  for (const imageData of validImages) {
+                    const filename = basename(imageData.path)
+                    onImagePaste(
+                      imageData.base64,
+                      imageData.mediaType,
+                      filename,
+                      imageData.dimensions,
+                      imageData.path,
+                    )
+                  }
+                  // If some paths weren't images, paste them as text
+                  const nonImageLines = lines.filter(
+                    line => !isImageFilePath(line),
+                  )
+                  if (nonImageLines.length > 0 && onPaste) {
+                    onPaste(nonImageLines.join('\n'))
+                  }
+                  setIsPasting(false)
+                } else if (isTempScreenshot && isMacOS) {
+                  // For temporary screenshot files that no longer exist, try clipboard
+                  checkClipboardForImage()
+                } else {
+                  if (onPaste) {
+                    onPaste(pastedText)
+                  }
+                  setIsPasting(false)
+                }
+              })
+              return { chunks: [], timeoutId: null }
+            }
+
+            // If paste is empty (common when trying to paste images with Cmd+V),
+            // check if clipboard has an image (macOS only)
+            if (isMacOS && onImagePaste && pastedText.length === 0) {
+              checkClipboardForImage()
+              return { chunks: [], timeoutId: null }
+            }
+
+            // Handle regular paste
+            if (onPaste) {
+              onPaste(pastedText)
+            }
+            // Reset isPasting state after paste is complete
+            setIsPasting(false)
+            return { chunks: [], timeoutId: null }
+          })
+        },
+        PASTE_COMPLETION_TIMEOUT_MS,
+        setPasteState,
+        onImagePaste,
+        onPaste,
+        setIsPasting,
+        checkClipboardForImage,
+        isMacOS,
+        pastePendingRef,
+      )
+    },
+    [checkClipboardForImage, isMacOS, onImagePaste, onPaste],
+  )
+
+  // Paste detection is now done via the InputEvent's keypress.isPasted flag,
+  // which is set by the keypress parser when it detects bracketed paste mode.
+  // This avoids the race condition caused by having multiple listeners on stdin.
+  // Previously, we had a stdin.on('data') listener here which competed with
+  // the 'readable' listener in App.tsx, causing dropped characters.
+
+  const wrappedOnInput = (input: string, key: Key, event: InputEvent): void => {
+    // Detect paste from the parsed keypress event.
+    // The keypress parser sets isPasted=true for content within bracketed paste.
+    const isFromPaste = event.keypress.isPasted
+
+    // If this is pasted content, set isPasting state for UI feedback
+    if (isFromPaste) {
+      setIsPasting(true)
+    }
+
+    // Handle large pastes (>PASTE_THRESHOLD chars)
+    // Usually we get one or two input characters at a time. If we
+    // get more than the threshold, the user has probably pasted.
+    // Unfortunately node batches long pastes, so it's possible
+    // that we would see e.g. 1024 characters and then just a few
+    // more in the next frame that belong with the original paste.
+    // This batching number is not consistent.
+
+    // Handle potential image filenames (even if they're shorter than paste threshold)
+    // When dragging multiple images, they may come as newline-separated or
+    // space-separated paths. Split on spaces preceding absolute paths:
+    // - Unix: ` /` - Windows: ` C:\` etc.
+    const hasImageFilePath = input
+      .split(/ (?=\/|[A-Za-z]:\\)/)
+      .flatMap(part => part.split('\n'))
+      .some(line => isImageFilePath(line.trim()))
+
+    // Handle empty paste (clipboard image on macOS)
+    // When the user pastes an image with Cmd+V, the terminal sends an empty
+    // bracketed paste sequence. The keypress parser emits this as isPasted=true
+    // with empty input.
+    if (isFromPaste && input.length === 0 && isMacOS && onImagePaste) {
+      checkClipboardForImage()
+      // Reset isPasting since there's no text content to process
+      setIsPasting(false)
+      return
+    }
+
+    // Check if we should handle as paste (from bracketed paste, large input, or continuation)
+    const shouldHandleAsPaste =
+      onPaste &&
+      (input.length > PASTE_THRESHOLD ||
+        pastePendingRef.current ||
+        hasImageFilePath ||
+        isFromPaste)
+
+    if (shouldHandleAsPaste) {
+      pastePendingRef.current = true
+      setPasteState(({ chunks, timeoutId }) => {
+        return {
+          chunks: [...chunks, input],
+          timeoutId: resetPasteTimeout(timeoutId),
+        }
+      })
+      return
+    }
+    onInput(input, key)
+    if (input.length > 10) {
+      // Ensure that setIsPasting is turned off on any other multicharacter
+      // input, because the stdin buffer may chunk at arbitrary points and split
+      // the closing escape sequence if the input length is too long for the
+      // stdin buffer.
+      setIsPasting(false)
+    }
+  }
+
+  return {
+    wrappedOnInput,
+    pasteState,
+    isPasting,
+  }
+}

package/src/hooks/usePluginRecommendationBase.tsx

@@ -0,0 +1,105 @@
+import { c as _c } from "react/compiler-runtime";
+/**
+ * Shared state machine + install helper for plugin-recommendation hooks
+ * (LSP, claude-code-hint). Centralizes the gate chain, async-guard,
+ * and success/failure notification JSX so new sources stay small.
+ */
+
+import figures from 'figures';
+import * as React from 'react';
+import { getIsRemoteMode } from '../bootstrap/state.js';
+import type { useNotifications } from '../context/notifications.js';
+import { Text } from '../ink.js';
+import { logError } from '../utils/log.js';
+import { getPluginById } from '../utils/plugins/marketplaceManager.js';
+type AddNotification = ReturnType<typeof useNotifications>['addNotification'];
+type PluginData = NonNullable<Awaited<ReturnType<typeof getPluginById>>>;
+
+/**
+ * Call tryResolve inside a useEffect; it applies standard gates (remote
+ * mode, already-showing, in-flight) then runs resolve(). Non-null return
+ * becomes the recommendation. Include tryResolve in effect deps — its
+ * identity tracks recommendation, so clearing re-triggers resolution.
+ */
+export function usePluginRecommendationBase() {
+  const $ = _c(6);
+  const [recommendation, setRecommendation] = React.useState(null);
+  const isCheckingRef = React.useRef(false);
+  let t0;
+  if ($[0] !== recommendation) {
+    t0 = resolve => {
+      if (getIsRemoteMode()) {
+        return;
+      }
+      if (recommendation) {
+        return;
+      }
+      if (isCheckingRef.current) {
+        return;
+      }
+      isCheckingRef.current = true;
+      resolve().then(rec => {
+        if (rec) {
+          setRecommendation(rec);
+        }
+      }).catch(logError).finally(() => {
+        isCheckingRef.current = false;
+      });
+    };
+    $[0] = recommendation;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const tryResolve = t0;
+  let t1;
+  if ($[2] === Symbol.for("react.memo_cache_sentinel")) {
+    t1 = () => setRecommendation(null);
+    $[2] = t1;
+  } else {
+    t1 = $[2];
+  }
+  const clearRecommendation = t1;
+  let t2;
+  if ($[3] !== recommendation || $[4] !== tryResolve) {
+    t2 = {
+      recommendation,
+      clearRecommendation,
+      tryResolve
+    };
+    $[3] = recommendation;
+    $[4] = tryResolve;
+    $[5] = t2;
+  } else {
+    t2 = $[5];
+  }
+  return t2;
+}
+
+/** Look up plugin, run install(), emit standard success/failure notification. */
+export async function installPluginAndNotify(pluginId: string, pluginName: string, keyPrefix: string, addNotification: AddNotification, install: (pluginData: PluginData) => Promise<void>): Promise<void> {
+  try {
+    const pluginData = await getPluginById(pluginId);
+    if (!pluginData) {
+      throw new Error(`Plugin ${pluginId} not found in marketplace`);
+    }
+    await install(pluginData);
+    addNotification({
+      key: `${keyPrefix}-installed`,
+      jsx: <Text color="success">
+          {figures.tick} {pluginName} installed · restart to apply
+        </Text>,
+      priority: 'immediate',
+      timeoutMs: 5000
+    });
+  } catch (error) {
+    logError(error);
+    addNotification({
+      key: `${keyPrefix}-install-failed`,
+      jsx: <Text color="error">Failed to install {pluginName}</Text>,
+      priority: 'immediate',
+      timeoutMs: 5000
+    });
+  }
+}
+//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJmaWd1cmVzIiwiUmVhY3QiLCJnZXRJc1JlbW90ZU1vZGUiLCJ1c2VOb3RpZmljYXRpb25zIiwiVGV4dCIsImxvZ0Vycm9yIiwiZ2V0UGx1Z2luQnlJZCIsIkFkZE5vdGlmaWNhdGlvbiIsIlJldHVyblR5cGUiLCJQbHVnaW5EYXRhIiwiTm9uTnVsbGFibGUiLCJBd2FpdGVkIiwidXNlUGx1Z2luUmVjb21tZW5kYXRpb25CYXNlIiwiJCIsIl9jIiwicmVjb21tZW5kYXRpb24iLCJzZXRSZWNvbW1lbmRhdGlvbiIsInVzZVN0YXRlIiwiaXNDaGVja2luZ1JlZiIsInVzZVJlZiIsInQwIiwicmVzb2x2ZSIsImN1cnJlbnQiLCJ0aGVuIiwicmVjIiwiY2F0Y2giLCJmaW5hbGx5IiwidHJ5UmVzb2x2ZSIsInQxIiwiU3ltYm9sIiwiZm9yIiwiY2xlYXJSZWNvbW1lbmRhdGlvbiIsInQyIiwiaW5zdGFsbFBsdWdpbkFuZE5vdGlmeSIsInBsdWdpbklkIiwicGx1Z2luTmFtZSIsImtleVByZWZpeCIsImFkZE5vdGlmaWNhdGlvbiIsImluc3RhbGwiLCJwbHVnaW5EYXRhIiwiUHJvbWlzZSIsIkVycm9yIiwia2V5IiwianN4IiwidGljayIsInByaW9yaXR5IiwidGltZW91dE1zIiwiZXJyb3IiXSwic291cmNlcyI6WyJ1c2VQbHVnaW5SZWNvbW1lbmRhdGlvbkJhc2UudHN4Il0sInNvdXJjZXNDb250ZW50IjpbIi8qKlxuICogU2hhcmVkIHN0YXRlIG1hY2hpbmUgKyBpbnN0YWxsIGhlbHBlciBmb3IgcGx1Z2luLXJlY29tbWVuZGF0aW9uIGhvb2tzXG4gKiAoTFNQLCBjbGF1ZGUtY29kZS1oaW50KS4gQ2VudHJhbGl6ZXMgdGhlIGdhdGUgY2hhaW4sIGFzeW5jLWd1YXJkLFxuICogYW5kIHN1Y2Nlc3MvZmFpbHVyZSBub3RpZmljYXRpb24gSlNYIHNvIG5ldyBzb3VyY2VzIHN0YXkgc21hbGwuXG4gKi9cblxuaW1wb3J0IGZpZ3VyZXMgZnJvbSAnZmlndXJlcydcbmltcG9ydCAqIGFzIFJlYWN0IGZyb20gJ3JlYWN0J1xuaW1wb3J0IHsgZ2V0SXNSZW1vdGVNb2RlIH0gZnJvbSAnLi4vYm9vdHN0cmFwL3N0YXRlLmpzJ1xuaW1wb3J0IHR5cGUgeyB1c2VOb3RpZmljYXRpb25zIH0gZnJvbSAnLi4vY29udGV4dC9ub3RpZmljYXRpb25zLmpzJ1xuaW1wb3J0IHsgVGV4dCB9IGZyb20gJy4uL2luay5qcydcbmltcG9ydCB7IGxvZ0Vycm9yIH0gZnJvbSAnLi4vdXRpbHMvbG9nLmpzJ1xuaW1wb3J0IHsgZ2V0UGx1Z2luQnlJZCB9IGZyb20gJy4uL3V0aWxzL3BsdWdpbnMvbWFya2V0cGxhY2VNYW5hZ2VyLmpzJ1xuXG50eXBlIEFkZE5vdGlmaWNhdGlvbiA9IFJldHVyblR5cGU8dHlwZW9mIHVzZU5vdGlmaWNhdGlvbnM+WydhZGROb3RpZmljYXRpb24nXVxudHlwZSBQbHVnaW5EYXRhID0gTm9uTnVsbGFibGU8QXdhaXRlZDxSZXR1cm5UeXBlPHR5cGVvZiBnZXRQbHVnaW5CeUlkPj4+XG5cbi8qKlxuICogQ2FsbCB0cnlSZXNvbHZlIGluc2lkZSBhIHVzZUVmZmVjdDsgaXQgYXBwbGllcyBzdGFuZGFyZCBnYXRlcyAocmVtb3RlXG4gKiBtb2RlLCBhbHJlYWR5LXNob3dpbmcsIGluLWZsaWdodCkgdGhlbiBydW5zIHJlc29sdmUoKS4gTm9uLW51bGwgcmV0dXJuXG4gKiBiZWNvbWVzIHRoZSByZWNvbW1lbmRhdGlvbi4gSW5jbHVkZSB0cnlSZXNvbHZlIGluIGVmZmVjdCBkZXBzIOKAlCBpdHNcbiAqIGlkZW50aXR5IHRyYWNrcyByZWNvbW1lbmRhdGlvbiwgc28gY2xlYXJpbmcgcmUtdHJpZ2dlcnMgcmVzb2x1dGlvbi5cbiAqL1xuZXhwb3J0IGZ1bmN0aW9uIHVzZVBsdWdpblJlY29tbWVuZGF0aW9uQmFzZTxUPigpOiB7XG4gIHJlY29tbWVuZGF0aW9uOiBUIHwgbnVsbFxuICBjbGVhclJlY29tbWVuZGF0aW9uOiAoKSA9PiB2b2lkXG4gIHRyeVJlc29sdmU6IChyZXNvbHZlOiAoKSA9PiBQcm9taXNlPFQgfCBudWxsPikgPT4gdm9pZFxufSB7XG4gIGNvbnN0IFtyZWNvbW1lbmRhdGlvbiwgc2V0UmVjb21tZW5kYXRpb25dID0gUmVhY3QudXNlU3RhdGU8VCB8IG51bGw+KG51bGwpXG4gIGNvbnN0IGlzQ2hlY2tpbmdSZWYgPSBSZWFjdC51c2VSZWYoZmFsc2UpXG5cbiAgY29uc3QgdHJ5UmVzb2x2ZSA9IFJlYWN0LnVzZUNhbGxiYWNrKFxuICAgIChyZXNvbHZlOiAoKSA9PiBQcm9taXNlPFQgfCBudWxsPikgPT4ge1xuICAgICAgaWYgKGdldElzUmVtb3RlTW9kZSgpKSByZXR1cm5cbiAgICAgIGlmIChyZWNvbW1lbmRhdGlvbikgcmV0dXJuXG4gICAgICBpZiAoaXNDaGVja2luZ1JlZi5jdXJyZW50KSByZXR1cm5cblxuICAgICAgaXNDaGVja2luZ1JlZi5jdXJyZW50ID0gdHJ1ZVxuICAgICAgdm9pZCByZXNvbHZlKClcbiAgICAgICAgLnRoZW4ocmVjID0+IHtcbiAgICAgICAgICBpZiAocmVjKSBzZXRSZWNvbW1lbmRhdGlvbihyZWMpXG4gICAgICAgIH0pXG4gICAgICAgIC5jYXRjaChsb2dFcnJvcilcbiAgICAgICAgLmZpbmFsbHkoKCkgPT4ge1xuICAgICAgICAgIGlzQ2hlY2tpbmdSZWYuY3VycmVudCA9IGZhbHNlXG4gICAgICAgIH0pXG4gICAgfSxcbiAgICBbcmVjb21tZW5kYXRpb25dLFxuICApXG5cbiAgY29uc3QgY2xlYXJSZWNvbW1lbmRhdGlvbiA9IFJlYWN0LnVzZUNhbGxiYWNrKFxuICAgICgpID0+IHNldFJlY29tbWVuZGF0aW9uKG51bGwpLFxuICAgIFtdLFxuICApXG5cbiAgcmV0dXJuIHsgcmVjb21tZW5kYXRpb24sIGNsZWFyUmVjb21tZW5kYXRpb24sIHRyeVJlc29sdmUgfVxufVxuXG4vKiogTG9vayB1cCBwbHVnaW4sIHJ1biBpbnN0YWxsKCksIGVtaXQgc3RhbmRhcmQgc3VjY2Vzcy9mYWlsdXJlIG5vdGlmaWNhdGlvbi4gKi9cbmV4cG9ydCBhc3luYyBmdW5jdGlvbiBpbnN0YWxsUGx1Z2luQW5kTm90aWZ5KFxuICBwbHVnaW5JZDogc3RyaW5nLFxuICBwbHVnaW5OYW1lOiBzdHJpbmcsXG4gIGtleVByZWZpeDogc3RyaW5nLFxuICBhZGROb3RpZmljYXRpb246IEFkZE5vdGlmaWNhdGlvbixcbiAgaW5zdGFsbDogKHBsdWdpbkRhdGE6IFBsdWdpbkRhdGEpID0+IFByb21pc2U8dm9pZD4sXG4pOiBQcm9taXNlPHZvaWQ+IHtcbiAgdHJ5IHtcbiAgICBjb25zdCBwbHVnaW5EYXRhID0gYXdhaXQgZ2V0UGx1Z2luQnlJZChwbHVnaW5JZClcbiAgICBpZiAoIXBsdWdpbkRhdGEpIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcihgUGx1Z2luICR7cGx1Z2luSWR9IG5vdCBmb3VuZCBpbiBtYXJrZXRwbGFjZWApXG4gICAgfVxuICAgIGF3YWl0IGluc3RhbGwocGx1Z2luRGF0YSlcbiAgICBhZGROb3RpZmljYXRpb24oe1xuICAgICAga2V5OiBgJHtrZXlQcmVmaXh9LWluc3RhbGxlZGAsXG4gICAgICBqc3g6IChcbiAgICAgICAgPFRleHQgY29sb3I9XCJzdWNjZXNzXCI+XG4gICAgICAgICAge2ZpZ3VyZXMudGlja30ge3BsdWdpbk5hbWV9IGluc3RhbGxlZCDCtyByZXN0YXJ0IHRvIGFwcGx5XG4gICAgICAgIDwvVGV4dD5cbiAgICAgICksXG4gICAgICBwcmlvcml0eTogJ2ltbWVkaWF0ZScsXG4gICAgICB0aW1lb3V0TXM6IDUwMDAsXG4gICAgfSlcbiAgfSBjYXRjaCAoZXJyb3IpIHtcbiAgICBsb2dFcnJvcihlcnJvcilcbiAgICBhZGROb3RpZmljYXRpb24oe1xuICAgICAga2V5OiBgJHtrZXlQcmVmaXh9LWluc3RhbGwtZmFpbGVkYCxcbiAgICAgIGpzeDogPFRleHQgY29sb3I9XCJlcnJvclwiPkZhaWxlZCB0byBpbnN0YWxsIHtwbHVnaW5OYW1lfTwvVGV4dD4sXG4gICAgICBwcmlvcml0eTogJ2ltbWVkaWF0ZScsXG4gICAgICB0aW1lb3V0TXM6IDUwMDAsXG4gICAgfSlcbiAgfVxufVxuIl0sIm1hcHBpbmdzIjoiO0FBQUE7QUFDQTtBQUNBO0FBQ0E7QUFDQTs7QUFFQSxPQUFPQSxPQUFPLE1BQU0sU0FBUztBQUM3QixPQUFPLEtBQUtDLEtBQUssTUFBTSxPQUFPO0FBQzlCLFNBQVNDLGVBQWUsUUFBUSx1QkFBdUI7QUFDdkQsY0FBY0MsZ0JBQWdCLFFBQVEsNkJBQTZCO0FBQ25FLFNBQVNDLElBQUksUUFBUSxXQUFXO0FBQ2hDLFNBQVNDLFFBQVEsUUFBUSxpQkFBaUI7QUFDMUMsU0FBU0MsYUFBYSxRQUFRLHdDQUF3QztBQUV0RSxLQUFLQyxlQUFlLEdBQUdDLFVBQVUsQ0FBQyxPQUFPTCxnQkFBZ0IsQ0FBQyxDQUFDLGlCQUFpQixDQUFDO0FBQzdFLEtBQUtNLFVBQVUsR0FBR0MsV0FBVyxDQUFDQyxPQUFPLENBQUNILFVBQVUsQ0FBQyxPQUFPRixhQUFhLENBQUMsQ0FBQyxDQUFDOztBQUV4RTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQSxPQUFPLFNBQUFNLDRCQUFBO0VBQUEsTUFBQUMsQ0FBQSxHQUFBQyxFQUFBO0VBS0wsT0FBQUMsY0FBQSxFQUFBQyxpQkFBQSxJQUE0Q2YsS0FBSyxDQUFBZ0IsUUFBUyxDQUFXLElBQUksQ0FBQztFQUMxRSxNQUFBQyxhQUFBLEdBQXNCakIsS0FBSyxDQUFBa0IsTUFBTyxDQUFDLEtBQUssQ0FBQztFQUFBLElBQUFDLEVBQUE7RUFBQSxJQUFBUCxDQUFBLFFBQUFFLGNBQUE7SUFHdkNLLEVBQUEsR0FBQUMsT0FBQTtNQUNFLElBQUluQixlQUFlLENBQUMsQ0FBQztRQUFBO01BQUE7TUFDckIsSUFBSWEsY0FBYztRQUFBO01BQUE7TUFDbEIsSUFBSUcsYUFBYSxDQUFBSSxPQUFRO1FBQUE7TUFBQTtNQUV6QkosYUFBYSxDQUFBSSxPQUFBLEdBQVcsSUFBSDtNQUNoQkQsT0FBTyxDQUFDLENBQUMsQ0FBQUUsSUFDUCxDQUFDQyxHQUFBO1FBQ0osSUFBSUEsR0FBRztVQUFFUixpQkFBaUIsQ0FBQ1EsR0FBRyxDQUFDO1FBQUE7TUFBQSxDQUNoQyxDQUFDLENBQUFDLEtBQ0ksQ0FBQ3BCLFFBQVEsQ0FBQyxDQUFBcUIsT0FDUixDQUFDO1FBQ1BSLGFBQWEsQ0FBQUksT0FBQSxHQUFXLEtBQUg7TUFBQSxDQUN0QixDQUFDO0lBQUEsQ0FDTDtJQUFBVCxDQUFBLE1BQUFFLGNBQUE7SUFBQUYsQ0FBQSxNQUFBTyxFQUFBO0VBQUE7SUFBQUEsRUFBQSxHQUFBUCxDQUFBO0VBQUE7RUFmSCxNQUFBYyxVQUFBLEdBQW1CUCxFQWlCbEI7RUFBQSxJQUFBUSxFQUFBO0VBQUEsSUFBQWYsQ0FBQSxRQUFBZ0IsTUFBQSxDQUFBQyxHQUFBO0lBR0NGLEVBQUEsR0FBQUEsQ0FBQSxLQUFNWixpQkFBaUIsQ0FBQyxJQUFJLENBQUM7SUFBQUgsQ0FBQSxNQUFBZSxFQUFBO0VBQUE7SUFBQUEsRUFBQSxHQUFBZixDQUFBO0VBQUE7RUFEL0IsTUFBQWtCLG1CQUFBLEdBQTRCSCxFQUczQjtFQUFBLElBQUFJLEVBQUE7RUFBQSxJQUFBbkIsQ0FBQSxRQUFBRSxjQUFBLElBQUFGLENBQUEsUUFBQWMsVUFBQTtJQUVNSyxFQUFBO01BQUFqQixjQUFBO01BQUFnQixtQkFBQTtNQUFBSjtJQUFrRCxDQUFDO0lBQUFkLENBQUEsTUFBQUUsY0FBQTtJQUFBRixDQUFBLE1BQUFjLFVBQUE7SUFBQWQsQ0FBQSxNQUFBbUIsRUFBQTtFQUFBO0lBQUFBLEVBQUEsR0FBQW5CLENBQUE7RUFBQTtFQUFBLE9BQW5EbUIsRUFBbUQ7QUFBQTs7QUFHNUQ7QUFDQSxPQUFPLGVBQWVDLHNCQUFzQkEsQ0FDMUNDLFFBQVEsRUFBRSxNQUFNLEVBQ2hCQyxVQUFVLEVBQUUsTUFBTSxFQUNsQkMsU0FBUyxFQUFFLE1BQU0sRUFDakJDLGVBQWUsRUFBRTlCLGVBQWUsRUFDaEMrQixPQUFPLEVBQUUsQ0FBQ0MsVUFBVSxFQUFFOUIsVUFBVSxFQUFFLEdBQUcrQixPQUFPLENBQUMsSUFBSSxDQUFDLENBQ25ELEVBQUVBLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQztFQUNmLElBQUk7SUFDRixNQUFNRCxVQUFVLEdBQUcsTUFBTWpDLGFBQWEsQ0FBQzRCLFFBQVEsQ0FBQztJQUNoRCxJQUFJLENBQUNLLFVBQVUsRUFBRTtNQUNmLE1BQU0sSUFBSUUsS0FBSyxDQUFDLFVBQVVQLFFBQVEsMkJBQTJCLENBQUM7SUFDaEU7SUFDQSxNQUFNSSxPQUFPLENBQUNDLFVBQVUsQ0FBQztJQUN6QkYsZUFBZSxDQUFDO01BQ2RLLEdBQUcsRUFBRSxHQUFHTixTQUFTLFlBQVk7TUFDN0JPLEdBQUcsRUFDRCxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsU0FBUztBQUM3QixVQUFVLENBQUMzQyxPQUFPLENBQUM0QyxJQUFJLENBQUMsQ0FBQyxDQUFDVCxVQUFVLENBQUM7QUFDckMsUUFBUSxFQUFFLElBQUksQ0FDUDtNQUNEVSxRQUFRLEVBQUUsV0FBVztNQUNyQkMsU0FBUyxFQUFFO0lBQ2IsQ0FBQyxDQUFDO0VBQ0osQ0FBQyxDQUFDLE9BQU9DLEtBQUssRUFBRTtJQUNkMUMsUUFBUSxDQUFDMEMsS0FBSyxDQUFDO0lBQ2ZWLGVBQWUsQ0FBQztNQUNkSyxHQUFHLEVBQUUsR0FBR04sU0FBUyxpQkFBaUI7TUFDbENPLEdBQUcsRUFBRSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDUixVQUFVLENBQUMsRUFBRSxJQUFJLENBQUM7TUFDOURVLFFBQVEsRUFBRSxXQUFXO01BQ3JCQyxTQUFTLEVBQUU7SUFDYixDQUFDLENBQUM7RUFDSjtBQUNGIiwiaWdub3JlTGlzdCI6W119

package/src/hooks/usePrStatus.ts

@@ -0,0 +1,106 @@
+import { useEffect, useRef, useState } from 'react'
+import { getLastInteractionTime } from '../bootstrap/state.js'
+import { fetchPrStatus, type PrReviewState } from '../utils/ghPrStatus.js'
+
+const POLL_INTERVAL_MS = 60_000
+const SLOW_GH_THRESHOLD_MS = 4_000
+const IDLE_STOP_MS = 60 * 60_000 // stop polling after 60 min idle
+
+export type PrStatusState = {
+  number: number | null
+  url: string | null
+  reviewState: PrReviewState | null
+  lastUpdated: number
+}
+
+const INITIAL_STATE: PrStatusState = {
+  number: null,
+  url: null,
+  reviewState: null,
+  lastUpdated: 0,
+}
+
+/**
+ * Polls PR review status every 60s while the session is active.
+ * When no interaction is detected for 60 minutes, the loop stops — no
+ * timers remain. React re-runs the effect when isLoading changes
+ * (turn starts/ends), restarting the loop. Effect setup schedules
+ * the next poll relative to the last fetch time so turn boundaries
+ * don't spawn `gh` more than once per interval. Disables permanently
+ * if a fetch exceeds 4s.
+ *
+ * Pass `enabled: false` to skip polling entirely (hook still must be
+ * called unconditionally to satisfy the rules of hooks).
+ */
+export function usePrStatus(isLoading: boolean, enabled = true): PrStatusState {
+  const [prStatus, setPrStatus] = useState<PrStatusState>(INITIAL_STATE)
+  const timeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null)
+  const disabledRef = useRef(false)
+  const lastFetchRef = useRef(0)
+
+  useEffect(() => {
+    if (!enabled) return
+    if (disabledRef.current) return
+
+    let cancelled = false
+    let lastSeenInteractionTime = -1
+    let lastActivityTimestamp = Date.now()
+
+    async function poll() {
+      if (cancelled) return
+
+      const currentInteractionTime = getLastInteractionTime()
+      if (lastSeenInteractionTime !== currentInteractionTime) {
+        lastSeenInteractionTime = currentInteractionTime
+        lastActivityTimestamp = Date.now()
+      } else if (Date.now() - lastActivityTimestamp >= IDLE_STOP_MS) {
+        return
+      }
+
+      const start = Date.now()
+      const result = await fetchPrStatus()
+      if (cancelled) return
+      lastFetchRef.current = start
+
+      setPrStatus(prev => {
+        const newNumber = result?.number ?? null
+        const newReviewState = result?.reviewState ?? null
+        if (prev.number === newNumber && prev.reviewState === newReviewState) {
+          return prev
+        }
+        return {
+          number: newNumber,
+          url: result?.url ?? null,
+          reviewState: newReviewState,
+          lastUpdated: Date.now(),
+        }
+      })
+
+      if (Date.now() - start > SLOW_GH_THRESHOLD_MS) {
+        disabledRef.current = true
+        return
+      }
+
+      if (!cancelled) {
+        timeoutRef.current = setTimeout(poll, POLL_INTERVAL_MS)
+      }
+    }
+
+    const elapsed = Date.now() - lastFetchRef.current
+    if (elapsed >= POLL_INTERVAL_MS) {
+      void poll()
+    } else {
+      timeoutRef.current = setTimeout(poll, POLL_INTERVAL_MS - elapsed)
+    }
+
+    return () => {
+      cancelled = true
+      if (timeoutRef.current) {
+        clearTimeout(timeoutRef.current)
+        timeoutRef.current = null
+      }
+    }
+  }, [isLoading, enabled])
+
+  return prStatus
+}

package/src/hooks/usePromptSuggestion.ts

@@ -0,0 +1,177 @@
+import { useCallback, useRef } from 'react'
+import { useTerminalFocus } from '../ink/hooks/use-terminal-focus.js'
+import {
+  type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+  logEvent,
+} from '../services/analytics/index.js'
+import { abortSpeculation } from '../services/PromptSuggestion/speculation.js'
+import { useAppState, useSetAppState } from '../state/AppState.js'
+
+type Props = {
+  inputValue: string
+  isAssistantResponding: boolean
+}
+
+export function usePromptSuggestion({
+  inputValue,
+  isAssistantResponding,
+}: Props): {
+  suggestion: string | null
+  markAccepted: () => void
+  markShown: () => void
+  logOutcomeAtSubmission: (
+    finalInput: string,
+    opts?: { skipReset: boolean },
+  ) => void
+} {
+  const promptSuggestion = useAppState(s => s.promptSuggestion)
+  const setAppState = useSetAppState()
+  const isTerminalFocused = useTerminalFocus()
+  const {
+    text: suggestionText,
+    promptId,
+    shownAt,
+    acceptedAt,
+    generationRequestId,
+  } = promptSuggestion
+
+  const suggestion =
+    isAssistantResponding || inputValue.length > 0 ? null : suggestionText
+
+  const isValidSuggestion = suggestionText && shownAt > 0
+
+  // Track engagement depth for telemetry
+  const firstKeystrokeAt = useRef<number>(0)
+  const wasFocusedWhenShown = useRef<boolean>(true)
+  const prevShownAt = useRef<number>(0)
+
+  // Capture focus state when a new suggestion appears (shownAt changes)
+  if (shownAt > 0 && shownAt !== prevShownAt.current) {
+    prevShownAt.current = shownAt
+    wasFocusedWhenShown.current = isTerminalFocused
+    firstKeystrokeAt.current = 0
+  } else if (shownAt === 0) {
+    prevShownAt.current = 0
+  }
+
+  // Record first keystroke while suggestion is visible
+  if (
+    inputValue.length > 0 &&
+    firstKeystrokeAt.current === 0 &&
+    isValidSuggestion
+  ) {
+    firstKeystrokeAt.current = Date.now()
+  }
+
+  const resetSuggestion = useCallback(() => {
+    abortSpeculation(setAppState)
+
+    setAppState(prev => ({
+      ...prev,
+      promptSuggestion: {
+        text: null,
+        promptId: null,
+        shownAt: 0,
+        acceptedAt: 0,
+        generationRequestId: null,
+      },
+    }))
+  }, [setAppState])
+
+  const markAccepted = useCallback(() => {
+    if (!isValidSuggestion) return
+    setAppState(prev => ({
+      ...prev,
+      promptSuggestion: {
+        ...prev.promptSuggestion,
+        acceptedAt: Date.now(),
+      },
+    }))
+  }, [isValidSuggestion, setAppState])
+
+  const markShown = useCallback(() => {
+    // Check shownAt inside setAppState callback to avoid depending on it
+    // (depending on shownAt causes infinite loop when this callback is called)
+    setAppState(prev => {
+      // Only mark shown if not already shown and suggestion exists
+      if (prev.promptSuggestion.shownAt !== 0 || !prev.promptSuggestion.text) {
+        return prev
+      }
+      return {
+        ...prev,
+        promptSuggestion: {
+          ...prev.promptSuggestion,
+          shownAt: Date.now(),
+        },
+      }
+    })
+  }, [setAppState])
+
+  const logOutcomeAtSubmission = useCallback(
+    (finalInput: string, opts?: { skipReset: boolean }) => {
+      if (!isValidSuggestion) return
+
+      // Determine if accepted: either Tab was pressed (acceptedAt set) OR
+      // final input matches suggestion (empty Enter case)
+      const tabWasPressed = acceptedAt > shownAt
+      const wasAccepted = tabWasPressed || finalInput === suggestionText
+      const timeMs = wasAccepted ? acceptedAt || Date.now() : Date.now()
+
+      logEvent('tengu_prompt_suggestion', {
+        source:
+          'cli' as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        outcome: (wasAccepted
+          ? 'accepted'
+          : 'ignored') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        prompt_id:
+          promptId as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        ...(generationRequestId && {
+          generationRequestId:
+            generationRequestId as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        }),
+        ...(wasAccepted && {
+          acceptMethod: (tabWasPressed
+            ? 'tab'
+            : 'enter') as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        }),
+        ...(wasAccepted && {
+          timeToAcceptMs: timeMs - shownAt,
+        }),
+        ...(!wasAccepted && {
+          timeToIgnoreMs: timeMs - shownAt,
+        }),
+        ...(firstKeystrokeAt.current > 0 && {
+          timeToFirstKeystrokeMs: firstKeystrokeAt.current - shownAt,
+        }),
+        wasFocusedWhenShown: wasFocusedWhenShown.current,
+        similarity:
+          Math.round(
+            (finalInput.length / (suggestionText?.length || 1)) * 100,
+          ) / 100,
+        ...(process.env.USER_TYPE === 'ant' && {
+          suggestion:
+            suggestionText as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+          userInput:
+            finalInput as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
+        }),
+      })
+      if (!opts?.skipReset) resetSuggestion()
+    },
+    [
+      isValidSuggestion,
+      acceptedAt,
+      shownAt,
+      suggestionText,
+      promptId,
+      generationRequestId,
+      resetSuggestion,
+    ],
+  )
+
+  return {
+    suggestion,
+    markAccepted,
+    markShown,
+    logOutcomeAtSubmission,
+  }
+}