wave-code 0.19.9 → 1.0.1

package/src/hooks/useInputManager.ts

@@ -1,10 +1,13 @@
-import { useEffect, useReducer, useCallback } from "react";
+import { useEffect, useReducer, useCallback, useRef } from "react";
 import { Key } from "ink";
 import {
   inputReducer,
   initialState,
   InputManagerCallbacks,
+  ESC_DOUBLE_PRESS_TIMEOUT_MS,
+  btwOverlayActiveRef,
 } from "../managers/inputReducer.js";
+import { createBracketedPasteDetector } from "../utils/bracketedPaste.js";
 import {
   searchFiles as searchFilesUtil,
   PermissionMode,
@@ -18,6 +21,17 @@ export const useInputManager = (
 ) => {
   const [state, dispatch] = useReducer(inputReducer, initialState);
 
+  // Detects bracketed paste (DECSET 2004) markers so pasted text is inserted
+  // without triggering submit — a pasted trailing \r must not be treated as
+  // Enter (see utils/bracketedPaste.ts).
+  const pasteDetectorRef = useRef(createBracketedPasteDetector());
+
+  // Abort plumbing for the /btw side question: ABORT_BTW (set when the overlay
+  // is dismissed while loading) aborts the in-flight onAskBtw call; the
+  // dismissed ref suppresses the late SET_BTW_STATE dispatch / error logging.
+  const btwAbortRef = useRef<AbortController | null>(null);
+  const btwDismissedRef = useRef(false);
+
   const {
     onInputTextChange,
     onCursorPositionChange,
@@ -47,7 +61,7 @@ export const useInputManager = (
     onRecallQueuedMessage,
     onClearMessages,
     onCompact,
-    onGoalCommand,
+    isIdle: isIdleProp,
   } = callbacks;
 
   // Handle debounced file search
@@ -70,25 +84,15 @@ export const useInputManager = (
     }
   }, [state.showFileSelector, state.fileSearchQuery]);
 
-  // Handle paste debouncing
+  // Auto-expire the double-Esc clear pending flag after the timeout window
+  // (aligned with Claude Code's useDoublePress timeout-based expiry).
   useEffect(() => {
-    if (state.isPasting) {
-      const pasteDebounceDelay = parseInt(
-        process.env.PASTE_DEBOUNCE_MS || "30",
-        10,
-      );
-      const timer = setTimeout(() => {
-        const processedInput = state.pasteBuffer.replace(/\r/g, "\n");
-        dispatch({
-          type: "INSERT_TEXT_WITH_PLACEHOLDER",
-          payload: processedInput,
-        });
-        dispatch({ type: "END_PASTE" });
-        dispatch({ type: "RESET_HISTORY_NAVIGATION" });
-      }, pasteDebounceDelay);
-      return () => clearTimeout(timer);
-    }
-  }, [state.isPasting, state.pasteBuffer]);
+    if (!state.escClearPending) return;
+    const timer = setTimeout(() => {
+      dispatch({ type: "RESET_ESC_CLEAR_PENDING" });
+    }, ESC_DOUBLE_PRESS_TIMEOUT_MS);
+    return () => clearTimeout(timer);
+  }, [state.escClearPending]);
 
   // Sync state changes with callbacks
   useEffect(() => {
@@ -127,28 +131,66 @@ export const useInputManager = (
           case "ABORT_MESSAGE":
             onAbortMessage?.();
             break;
+          case "SAVE_PROMPT_HISTORY":
+            PromptHistoryManager.addEntry(
+              effect.content,
+              sessionId,
+              effect.longTextMap,
+              workdir,
+            ).catch((err: unknown) => {
+              logger?.error("Failed to save prompt history", err);
+            });
+            break;
           case "BACKGROUND_CURRENT_TASK":
             onBackgroundCurrentTask?.();
             break;
-          case "ASK_BTW":
+          case "ASK_BTW": {
+            const controller = new AbortController();
+            btwAbortRef.current = controller;
+            btwDismissedRef.current = false;
             try {
-              const answer = await onAskBtw?.(effect.question);
-              dispatch({
-                type: "SET_BTW_STATE",
-                payload: { answer, isLoading: false },
-              });
-            } catch (error) {
-              console.error("Failed to ask side question:", error);
-              dispatch({
-                type: "SET_BTW_STATE",
-                payload: {
-                  answer:
-                    "Error: Failed to get an answer for your side question.",
-                  isLoading: false,
+              const answer = await onAskBtw?.(
+                effect.question,
+                controller.signal,
+                (content: string) => {
+                  // Stream partial answers into the overlay so the user sees
+                  // the response grow in real time (same visual language as
+                  // assistant text / thinking blocks) instead of a static
+                  // "Answering..." indicator.
+                  if (!btwDismissedRef.current) {
+                    dispatch({
+                      type: "SET_BTW_STATE",
+                      payload: { answer: content, isLoading: true },
+                    });
+                  }
                 },
-              });
+              );
+              if (!btwDismissedRef.current) {
+                dispatch({
+                  type: "SET_BTW_STATE",
+                  payload: { answer, isLoading: false },
+                });
+              }
+            } catch (error) {
+              if (!btwDismissedRef.current) {
+                console.error("Failed to ask side question:", error);
+                dispatch({
+                  type: "SET_BTW_STATE",
+                  payload: {
+                    answer:
+                      "Error: Failed to get an answer for your side question.",
+                    isLoading: false,
+                  },
+                });
+              }
             }
             break;
+          }
+          case "ABORT_BTW":
+            btwDismissedRef.current = true;
+            btwAbortRef.current?.abort();
+            btwAbortRef.current = null;
+            break;
           case "PERMISSION_MODE_CHANGE":
             onPermissionModeChange?.(effect.mode);
             break;
@@ -209,6 +251,17 @@ export const useInputManager = (
                 dispatch({ type: "SET_SHOW_LOGIN_COMMAND", payload: true });
               } else if (command === "logout") {
                 dispatch({ type: "SET_SHOW_LOGIN_COMMAND", payload: true });
+              } else if (command === "btw") {
+                // Bare /btw executed via the command selector — show usage
+                // (aligned with Claude Code's empty-args message).
+                dispatch({
+                  type: "SET_BTW_STATE",
+                  payload: {
+                    question: "",
+                    isLoading: false,
+                    answer: "Usage: /btw <your question>",
+                  },
+                });
               } else if (command === "plugin") {
                 dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
               } else if (command === "model") {
@@ -219,8 +272,6 @@ export const useInputManager = (
                 await onClearMessages?.();
               } else if (command === "compact") {
                 await onCompact?.(effect.args);
-              } else if (command === "goal") {
-                await onGoalCommand?.(effect.args);
               }
             }
             break;
@@ -249,7 +300,6 @@ export const useInputManager = (
     onRecallQueuedMessage,
     onClearMessages,
     onCompact,
-    onGoalCommand,
   ]);
 
   useEffect(() => {
@@ -327,7 +377,13 @@ export const useInputManager = (
     onImagesStateChange?.(state.attachedImages);
   }, [state.attachedImages, onImagesStateChange]);
 
-  // /btw side question is handled via pendingEffect "ASK_BTW" above
+  // Keep the shared overlay-active flag in sync so App's Ctrl+C exit handler
+  // can defer to the /btw overlay (ink runs ALL useInput handlers per keypress
+  // with no propagation control).
+  useEffect(() => {
+    btwOverlayActiveRef.current =
+      state.btwState.question !== "" || state.btwState.answer !== undefined;
+  }, [state.btwState.question, state.btwState.answer]);
 
   // Methods
   const insertTextAtCursor = useCallback((text: string) => {
@@ -501,10 +557,11 @@ export const useInputManager = (
           key: {} as Key,
           hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
           hasQueuedMessages: hasQueuedMessagesProp ?? false,
+          isIdle: isIdleProp ?? false,
         },
       });
     },
-    [onHasSlashCommand, hasQueuedMessagesProp],
+    [onHasSlashCommand, hasQueuedMessagesProp, isIdleProp],
   );
 
   const handleSubmit = useCallback(async () => {
@@ -515,9 +572,10 @@ export const useInputManager = (
         key: { return: true } as Key,
         hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
         hasQueuedMessages: hasQueuedMessagesProp ?? false,
+        isIdle: isIdleProp ?? false,
       },
     });
-  }, [onHasSlashCommand, hasQueuedMessagesProp]);
+  }, [onHasSlashCommand, hasQueuedMessagesProp, isIdleProp]);
 
   const expandLongTextPlaceholders = useCallback(
     (text: string) => {
@@ -532,18 +590,51 @@ export const useInputManager = (
 
   const handleInput = useCallback(
     async (input: string, key: Key) => {
+      const result = pasteDetectorRef.current.process(input);
+
+      if (result.kind === "consume") {
+        // Content of an in-flight bracketed paste (or an empty paste):
+        // hold it, never submit or insert prematurely.
+        return true;
+      }
+
+      if (result.kind === "paste") {
+        if (result.leadingInput) {
+          dispatch({
+            type: "HANDLE_KEY",
+            payload: {
+              input: result.leadingInput,
+              key,
+              hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
+              hasQueuedMessages: hasQueuedMessagesProp ?? false,
+              isIdle: isIdleProp ?? false,
+            },
+          });
+        }
+        if (result.text !== "") {
+          // Insert-only: \r → \n normalizes CRLF terminals, matching the
+          // canonical paste path (inputHandlers.handlePasteInput).
+          dispatch({
+            type: "INSERT_TEXT_WITH_PLACEHOLDER",
+            payload: result.text.replace(/\r/g, "\n"),
+          });
+        }
+        return true;
+      }
+
       dispatch({
         type: "HANDLE_KEY",
         payload: {
-          input,
+          input: result.input,
           key,
           hasSlashCommand: (cmd) => !!onHasSlashCommand?.(cmd),
           hasQueuedMessages: hasQueuedMessagesProp ?? false,
+          isIdle: isIdleProp ?? false,
         },
       });
       return true;
     },
-    [onHasSlashCommand, hasQueuedMessagesProp],
+    [onHasSlashCommand, hasQueuedMessagesProp, isIdleProp],
   );
 
   return {
@@ -572,6 +663,7 @@ export const useInputManager = (
     permissionMode: state.permissionMode,
     attachedImages: state.attachedImages,
     btwState: state.btwState,
+    escClearPending: state.escClearPending,
     isManagerReady: true,
 
     // Methods

package/src/index.ts

@@ -53,6 +53,12 @@ export async function main() {
         default: false,
         global: false,
       })
+      .option("daemon", {
+        description:
+          "Start as a background daemon (JSON-RPC over a unix socket at PATH)",
+        type: "string",
+        global: false,
+      })
       .option("show-stats", {
         description: "Show timing and usage statistics in print mode",
         type: "boolean",
@@ -402,6 +408,12 @@ export async function main() {
       return startStdioCli();
     }
 
+    // Handle daemon mode (remote background sessions)
+    if (typeof argv.daemon === "string") {
+      const { startDaemonCli } = await import("./daemon-cli.js");
+      return startDaemonCli(argv.daemon);
+    }
+
     await startCli({
       restoreSessionId: argv.restore as string | undefined,
       continueLastSession: argv.continue as boolean | undefined,

package/src/managers/inputHandlers.ts

@@ -63,16 +63,26 @@ export const handleSubmit = async (
         ? contentWithPlaceholders.substring(5).trim()
         : "";
 
-      const payload = {
-        isActive: true,
-        question,
-        isLoading: question !== "",
-      };
-
-      dispatch({
-        type: "SET_BTW_STATE",
-        payload,
-      });
+      if (question) {
+        dispatch({
+          type: "SET_BTW_STATE",
+          payload: {
+            question,
+            isLoading: true,
+            answer: undefined,
+          },
+        });
+      } else {
+        // Bare /btw — show usage (aligned with Claude Code)
+        dispatch({
+          type: "SET_BTW_STATE",
+          payload: {
+            question: "",
+            isLoading: false,
+            answer: "Usage: /btw <your question>",
+          },
+        });
+      }
       dispatch({ type: "CLEAR_INPUT" });
       dispatch({ type: "RESET_HISTORY_NAVIGATION" });
       dispatch({ type: "CLEAR_LONG_TEXT_MAP" });
@@ -292,17 +302,13 @@ export const handlePasteInput = (
   input: string,
 ): void => {
   const inputString = input;
-  const isPasteOperation =
-    inputString.length > 1 ||
-    inputString.includes("\n") ||
-    inputString.includes("\r");
-
-  if (isPasteOperation) {
-    // Dispatch a single action type; the reducer determines start vs append
-    // by checking pasteBuffer, avoiding stale state issues
+
+  if (inputString.length > 1) {
+    // Multi-char chunk: insert immediately (\r → \n normalizes CRLF
+    // terminals), matching the reducer's HANDLE_KEY path.
     dispatch({
-      type: "APPEND_PASTE_CHUNK",
-      payload: { chunk: inputString, cursorPosition: state.cursorPosition },
+      type: "INSERT_TEXT_WITH_PLACEHOLDER",
+      payload: inputString.replace(/\r/g, "\n"),
     });
   } else {
     let char = inputString;
@@ -367,6 +373,17 @@ export const handleCommandSelect = (
           dispatch({ type: "SET_SHOW_HELP", payload: true });
         } else if (command === "status") {
           dispatch({ type: "SET_SHOW_STATUS_COMMAND", payload: true });
+        } else if (command === "btw") {
+          // Bare /btw executed via the command selector — show usage
+          // (aligned with Claude Code's empty-args message).
+          dispatch({
+            type: "SET_BTW_STATE",
+            payload: {
+              question: "",
+              isLoading: false,
+              answer: "Usage: /btw <your question>",
+            },
+          });
         } else if (command === "plugin") {
           dispatch({ type: "SET_SHOW_PLUGIN_MANAGER", payload: true });
         } else if (command === "model") {
@@ -377,8 +394,6 @@ export const handleCommandSelect = (
           await callbacks.onClearMessages?.();
         } else if (command === "compact") {
           await callbacks.onCompact?.();
-        } else if (command === "goal") {
-          await callbacks.onGoalCommand?.();
         }
       }
     })();
@@ -702,16 +717,24 @@ export const handleInput = async (
   key: Key,
   clearImages?: () => void,
 ): Promise<boolean> => {
-  // Handle ESC to dismiss btw answer
-  if (key.escape && state.btwState.question && !state.btwState.isLoading) {
-    dispatch({
-      type: "SET_BTW_STATE",
-      payload: {
-        question: "",
-        answer: undefined,
-        isLoading: false,
-      },
-    });
+  // /btw overlay handling (mirrors inputReducer's HANDLE_KEY block; this
+  // handler is not wired, kept in sync for consistency). Active while a
+  // question is displayed, or the bare-/btw usage message shows. Only
+  // Escape dismisses; every other key is ignored.
+  if (state.btwState.question || state.btwState.answer) {
+    if (key.escape) {
+      dispatch({
+        type: "SET_BTW_STATE",
+        payload: {
+          question: "",
+          answer: undefined,
+          isLoading: false,
+        },
+      });
+      return true;
+    }
+
+    // Any other key while the overlay is up is ignored
     return true;
   }