pi-studio 0.9.16 → 0.9.18

package/index.ts

@@ -162,6 +162,7 @@ interface PreparedStudioPdfExport {
 	createdAt: number;
 	filePath?: string;
 	tempDirPath?: string;
+	persistent?: boolean;
 }
 
 interface PreparedStudioHtmlExport {
@@ -171,6 +172,7 @@ interface PreparedStudioHtmlExport {
 	createdAt: number;
 	filePath?: string;
 	tempDirPath?: string;
+	persistent?: boolean;
 }
 
 interface StudioHtmlAnnotationPlaceholder {
@@ -320,6 +322,13 @@ interface CompletionSuggestionRequestMessage {
 	language?: string;
 	label?: string;
 	path?: string;
+	contextMode?: "cursor" | "session";
+	contextText?: string;
+}
+
+interface CompletionSuggestionCancelRequestMessage {
+	type: "completion_suggestion_cancel_request";
+	requestId: string;
 }
 
 interface QuizGenerateRequestMessage {
@@ -463,6 +472,7 @@ type IncomingStudioMessage =
 	| AnnotationRequestMessage
 	| SendRunRequestMessage
 	| CompletionSuggestionRequestMessage
+	| CompletionSuggestionCancelRequestMessage
 	| QuizGenerateRequestMessage
 	| QuizAnswerRequestMessage
 	| QuizDiscussRequestMessage
@@ -485,6 +495,7 @@ type IncomingStudioMessage =
 const REQUEST_TIMEOUT_MS = 5 * 60 * 1000;
 const PREVIEW_RENDER_MAX_CHARS = 400_000;
 const STUDIO_COMPLETION_MAX_TEXT_CHARS = 250_000;
+const STUDIO_COMPLETION_MAX_CONTEXT_CHARS = 12_000;
 const STUDIO_COMPLETION_PREFIX_CHARS = 12_000;
 const STUDIO_COMPLETION_SUFFIX_CHARS = 6_000;
 const PDF_EXPORT_MAX_CHARS = 400_000;
@@ -2248,6 +2259,8 @@ function inferStudioPdfLanguageFromPath(pathInput: string): string | undefined {
 		".yml": "yaml",
 		".toml": "toml",
 		".lua": "lua",
+		".csv": "csv",
+		".tsv": "tsv",
 		".txt": "text",
 		".rst": "text",
 		".adoc": "text",
@@ -2290,6 +2303,33 @@ function buildStudioResponseExportOutputPath(cwd: string, extension: "pdf" | "ht
 	return join(cwd || process.cwd(), `studio-response-${formatStudioExportTimestamp()}.studio.${extension}`);
 }
 
+function buildStudioPreviewExportPath(sourcePath: string | undefined, resourceDir: string | undefined, fallbackCwd: string, filename: string): string | null {
+	const cleanFilename = String(filename || "").trim();
+	if (!cleanFilename) return null;
+	const source = typeof sourcePath === "string" ? sourcePath.trim() : "";
+	if (source) {
+		const expanded = recoverLikelyDroppedLeadingSlashPath(expandHome(source));
+		return join(dirname(isAbsolute(expanded) ? expanded : resolve(fallbackCwd, expanded)), cleanFilename);
+	}
+	const resource = normalizeStudioResourceDirectoryInput(typeof resourceDir === "string" ? resourceDir : "");
+	if (resource) {
+		const expanded = recoverLikelyDroppedLeadingSlashPath(expandHome(resource));
+		return join(isAbsolute(expanded) ? expanded : resolve(fallbackCwd, expanded), cleanFilename);
+	}
+	return join(fallbackCwd || process.cwd(), cleanFilename);
+}
+
+function writeStudioPreviewExportFile(path: string | null, data: Buffer): { filePath: string | null; error: string | null } {
+	if (!path) return { filePath: null, error: "No export path was resolved." };
+	try {
+		mkdirSync(dirname(path), { recursive: true });
+		writeFileSync(path, data);
+		return { filePath: path, error: null };
+	} catch (error) {
+		return { filePath: null, error: error instanceof Error ? error.message : String(error) };
+	}
+}
+
 function writeStudioFile(pathArg: string, cwd: string, content: string):
 	| { ok: true; label: string; resolvedPath: string }
 	| { ok: false; message: string } {
@@ -2457,6 +2497,7 @@ const STUDIO_LOCAL_LINK_TEXT_EXTENSIONS = new Set([
 	".diff", ".patch",
 ]);
 const STUDIO_LOCAL_LINK_IMAGE_EXTENSIONS = new Set([".png", ".jpg", ".jpeg", ".gif", ".webp"]);
+const STUDIO_LOCAL_LINK_OFFICE_EXTENSIONS = new Set([".docx", ".odt"]);
 const STUDIO_LOCAL_LINK_TEXT_FILENAMES = new Set([
 	".dockerignore", ".editorconfig", ".env", ".env.example", ".eslintignore", ".gitattributes",
 	".gitignore", ".gitmodules", ".npmignore", ".prettierignore", "dockerfile", "gemfile",
@@ -2468,7 +2509,7 @@ const STUDIO_FILE_BROWSER_IGNORED_DIRS = new Set([
 	"__pycache__", ".venv", "venv", ".mypy_cache", ".pytest_cache", ".ruff_cache",
 ]);
 
-type StudioLocalPreviewResourceKind = "pdf" | "text" | "image" | "other";
+type StudioLocalPreviewResourceKind = "pdf" | "text" | "image" | "office" | "other";
 
 interface StudioLocalPreviewResource {
 	filePath: string;
@@ -2564,6 +2605,7 @@ function getStudioLocalPreviewResourceKind(extension: string, filePathOrName?: s
 	if (ext === ".pdf") return "pdf";
 	if (STUDIO_LOCAL_LINK_TEXT_EXTENSIONS.has(ext) || STUDIO_LOCAL_LINK_TEXT_FILENAMES.has(name)) return "text";
 	if (STUDIO_LOCAL_LINK_IMAGE_EXTENSIONS.has(ext)) return "image";
+	if (STUDIO_LOCAL_LINK_OFFICE_EXTENSIONS.has(ext)) return "office";
 	return "other";
 }
 
@@ -4295,6 +4337,125 @@ function wrapStudioCodeAsMarkdown(code: string, language?: string): string {
 	return `${marker}${lang}\n${source}\n${marker}`;
 }
 
+const STUDIO_DELIMITED_PREVIEW_MAX_DATA_ROWS = 200;
+const STUDIO_DELIMITED_PREVIEW_MAX_COLUMNS = 50;
+const STUDIO_DELIMITED_PREVIEW_MAX_CELL_CHARS = 500;
+
+function getStudioDelimitedTextConfig(language?: string): { label: string; delimiter: string } | null {
+	const normalized = normalizeStudioEditorLanguage(language);
+	if (normalized === "csv") return { label: "CSV", delimiter: "," };
+	if (normalized === "tsv") return { label: "TSV", delimiter: "\t" };
+	return null;
+}
+
+function parseStudioDelimitedTextRows(text: string, delimiter: string, maxRows: number): { rows: string[][]; truncatedRows: boolean } {
+	const source = String(text ?? "").replace(/^\uFEFF/, "");
+	const limit = Math.max(1, Math.floor(maxRows));
+	const rows: string[][] = [];
+	let row: string[] = [];
+	let cell = "";
+	let inQuotes = false;
+	let truncatedRows = false;
+
+	const pushCell = () => {
+		row.push(cell);
+		cell = "";
+	};
+	const pushRow = (index: number): boolean => {
+		pushCell();
+		rows.push(row);
+		row = [];
+		if (rows.length >= limit) {
+			truncatedRows = index < source.length - 1;
+			return true;
+		}
+		return false;
+	};
+
+	for (let i = 0; i < source.length; i += 1) {
+		if (rows.length >= limit) {
+			truncatedRows = true;
+			break;
+		}
+		const ch = source[i];
+		if (inQuotes) {
+			if (ch === '"') {
+				if (source[i + 1] === '"') {
+					cell += '"';
+					i += 1;
+				} else {
+					inQuotes = false;
+				}
+			} else {
+				cell += ch;
+			}
+			continue;
+		}
+		if (ch === '"' && cell === "") {
+			inQuotes = true;
+			continue;
+		}
+		if (ch === delimiter) {
+			pushCell();
+			continue;
+		}
+		if (ch === "\n") {
+			if (pushRow(i)) break;
+			continue;
+		}
+		if (ch === "\r") {
+			if (source[i + 1] === "\n") i += 1;
+			if (pushRow(i)) break;
+			continue;
+		}
+		cell += ch;
+	}
+
+	if (!truncatedRows && rows.length < limit && (cell.length > 0 || row.length > 0)) {
+		pushCell();
+		rows.push(row);
+	}
+
+	return { rows, truncatedRows };
+}
+
+function formatStudioDelimitedMarkdownCell(value: string | undefined): string {
+	const raw = String(value ?? "").replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+	const shortened = raw.length > STUDIO_DELIMITED_PREVIEW_MAX_CELL_CHARS
+		? `${raw.slice(0, STUDIO_DELIMITED_PREVIEW_MAX_CELL_CHARS)}…`
+		: raw;
+	return shortened.replace(/\n/g, "<br>").replace(/\|/g, "\\|").trim() || " ";
+}
+
+function formatStudioDelimitedTextAsMarkdown(text: string, language?: string): string | null {
+	const config = getStudioDelimitedTextConfig(language);
+	if (!config) return null;
+	const parsed = parseStudioDelimitedTextRows(text, config.delimiter, STUDIO_DELIMITED_PREVIEW_MAX_DATA_ROWS + 1);
+	const rows = parsed.rows;
+	if (!rows.length) return `_${config.label} file has no tabular data to preview._`;
+	const rawColumnCount = rows.reduce((max, row) => Math.max(max, row.length), 0);
+	const columnCount = Math.min(rawColumnCount, STUDIO_DELIMITED_PREVIEW_MAX_COLUMNS);
+	if (columnCount <= 0) return `_${config.label} file has no tabular data to preview._`;
+	const header = rows[0] ?? [];
+	const dataRows = rows.slice(1);
+	const columnIndexes = Array.from({ length: columnCount }, (_value, index) => index);
+	const lines: string[] = [`**${config.label} preview**`, ""];
+	const notices: string[] = [];
+	if (parsed.truncatedRows) notices.push(`showing first ${Math.max(0, dataRows.length)} data rows`);
+	if (rawColumnCount > columnCount) notices.push(`showing first ${columnCount} of ${rawColumnCount} columns`);
+	if (notices.length) lines.push(`_${notices.join("; ")}._`, "");
+	lines.push(`| ${columnIndexes.map((index) => formatStudioDelimitedMarkdownCell(header[index] || `Column ${index + 1}`)).join(" | ")} |`);
+	lines.push(`| ${columnIndexes.map(() => "---").join(" | ")} |`);
+	if (dataRows.length) {
+		dataRows.forEach((row) => {
+			lines.push(`| ${columnIndexes.map((index) => formatStudioDelimitedMarkdownCell(row[index])).join(" | ")} |`);
+		});
+	} else {
+		lines.push(`| ${columnIndexes.map(() => " ").join(" | ")} |`);
+	}
+	return lines.join("\n");
+}
+
 function extractStudioFenceInfoLanguage(info: string): string | undefined {
 	const firstToken = String(info ?? "").trim().split(/\s+/)[0]?.replace(/^\./, "") ?? "";
 	return normalizeStudioEditorLanguage(firstToken || undefined);
@@ -5245,11 +5406,13 @@ function buildStudioLiteralTextPdfTexConfig(options?: StudioPdfRenderOptions): {
 
 function prepareStudioPdfMarkdown(markdown: string, isLatex?: boolean, editorLanguage?: string): string {
 	if (isLatex) return markdown;
-	const effectiveEditorLanguage = inferStudioPdfLanguage(markdown, editorLanguage);
+	const delimitedMarkdown = formatStudioDelimitedTextAsMarkdown(markdown, editorLanguage);
+	const input = delimitedMarkdown ?? markdown;
+	const effectiveEditorLanguage = delimitedMarkdown ? "markdown" : inferStudioPdfLanguage(input, editorLanguage);
 	const source = effectiveEditorLanguage && effectiveEditorLanguage !== "markdown" && effectiveEditorLanguage !== "latex"
-		&& !isStudioSingleFencedCodeBlock(markdown)
-		? wrapStudioCodeAsMarkdown(markdown, effectiveEditorLanguage)
-		: markdown;
+		&& !isStudioSingleFencedCodeBlock(input)
+		? wrapStudioCodeAsMarkdown(input, effectiveEditorLanguage)
+		: input;
 	const annotationReadySource = !effectiveEditorLanguage || effectiveEditorLanguage === "markdown" || effectiveEditorLanguage === "latex"
 		? replaceStudioAnnotationMarkersForPdf(source)
 		: source;
@@ -5970,17 +6133,19 @@ async function renderStudioStandaloneHtmlWithPandoc(
 	sourcePath?: string,
 	options?: StudioHtmlRenderOptions,
 ): Promise<{ html: Buffer; warning?: string }> {
-	const effectiveEditorLanguage = inferStudioPdfLanguage(markdown, editorLanguage);
-	if (!isLatex && isLikelyStandaloneStudioHtml(markdown, effectiveEditorLanguage)) {
-		return { html: Buffer.from(String(markdown ?? ""), "utf-8") };
+	const delimitedMarkdown = isLatex ? null : formatStudioDelimitedTextAsMarkdown(markdown, editorLanguage);
+	const input = delimitedMarkdown ?? markdown;
+	const effectiveEditorLanguage = delimitedMarkdown ? "markdown" : inferStudioPdfLanguage(input, editorLanguage);
+	if (!isLatex && isLikelyStandaloneStudioHtml(input, effectiveEditorLanguage)) {
+		return { html: Buffer.from(String(input ?? ""), "utf-8") };
 	}
 	const source = !isLatex
 		&& effectiveEditorLanguage
 		&& effectiveEditorLanguage !== "markdown"
 		&& effectiveEditorLanguage !== "latex"
-		&& !isStudioSingleFencedCodeBlock(markdown)
-		? wrapStudioCodeAsMarkdown(markdown, effectiveEditorLanguage)
-		: markdown;
+		&& !isStudioSingleFencedCodeBlock(input)
+		? wrapStudioCodeAsMarkdown(input, effectiveEditorLanguage)
+		: input;
 	const annotationPrepared = prepareStudioAnnotationMarkersForHtml(source);
 	const pdfPrepared = prepareStudioPdfBlocksForHtml(annotationPrepared.markdown);
 	let renderedHtml = await renderStudioMarkdownWithPandoc(pdfPrepared.markdown, isLatex, resourcePath, sourcePath);
@@ -6651,7 +6816,74 @@ function respondHtmlPreviewResourceJson(req: IncomingMessage, res: ServerRespons
 	});
 }
 
-function respondLocalPreviewLinkJson(req: IncomingMessage, res: ServerResponse, requestUrl: URL, resource: StudioLocalPreviewResource, serverState: StudioServerState): void {
+function formatStudioMarkdownAngleTarget(pathText: string): string {
+	return `<${String(pathText || "").replace(/>/g, "%3E")}>`;
+}
+
+function sanitizeStudioPreviewBlockLine(value: string): string {
+	return String(value || "").replace(/[\r\n]+/g, " ").trim();
+}
+
+function buildStudioLocalResourcePreviewDocument(resource: StudioLocalPreviewResource): InitialStudioDocument {
+	const label = basename(resource.filePath) || resource.label || "local preview";
+	const resourcePath = resource.label || basename(resource.filePath) || resource.filePath;
+	const title = sanitizeStudioPreviewBlockLine(label);
+	let text = "";
+	if (resource.kind === "pdf") {
+		text = "```studio-pdf\n"
+			+ `path: ${sanitizeStudioPreviewBlockLine(resourcePath)}\n`
+			+ `title: ${title || "PDF preview"}\n`
+			+ "height: 820\n"
+			+ "```\n";
+	} else if (resource.kind === "image") {
+		text = `![${title || "Image preview"}](${formatStudioMarkdownAngleTarget(resourcePath)})\n`;
+	} else {
+		throw new Error("This local resource cannot be opened as a preview document.");
+	}
+	return {
+		text,
+		label: `${label} preview`,
+		source: "blank",
+		resourceDir: resource.resourceDir,
+	};
+}
+
+function getStudioOfficePandocInputFormat(extension: string): string {
+	const ext = String(extension || "").toLowerCase();
+	if (ext === ".docx") return "docx";
+	if (ext === ".odt") return "odt";
+	return ext.replace(/^\./, "") || "docx";
+}
+
+async function convertStudioOfficeDocumentToMarkdown(resource: StudioLocalPreviewResource): Promise<{ text: string; label: string }> {
+	if (resource.kind !== "office") throw new Error("This local resource is not a supported convertible document.");
+	const pandocCommand = process.env.PANDOC_PATH?.trim() || "pandoc";
+	const inputFormat = getStudioOfficePandocInputFormat(resource.extension);
+	const result = await runStudioSubprocess(pandocCommand, [
+		"-f", inputFormat,
+		"-t", "markdown",
+		"--wrap=none",
+		resource.filePath,
+	], {
+		cwd: dirname(resource.filePath),
+		timeoutMs: STUDIO_PANDOC_TIMEOUT_MS,
+		stdoutMaxBytes: STUDIO_SUBPROCESS_OUTPUT_MAX_BYTES,
+		label: "pandoc document conversion",
+		notFoundMessage: "pandoc was not found. Install pandoc or set PANDOC_PATH to convert DOCX/ODT documents in Studio.",
+	});
+	if (result.code !== 0) {
+		throw new Error(`pandoc failed with exit code ${result.code}${result.stderr ? `: ${result.stderr}` : ""}`);
+	}
+	if (result.stdoutTruncated) {
+		throw new Error("Converted document exceeded Studio's import size limit.");
+	}
+	const label = `converted: ${resource.label || basename(resource.filePath) || "document"}`;
+	const note = `<!-- ${label} from ${resource.filePath}. This is a Markdown conversion; saving will not update the original ${resource.extension || "document"} file. -->`;
+	const body = result.stdout.trim();
+	return { text: `${note}\n\n${body}\n`, label };
+}
+
+async function respondLocalPreviewLinkJson(req: IncomingMessage, res: ServerResponse, requestUrl: URL, resource: StudioLocalPreviewResource, serverState: StudioServerState): Promise<void> {
 	const method = (req.method ?? "GET").toUpperCase();
 	if (method !== "GET" && method !== "HEAD") {
 		res.setHeader("Allow", "GET, HEAD");
@@ -6675,32 +6907,72 @@ function respondLocalPreviewLinkJson(req: IncomingMessage, res: ServerResponse,
 		return;
 	}
 
+	if (action === "preview-url") {
+		if (resource.kind !== "pdf" && resource.kind !== "image") {
+			respondJson(res, 400, { ok: false, error: "This local resource cannot be opened in a Studio preview tab." });
+			return;
+		}
+		const document = buildStudioLocalResourcePreviewDocument(resource);
+		const docId = storeTransientStudioDocument(document);
+		const url = buildStudioUrl(serverState.port, serverState.token, "editor-only", document, docId);
+		const parsedUrl = new URL(url);
+		respondJson(res, 200, {
+			...basePayload,
+			url,
+			relativeUrl: `${parsedUrl.pathname}${parsedUrl.search}`,
+		});
+		return;
+	}
+
 	if (action !== "document" && action !== "editor-url") {
 		respondJson(res, 400, { ok: false, error: "Unsupported local link action." });
 		return;
 	}
-	if (resource.kind !== "text") {
-		respondJson(res, 400, { ok: false, error: "This local resource is not a text document Studio can load into the editor." });
+	if (resource.kind !== "text" && resource.kind !== "office") {
+		respondJson(res, 400, { ok: false, error: "This local resource is not a document Studio can load into the editor." });
 		return;
 	}
 
-	const file = readStudioFile(resource.filePath, dirname(resource.filePath));
-	if (file.ok === false) {
-		respondJson(res, 400, { ok: false, error: file.message });
-		return;
+	let document: InitialStudioDocument;
+	let responseText = "";
+	let converted = false;
+	if (resource.kind === "office") {
+		let conversion: { text: string; label: string };
+		try {
+			conversion = await convertStudioOfficeDocumentToMarkdown(resource);
+		} catch (error) {
+			respondJson(res, 400, { ok: false, error: `Document conversion failed: ${error instanceof Error ? error.message : String(error)}` });
+			return;
+		}
+		converted = true;
+		responseText = conversion.text;
+		document = {
+			text: conversion.text,
+			label: conversion.label,
+			source: "blank",
+			resourceDir: resource.resourceDir,
+		};
+	} else {
+		const file = readStudioFile(resource.filePath, dirname(resource.filePath));
+		if (file.ok === false) {
+			respondJson(res, 400, { ok: false, error: file.message });
+			return;
+		}
+		responseText = file.text;
+		document = {
+			text: file.text,
+			label: resource.label || file.label,
+			source: "file",
+			path: file.resolvedPath,
+			resourceDir: resource.resourceDir,
+		};
 	}
-
-	const document: InitialStudioDocument = {
-		text: file.text,
-		label: resource.label || file.label,
-		source: "file",
-		path: file.resolvedPath,
-		resourceDir: resource.resourceDir,
-	};
 	if (action === "document") {
 		respondJson(res, 200, {
 			...basePayload,
-			text: file.text,
+			text: responseText,
+			label: document.label,
+			converted,
 			resourceDir: resource.resourceDir,
 		});
 		return;
@@ -6711,6 +6983,7 @@ function respondLocalPreviewLinkJson(req: IncomingMessage, res: ServerResponse,
 	const parsedUrl = new URL(url);
 	respondJson(res, 200, {
 		...basePayload,
+		converted,
 		url,
 		relativeUrl: `${parsedUrl.pathname}${parsedUrl.search}`,
 	});
@@ -7217,6 +7490,8 @@ function buildStudioCompletionSuggestionPrompt(options: {
 	language?: string;
 	label?: string;
 	path?: string;
+	contextMode?: "cursor" | "session";
+	contextText?: string;
 }): string {
 	const text = String(options.text || "");
 	const start = Math.max(0, Math.min(Math.floor(options.selectionStart || 0), text.length));
@@ -7226,17 +7501,23 @@ function buildStudioCompletionSuggestionPrompt(options: {
 	const suffix = text.slice(end, Math.min(text.length, end + STUDIO_COMPLETION_SUFFIX_CHARS));
 	const language = String(options.language || "").trim() || "unknown";
 	const label = String(options.label || options.path || "Studio editor").trim();
+	const contextText = String(options.contextText || "").trim().slice(-STUDIO_COMPLETION_MAX_CONTEXT_CHARS);
 	return [
 		"Generate an inline completion for the current editor cursor position.",
 		"Return only the exact text to insert. Do not wrap it in Markdown fences. Do not explain.",
 		"Match the surrounding language, style, indentation, and register.",
 		"Keep the suggestion short unless the context clearly asks for a longer continuation.",
+		contextText
+			? "Use the extra session context only as background. Do not continue the extra context directly unless the editor cursor calls for it."
+			: "Use only the cursor-local editor context below.",
 		selected
 			? "The selected text will be replaced by the completion."
 			: "The completion will be inserted at the cursor.",
 		"",
 		`File/context label: ${label}`,
 		`Language mode: ${language}`,
+		`Suggestion context mode: ${contextText ? "editor plus latest response" : "editor only"}`,
+		contextText ? ["", "<extra_context>", contextText, "</extra_context>"].join("\n") : "",
 		"",
 		"<prefix>",
 		prefix,
@@ -7262,6 +7543,9 @@ async function runStudioCompletionSuggestion(ctx: StudioModelRequestContext, opt
 	language?: string;
 	label?: string;
 	path?: string;
+	contextMode?: "cursor" | "session";
+	contextText?: string;
+	signal?: AbortSignal;
 }): Promise<string> {
 	const prompt = buildStudioCompletionSuggestionPrompt(options);
 	// Intentionally omit `reasoning`: pi-ai treats absent reasoning as off/disabled
@@ -7271,6 +7555,7 @@ async function runStudioCompletionSuggestion(ctx: StudioModelRequestContext, opt
 		maxTokens: 650,
 		timeoutMs: 60_000,
 		trim: false,
+		signal: options.signal,
 	}));
 	if (!suggestion.trim()) throw new Error("Model returned an empty completion suggestion.");
 	return suggestion;
@@ -7685,12 +7970,20 @@ function parseIncomingMessage(data: RawData): IncomingStudioMessage | null {
 		};
 	}
 
+	if (msg.type === "completion_suggestion_cancel_request" && typeof msg.requestId === "string") {
+		return {
+			type: "completion_suggestion_cancel_request",
+			requestId: msg.requestId,
+		};
+	}
+
 	if (msg.type === "completion_suggestion_request" && typeof msg.requestId === "string" && typeof msg.text === "string") {
 		const textLength = msg.text.length;
 		const rawStart = typeof msg.selectionStart === "number" && Number.isFinite(msg.selectionStart) ? msg.selectionStart : textLength;
 		const rawEnd = typeof msg.selectionEnd === "number" && Number.isFinite(msg.selectionEnd) ? msg.selectionEnd : rawStart;
 		const selectionStart = Math.max(0, Math.min(Math.floor(rawStart), textLength));
 		const selectionEnd = Math.max(selectionStart, Math.min(Math.floor(rawEnd), textLength));
+		const contextMode = msg.contextMode === "session" ? "session" : "cursor";
 		return {
 			type: "completion_suggestion_request",
 			requestId: msg.requestId,
@@ -7700,6 +7993,8 @@ function parseIncomingMessage(data: RawData): IncomingStudioMessage | null {
 			language: typeof msg.language === "string" ? msg.language : undefined,
 			label: typeof msg.label === "string" ? msg.label : undefined,
 			path: typeof msg.path === "string" ? msg.path : undefined,
+			contextMode,
+			contextText: contextMode === "session" && typeof msg.contextText === "string" ? msg.contextText.slice(-STUDIO_COMPLETION_MAX_CONTEXT_CHARS) : undefined,
 		};
 	}
 
@@ -9582,6 +9877,11 @@ ${cssVarsBlock}
             <div class="source-actions-row">
               <button id="copyDraftBtn" type="button" title="Copy the current editor text to the clipboard.">Copy</button>
               <button id="suggestCompletionBtn" type="button" title="Ask the current model for a short completion at the editor cursor. Shortcut: Option/Alt+Tab where available, or Cmd/Ctrl+Shift+Space from the editor.">Suggest</button>
+              <button id="suggestCompletionOptionsBtn" type="button" hidden title="Suggestion context options">▾</button>
+              <select id="completionContextSelect" hidden aria-label="Suggestion context mode" title="Choose how much context Suggest includes.">
+                <option value="cursor" selected>Context: editor only</option>
+                <option value="session">Context: editor + latest response</option>
+              </select>
               <button id="openCompanionBtn" type="button" title="Open a detached copy of the current editor text in a new editor-only Studio tab.">New editor</button>
               <button id="sendEditorBtn" type="button">Send to pi editor</button>
             </div>
@@ -9608,6 +9908,7 @@ ${cssVarsBlock}
                 <option value="c">Syntax highlight: C</option>
                 <option value="cpp">Syntax highlight: C++</option>
                 <option value="css">Syntax highlight: CSS</option>
+                <option value="csv">Syntax highlight: CSV</option>
                 <option value="diff">Syntax highlight: Diff</option>
                 <option value="fortran">Syntax highlight: Fortran</option>
                 <option value="go">Syntax highlight: Go</option>
@@ -9626,6 +9927,7 @@ ${cssVarsBlock}
                 <option value="rust">Syntax highlight: Rust</option>
                 <option value="swift">Syntax highlight: Swift</option>
                 <option value="toml">Syntax highlight: TOML</option>
+                <option value="tsv">Syntax highlight: TSV</option>
                 <option value="typescript">Syntax highlight: TypeScript</option>
                 <option value="xml">Syntax highlight: XML</option>
                 <option value="yaml">Syntax highlight: YAML</option>
@@ -9811,11 +10113,11 @@ ${cssVarsBlock}
       <div class="shortcuts-header">
         <div>
           <h2 id="shortcutsTitle">Keyboard shortcuts</h2>
-          <p class="shortcuts-description">Studio navigation and high-frequency actions.</p>
+          <p class="shortcuts-description">Studio navigation and high-frequency actions. Use arrow keys, Page Up/Down, Home/End, or mouse/trackpad to scroll.</p>
         </div>
         <button id="shortcutsCloseBtn" class="shortcuts-close-btn" type="button" aria-label="Close keyboard shortcuts">Close</button>
       </div>
-      <div class="shortcuts-body">
+      <div id="shortcutsBody" class="shortcuts-body" tabindex="0" aria-label="Keyboard shortcuts list">
         <section class="shortcuts-group">
           <h3>Navigation</h3>
           <dl>
@@ -9844,6 +10146,7 @@ ${cssVarsBlock}
             <div><dt>Cmd/Ctrl+Enter</dt><dd>Run editor text, or queue steering during an active run</dd></div>
             <div><dt>Option/Alt+Tab or Cmd/Ctrl+Shift+Space</dt><dd>Suggest a completion at the editor cursor</dd></div>
             <div><dt>Tab</dt><dd>Insert a visible completion suggestion; otherwise indent selected editor text</dd></div>
+            <div><dt>Esc</dt><dd>Dismiss a visible completion suggestion, close overlays, exit pane focus, or stop an active request</dd></div>
             <div><dt>Shift+Tab</dt><dd>Unindent selected editor text</dd></div>
           </dl>
         </section>
@@ -9938,6 +10241,7 @@ export default function (pi: ExtensionAPI) {
 	let studioReplActiveSessionName: string | null = null;
 	let compactInProgress = false;
 	let compactRequestId: string | null = null;
+	const activeCompletionSuggestions = new Map<string, AbortController>();
 
 	const selectStudioReplSessionForTool = (params: { sessionName?: string; target?: string }): { session: StudioReplSessionInfo | null; error?: string; sessions: StudioReplSessionInfo[] } => {
 		const state = listStudioReplSessions();
@@ -11424,6 +11728,21 @@ export default function (pi: ExtensionAPI) {
 			return;
 		}
 
+		if (msg.type === "completion_suggestion_cancel_request") {
+			if (!isValidRequestId(msg.requestId)) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "Invalid request ID." });
+				return;
+			}
+			const controller = activeCompletionSuggestions.get(msg.requestId);
+			if (!controller) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "No matching suggestion request is running." });
+				return;
+			}
+			controller.abort();
+			sendToClient(client, { type: "completion_suggestion_progress", requestId: msg.requestId, message: "Stopping suggestion…" });
+			return;
+		}
+
 		if (msg.type === "completion_suggestion_request") {
 			if (!isValidRequestId(msg.requestId)) {
 				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "Invalid request ID." });
@@ -11443,6 +11762,8 @@ export default function (pi: ExtensionAPI) {
 				return;
 			}
 			sendToClient(client, { type: "completion_suggestion_progress", requestId: msg.requestId, message: "Generating suggestion…" });
+			const completionController = new AbortController();
+			activeCompletionSuggestions.set(msg.requestId, completionController);
 			void (async () => {
 				try {
 					const suggestion = await runStudioCompletionSuggestion(ctx, {
@@ -11452,6 +11773,9 @@ export default function (pi: ExtensionAPI) {
 						language: msg.language,
 						label: msg.label,
 						path: msg.path,
+						contextMode: msg.contextMode,
+						contextText: msg.contextText,
+						signal: completionController.signal,
 					});
 					sendToClient(client, {
 						type: "completion_suggestion_result",
@@ -11464,8 +11788,12 @@ export default function (pi: ExtensionAPI) {
 					sendToClient(client, {
 						type: "completion_suggestion_error",
 						requestId: msg.requestId,
-						message: `Suggestion failed: ${error instanceof Error ? error.message : String(error)}`,
+						message: completionController.signal.aborted
+							? "Suggestion stopped."
+							: `Suggestion failed: ${error instanceof Error ? error.message : String(error)}`,
 					});
+				} finally {
+					activeCompletionSuggestions.delete(msg.requestId);
 				}
 			})();
 			return;
@@ -12011,6 +12339,7 @@ export default function (pi: ExtensionAPI) {
 	};
 
 	const disposePreparedPdfExport = (entry: PreparedStudioPdfExport | null | undefined) => {
+		if (entry?.persistent) return;
 		if (!entry?.tempDirPath) return;
 		void rm(entry.tempDirPath, { recursive: true, force: true }).catch(() => undefined);
 	};
@@ -12039,7 +12368,7 @@ export default function (pi: ExtensionAPI) {
 		}
 	};
 
-	const storePreparedPdfExport = (pdf: Buffer, filename: string, warning?: string): string => {
+	const storePreparedPdfExport = (pdf: Buffer, filename: string, warning?: string, filePath?: string): string => {
 		prunePreparedPdfExports();
 		const exportId = randomUUID();
 		preparedPdfExports.set(exportId, {
@@ -12047,6 +12376,8 @@ export default function (pi: ExtensionAPI) {
 			filename,
 			warning,
 			createdAt: Date.now(),
+			filePath,
+			persistent: Boolean(filePath),
 		});
 		return exportId;
 	};
@@ -12055,7 +12386,7 @@ export default function (pi: ExtensionAPI) {
 		prunePreparedPdfExports();
 		const entry = preparedPdfExports.get(exportId);
 		if (!entry) return null;
-		if (entry.filePath && entry.tempDirPath) return entry;
+		if (entry.filePath && (entry.tempDirPath || entry.persistent)) return entry;
 
 		const tempDirPath = join(tmpdir(), `pi-studio-prepared-pdf-${Date.now()}-${randomUUID()}`);
 		const filePath = join(tempDirPath, sanitizePdfFilename(entry.filename));
@@ -12105,6 +12436,7 @@ export default function (pi: ExtensionAPI) {
 	};
 
 	const disposePreparedHtmlExport = (entry: PreparedStudioHtmlExport | null | undefined) => {
+		if (entry?.persistent) return;
 		if (!entry?.tempDirPath) return;
 		void rm(entry.tempDirPath, { recursive: true, force: true }).catch(() => undefined);
 	};
@@ -12133,7 +12465,7 @@ export default function (pi: ExtensionAPI) {
 		}
 	};
 
-	const storePreparedHtmlExport = (html: Buffer, filename: string, warning?: string): string => {
+	const storePreparedHtmlExport = (html: Buffer, filename: string, warning?: string, filePath?: string): string => {
 		prunePreparedHtmlExports();
 		const exportId = randomUUID();
 		preparedHtmlExports.set(exportId, {
@@ -12141,6 +12473,8 @@ export default function (pi: ExtensionAPI) {
 			filename,
 			warning,
 			createdAt: Date.now(),
+			filePath,
+			persistent: Boolean(filePath),
 		});
 		return exportId;
 	};
@@ -12149,7 +12483,7 @@ export default function (pi: ExtensionAPI) {
 		prunePreparedHtmlExports();
 		const entry = preparedHtmlExports.get(exportId);
 		if (!entry) return null;
-		if (entry.filePath && entry.tempDirPath) return entry;
+		if (entry.filePath && (entry.tempDirPath || entry.persistent)) return entry;
 
 		const tempDirPath = join(tmpdir(), `pi-studio-prepared-html-${Date.now()}-${randomUUID()}`);
 		const filePath = join(tempDirPath, sanitizeHtmlFilename(entry.filename));
@@ -12570,7 +12904,8 @@ export default function (pi: ExtensionAPI) {
 
 		try {
 			const { pdf, warning } = await renderStudioPdfWithPandoc(markdown, isLatex, resourcePath, editorPdfLanguage, sourcePath || undefined);
-			const exportId = storePreparedPdfExport(pdf, filename, warning);
+			const writeResult = writeStudioPreviewExportFile(buildStudioPreviewExportPath(sourcePath || undefined, userResourceDir || undefined, studioCwd, filename), pdf);
+			const exportId = storePreparedPdfExport(pdf, filename, warning, writeResult.filePath ?? undefined);
 			const token = serverState?.token ?? "";
 			let openedExternal = false;
 			let openError: string | null = null;
@@ -12587,6 +12922,8 @@ export default function (pi: ExtensionAPI) {
 			respondJson(res, 200, {
 				ok: true,
 				filename,
+				path: writeResult.filePath,
+				writeError: writeResult.error,
 				warning: warning ?? null,
 				openedExternal,
 				openError,
@@ -12681,7 +13018,8 @@ export default function (pi: ExtensionAPI) {
 					themeVars,
 				},
 			);
-			const exportId = storePreparedHtmlExport(html, filename, warning);
+			const writeResult = writeStudioPreviewExportFile(buildStudioPreviewExportPath(sourcePath || undefined, userResourceDir || undefined, studioCwd, filename), html);
+			const exportId = storePreparedHtmlExport(html, filename, warning, writeResult.filePath ?? undefined);
 			const token = serverState?.token ?? "";
 			let openedExternal = false;
 			let openError: string | null = null;
@@ -12698,6 +13036,8 @@ export default function (pi: ExtensionAPI) {
 			respondJson(res, 200, {
 				ok: true,
 				filename,
+				path: writeResult.filePath,
+				writeError: writeResult.error,
 				warning: warning ?? null,
 				openedExternal,
 				openError,
@@ -12990,17 +13330,19 @@ export default function (pi: ExtensionAPI) {
 				return;
 			}
 
-			try {
-				const resource = resolveStudioLocalPreviewResourcePath(
-					requestUrl.searchParams.get("path") ?? "",
-					requestUrl.searchParams.get("sourcePath") ?? undefined,
-					requestUrl.searchParams.get("resourceDir") ?? undefined,
-					studioCwd,
-				);
-				respondLocalPreviewLinkJson(req, res, requestUrl, resource, serverState);
-			} catch (error) {
-				respondJson(res, 404, { ok: false, error: `Local resource unavailable: ${error instanceof Error ? error.message : String(error)}` });
-			}
+			void (async () => {
+				try {
+					const resource = resolveStudioLocalPreviewResourcePath(
+						requestUrl.searchParams.get("path") ?? "",
+						requestUrl.searchParams.get("sourcePath") ?? undefined,
+						requestUrl.searchParams.get("resourceDir") ?? undefined,
+						studioCwd,
+					);
+					await respondLocalPreviewLinkJson(req, res, requestUrl, resource, serverState);
+				} catch (error) {
+					respondJson(res, 404, { ok: false, error: `Local resource unavailable: ${error instanceof Error ? error.message : String(error)}` });
+				}
+			})();
 			return;
 		}