| ... | ... | @@ -13,6 +13,7 @@ import { |
| 13 | 13 | XIcon, |
| 14 | 14 | } from "lucide-react"; |
| 15 | 15 | import { |
| 16 | type ContextMenuItem, |
| 16 | 17 | type ResolvedKeybindingsConfig, |
| 17 | 18 | type ScopedThreadRef, |
| 18 | 19 | type ThreadId, |
| ... | ... | @@ -32,7 +33,7 @@ import { |
| 32 | 33 | } from "react"; |
| 33 | 34 | import { Popover, PopoverPopup, PopoverTrigger } from "~/components/ui/popover"; |
| 34 | 35 | import { Button } from "~/components/ui/button"; |
| 35 | | import { writeTextToClipboard } from "~/hooks/useCopyToClipboard"; |
| 36 | import { readTextFromClipboard, writeTextToClipboard } from "~/hooks/useCopyToClipboard"; |
| 36 | 37 | import { cn } from "~/lib/utils"; |
| 37 | 38 | import { type TerminalContextSelection } from "~/lib/terminalContext"; |
| 38 | 39 | import { |
| ... | ... | @@ -255,6 +256,49 @@ export function terminalSelectionLineRange(position: { |
| 255 | 256 | }; |
| 256 | 257 | } |
| 257 | 258 | |
| 259 | export type TerminalContextMenuAction = "add-to-chat" | "copy" | "paste"; |
| 260 | |
| 261 | /** Post-selection popup: just the two selection actions, always enabled. */ |
| 262 | export function terminalSelectionMenuItems(): ContextMenuItem<"add-to-chat" | "copy">[] { |
| 263 | return [ |
| 264 | { id: "add-to-chat", label: "Add to chat" }, |
| 265 | { id: "copy", label: "Copy" }, |
| 266 | ]; |
| 267 | } |
| 268 | |
| 269 | /** |
| 270 | * Right-click menu for the terminal canvas: the selection actions (disabled |
| 271 | * until a selection exists) plus Paste. Paste is always offered: the browser |
| 272 | * (and Electron's default editing menu) can only paste into an editable |
| 273 | * element, so a canvas terminal never gets a usable entry from them. |
| 274 | */ |
| 275 | export function terminalContextMenuItems(options: { |
| 276 | hasSelection: boolean; |
| 277 | }): ContextMenuItem<TerminalContextMenuAction>[] { |
| 278 | return [ |
| 279 | ...terminalSelectionMenuItems().map((item) => ({ |
| 280 | ...item, |
| 281 | disabled: !options.hasSelection, |
| 282 | })), |
| 283 | { id: "paste", label: "Paste" }, |
| 284 | ]; |
| 285 | } |
| 286 | |
| 287 | /** |
| 288 | * An empty selection change may only cancel a selection-action flow that is |
| 289 | * still current: a pending popup timer, or an open popup whose request id has |
| 290 | * not been superseded. A popup already superseded by a right-click keeps its |
| 291 | * menu promise unsettled for a moment; treating it as active would cancel the |
| 292 | * newer context-menu flow instead. |
| 293 | */ |
| 294 | export function shouldClearTerminalSelectionAction(options: { |
| 295 | timerPending: boolean; |
| 296 | openMenuRequestId: number | null; |
| 297 | currentRequestId: number; |
| 298 | }): boolean { |
| 299 | return options.timerPending || options.openMenuRequestId === options.currentRequestId; |
| 300 | } |
| 301 | |
| 258 | 302 | export function shouldHandleTerminalExit( |
| 259 | 303 | current: TerminalSessionState["status"], |
| 260 | 304 | synchronized: TerminalSessionState["status"], |
| ... | ... | @@ -328,7 +372,10 @@ export function TerminalViewport({ |
| 328 | 372 | const selectionPointerRef = useRef<{ x: number; y: number } | null>(null); |
| 329 | 373 | const selectionGestureActiveRef = useRef(false); |
| 330 | 374 | const selectionActionRequestIdRef = useRef(0); |
| 331 | | const selectionActionMenuOpenRef = useRef(false); |
| 375 | // Holds the request id of the selection popup currently on screen, so a |
| 376 | // popup that was superseded (but whose menu promise has not settled yet) |
| 377 | // cannot be mistaken for the active flow. |
| 378 | const openSelectionMenuRequestIdRef = useRef<number | null>(null); |
| 332 | 379 | const selectionActionTimerRef = useRef<number | null>(null); |
| 333 | 380 | const keybindingsRef = useRef(keybindings); |
| 334 | 381 | const runtimeEnvKey = useMemo(() => runtimeEnvSignature(runtimeEnv), [runtimeEnv]); |
| ... | ... | @@ -443,6 +490,12 @@ export function TerminalViewport({ |
| 443 | 490 | onSelectionChange: () => handleSelectionChange(), |
| 444 | 491 | beforeKey: (event) => handleBeforeKey(event), |
| 445 | 492 | onLinkActivate: (text, event) => handleLinkActivate(text, event), |
| 493 | // The surface listens from construction, so a right-click can land |
| 494 | // while `create` is still awaiting WASM — before the handler below it |
| 495 | // exists. The ref is only assigned once that setup has run. |
| 496 | onContextMenu: (event) => { |
| 497 | if (terminalRef.current) void showTerminalContextMenu(event); |
| 498 | }, |
| 446 | 499 | }; |
| 447 | 500 | const terminal = await GhosttyTerminalSurface.create(mount, terminalOptions); |
| 448 | 501 | if (cancelled) { |
| ... | ... | @@ -518,12 +571,98 @@ export function TerminalViewport({ |
| 518 | 571 | }; |
| 519 | 572 | }; |
| 520 | 573 | |
| 574 | const addSelectionToChat = (selection: TerminalContextSelection) => { |
| 575 | handleAddTerminalContext(selection); |
| 576 | terminalRef.current?.clearSelection(); |
| 577 | terminalRef.current?.focus(); |
| 578 | }; |
| 579 | |
| 580 | // A selection-action flow that was superseded while its async work ran |
| 581 | // must go silent: no error message, no focus steal. |
| 582 | const reportIfCurrent = (requestId: number, error: unknown, fallback: string) => { |
| 583 | if (requestId !== selectionActionRequestIdRef.current) return; |
| 584 | const activeTerminal = terminalRef.current; |
| 585 | if (activeTerminal) { |
| 586 | writeSystemMessage(activeTerminal, error instanceof Error ? error.message : fallback); |
| 587 | } |
| 588 | }; |
| 589 | |
| 590 | const focusIfCurrent = (requestId: number) => { |
| 591 | if (requestId === selectionActionRequestIdRef.current) { |
| 592 | terminalRef.current?.focus(); |
| 593 | } |
| 594 | }; |
| 595 | |
| 596 | const copySelection = async (text: string, requestId: number) => { |
| 597 | try { |
| 598 | await writeTextToClipboard(text, "terminal selection"); |
| 599 | } catch (error) { |
| 600 | reportIfCurrent(requestId, error, "Unable to copy terminal selection"); |
| 601 | } |
| 602 | focusIfCurrent(requestId); |
| 603 | }; |
| 604 | |
| 605 | const pasteFromClipboard = async (requestId: number) => { |
| 606 | const activeTerminal = terminalRef.current; |
| 607 | if (!activeTerminal) return; |
| 608 | try { |
| 609 | // The surface owns the read so it can claim the paste race before it |
| 610 | // starts: a paste shortcut fired while the menu read is in flight |
| 611 | // supersedes this paste instead of landing alongside it. |
| 612 | await activeTerminal.pasteFromClipboard( |
| 613 | () => readTextFromClipboard("terminal input"), |
| 614 | () => requestId === selectionActionRequestIdRef.current, |
| 615 | ); |
| 616 | } catch (error) { |
| 617 | reportIfCurrent(requestId, error, "Unable to read the clipboard"); |
| 618 | return; |
| 619 | } |
| 620 | focusIfCurrent(requestId); |
| 621 | }; |
| 622 | |
| 623 | const showTerminalContextMenu = async (event: MouseEvent) => { |
| 624 | if (!localApi || !terminalRef.current) return; |
| 625 | // Own the gesture before anything async: leaving the default alive lets |
| 626 | // the browser (or Electron's editing menu) answer with a Paste entry |
| 627 | // that is permanently disabled over the terminal canvas. |
| 628 | event.preventDefault(); |
| 629 | // A right-click supersedes a selection popup that is pending or open. |
| 630 | clearSelectionAction(); |
| 631 | const selectionAction = readSelectionAction(); |
| 632 | const requestId = selectionActionRequestIdRef.current; |
| 633 | let clicked: TerminalContextMenuAction | null; |
| 634 | try { |
| 635 | clicked = await localApi.contextMenu.show( |
| 636 | terminalContextMenuItems({ hasSelection: selectionAction !== null }), |
| 637 | { x: event.clientX, y: event.clientY }, |
| 638 | ); |
| 639 | } catch (error) { |
| 640 | reportIfCurrent(requestId, error, "Unable to open the terminal context menu"); |
| 641 | focusIfCurrent(requestId); |
| 642 | return; |
| 643 | } |
| 644 | if (requestId !== selectionActionRequestIdRef.current || clicked === null) { |
| 645 | return; |
| 646 | } |
| 647 | switch (clicked) { |
| 648 | case "add-to-chat": |
| 649 | if (selectionAction) addSelectionToChat(selectionAction.selection); |
| 650 | return; |
| 651 | case "copy": |
| 652 | if (selectionAction) await copySelection(selectionAction.clipboardText, requestId); |
| 653 | return; |
| 654 | case "paste": |
| 655 | await pasteFromClipboard(requestId); |
| 656 | return; |
| 657 | } |
| 658 | }; |
| 659 | |
| 521 | 660 | const showSelectionAction = async () => { |
| 522 | 661 | if (!localApi) { |
| 523 | 662 | clearSelectionAction(); |
| 524 | 663 | return; |
| 525 | 664 | } |
| 526 | | if (selectionActionMenuOpenRef.current) { |
| 665 | if (openSelectionMenuRequestIdRef.current !== null) { |
| 527 | 666 | return; |
| 528 | 667 | } |
| 529 | 668 | const nextAction = readSelectionAction(); |
| ... | ... | @@ -532,45 +671,23 @@ export function TerminalViewport({ |
| 532 | 671 | return; |
| 533 | 672 | } |
| 534 | 673 | const requestId = ++selectionActionRequestIdRef.current; |
| 535 | | selectionActionMenuOpenRef.current = true; |
| 674 | openSelectionMenuRequestIdRef.current = requestId; |
| 536 | 675 | const clicked = await localApi.contextMenu |
| 537 | | .show( |
| 538 | | [ |
| 539 | | { id: "add-to-chat", label: "Add to chat" }, |
| 540 | | { id: "copy", label: "Copy" }, |
| 541 | | ], |
| 542 | | nextAction.position, |
| 543 | | ) |
| 676 | .show(terminalSelectionMenuItems(), nextAction.position) |
| 544 | 677 | .finally(() => { |
| 545 | | selectionActionMenuOpenRef.current = false; |
| 678 | if (openSelectionMenuRequestIdRef.current === requestId) { |
| 679 | openSelectionMenuRequestIdRef.current = null; |
| 680 | } |
| 546 | 681 | }); |
| 547 | 682 | if (requestId !== selectionActionRequestIdRef.current || clicked === null) { |
| 548 | 683 | return; |
| 549 | 684 | } |
| 550 | 685 | switch (clicked) { |
| 551 | 686 | case "add-to-chat": |
| 552 | | handleAddTerminalContext(nextAction.selection); |
| 553 | | terminalRef.current?.clearSelection(); |
| 554 | | terminalRef.current?.focus(); |
| 687 | addSelectionToChat(nextAction.selection); |
| 555 | 688 | return; |
| 556 | 689 | case "copy": |
| 557 | | try { |
| 558 | | await writeTextToClipboard(nextAction.clipboardText, "terminal selection"); |
| 559 | | } catch (error) { |
| 560 | | if (requestId !== selectionActionRequestIdRef.current) { |
| 561 | | return; |
| 562 | | } |
| 563 | | const activeTerminal = terminalRef.current; |
| 564 | | if (activeTerminal) { |
| 565 | | writeSystemMessage( |
| 566 | | activeTerminal, |
| 567 | | error instanceof Error ? error.message : "Unable to copy terminal selection", |
| 568 | | ); |
| 569 | | } |
| 570 | | } |
| 571 | | if (requestId === selectionActionRequestIdRef.current) { |
| 572 | | terminalRef.current?.focus(); |
| 573 | | } |
| 690 | await copySelection(nextAction.clipboardText, requestId); |
| 574 | 691 | return; |
| 575 | 692 | } |
| 576 | 693 | }; |
| ... | ... | @@ -684,11 +801,17 @@ export function TerminalViewport({ |
| 684 | 801 | if (terminalRef.current?.hasSelection()) { |
| 685 | 802 | return; |
| 686 | 803 | } |
| 804 | const shouldClear = shouldClearTerminalSelectionAction({ |
| 805 | timerPending: selectionActionTimerRef.current !== null, |
| 806 | openMenuRequestId: openSelectionMenuRequestIdRef.current, |
| 807 | currentRequestId: selectionActionRequestIdRef.current, |
| 808 | }); |
| 809 | if (!shouldClear) return; |
| 687 | 810 | clearSelectionAction(); |
| 688 | 811 | // A copy shortcut that clears the selection (Ctrl+C) must also close |
| 689 | 812 | // the context menu that appears with the selection, but a clear that |
| 690 | 813 | // never opened a menu must not dismiss an unrelated one. |
| 691 | | if (selectionActionMenuOpenRef.current) { |
| 814 | if (openSelectionMenuRequestIdRef.current !== null) { |
| 692 | 815 | void localApi?.contextMenu.close(); |
| 693 | 816 | } |
| 694 | 817 | } |