refac

2026-08-23 14:40:48 -04:00
parent fb4f476316
commit 78f48a21ee
6 changed files with 434 additions and 7 deletions
@@ -1,6 +1,7 @@
 <script lang="ts">
 	import Collapsible from '$lib/components/common/Collapsible.svelte';
 	import ToolCallDisplay from '$lib/components/common/ToolCallDisplay.svelte';
+	import TerminalOutputFile from './TerminalOutputFile.svelte';
 	import { resolveChatMessageToolCall } from '$lib/apis/chats';
 	import { settings } from '$lib/stores';
 	import { toast } from 'svelte-sonner';
@@ -161,6 +162,8 @@
 				{/each}
 			</div>
 		</ConsecutiveDetailsGroup>
+	{:else if displayItem.type === 'file'}
+		<TerminalOutputFile item={displayItem.item} {chatId} />
 	{:else}
 		{@const detailToken = displayItem.token}
 		{#if detailToken.attributes?.type === 'tool_calls'}
@@ -0,0 +1,265 @@
+<script lang="ts">
+	import { getContext, onDestroy } from 'svelte';
+	import type { Writable } from 'svelte/store';
+	import type { i18n as i18nType } from 'i18next';
+	import { toast } from 'svelte-sonner';
+	import { settings, selectedTerminalId, showControls, showFileNavPath, terminalServers } from '$lib/stores';
+	import { downloadFileBlob, readFile } from '$lib/apis/terminal';
+	import FilePreview from '$lib/components/chat/FileNav/FilePreview.svelte';
+	import Icon from '$lib/components/chat/FileNav/Icon.svelte';
+	import { fileIconName } from '$lib/components/chat/FileNav/fileIcon';
+
+	export let item: any;
+	export let chatId = '';
+
+	const i18n = getContext<Writable<i18nType>>('i18n');
+
+	let expanded = true;
+	let loading = false;
+	let loadedKey = '';
+	let error = '';
+	let objectUrls: string[] = [];
+
+	let fileImageUrl: string | null = null;
+	let fileVideoUrl: string | null = null;
+	let fileAudioUrl: string | null = null;
+	let filePdfData: ArrayBuffer | null = null;
+	let fileSqliteData: ArrayBuffer | null = null;
+	let fileDocxData: ArrayBuffer | null = null;
+	let fileContent: string | null = null;
+	let fileOfficeHtml: string | null = null;
+	let fileOfficeSlides: string[] | null = null;
+	let currentSlide = 0;
+	let excelSheetNames: string[] = [];
+	let selectedExcelSheet = '';
+	let excelWorkbook: any = null;
+
+	const IMAGE_EXTS = new Set(['jpg', 'jpeg', 'png', 'gif', 'webp', 'bmp', 'ico', 'avif']);
+	const VIDEO_EXTS = new Set(['mp4', 'webm', 'mov', 'ogv', 'avi', 'mkv']);
+	const AUDIO_EXTS = new Set(['mp3', 'wav', 'ogg', 'oga', 'flac', 'm4a', 'aac', 'wma', 'opus']);
+	const SQLITE_EXTS = new Set(['db', 'sqlite', 'sqlite3', 'db3']);
+	const OFFICE_EXTS = new Set(['docx', 'xlsx', 'xls', 'pptx']);
+
+	$: path = String(item?.full_path || item?.path || '');
+	$: name = String(item?.name || path.split('/').filter(Boolean).at(-1) || 'file');
+	$: selector = item?.terminal_selector;
+	$: terminal = resolveTerminal();
+	$: unavailable = !terminal;
+	$: previewKey = terminal ? `${terminal.url}|${terminal.key}|${path}|${chatId}` : '';
+	$: previewClass = isImage(path)
+		? 'h-[22rem]'
+		: isPdf(path) || isOffice(path) || getExt(path) === 'html' || getExt(path) === 'htm' || getExt(path) === 'svg'
+			? 'h-96'
+			: 'h-72';
+	$: if (expanded && terminal && path && previewKey !== loadedKey && !loading) {
+		void loadPreview(previewKey);
+	}
+
+	const getExt = (value: string) => value.split('.').pop()?.toLowerCase() ?? '';
+	const isImage = (value: string) => IMAGE_EXTS.has(getExt(value));
+	const isVideo = (value: string) => VIDEO_EXTS.has(getExt(value));
+	const isAudio = (value: string) => AUDIO_EXTS.has(getExt(value));
+	const isSqlite = (value: string) => SQLITE_EXTS.has(getExt(value));
+	const isPdf = (value: string) => getExt(value) === 'pdf';
+	const isOffice = (value: string) => OFFICE_EXTS.has(getExt(value));
+
+	const t = (key: string, vars?: Record<string, unknown>) => $i18n?.t?.(key, vars) ?? key;
+
+	function resolveTerminal(): { url: string; key: string } | null {
+		if (!selector || $selectedTerminalId !== selector) return null;
+
+		const systemTerminal = ($terminalServers ?? []).find((server: any) => server.id === selector);
+		if (systemTerminal?.url) {
+			return { url: systemTerminal.url, key: localStorage.token };
+		}
+
+		const directTerminal = (($settings as any)?.terminalServers ?? []).find(
+			(server: any) => server.url === selector && server.enabled
+		);
+		if (directTerminal?.url) {
+			return { url: directTerminal.url, key: directTerminal.key ?? '' };
+		}
+
+		return null;
+	}
+
+	function clearPreview() {
+		for (const url of objectUrls) URL.revokeObjectURL(url);
+		objectUrls = [];
+		fileImageUrl = null;
+		fileVideoUrl = null;
+		fileAudioUrl = null;
+		filePdfData = null;
+		fileSqliteData = null;
+		fileDocxData = null;
+		fileContent = null;
+		fileOfficeHtml = null;
+		fileOfficeSlides = null;
+		currentSlide = 0;
+		excelSheetNames = [];
+		selectedExcelSheet = '';
+		excelWorkbook = null;
+	}
+
+	async function blobForPreview() {
+		if (!terminal) return null;
+		return downloadFileBlob(terminal.url, terminal.key, path, chatId || undefined);
+	}
+
+	async function loadExcelSheet(sheet: string) {
+		if (!excelWorkbook) return;
+		selectedExcelSheet = sheet;
+		const { excelToTable } = await import('$lib/utils/excelToTable');
+		const result = await excelToTable(excelWorkbook.Sheets[selectedExcelSheet]);
+		const DOMPurify = (await import('dompurify')).default;
+		fileOfficeHtml = DOMPurify.sanitize(result.html);
+	}
+
+	async function loadPreview(key: string) {
+		loadedKey = key;
+		loading = true;
+		error = '';
+		clearPreview();
+
+		try {
+			if (isImage(path) || isVideo(path) || isAudio(path)) {
+				const result = await blobForPreview();
+				if (!result) throw new Error(t('Preview failed'));
+				const url = URL.createObjectURL(result.blob);
+				objectUrls = [...objectUrls, url];
+				if (isImage(path)) fileImageUrl = url;
+				else if (isVideo(path)) fileVideoUrl = url;
+				else fileAudioUrl = url;
+			} else if (isPdf(path) || isSqlite(path) || isOffice(path)) {
+				const result = await blobForPreview();
+				if (!result) throw new Error(t('Preview failed'));
+				const arrayBuffer = await result.blob.arrayBuffer();
+				const ext = getExt(path);
+
+				if (isPdf(path)) {
+					filePdfData = arrayBuffer;
+				} else if (isSqlite(path)) {
+					fileSqliteData = arrayBuffer;
+				} else if (ext === 'docx') {
+					fileDocxData = arrayBuffer;
+				} else if (ext === 'xlsx' || ext === 'xls') {
+					const XLSX = await import('xlsx');
+					excelWorkbook = XLSX.read(new Uint8Array(arrayBuffer), { type: 'array' });
+					excelSheetNames = excelWorkbook.SheetNames;
+					if (excelSheetNames.length > 0) await loadExcelSheet(excelSheetNames[0]);
+				} else if (ext === 'pptx') {
+					const { pptxToImages } = await import('$lib/utils/pptxToHtml');
+					const result = await pptxToImages(arrayBuffer);
+					fileOfficeSlides = result.images;
+				}
+			} else if (terminal) {
+				fileContent = await readFile(terminal.url, terminal.key, path, chatId || undefined);
+			}
+		} catch (e) {
+			error = e instanceof Error ? e.message : t('Preview failed');
+		} finally {
+			loading = false;
+		}
+	}
+
+	function openInFiles() {
+		if (unavailable || !path) return;
+		showControls.set(true);
+		showFileNavPath.set(path);
+	}
+
+	async function downloadFile() {
+		if (!terminal || !path) return;
+		const result = await downloadFileBlob(terminal.url, terminal.key, path, chatId || undefined);
+		if (!result) {
+			toast.error(t('Download failed'));
+			return;
+		}
+
+		const url = URL.createObjectURL(result.blob);
+		const a = document.createElement('a');
+		a.href = url;
+		a.download = result.filename;
+		a.click();
+		URL.revokeObjectURL(url);
+	}
+
+	onDestroy(clearPreview);
+</script>
+
+<div
+	class="my-2 w-full overflow-hidden rounded-xl border border-gray-200 bg-white dark:border-white/8 dark:bg-gray-950/20"
+>
+	<div class="flex h-8 items-center {expanded ? 'border-b border-gray-100 dark:border-white/8' : ''}">
+		<button
+			type="button"
+			class="flex h-full min-w-0 flex-1 items-center gap-2 px-2.5 text-left"
+			on:click={() => (expanded = !expanded)}
+			aria-expanded={expanded}
+		>
+			<div class="flex size-5 shrink-0 items-center justify-center text-gray-500 dark:text-gray-400">
+				<Icon name={fileIconName(name || path || '', 'file')} size={14} />
+			</div>
+			<div class="min-w-0 flex-1 truncate text-xs font-medium text-gray-800 dark:text-gray-100">
+				{name}
+			</div>
+		</button>
+
+		<button
+			type="button"
+			class="mr-1 flex size-6 shrink-0 items-center justify-center rounded text-gray-400 transition-colors hover:text-gray-700 disabled:opacity-40 disabled:hover:text-gray-400 dark:text-gray-500 dark:hover:text-gray-200 dark:disabled:hover:text-gray-500"
+			disabled={unavailable}
+			on:click|stopPropagation={downloadFile}
+			aria-label={t('Download')}
+		>
+			<Icon name="download" size={13} />
+		</button>
+		<button
+			type="button"
+			class="mr-1 flex size-6 shrink-0 items-center justify-center rounded text-gray-400 transition-colors hover:text-gray-700 disabled:opacity-40 disabled:hover:text-gray-400 dark:text-gray-500 dark:hover:text-gray-200 dark:disabled:hover:text-gray-500"
+			disabled={unavailable}
+			on:click|stopPropagation={openInFiles}
+			aria-label={t('Open')}
+		>
+			<Icon name="external-link" size={13} />
+		</button>
+	</div>
+
+	{#if expanded}
+		{#if unavailable}
+			<div class="px-4 py-3 text-xs text-gray-500 dark:text-gray-400">
+				{t('Terminal unavailable')}
+			</div>
+		{:else if error}
+			<div class="px-4 py-3 text-xs text-red-500 dark:text-red-400">{error}</div>
+		{:else}
+			<div class="{previewClass} max-h-[75vh] min-h-24 resize-y overflow-hidden bg-gray-50 dark:bg-gray-950">
+				<FilePreview
+					selectedFile={path}
+					fileLoading={loading}
+					{fileImageUrl}
+					{fileVideoUrl}
+					{fileAudioUrl}
+					{filePdfData}
+					{fileSqliteData}
+					{fileDocxData}
+					{fileContent}
+					baseUrl={terminal?.url ?? ''}
+					apiKey={terminal?.key ?? ''}
+					{fileOfficeHtml}
+					{fileOfficeSlides}
+					{currentSlide}
+					{excelSheetNames}
+					{selectedExcelSheet}
+					onSheetChange={loadExcelSheet}
+					readOnly={true}
+				/>
+				{#if !loading && fileImageUrl === null && fileVideoUrl === null && fileAudioUrl === null && filePdfData === null && fileSqliteData === null && fileDocxData === null && fileContent === null && fileOfficeHtml === null && fileOfficeSlides === null}
+					<div class="flex h-full items-center justify-center px-3 text-xs text-gray-500 dark:text-gray-400">
+						{t('No preview available')}
+					</div>
+				{/if}
+			</div>
+		{/if}
+	{/if}
+</div>
@@ -57,6 +57,11 @@ export type OutputDisplayItem =
 			type: 'detail_group';
 			id: string;
 			tokens: OutputDetailToken[];
+	  }
+	| {
+			type: 'file';
+			id: string;
+			item: Record<string, unknown>;
 	  };

 type ResponseStreamEvent = {
@@ -142,6 +147,48 @@ function getToolResultText(item?: OutputItem): string {
 		.join('');
 }

+function parseJSONStringValue(value: unknown): unknown {
+	if (typeof value !== 'string') {
+		return value;
+	}
+
+	let parsed: unknown = value.trim();
+	while (typeof parsed === 'string') {
+		try {
+			parsed = JSON.parse(parsed);
+		} catch {
+			break;
+		}
+	}
+	return parsed;
+}
+
+function getInlineFileFromToolOutput(callItem?: OutputItem, resultItem?: OutputItem) {
+	if (!callItem || !resultItem || callItem.name !== 'display_file') {
+		return null;
+	}
+
+	const args = parseJSONStringValue(callItem.arguments) as Record<string, unknown>;
+	if (!args || typeof args !== 'object' || args.inline !== true) {
+		return null;
+	}
+
+	const result = parseJSONStringValue(getToolResultText(resultItem)) as Record<string, unknown>;
+	if (
+		!result ||
+		typeof result !== 'object' ||
+		result.type !== 'file' ||
+		result.source !== 'open_terminal' ||
+		result.exists === false ||
+		!result.path ||
+		!result.terminal_selector
+	) {
+		return null;
+	}
+
+	return result;
+}
+
 function buildToolCallToken(item: OutputItem, toolOutputByCallId: Record<string, OutputItem>) {
 	const callId = item.call_id ?? item.id ?? '';
 	const resultItem = toolOutputByCallId[callId];
@@ -297,10 +344,13 @@ export function buildOutputDisplayItems(output: OutputItem[] = []): OutputDispla
 	const displayItems: OutputDisplayItem[] = [];
 	const currentDetailTokens: OutputDetailToken[] = [];
 	const toolOutputByCallId: Record<string, OutputItem> = {};
+	const toolCallByCallId: Record<string, OutputItem> = {};

 	for (const item of output) {
 		if (item?.type === 'function_call_output' && item.call_id) {
 			toolOutputByCallId[item.call_id] = item;
+		} else if (item?.type === 'function_call' && (item.call_id || item.id)) {
+			toolCallByCallId[item.call_id ?? item.id ?? ''] = item;
 		}
 	}

@@ -323,6 +373,18 @@ export function buildOutputDisplayItems(output: OutputItem[] = []): OutputDispla

 	output.forEach((item, index) => {
 		if (item?.type === 'function_call_output') {
+			const inlineFile = getInlineFileFromToolOutput(
+				toolCallByCallId[item.call_id ?? ''],
+				item
+			);
+			if (inlineFile) {
+				flushDetails();
+				displayItems.push({
+					type: 'file',
+					id: item.id ?? `file-${index}`,
+					item: inlineFile
+				});
+			}
 			return;
 		}