waveterm/frontend/app/view/preview/preview.tsx

1312 lines
49 KiB
TypeScript

// Copyright 2025, Command Line Inc.
// SPDX-License-Identifier: Apache-2.0
import { BlockNodeModel } from "@/app/block/blocktypes";
import { Button } from "@/app/element/button";
import { CopyButton } from "@/app/element/copybutton";
import { CenteredDiv } from "@/app/element/quickelems";
import { TypeAheadModal } from "@/app/modals/typeaheadmodal";
import { ContextMenuModel } from "@/app/store/contextmenu";
import { tryReinjectKey } from "@/app/store/keymodel";
import { RpcApi } from "@/app/store/wshclientapi";
import { TabRpcClient } from "@/app/store/wshrpcutil";
import { BlockHeaderSuggestionControl } from "@/app/suggestion/suggestion";
import { CodeEditor } from "@/app/view/codeeditor/codeeditor";
import { Markdown } from "@/element/markdown";
import {
atoms,
createBlock,
getApi,
getConnStatusAtom,
getOverrideConfigAtom,
getSettingsKeyAtom,
globalStore,
PLATFORM,
refocusNode,
} from "@/store/global";
import * as services from "@/store/services";
import * as WOS from "@/store/wos";
import { getWebServerEndpoint } from "@/util/endpoints";
import { goHistory, goHistoryBack, goHistoryForward } from "@/util/historyutil";
import { adaptFromReactOrNativeKeyEvent, checkKeyPressed, keydownWrapper } from "@/util/keyutil";
import {
base64ToString,
fireAndForget,
isBlank,
jotaiLoadableValue,
makeConnRoute,
makeNativeLabel,
stringToBase64,
} from "@/util/util";
import { formatRemoteUri } from "@/util/waveutil";
import { Monaco } from "@monaco-editor/react";
import clsx from "clsx";
import { Atom, atom, Getter, PrimitiveAtom, useAtom, useAtomValue, useSetAtom, WritableAtom } from "jotai";
import { loadable } from "jotai/utils";
import type * as MonacoTypes from "monaco-editor/esm/vs/editor/editor.api";
import { OverlayScrollbarsComponent } from "overlayscrollbars-react";
import { createRef, memo, useCallback, useEffect, useMemo, useState } from "react";
import { TransformComponent, TransformWrapper, useControls } from "react-zoom-pan-pinch";
import { CSVView } from "./csvview";
import { DirectoryPreview } from "./directorypreview";
import "./preview.scss";
const MaxFileSize = 1024 * 1024 * 10; // 10MB
const MaxCSVSize = 1024 * 1024 * 1; // 1MB
// TODO drive this using config
const BOOKMARKS: { label: string; path: string }[] = [
{ label: "Home", path: "~" },
{ label: "Desktop", path: "~/Desktop" },
{ label: "Downloads", path: "~/Downloads" },
{ label: "Documents", path: "~/Documents" },
{ label: "Root", path: "/" },
];
type SpecializedViewProps = {
model: PreviewModel;
parentRef: React.RefObject<HTMLDivElement>;
};
const SpecializedViewMap: { [view: string]: ({ model }: SpecializedViewProps) => React.JSX.Element } = {
streaming: StreamingPreview,
markdown: MarkdownPreview,
codeedit: CodeEditPreview,
csv: CSVViewPreview,
directory: DirectoryPreview,
};
const textApplicationMimetypes = [
"application/sql",
"application/x-php",
"application/x-pem-file",
"application/x-httpd-php",
"application/liquid",
"application/graphql",
"application/javascript",
"application/typescript",
"application/x-javascript",
"application/x-typescript",
"application/dart",
"application/vnd.dart",
"application/x-ruby",
"application/sql",
"application/wasm",
"application/x-latex",
"application/x-sh",
"application/x-python",
"application/x-awk",
];
function isTextFile(mimeType: string): boolean {
if (mimeType == null) {
return false;
}
return (
mimeType.startsWith("text/") ||
textApplicationMimetypes.includes(mimeType) ||
(mimeType.startsWith("application/") &&
(mimeType.includes("json") || mimeType.includes("yaml") || mimeType.includes("toml"))) ||
mimeType.includes("xml")
);
}
function canPreview(mimeType: string): boolean {
if (mimeType == null) {
return false;
}
return mimeType.startsWith("text/markdown") || mimeType.startsWith("text/csv");
}
function isStreamingType(mimeType: string): boolean {
if (mimeType == null) {
return false;
}
return (
mimeType.startsWith("application/pdf") ||
mimeType.startsWith("video/") ||
mimeType.startsWith("audio/") ||
mimeType.startsWith("image/")
);
}
export class PreviewModel implements ViewModel {
viewType: string;
blockId: string;
nodeModel: BlockNodeModel;
noPadding?: Atom<boolean>;
blockAtom: Atom<Block>;
viewIcon: Atom<string | IconButtonDecl>;
viewName: Atom<string>;
viewText: Atom<HeaderElem[]>;
preIconButton: Atom<IconButtonDecl>;
endIconButtons: Atom<IconButtonDecl[]>;
previewTextRef: React.RefObject<HTMLDivElement>;
editMode: Atom<boolean>;
canPreview: PrimitiveAtom<boolean>;
specializedView: Atom<Promise<{ specializedView?: string; errorStr?: string }>>;
loadableSpecializedView: Atom<Loadable<{ specializedView?: string; errorStr?: string }>>;
manageConnection: Atom<boolean>;
connStatus: Atom<ConnStatus>;
filterOutNowsh?: Atom<boolean>;
metaFilePath: Atom<string>;
statFilePath: Atom<Promise<string>>;
normFilePath: Atom<Promise<string>>;
loadableStatFilePath: Atom<Loadable<string>>;
loadableFileInfo: Atom<Loadable<FileInfo>>;
connection: Atom<Promise<string>>;
connectionImmediate: Atom<string>;
statFile: Atom<Promise<FileInfo>>;
fullFile: Atom<Promise<FileData>>;
fileMimeType: Atom<Promise<string>>;
fileMimeTypeLoadable: Atom<Loadable<string>>;
fileContentSaved: PrimitiveAtom<string | null>;
fileContent: WritableAtom<Promise<string>, [string], void>;
newFileContent: PrimitiveAtom<string | null>;
connectionError: PrimitiveAtom<string>;
errorMsgAtom: PrimitiveAtom<ErrorMsg>;
openFileModal: PrimitiveAtom<boolean>;
openFileModalDelay: PrimitiveAtom<boolean>;
openFileError: PrimitiveAtom<string>;
openFileModalGiveFocusRef: React.MutableRefObject<() => boolean>;
markdownShowToc: PrimitiveAtom<boolean>;
monacoRef: React.MutableRefObject<MonacoTypes.editor.IStandaloneCodeEditor>;
showHiddenFiles: PrimitiveAtom<boolean>;
refreshVersion: PrimitiveAtom<number>;
refreshCallback: () => void;
directoryKeyDownHandler: (waveEvent: WaveKeyboardEvent) => boolean;
codeEditKeyDownHandler: (waveEvent: WaveKeyboardEvent) => boolean;
showS3 = atom(true);
constructor(blockId: string, nodeModel: BlockNodeModel) {
this.viewType = "preview";
this.blockId = blockId;
this.nodeModel = nodeModel;
let showHiddenFiles = globalStore.get(getSettingsKeyAtom("preview:showhiddenfiles")) ?? true;
this.showHiddenFiles = atom<boolean>(showHiddenFiles);
this.refreshVersion = atom(0);
this.previewTextRef = createRef();
this.openFileModal = atom(false);
this.openFileModalDelay = atom(false);
this.openFileError = atom(null) as PrimitiveAtom<string>;
this.openFileModalGiveFocusRef = createRef();
this.manageConnection = atom(true);
this.blockAtom = WOS.getWaveObjectAtom<Block>(`block:${blockId}`);
this.markdownShowToc = atom(false);
this.filterOutNowsh = atom(true);
this.monacoRef = createRef();
this.connectionError = atom("");
this.errorMsgAtom = atom(null) as PrimitiveAtom<ErrorMsg | null>;
this.viewIcon = atom((get) => {
const blockData = get(this.blockAtom);
if (blockData?.meta?.icon) {
return blockData.meta.icon;
}
const connStatus = get(this.connStatus);
if (connStatus?.status != "connected") {
return null;
}
const mimeTypeLoadable = get(this.fileMimeTypeLoadable);
const mimeType = jotaiLoadableValue(mimeTypeLoadable, "");
if (mimeType == "directory") {
return {
elemtype: "iconbutton",
icon: "folder-open",
longClick: (e: React.MouseEvent<any>) => {
const menuItems: ContextMenuItem[] = BOOKMARKS.map((bookmark) => ({
label: `Go to ${bookmark.label} (${bookmark.path})`,
click: () => this.goHistory(bookmark.path),
}));
ContextMenuModel.showContextMenu(menuItems, e);
},
};
}
return iconForFile(mimeType);
});
this.editMode = atom((get) => {
const blockData = get(this.blockAtom);
return blockData?.meta?.edit ?? false;
});
this.viewName = atom("Preview");
this.viewText = atom((get) => {
let headerPath = get(this.metaFilePath);
const connStatus = get(this.connStatus);
if (connStatus?.status != "connected") {
return [
{
elemtype: "text",
text: headerPath,
className: "preview-filename",
},
];
}
const loadableSV = get(this.loadableSpecializedView);
const isCeView = loadableSV.state == "hasData" && loadableSV.data.specializedView == "codeedit";
const loadableFileInfo = get(this.loadableFileInfo);
if (loadableFileInfo.state == "hasData") {
headerPath = loadableFileInfo.data?.path;
if (headerPath == "~") {
headerPath = `~ (${loadableFileInfo.data?.dir + "/" + loadableFileInfo.data?.name})`;
}
}
const viewTextChildren: HeaderElem[] = [
{
elemtype: "text",
text: headerPath,
ref: this.previewTextRef,
className: "preview-filename",
onClick: () => this.toggleOpenFileModal(),
},
];
let saveClassName = "grey";
if (get(this.newFileContent) !== null) {
saveClassName = "green";
}
if (isCeView) {
const fileInfo = globalStore.get(this.loadableFileInfo);
if (fileInfo.state != "hasData") {
viewTextChildren.push({
elemtype: "textbutton",
text: "Loading ...",
className: clsx(
`grey warning border-radius-4 vertical-padding-2 horizontal-padding-10 font-size-11 font-weight-500`
),
onClick: () => {},
});
} else if (fileInfo.data.readonly) {
viewTextChildren.push({
elemtype: "textbutton",
text: "Read Only",
className: clsx(
`yellow warning border-radius-4 vertical-padding-2 horizontal-padding-10 font-size-11 font-weight-500`
),
onClick: () => {},
});
} else {
viewTextChildren.push({
elemtype: "textbutton",
text: "Save",
className: clsx(
`${saveClassName} warning border-radius-4 vertical-padding-2 horizontal-padding-10 font-size-11 font-weight-500`
),
onClick: () => fireAndForget(this.handleFileSave.bind(this)),
});
}
if (get(this.canPreview)) {
viewTextChildren.push({
elemtype: "textbutton",
text: "Preview",
className:
"grey border-radius-4 vertical-padding-2 horizontal-padding-10 font-size-11 font-weight-500",
onClick: () => fireAndForget(() => this.setEditMode(false)),
});
}
} else if (get(this.canPreview)) {
viewTextChildren.push({
elemtype: "textbutton",
text: "Edit",
className:
"grey border-radius-4 vertical-padding-2 horizontal-padding-10 font-size-11 font-weight-500",
onClick: () => fireAndForget(() => this.setEditMode(true)),
});
}
return [
{
elemtype: "div",
children: viewTextChildren,
},
] as HeaderElem[];
});
this.preIconButton = atom((get) => {
const connStatus = get(this.connStatus);
if (connStatus?.status != "connected") {
return null;
}
const mimeType = jotaiLoadableValue(get(this.fileMimeTypeLoadable), "");
const metaPath = get(this.metaFilePath);
if (mimeType == "directory" && metaPath == "/") {
return null;
}
return {
elemtype: "iconbutton",
icon: "chevron-left",
click: this.goParentDirectory.bind(this),
};
});
this.endIconButtons = atom((get) => {
const connStatus = get(this.connStatus);
if (connStatus?.status != "connected") {
return null;
}
const mimeType = jotaiLoadableValue(get(this.fileMimeTypeLoadable), "");
const loadableSV = get(this.loadableSpecializedView);
const isCeView = loadableSV.state == "hasData" && loadableSV.data.specializedView == "codeedit";
if (mimeType == "directory") {
const showHiddenFiles = get(this.showHiddenFiles);
const settings = get(atoms.settingsAtom);
return [
{
elemtype: "iconbutton",
icon: showHiddenFiles ? "eye" : "eye-slash",
click: () => {
globalStore.set(this.showHiddenFiles, (prev) => !prev);
},
},
{
elemtype: "iconbutton",
icon: "arrows-rotate",
click: () => this.refreshCallback?.(),
},
] as IconButtonDecl[];
} else if (!isCeView && mimeType?.startsWith("text/markdown")) {
return [
{
elemtype: "iconbutton",
icon: "book",
title: "Table of Contents",
click: () => this.markdownShowTocToggle(),
},
] as IconButtonDecl[];
}
return null;
});
this.metaFilePath = atom<string>((get) => {
const file = get(this.blockAtom)?.meta?.file;
if (isBlank(file)) {
return "~";
}
return file;
});
this.statFilePath = atom<Promise<string>>(async (get) => {
const fileInfo = await get(this.statFile);
return fileInfo?.path;
});
this.normFilePath = atom<Promise<string>>(async (get) => {
const fileInfo = await get(this.statFile);
return fileInfo?.path;
});
this.loadableStatFilePath = loadable(this.statFilePath);
this.connection = atom<Promise<string>>(async (get) => {
const connName = get(this.blockAtom)?.meta?.connection;
try {
await RpcApi.ConnEnsureCommand(TabRpcClient, { connname: connName }, { timeout: 60000 });
globalStore.set(this.connectionError, "");
} catch (e) {
globalStore.set(this.connectionError, e as string);
}
return connName;
});
this.connectionImmediate = atom<string>((get) => {
return get(this.blockAtom)?.meta?.connection;
});
this.statFile = atom<Promise<FileInfo>>(async (get) => {
const fileName = get(this.metaFilePath);
console.log("stat file", fileName);
const path = await this.formatRemoteUri(fileName, get);
if (fileName == null) {
return null;
}
const statFile = await RpcApi.FileInfoCommand(TabRpcClient, {
info: {
path,
},
});
console.log("stat file", statFile);
return statFile;
});
this.fileMimeType = atom<Promise<string>>(async (get) => {
const fileInfo = await get(this.statFile);
return fileInfo?.mimetype;
});
this.fileMimeTypeLoadable = loadable(this.fileMimeType);
this.newFileContent = atom(null) as PrimitiveAtom<string | null>;
this.goParentDirectory = this.goParentDirectory.bind(this);
const fullFileAtom = atom<Promise<FileData>>(async (get) => {
const fileName = get(this.metaFilePath);
const path = await this.formatRemoteUri(fileName, get);
if (fileName == null) {
return null;
}
console.log("full file path", path);
const file = await RpcApi.FileReadCommand(TabRpcClient, {
info: {
path,
},
});
console.log("full file", file);
return file;
});
this.fileContentSaved = atom(null) as PrimitiveAtom<string | null>;
const fileContentAtom = atom(
async (get) => {
const newContent = get(this.newFileContent);
if (newContent != null) {
return newContent;
}
const savedContent = get(this.fileContentSaved);
if (savedContent != null) {
return savedContent;
}
const fullFile = await get(fullFileAtom);
return base64ToString(fullFile?.data64);
},
(_, set, update: string) => {
set(this.fileContentSaved, update);
}
);
this.fullFile = fullFileAtom;
this.fileContent = fileContentAtom;
this.specializedView = atom<Promise<{ specializedView?: string; errorStr?: string }>>(async (get) => {
return this.getSpecializedView(get);
});
this.loadableSpecializedView = loadable(this.specializedView);
this.canPreview = atom(false);
this.loadableFileInfo = loadable(this.statFile);
this.connStatus = atom((get) => {
const blockData = get(this.blockAtom);
const connName = blockData?.meta?.connection;
const connAtom = getConnStatusAtom(connName);
return get(connAtom);
});
this.noPadding = atom(true);
}
markdownShowTocToggle() {
globalStore.set(this.markdownShowToc, !globalStore.get(this.markdownShowToc));
}
get viewComponent(): ViewComponent {
return PreviewView;
}
async getSpecializedView(getFn: Getter): Promise<{ specializedView?: string; errorStr?: string }> {
const mimeType = await getFn(this.fileMimeType);
const fileInfo = await getFn(this.statFile);
const fileName = await getFn(this.statFilePath);
const connErr = getFn(this.connectionError);
const editMode = getFn(this.editMode);
const parentFileInfo = await this.getParentInfo(fileInfo);
if (connErr != "") {
return { errorStr: `Connection Error: ${connErr}` };
}
if (parentFileInfo?.notfound ?? false) {
return { errorStr: `Parent Directory Not Found: ${fileInfo.path}` };
}
if (fileInfo?.notfound) {
return { specializedView: "codeedit" };
}
if (mimeType == null) {
return { errorStr: `Unable to determine mimetype for: ${fileInfo.path}` };
}
if (isStreamingType(mimeType)) {
return { specializedView: "streaming" };
}
if (!fileInfo) {
const fileNameStr = fileName ? " " + JSON.stringify(fileName) : "";
return { errorStr: "File Not Found" + fileNameStr };
}
if (fileInfo.size > MaxFileSize) {
return { errorStr: "File Too Large to Preiview (10 MB Max)" };
}
if (mimeType == "text/csv" && fileInfo.size > MaxCSVSize) {
return { errorStr: "CSV File Too Large to Preiview (1 MB Max)" };
}
if (mimeType == "directory") {
return { specializedView: "directory" };
}
if (mimeType == "text/csv") {
if (editMode) {
return { specializedView: "codeedit" };
}
return { specializedView: "csv" };
}
if (mimeType.startsWith("text/markdown")) {
if (editMode) {
return { specializedView: "codeedit" };
}
return { specializedView: "markdown" };
}
if (isTextFile(mimeType) || fileInfo.size == 0) {
return { specializedView: "codeedit" };
}
return { errorStr: `Preview (${mimeType})` };
}
updateOpenFileModalAndError(isOpen, errorMsg = null) {
globalStore.set(this.openFileModal, isOpen);
globalStore.set(this.openFileError, errorMsg);
if (isOpen) {
globalStore.set(this.openFileModalDelay, true);
} else {
const delayVal = globalStore.get(this.openFileModalDelay);
if (delayVal) {
setTimeout(() => {
globalStore.set(this.openFileModalDelay, false);
}, 200);
}
}
}
toggleOpenFileModal() {
const modalOpen = globalStore.get(this.openFileModal);
const delayVal = globalStore.get(this.openFileModalDelay);
if (!modalOpen && delayVal) {
return;
}
this.updateOpenFileModalAndError(!modalOpen);
}
async goHistory(newPath: string) {
let fileName = globalStore.get(this.metaFilePath);
if (fileName == null) {
fileName = "";
}
const blockMeta = globalStore.get(this.blockAtom)?.meta;
const updateMeta = goHistory("file", fileName, newPath, blockMeta);
if (updateMeta == null) {
return;
}
const blockOref = WOS.makeORef("block", this.blockId);
await services.ObjectService.UpdateObjectMeta(blockOref, updateMeta);
// Clear the saved file buffers
globalStore.set(this.fileContentSaved, null);
globalStore.set(this.newFileContent, null);
}
async getParentInfo(fileInfo: FileInfo): Promise<FileInfo | undefined> {
try {
const parentFileInfo = await RpcApi.FileInfoCommand(TabRpcClient, {
info: {
path: await this.formatRemoteUri(fileInfo.dir, globalStore.get),
},
});
console.log("parent file info", parentFileInfo);
return parentFileInfo;
} catch {
return undefined;
}
}
async goParentDirectory({ fileInfo = null }: { fileInfo?: FileInfo | null }) {
// optional parameter needed for recursive case
const defaultFileInfo = await globalStore.get(this.statFile);
if (fileInfo === null) {
fileInfo = defaultFileInfo;
}
if (fileInfo == null) {
this.updateOpenFileModalAndError(false);
return true;
}
try {
const newFileInfo = await RpcApi.FileInfoCommand(TabRpcClient, {
info: {
path: await this.formatRemoteUri(fileInfo.dir, globalStore.get),
},
});
if (newFileInfo.path != "" && newFileInfo.notfound) {
console.log("parent does not exist, ", newFileInfo.path);
this.goParentDirectory({ fileInfo: newFileInfo });
return;
}
this.updateOpenFileModalAndError(false);
await this.goHistory(newFileInfo.path);
refocusNode(this.blockId);
} catch (e) {
globalStore.set(this.openFileError, e.message);
console.error("Error opening file", fileInfo.dir, e);
}
}
async goHistoryBack() {
const blockMeta = globalStore.get(this.blockAtom)?.meta;
const curPath = globalStore.get(this.metaFilePath);
const updateMeta = goHistoryBack("file", curPath, blockMeta, true);
if (updateMeta == null) {
return;
}
updateMeta.edit = false;
const blockOref = WOS.makeORef("block", this.blockId);
await services.ObjectService.UpdateObjectMeta(blockOref, updateMeta);
}
async goHistoryForward() {
const blockMeta = globalStore.get(this.blockAtom)?.meta;
const curPath = globalStore.get(this.metaFilePath);
const updateMeta = goHistoryForward("file", curPath, blockMeta);
if (updateMeta == null) {
return;
}
updateMeta.edit = false;
const blockOref = WOS.makeORef("block", this.blockId);
await services.ObjectService.UpdateObjectMeta(blockOref, updateMeta);
}
async setEditMode(edit: boolean) {
const blockMeta = globalStore.get(this.blockAtom)?.meta;
const blockOref = WOS.makeORef("block", this.blockId);
await services.ObjectService.UpdateObjectMeta(blockOref, { ...blockMeta, edit });
}
async handleFileSave() {
const filePath = await globalStore.get(this.statFilePath);
if (filePath == null) {
return;
}
const newFileContent = globalStore.get(this.newFileContent);
if (newFileContent == null) {
console.log("not saving file, newFileContent is null");
return;
}
try {
await RpcApi.FileWriteCommand(TabRpcClient, {
info: {
path: await this.formatRemoteUri(filePath, globalStore.get),
},
data64: stringToBase64(newFileContent),
});
globalStore.set(this.fileContent, newFileContent);
globalStore.set(this.newFileContent, null);
console.log("saved file", filePath);
} catch (error) {
console.error("Error saving file:", error);
}
}
async handleFileRevert() {
const fileContent = await globalStore.get(this.fileContent);
this.monacoRef.current?.setValue(fileContent);
globalStore.set(this.newFileContent, null);
}
async handleOpenFile(filePath: string) {
const fileInfo = await globalStore.get(this.statFile);
this.updateOpenFileModalAndError(false);
if (fileInfo == null) {
return true;
}
try {
this.goHistory(filePath);
refocusNode(this.blockId);
} catch (e) {
globalStore.set(this.openFileError, e.message);
console.error("Error opening file", filePath, e);
}
}
isSpecializedView(sv: string): boolean {
const loadableSV = globalStore.get(this.loadableSpecializedView);
return loadableSV.state == "hasData" && loadableSV.data.specializedView == sv;
}
getSettingsMenuItems(): ContextMenuItem[] {
const menuItems: ContextMenuItem[] = [];
menuItems.push({
label: "Copy Full Path",
click: () =>
fireAndForget(async () => {
const filePath = await globalStore.get(this.normFilePath);
if (filePath == null) {
return;
}
const conn = await globalStore.get(this.connection);
if (conn) {
// remote path
await navigator.clipboard.writeText(formatRemoteUri(filePath, conn));
} else {
// local path
await navigator.clipboard.writeText(filePath);
}
}),
});
menuItems.push({
label: "Copy File Name",
click: () =>
fireAndForget(async () => {
const fileInfo = await globalStore.get(this.statFile);
if (fileInfo == null || fileInfo.name == null) {
return;
}
await navigator.clipboard.writeText(fileInfo.name);
}),
});
const mimeType = jotaiLoadableValue(globalStore.get(this.fileMimeTypeLoadable), "");
if (mimeType == "directory") {
menuItems.push({
label: "Open Terminal in New Block",
click: () =>
fireAndForget(async () => {
const conn = await globalStore.get(this.connection);
const fileInfo = await globalStore.get(this.statFile);
const termBlockDef: BlockDef = {
meta: {
view: "term",
controller: "shell",
"cmd:cwd": fileInfo.path,
connection: conn,
},
};
await createBlock(termBlockDef);
}),
});
const conn = globalStore.get(this.connectionImmediate);
if (!conn) {
menuItems.push({
label: makeNativeLabel(PLATFORM, true, true),
click: async () => {
const fileInfo = await globalStore.get(this.statFile);
getApi().openNativePath(fileInfo.path);
},
});
}
} else {
const conn = globalStore.get(this.connectionImmediate);
if (!conn) {
menuItems.push({
label: makeNativeLabel(PLATFORM, false, false),
click: async () => {
const fileInfo = await globalStore.get(this.statFile);
getApi().openNativePath(fileInfo.path);
},
});
}
}
const loadableSV = globalStore.get(this.loadableSpecializedView);
const wordWrapAtom = getOverrideConfigAtom(this.blockId, "editor:wordwrap");
const wordWrap = globalStore.get(wordWrapAtom) ?? false;
if (loadableSV.state == "hasData") {
if (loadableSV.data.specializedView == "codeedit") {
if (globalStore.get(this.newFileContent) != null) {
menuItems.push({ type: "separator" });
menuItems.push({
label: "Save File",
click: () => fireAndForget(this.handleFileSave.bind(this)),
});
menuItems.push({
label: "Revert File",
click: () => fireAndForget(this.handleFileRevert.bind(this)),
});
}
menuItems.push({ type: "separator" });
menuItems.push({
label: "Word Wrap",
type: "checkbox",
checked: wordWrap,
click: () =>
fireAndForget(async () => {
const blockOref = WOS.makeORef("block", this.blockId);
await services.ObjectService.UpdateObjectMeta(blockOref, {
"editor:wordwrap": !wordWrap,
});
}),
});
}
}
return menuItems;
}
giveFocus(): boolean {
const openModalOpen = globalStore.get(this.openFileModal);
if (openModalOpen) {
this.openFileModalGiveFocusRef.current?.();
return true;
}
if (this.monacoRef.current) {
this.monacoRef.current.focus();
return true;
}
return false;
}
keyDownHandler(e: WaveKeyboardEvent): boolean {
if (checkKeyPressed(e, "Cmd:ArrowLeft")) {
fireAndForget(this.goHistoryBack.bind(this));
return true;
}
if (checkKeyPressed(e, "Cmd:ArrowRight")) {
fireAndForget(this.goHistoryForward.bind(this));
return true;
}
if (checkKeyPressed(e, "Cmd:ArrowUp")) {
// handle up directory
fireAndForget(() => this.goParentDirectory({}));
return true;
}
if (checkKeyPressed(e, "Cmd:o")) {
this.toggleOpenFileModal();
return true;
}
const canPreview = globalStore.get(this.canPreview);
if (canPreview) {
if (checkKeyPressed(e, "Cmd:e")) {
const editMode = globalStore.get(this.editMode);
fireAndForget(() => this.setEditMode(!editMode));
return true;
}
}
if (this.directoryKeyDownHandler) {
const handled = this.directoryKeyDownHandler(e);
if (handled) {
return true;
}
}
if (this.codeEditKeyDownHandler) {
const handled = this.codeEditKeyDownHandler(e);
if (handled) {
return true;
}
}
return false;
}
async formatRemoteUri(path: string, get: Getter): Promise<string> {
return formatRemoteUri(path, await get(this.connection));
}
}
function MarkdownPreview({ model }: SpecializedViewProps) {
const connName = useAtomValue(model.connection);
const fileInfo = useAtomValue(model.statFile);
const fontSizeOverride = useAtomValue(getOverrideConfigAtom(model.blockId, "markdown:fontsize"));
const fixedFontSizeOverride = useAtomValue(getOverrideConfigAtom(model.blockId, "markdown:fixedfontsize"));
const resolveOpts: MarkdownResolveOpts = useMemo<MarkdownResolveOpts>(() => {
return {
connName: connName,
baseDir: fileInfo.dir,
};
}, [connName, fileInfo.dir]);
return (
<div className="view-preview view-preview-markdown">
<Markdown
textAtom={model.fileContent}
showTocAtom={model.markdownShowToc}
resolveOpts={resolveOpts}
fontSizeOverride={fontSizeOverride}
fixedFontSizeOverride={fixedFontSizeOverride}
/>
</div>
);
}
function ImageZooomControls() {
const { zoomIn, zoomOut, resetTransform } = useControls();
return (
<div className="tools">
<Button onClick={() => zoomIn()} title="Zoom In">
<i className="fa-sharp fa-plus" />
</Button>
<Button onClick={() => zoomOut()} title="Zoom Out">
<i className="fa-sharp fa-minus" />
</Button>
<Button onClick={() => resetTransform()} title="Reset Zoom">
<i className="fa-sharp fa-rotate-left" />
</Button>
</div>
);
}
function StreamingImagePreview({ url }: { url: string }) {
return (
<div className="view-preview view-preview-image">
<TransformWrapper initialScale={1} centerOnInit pinch={{ step: 10 }}>
{({ zoomIn, zoomOut, resetTransform, ...rest }) => (
<>
<ImageZooomControls />
<TransformComponent>
<img src={url} />
</TransformComponent>
</>
)}
</TransformWrapper>
</div>
);
}
function StreamingPreview({ model }: SpecializedViewProps) {
const conn = useAtomValue(model.connection);
const fileInfo = useAtomValue(model.statFile);
const filePath = fileInfo.path;
const remotePath = formatRemoteUri(filePath, conn);
const usp = new URLSearchParams();
usp.set("path", remotePath);
if (conn != null) {
usp.set("connection", conn);
}
const streamingUrl = `${getWebServerEndpoint()}/wave/stream-file?${usp.toString()}`;
if (fileInfo.mimetype === "application/pdf") {
return (
<div className="view-preview view-preview-pdf">
<iframe src={streamingUrl} width="100%" height="100%" name="pdfview" />
</div>
);
}
if (fileInfo.mimetype.startsWith("video/")) {
return (
<div className="view-preview view-preview-video">
<video controls>
<source src={streamingUrl} />
</video>
</div>
);
}
if (fileInfo.mimetype.startsWith("audio/")) {
return (
<div className="view-preview view-preview-audio">
<audio controls>
<source src={streamingUrl} />
</audio>
</div>
);
}
if (fileInfo.mimetype.startsWith("image/")) {
return <StreamingImagePreview url={streamingUrl} />;
}
return <CenteredDiv>Preview Not Supported</CenteredDiv>;
}
function CodeEditPreview({ model }: SpecializedViewProps) {
const fileContent = useAtomValue(model.fileContent);
const setNewFileContent = useSetAtom(model.newFileContent);
const fileName = useAtomValue(model.statFilePath);
const fileInfo = useAtomValue(model.statFile);
const blockMeta = useAtomValue(model.blockAtom)?.meta;
function codeEditKeyDownHandler(e: WaveKeyboardEvent): boolean {
if (checkKeyPressed(e, "Cmd:e")) {
fireAndForget(() => model.setEditMode(false));
return true;
}
if (checkKeyPressed(e, "Cmd:s") || checkKeyPressed(e, "Ctrl:s")) {
fireAndForget(model.handleFileSave.bind(model));
return true;
}
if (checkKeyPressed(e, "Cmd:r")) {
fireAndForget(model.handleFileRevert.bind(model));
return true;
}
return false;
}
useEffect(() => {
model.codeEditKeyDownHandler = codeEditKeyDownHandler;
return () => {
model.codeEditKeyDownHandler = null;
model.monacoRef.current = null;
};
}, []);
function onMount(editor: MonacoTypes.editor.IStandaloneCodeEditor, monaco: Monaco): () => void {
model.monacoRef.current = editor;
editor.onKeyDown((e: MonacoTypes.IKeyboardEvent) => {
const waveEvent = adaptFromReactOrNativeKeyEvent(e.browserEvent);
const handled = tryReinjectKey(waveEvent);
if (handled) {
e.stopPropagation();
e.preventDefault();
}
});
const isFocused = globalStore.get(model.nodeModel.isFocused);
if (isFocused) {
editor.focus();
}
return null;
}
return (
<CodeEditor
blockId={model.blockId}
text={fileContent}
filename={fileName}
fileinfo={fileInfo}
meta={blockMeta}
onChange={(text) => setNewFileContent(text)}
onMount={onMount}
/>
);
}
function CSVViewPreview({ model, parentRef }: SpecializedViewProps) {
const fileContent = useAtomValue(model.fileContent);
const fileName = useAtomValue(model.statFilePath);
return <CSVView parentRef={parentRef} readonly={true} content={fileContent} filename={fileName} />;
}
function iconForFile(mimeType: string): string {
if (mimeType == null) {
mimeType = "unknown";
}
if (mimeType == "application/pdf") {
return "file-pdf";
} else if (mimeType.startsWith("image/")) {
return "image";
} else if (mimeType.startsWith("video/")) {
return "film";
} else if (mimeType.startsWith("audio/")) {
return "headphones";
} else if (mimeType.startsWith("text/markdown")) {
return "file-lines";
} else if (mimeType == "text/csv") {
return "file-csv";
} else if (
mimeType.startsWith("text/") ||
mimeType == "application/sql" ||
(mimeType.startsWith("application/") &&
(mimeType.includes("json") || mimeType.includes("yaml") || mimeType.includes("toml")))
) {
return "file-code";
} else {
return "file";
}
}
const SpecializedView = memo(({ parentRef, model }: SpecializedViewProps) => {
const specializedView = useAtomValue(model.specializedView);
const mimeType = useAtomValue(model.fileMimeType);
const setCanPreview = useSetAtom(model.canPreview);
const path = useAtomValue(model.statFilePath);
useEffect(() => {
setCanPreview(canPreview(mimeType));
}, [mimeType, setCanPreview]);
if (specializedView.errorStr != null) {
return <CenteredDiv>{specializedView.errorStr}</CenteredDiv>;
}
const SpecializedViewComponent = SpecializedViewMap[specializedView.specializedView];
if (!SpecializedViewComponent) {
return <CenteredDiv>Invalid Specialzied View Component ({specializedView.specializedView})</CenteredDiv>;
}
return <SpecializedViewComponent key={path} model={model} parentRef={parentRef} />;
});
const fetchSuggestions = async (
model: PreviewModel,
query: string,
reqContext: SuggestionRequestContext
): Promise<FetchSuggestionsResponse> => {
const fileInfo = await globalStore.get(model.statFile);
if (fileInfo == null) {
return null;
}
const conn = await globalStore.get(model.connection);
return await RpcApi.FetchSuggestionsCommand(
TabRpcClient,
{
suggestiontype: "file",
"file:cwd": fileInfo.path,
query: query,
widgetid: reqContext.widgetid,
reqnum: reqContext.reqnum,
},
{
route: makeConnRoute(conn),
}
);
};
function PreviewView({
blockRef,
contentRef,
model,
}: {
blockId: string;
blockRef: React.RefObject<HTMLDivElement>;
contentRef: React.RefObject<HTMLDivElement>;
model: PreviewModel;
}) {
const connStatus = useAtomValue(model.connStatus);
const [errorMsg, setErrorMsg] = useAtom(model.errorMsgAtom);
if (connStatus?.status != "connected") {
return null;
}
const handleSelect = (s: SuggestionType) => {
model.handleOpenFile(s["file:path"]);
};
const handleTab = (s: SuggestionType, query: string): string => {
if (s["mime:type"] == "directory") {
return s["file:name"] + "/";
} else {
return s["file:name"];
}
};
const fetchSuggestionsFn = async (query, ctx) => {
return await fetchSuggestions(model, query, ctx);
};
return (
<>
{/* <OpenFileModal blockId={blockId} model={model} blockRef={blockRef} /> */}
<div key="fullpreview" className="full-preview scrollbar-hide-until-hover">
{errorMsg && <ErrorOverlay errorMsg={errorMsg} resetOverlay={() => setErrorMsg(null)} />}
<div ref={contentRef} className="full-preview-content">
<SpecializedView parentRef={contentRef} model={model} />
</div>
</div>
<BlockHeaderSuggestionControl
blockRef={blockRef}
openAtom={model.openFileModal}
onClose={() => model.updateOpenFileModalAndError(false)}
onSelect={handleSelect}
onTab={handleTab}
fetchSuggestions={fetchSuggestionsFn}
placeholderText="Open File..."
/>
</>
);
}
const OpenFileModal = memo(
({
model,
blockRef,
blockId,
}: {
model: PreviewModel;
blockRef: React.RefObject<HTMLDivElement>;
blockId: string;
}) => {
const openFileModal = useAtomValue(model.openFileModal);
const curFileName = useAtomValue(model.metaFilePath);
const [filePath, setFilePath] = useState("");
const isNodeFocused = useAtomValue(model.nodeModel.isFocused);
const handleKeyDown = useCallback(
keydownWrapper((waveEvent: WaveKeyboardEvent): boolean => {
if (checkKeyPressed(waveEvent, "Escape")) {
model.updateOpenFileModalAndError(false);
return true;
}
const handleCommandOperations = async () => {
if (checkKeyPressed(waveEvent, "Enter")) {
await model.handleOpenFile(filePath);
return true;
}
return false;
};
handleCommandOperations().catch((error) => {
console.error("Error handling key down:", error);
model.updateOpenFileModalAndError(true, "An error occurred during operation.");
return false;
});
return false;
}),
[model, blockId, filePath, curFileName]
);
const handleFileSuggestionSelect = (value) => {
globalStore.set(model.openFileModal, false);
};
const handleFileSuggestionChange = (value) => {
setFilePath(value);
};
const handleBackDropClick = () => {
globalStore.set(model.openFileModal, false);
};
if (!openFileModal) {
return null;
}
return (
<TypeAheadModal
label="Open path"
blockRef={blockRef}
anchorRef={model.previewTextRef}
onKeyDown={handleKeyDown}
onSelect={handleFileSuggestionSelect}
onChange={handleFileSuggestionChange}
onClickBackdrop={handleBackDropClick}
autoFocus={isNodeFocused}
giveFocusRef={model.openFileModalGiveFocusRef}
/>
);
}
);
const ErrorOverlay = memo(({ errorMsg, resetOverlay }: { errorMsg: ErrorMsg; resetOverlay: () => void }) => {
const showDismiss = errorMsg.showDismiss ?? true;
const buttonClassName = "outlined grey font-size-11 vertical-padding-3 horizontal-padding-7";
let iconClass = "fa-solid fa-circle-exclamation text-[var(--error-color)] text-base";
if (errorMsg.level == "warning") {
iconClass = "fa-solid fa-triangle-exclamation text-[var(--warning-color)] text-base";
}
const handleCopyToClipboard = useCallback(async () => {
await navigator.clipboard.writeText(errorMsg.text);
}, [errorMsg.text]);
return (
<div className="absolute top-[0] left-1.5 right-1.5 z-[var(--zindex-block-mask-inner)] overflow-hidden bg-[var(--conn-status-overlay-bg-color)] backdrop-blur-[50px] rounded-md shadow-lg">
<div className="flex flex-row justify-between p-2.5 pl-3 font-[var(--base-font)] text-[var(--secondary-text-color)]">
<div
className={clsx("flex flex-row items-center gap-3 grow min-w-0", {
"items-start": true,
})}
>
<i className={iconClass}></i>
<div className="flex flex-col items-start gap-1 grow w-full">
<div className="max-w-full text-xs font-semibold leading-4 tracking-[0.11px] text-white">
{errorMsg.status}
</div>
<OverlayScrollbarsComponent
className="group text-xs font-normal leading-[15px] tracking-[0.11px] text-wrap max-h-20 rounded-lg py-1.5 pl-0 relative w-full"
options={{ scrollbars: { autoHide: "leave" } }}
>
<CopyButton
className="invisible group-hover:visible flex absolute top-0 right-1 rounded backdrop-blur-lg p-1 items-center justify-end gap-1"
onClick={handleCopyToClipboard}
title="Copy"
/>
<div>{errorMsg.text}</div>
</OverlayScrollbarsComponent>
{errorMsg.buttons?.map((buttonDef) => (
<Button
className={buttonClassName}
onClick={() => {
buttonDef.onClick();
resetOverlay();
}}
key={crypto.randomUUID()}
>
{buttonDef.text}
</Button>
))}
</div>
{showDismiss && (
<div className="flex items-start">
<Button
className={clsx(buttonClassName, "fa-xmark fa-solid")}
onClick={() => {
if (errorMsg.closeAction) {
errorMsg.closeAction();
}
resetOverlay();
}}
/>
</div>
)}
</div>
</div>
</div>
);
});
export { PreviewView };