// Copyright 2025, Command Line Inc. // SPDX-License-Identifier: Apache-2.0 import { blockViewToIcon, blockViewToName, ConnectionButton, getBlockHeaderIcon, Input } from "@/app/block/blockutil"; import { Button } from "@/app/element/button"; import { useDimensionsWithCallbackRef } from "@/app/hook/useDimensions"; import { ChangeConnectionBlockModal } from "@/app/modals/conntypeahead"; import { ContextMenuModel } from "@/app/store/contextmenu"; import { atoms, getBlockComponentModel, getConnStatusAtom, getSettingsKeyAtom, globalStore, recordTEvent, useBlockAtom, WOS, } from "@/app/store/global"; import { RpcApi } from "@/app/store/wshclientapi"; import { TabRpcClient } from "@/app/store/wshrpcutil"; import { ErrorBoundary } from "@/element/errorboundary"; import { IconButton, ToggleIconButton } from "@/element/iconbutton"; import { MagnifyIcon } from "@/element/magnify"; import { MenuButton } from "@/element/menubutton"; import { NodeModel } from "@/layout/index"; import * as util from "@/util/util"; import { computeBgStyleFromMeta } from "@/util/waveutil"; import clsx from "clsx"; import * as jotai from "jotai"; import { OverlayScrollbarsComponent } from "overlayscrollbars-react"; import * as React from "react"; import { CopyButton } from "../element/copybutton"; import { BlockFrameProps } from "./blocktypes"; const NumActiveConnColors = 8; function handleHeaderContextMenu( e: React.MouseEvent, blockData: Block, viewModel: ViewModel, magnified: boolean, onMagnifyToggle: () => void, onClose: () => void ) { e.preventDefault(); e.stopPropagation(); let menu: ContextMenuItem[] = [ { label: magnified ? "Un-Magnify Block" : "Magnify Block", click: () => { onMagnifyToggle(); }, }, // { // label: "Move to New Window", // click: () => { // const currentTabId = globalStore.get(atoms.staticTabId); // try { // services.WindowService.MoveBlockToNewWindow(currentTabId, blockData.oid); // } catch (e) { // console.error("error moving block to new window", e); // } // }, // }, { type: "separator" }, { label: "Copy BlockId", click: () => { navigator.clipboard.writeText(blockData.oid); }, }, ]; const extraItems = viewModel?.getSettingsMenuItems?.(); if (extraItems && extraItems.length > 0) menu.push({ type: "separator" }, ...extraItems); menu.push( { type: "separator" }, { label: "Close Block", click: onClose, } ); ContextMenuModel.showContextMenu(menu, e); } function getViewIconElem(viewIconUnion: string | IconButtonDecl, blockData: Block): React.ReactElement { if (viewIconUnion == null || typeof viewIconUnion === "string") { const viewIcon = viewIconUnion as string; return
{getBlockHeaderIcon(viewIcon, blockData)}
; } else { return ; } } const OptMagnifyButton = React.memo( ({ magnified, toggleMagnify, disabled }: { magnified: boolean; toggleMagnify: () => void; disabled: boolean }) => { const magnifyDecl: IconButtonDecl = { elemtype: "iconbutton", icon: , title: magnified ? "Minimize" : "Magnify", click: toggleMagnify, disabled, }; return ; } ); function computeEndIcons( viewModel: ViewModel, nodeModel: NodeModel, onContextMenu: (e: React.MouseEvent) => void ): React.ReactElement[] { const endIconsElem: React.ReactElement[] = []; const endIconButtons = util.useAtomValueSafe(viewModel?.endIconButtons); const magnified = jotai.useAtomValue(nodeModel.isMagnified); const ephemeral = jotai.useAtomValue(nodeModel.isEphemeral); const numLeafs = jotai.useAtomValue(nodeModel.numLeafs); const magnifyDisabled = numLeafs <= 1; if (endIconButtons && endIconButtons.length > 0) { endIconsElem.push(...endIconButtons.map((button, idx) => )); } const settingsDecl: IconButtonDecl = { elemtype: "iconbutton", icon: "cog", title: "Settings", click: onContextMenu, }; endIconsElem.push(); if (ephemeral) { const addToLayoutDecl: IconButtonDecl = { elemtype: "iconbutton", icon: "circle-plus", title: "Add to Layout", click: () => { nodeModel.addEphemeralNodeToLayout(); }, }; endIconsElem.push(); } else { endIconsElem.push( ); } const closeDecl: IconButtonDecl = { elemtype: "iconbutton", icon: "xmark-large", title: "Close", click: nodeModel.onClose, }; endIconsElem.push(); return endIconsElem; } const BlockFrame_Header = ({ nodeModel, viewModel, preview, connBtnRef, changeConnModalAtom, error, }: BlockFrameProps & { changeConnModalAtom: jotai.PrimitiveAtom; error?: Error }) => { const [blockData] = WOS.useWaveObjectValue(WOS.makeORef("block", nodeModel.blockId)); let viewName = util.useAtomValueSafe(viewModel?.viewName) ?? blockViewToName(blockData?.meta?.view); const showBlockIds = jotai.useAtomValue(getSettingsKeyAtom("blockheader:showblockids")); let viewIconUnion = util.useAtomValueSafe(viewModel?.viewIcon) ?? blockViewToIcon(blockData?.meta?.view); const preIconButton = util.useAtomValueSafe(viewModel?.preIconButton); let headerTextUnion = util.useAtomValueSafe(viewModel?.viewText); const magnified = jotai.useAtomValue(nodeModel.isMagnified); const prevMagifiedState = React.useRef(magnified); const manageConnection = util.useAtomValueSafe(viewModel?.manageConnection); const dragHandleRef = preview ? null : nodeModel.dragHandleRef; const connName = blockData?.meta?.connection; const connStatus = util.useAtomValueSafe(getConnStatusAtom(connName)); const wshProblem = connName && !connStatus?.wshenabled && connStatus?.status == "connected"; React.useEffect(() => { if (!magnified || preview || prevMagifiedState.current) { return; } RpcApi.ActivityCommand(TabRpcClient, { nummagnify: 1 }); recordTEvent("action:magnify", { "block:view": viewName }); }, [magnified]); if (blockData?.meta?.["frame:title"]) { viewName = blockData.meta["frame:title"]; } if (blockData?.meta?.["frame:icon"]) { viewIconUnion = blockData.meta["frame:icon"]; } if (blockData?.meta?.["frame:text"]) { headerTextUnion = blockData.meta["frame:text"]; } const onContextMenu = React.useCallback( (e: React.MouseEvent) => { handleHeaderContextMenu(e, blockData, viewModel, magnified, nodeModel.toggleMagnify, nodeModel.onClose); }, [magnified] ); const endIconsElem = computeEndIcons(viewModel, nodeModel, onContextMenu); const viewIconElem = getViewIconElem(viewIconUnion, blockData); let preIconButtonElem: React.ReactElement = null; if (preIconButton) { preIconButtonElem = ; } const headerTextElems: React.ReactElement[] = []; if (typeof headerTextUnion === "string") { if (!util.isBlank(headerTextUnion)) { headerTextElems.push(
‎{headerTextUnion}
); } } else if (Array.isArray(headerTextUnion)) { headerTextElems.push(...renderHeaderElements(headerTextUnion, preview)); } if (error != null) { const copyHeaderErr = () => { navigator.clipboard.writeText(error.message + "\n" + error.stack); }; headerTextElems.push(
); } const wshInstallButton: IconButtonDecl = { elemtype: "iconbutton", icon: "link-slash", title: "wsh is not installed for this connection", }; const showNoWshButton = manageConnection && wshProblem && !util.isBlank(connName) && !connName.startsWith("aws:"); return (
{preIconButtonElem}
{viewIconElem}
{viewName}
{showBlockIds &&
[{nodeModel.blockId.substring(0, 8)}]
}
{manageConnection && ( )} {showNoWshButton && }
{headerTextElems}
{endIconsElem}
); }; const HeaderTextElem = React.memo(({ elem, preview }: { elem: HeaderElem; preview: boolean }) => { if (elem.elemtype == "iconbutton") { return ; } else if (elem.elemtype == "toggleiconbutton") { return ; } else if (elem.elemtype == "input") { return ; } else if (elem.elemtype == "text") { return (
elem?.onClick(e)}> ‎{elem.text}
); } else if (elem.elemtype == "textbutton") { return ( ); } else if (elem.elemtype == "div") { return (
{elem.children.map((child, childIdx) => ( ))}
); } else if (elem.elemtype == "menubutton") { return ; } return null; }); function renderHeaderElements(headerTextUnion: HeaderElem[], preview: boolean): React.ReactElement[] { const headerTextElems: React.ReactElement[] = []; for (let idx = 0; idx < headerTextUnion.length; idx++) { const elem = headerTextUnion[idx]; const renderedElement = ; if (renderedElement) { headerTextElems.push(renderedElement); } } return headerTextElems; } const ConnStatusOverlay = React.memo( ({ nodeModel, viewModel, changeConnModalAtom, }: { nodeModel: NodeModel; viewModel: ViewModel; changeConnModalAtom: jotai.PrimitiveAtom; }) => { const [blockData] = WOS.useWaveObjectValue(WOS.makeORef("block", nodeModel.blockId)); const [connModalOpen] = jotai.useAtom(changeConnModalAtom); const connName = blockData.meta?.connection; const connStatus = jotai.useAtomValue(getConnStatusAtom(connName)); const isLayoutMode = jotai.useAtomValue(atoms.controlShiftDelayAtom); const [overlayRefCallback, _, domRect] = useDimensionsWithCallbackRef(30); const width = domRect?.width; const [showError, setShowError] = React.useState(false); const fullConfig = jotai.useAtomValue(atoms.fullConfigAtom); const [showWshError, setShowWshError] = React.useState(false); React.useEffect(() => { if (width) { const hasError = !util.isBlank(connStatus.error); const showError = hasError && width >= 250 && connStatus.status == "error"; setShowError(showError); } }, [width, connStatus, setShowError]); const handleTryReconnect = React.useCallback(() => { const prtn = RpcApi.ConnConnectCommand( TabRpcClient, { host: connName, logblockid: nodeModel.blockId }, { timeout: 60000 } ); prtn.catch((e) => console.log("error reconnecting", connName, e)); }, [connName]); const handleDisableWsh = React.useCallback(async () => { // using unknown is a hack. we need proper types for the // connection config on the frontend const metamaptype: unknown = { "conn:wshenabled": false, }; const data: ConnConfigRequest = { host: connName, metamaptype: metamaptype, }; try { await RpcApi.SetConnectionsConfigCommand(TabRpcClient, data); } catch (e) { console.log("problem setting connection config: ", e); } }, [connName]); const handleRemoveWshError = React.useCallback(async () => { try { await RpcApi.DismissWshFailCommand(TabRpcClient, connName); } catch (e) { console.log("unable to dismiss wsh error: ", e); } }, [connName]); let statusText = `Disconnected from "${connName}"`; let showReconnect = true; if (connStatus.status == "connecting") { statusText = `Connecting to "${connName}"...`; showReconnect = false; } if (connStatus.status == "connected") { showReconnect = false; } let reconDisplay = null; let reconClassName = "outlined grey"; if (width && width < 350) { reconDisplay = ; reconClassName = clsx(reconClassName, "font-size-12 vertical-padding-5 horizontal-padding-6"); } else { reconDisplay = "Reconnect"; reconClassName = clsx(reconClassName, "font-size-11 vertical-padding-3 horizontal-padding-7"); } const showIcon = connStatus.status != "connecting"; const wshConfigEnabled = fullConfig?.connections?.[connName]?.["conn:wshenabled"] ?? true; React.useEffect(() => { const showWshErrorTemp = connStatus.status == "connected" && connStatus.wsherror && connStatus.wsherror != "" && wshConfigEnabled; setShowWshError(showWshErrorTemp); }, [connStatus, wshConfigEnabled]); const handleCopy = React.useCallback( async (e: React.MouseEvent) => { const errTexts = []; if (showError) { errTexts.push(`error: ${connStatus.error}`); } if (showWshError) { errTexts.push(`unable to use wsh: ${connStatus.wsherror}`); } const textToCopy = errTexts.join("\n"); await navigator.clipboard.writeText(textToCopy); }, [showError, showWshError, connStatus.error, connStatus.wsherror] ); if (!showWshError && (isLayoutMode || connStatus.status == "connected" || connModalOpen)) { return null; } return (
{showIcon && }
{statusText}
{(showError || showWshError) && ( {showError ?
error: {connStatus.error}
: null} {showWshError ?
unable to use wsh: {connStatus.wsherror}
: null}
)} {showWshError && ( )}
{showReconnect ? (
) : null} {showWshError ? (
) : null}
); } ); const BlockMask = React.memo(({ nodeModel }: { nodeModel: NodeModel }) => { const isFocused = jotai.useAtomValue(nodeModel.isFocused); const blockNum = jotai.useAtomValue(nodeModel.blockNum); const isLayoutMode = jotai.useAtomValue(atoms.controlShiftDelayAtom); const [blockData] = WOS.useWaveObjectValue(WOS.makeORef("block", nodeModel.blockId)); const style: React.CSSProperties = {}; let showBlockMask = false; if (isFocused) { const tabData = jotai.useAtomValue(atoms.tabAtom); const tabActiveBorderColor = tabData?.meta?.["bg:activebordercolor"]; if (tabActiveBorderColor) { style.borderColor = tabActiveBorderColor; } if (blockData?.meta?.["frame:activebordercolor"]) { style.borderColor = blockData.meta["frame:activebordercolor"]; } } else { const tabData = jotai.useAtomValue(atoms.tabAtom); const tabBorderColor = tabData?.meta?.["bg:bordercolor"]; if (tabBorderColor) { style.borderColor = tabBorderColor; } if (blockData?.meta?.["frame:bordercolor"]) { style.borderColor = blockData.meta["frame:bordercolor"]; } } let innerElem = null; if (isLayoutMode) { showBlockMask = true; innerElem = (
{blockNum}
); } return (
{innerElem}
); }); const BlockFrame_Default_Component = (props: BlockFrameProps) => { const { nodeModel, viewModel, blockModel, preview, numBlocksInTab, children } = props; const [blockData] = WOS.useWaveObjectValue(WOS.makeORef("block", nodeModel.blockId)); const isFocused = jotai.useAtomValue(nodeModel.isFocused); const viewIconUnion = util.useAtomValueSafe(viewModel?.viewIcon) ?? blockViewToIcon(blockData?.meta?.view); const customBg = util.useAtomValueSafe(viewModel?.blockBg); const manageConnection = util.useAtomValueSafe(viewModel?.manageConnection); const changeConnModalAtom = useBlockAtom(nodeModel.blockId, "changeConn", () => { return jotai.atom(false); }) as jotai.PrimitiveAtom; const connModalOpen = jotai.useAtomValue(changeConnModalAtom); const isMagnified = jotai.useAtomValue(nodeModel.isMagnified); const isEphemeral = jotai.useAtomValue(nodeModel.isEphemeral); const [magnifiedBlockBlurAtom] = React.useState(() => getSettingsKeyAtom("window:magnifiedblockblurprimarypx")); const magnifiedBlockBlur = jotai.useAtomValue(magnifiedBlockBlurAtom); const [magnifiedBlockOpacityAtom] = React.useState(() => getSettingsKeyAtom("window:magnifiedblockopacity")); const magnifiedBlockOpacity = jotai.useAtomValue(magnifiedBlockOpacityAtom); const connBtnRef = React.useRef(null); const noHeader = util.useAtomValueSafe(viewModel?.noHeader); React.useEffect(() => { if (!manageConnection) { return; } const bcm = getBlockComponentModel(nodeModel.blockId); if (bcm != null) { bcm.openSwitchConnection = () => { globalStore.set(changeConnModalAtom, true); }; } return () => { const bcm = getBlockComponentModel(nodeModel.blockId); if (bcm != null) { bcm.openSwitchConnection = null; } }; }, [manageConnection]); React.useEffect(() => { // on mount, if manageConnection, call ConnEnsure if (!manageConnection || blockData == null || preview) { return; } const connName = blockData?.meta?.connection; if (!util.isBlank(connName)) { console.log("ensure conn", nodeModel.blockId, connName); RpcApi.ConnEnsureCommand( TabRpcClient, { connname: connName, logblockid: nodeModel.blockId }, { timeout: 60000 } ).catch((e) => { console.log("error ensuring connection", nodeModel.blockId, connName, e); }); } }, [manageConnection, blockData]); const viewIconElem = getViewIconElem(viewIconUnion, blockData); let innerStyle: React.CSSProperties = {}; if (!preview) { innerStyle = computeBgStyleFromMeta(customBg); } const previewElem =
{viewIconElem}
; const headerElem = ( ); const headerElemNoView = React.cloneElement(headerElem, { viewModel: null }); return (
{preview || viewModel == null ? null : ( )}
{noHeader || {headerElem}} {preview ? previewElem : children}
{preview || viewModel == null || !connModalOpen ? null : ( )}
); }; const BlockFrame_Default = React.memo(BlockFrame_Default_Component) as typeof BlockFrame_Default_Component; const BlockFrame = React.memo((props: BlockFrameProps) => { const blockId = props.nodeModel.blockId; const [blockData] = WOS.useWaveObjectValue(WOS.makeORef("block", blockId)); const tabData = jotai.useAtomValue(atoms.tabAtom); if (!blockId || !blockData) { return null; } const numBlocks = tabData?.blockids?.length ?? 0; return ; }); export { BlockFrame, NumActiveConnColors };