pi-studio 0.9.14 → 0.9.16

package/index.ts

@@ -311,6 +311,17 @@ interface SendRunRequestMessage {
 	text: string;
 }
 
+interface CompletionSuggestionRequestMessage {
+	type: "completion_suggestion_request";
+	requestId: string;
+	text: string;
+	selectionStart: number;
+	selectionEnd: number;
+	language?: string;
+	label?: string;
+	path?: string;
+}
+
 interface QuizGenerateRequestMessage {
 	type: "quiz_generate_request";
 	requestId: string;
@@ -451,6 +462,7 @@ type IncomingStudioMessage =
 	| CritiqueRequestMessage
 	| AnnotationRequestMessage
 	| SendRunRequestMessage
+	| CompletionSuggestionRequestMessage
 	| QuizGenerateRequestMessage
 	| QuizAnswerRequestMessage
 	| QuizDiscussRequestMessage
@@ -472,6 +484,9 @@ 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_PREFIX_CHARS = 12_000;
+const STUDIO_COMPLETION_SUFFIX_CHARS = 6_000;
 const PDF_EXPORT_MAX_CHARS = 400_000;
 const HTML_EXPORT_MAX_CHARS = 400_000;
 const HTML_PREVIEW_MATH_RENDER_MAX_ITEMS = 250;
@@ -2442,6 +2457,16 @@ 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_TEXT_FILENAMES = new Set([
+	".dockerignore", ".editorconfig", ".env", ".env.example", ".eslintignore", ".gitattributes",
+	".gitignore", ".gitmodules", ".npmignore", ".prettierignore", "dockerfile", "gemfile",
+	"justfile", "license", "makefile", "rakefile", "readme",
+]);
+const STUDIO_FILE_BROWSER_MAX_ENTRIES = 500;
+const STUDIO_FILE_BROWSER_IGNORED_DIRS = new Set([
+	".git", "node_modules", ".next", ".cache", "dist", "build", "coverage", "target",
+	"__pycache__", ".venv", "venv", ".mypy_cache", ".pytest_cache", ".ruff_cache",
+]);
 
 type StudioLocalPreviewResourceKind = "pdf" | "text" | "image" | "other";
 
@@ -2454,6 +2479,17 @@ interface StudioLocalPreviewResource {
 	resourceDir: string;
 }
 
+interface StudioFileBrowserEntry {
+	name: string;
+	path: string;
+	type: "directory" | "file";
+	extension: string;
+	kind: StudioLocalPreviewResourceKind | "directory";
+	size: number;
+	mtimeMs: number;
+	hidden: boolean;
+}
+
 function resolveStudioPdfResourcePath(pdfPath: string | undefined, sourcePath: string | undefined, resourceDir: string | undefined, fallbackCwd: string): string {
 	const rawPath = typeof pdfPath === "string" ? pdfPath.trim() : "";
 	if (!rawPath) throw new Error("Missing PDF path.");
@@ -2522,10 +2558,11 @@ function parseStudioLocalPreviewResourcePage(resourcePath: string): number | nul
 	return null;
 }
 
-function getStudioLocalPreviewResourceKind(extension: string): StudioLocalPreviewResourceKind {
+function getStudioLocalPreviewResourceKind(extension: string, filePathOrName?: string): StudioLocalPreviewResourceKind {
 	const ext = extension.toLowerCase();
+	const name = basename(String(filePathOrName || "")).toLowerCase();
 	if (ext === ".pdf") return "pdf";
-	if (STUDIO_LOCAL_LINK_TEXT_EXTENSIONS.has(ext)) return "text";
+	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";
 	return "other";
 }
@@ -2563,12 +2600,94 @@ function resolveStudioLocalPreviewResourcePath(
 		filePath: candidateReal,
 		label: rel && rel !== "" ? rel : basename(candidateReal),
 		extension,
-		kind: getStudioLocalPreviewResourceKind(extension),
+		kind: getStudioLocalPreviewResourceKind(extension, candidateReal),
 		page: parseStudioLocalPreviewResourcePage(rawPath),
 		resourceDir: boundaryReal,
 	};
 }
 
+function resolveStudioFileBrowserDirectory(
+	dirPath: string | undefined,
+	sourcePath: string | undefined,
+	resourceDir: string | undefined,
+	fallbackCwd: string,
+): { rootDir: string; currentDir: string; relativeDir: string; parentDir: string | null } {
+	const context = resolveStudioPreviewResourceContext(sourcePath, resourceDir, fallbackCwd);
+	const rootReal = realpathSync(context.boundaryDir);
+	const rawDir = typeof dirPath === "string" ? dirPath.trim() : "";
+	const baseDir = context.baseDir;
+	const requested = rawDir
+		? (isAbsolute(recoverLikelyDroppedLeadingSlashPath(expandHome(rawDir)))
+			? recoverLikelyDroppedLeadingSlashPath(expandHome(rawDir))
+			: resolve(baseDir, recoverLikelyDroppedLeadingSlashPath(expandHome(rawDir))))
+		: baseDir;
+	const currentReal = realpathSync(requested);
+	const currentStat = statSync(currentReal);
+	if (!currentStat.isDirectory()) throw new Error("File browser path does not refer to a directory.");
+	if (!isPathInsideOrEqualDirectory(currentReal, rootReal)) {
+		throw new Error("File browser path must stay within the current Studio resource directory.");
+	}
+	const parent = dirname(currentReal);
+	const parentDir = parent !== currentReal && isPathInsideOrEqualDirectory(parent, rootReal) ? parent : null;
+	const relativeDir = relative(rootReal, currentReal) || ".";
+	return { rootDir: rootReal, currentDir: currentReal, relativeDir, parentDir };
+}
+
+function listStudioFileBrowserDirectory(
+	dirPath: string | undefined,
+	sourcePath: string | undefined,
+	resourceDir: string | undefined,
+	fallbackCwd: string,
+): { rootDir: string; currentDir: string; relativeDir: string; parentDir: string | null; entries: StudioFileBrowserEntry[]; omitted: number; omittedIgnored: number } {
+	const context = resolveStudioFileBrowserDirectory(dirPath, sourcePath, resourceDir, fallbackCwd);
+	const entries: StudioFileBrowserEntry[] = [];
+	let omitted = 0;
+	let omittedIgnored = 0;
+	const dirents = readdirSync(context.currentDir, { withFileTypes: true });
+	for (const dirent of dirents) {
+		const name = dirent.name;
+		if (STUDIO_FILE_BROWSER_IGNORED_DIRS.has(name)) {
+			omittedIgnored += 1;
+			continue;
+		}
+		const candidate = join(context.currentDir, name);
+		try {
+			const real = realpathSync(candidate);
+			if (!isPathInsideOrEqualDirectory(real, context.rootDir)) {
+				omitted += 1;
+				continue;
+			}
+			const stat = statSync(real);
+			if (!stat.isDirectory() && !stat.isFile()) {
+				omitted += 1;
+				continue;
+			}
+			const type = stat.isDirectory() ? "directory" : "file";
+			const extension = type === "file" ? extname(real).toLowerCase() : "";
+			entries.push({
+				name,
+				path: real,
+				type,
+				extension,
+				kind: type === "directory" ? "directory" : getStudioLocalPreviewResourceKind(extension, real),
+				size: stat.size,
+				mtimeMs: stat.mtimeMs,
+				hidden: name.startsWith("."),
+			});
+		} catch {
+			omitted += 1;
+		}
+	}
+	entries.sort((a, b) => {
+		if (a.type !== b.type) return a.type === "directory" ? -1 : 1;
+		if (a.hidden !== b.hidden) return a.hidden ? 1 : -1;
+		return a.name.localeCompare(b.name, undefined, { sensitivity: "base", numeric: true });
+	});
+	const limitedEntries = entries.slice(0, STUDIO_FILE_BROWSER_MAX_ENTRIES);
+	omitted += Math.max(0, entries.length - limitedEntries.length);
+	return { ...context, entries: limitedEntries, omitted, omittedIgnored };
+}
+
 function resolveStudioHtmlPreviewResourcePath(
 	resourcePath: string | undefined,
 	sourcePath: string | undefined,
@@ -7014,7 +7133,7 @@ async function resolveStudioModelRequestAuth(ctx: StudioModelRequestContext, mod
 	if (typeof registry.getApiKey === "function") {
 		return { apiKey: await registry.getApiKey(model) };
 	}
-	throw new Error("Current pi model registry does not expose model credentials for Studio quiz.");
+	throw new Error("Current pi model registry does not expose model credentials for Studio model requests.");
 }
 
 function getStudioQuizReasoning(model: NonNullable<ExtensionContext["model"]>, thinking: StudioQuizThinking | undefined): ThinkingLevel | undefined {
@@ -7023,33 +7142,47 @@ function getStudioQuizReasoning(model: NonNullable<ExtensionContext["model"]>, t
 	return normalized === "off" ? undefined : normalized;
 }
 
-async function runStudioQuizModelText(ctx: StudioModelRequestContext, prompt: string, options?: { maxTokens?: number; signal?: AbortSignal; thinking?: StudioQuizThinking }): Promise<string> {
+async function runStudioModelText(
+	ctx: StudioModelRequestContext,
+	prompt: string,
+	options?: { systemPrompt?: string; maxTokens?: number; signal?: AbortSignal; reasoning?: ThinkingLevel; timeoutMs?: number; trim?: boolean },
+): Promise<string> {
 	if (!ctx.model) throw new Error("No active model selected.");
 	const auth = await resolveStudioModelRequestAuth(ctx, ctx.model);
 	const response = await completeSimple(
 		ctx.model,
 		{
-			systemPrompt: "You are an active tutor inside pi Studio. Ask and mark concise, probing quiz questions. Return exactly the requested format.",
+			systemPrompt: options?.systemPrompt ?? "You are a concise assistant inside pi Studio. Return exactly the requested format.",
 			messages: [{ role: "user", content: [{ type: "text", text: prompt }], timestamp: Date.now() }],
 		},
 		{
 			apiKey: auth.apiKey,
 			headers: auth.headers,
-			reasoning: getStudioQuizReasoning(ctx.model, options?.thinking),
+			reasoning: options?.reasoning,
 			maxTokens: options?.maxTokens ?? 2500,
 			signal: options?.signal,
-			timeoutMs: 120_000,
+			timeoutMs: options?.timeoutMs ?? 120_000,
 		},
 	);
-	const text = response.content
+	const rawText = response.content
 		.filter((part): part is { type: "text"; text: string } => part.type === "text")
 		.map((part) => part.text)
-		.join("\n")
-		.trim();
-	if (!text) throw new Error("Model returned no text response.");
+		.join("\n");
+	const text = options?.trim === false ? rawText : rawText.trim();
+	if (!text.trim()) throw new Error("Model returned no text response.");
 	return text;
 }
 
+async function runStudioQuizModelText(ctx: StudioModelRequestContext, prompt: string, options?: { maxTokens?: number; signal?: AbortSignal; thinking?: StudioQuizThinking }): Promise<string> {
+	return runStudioModelText(ctx, prompt, {
+		systemPrompt: "You are an active tutor inside pi Studio. Ask and mark concise, probing quiz questions. Return exactly the requested format.",
+		reasoning: ctx.model ? getStudioQuizReasoning(ctx.model, options?.thinking) : undefined,
+		maxTokens: options?.maxTokens ?? 2500,
+		signal: options?.signal,
+		timeoutMs: 120_000,
+	});
+}
+
 async function runStudioQuizModelJson(
 	ctx: StudioModelRequestContext,
 	prompt: string,
@@ -7077,6 +7210,72 @@ async function runStudioQuizModelJson(
 	throw lastError ?? new Error("Model did not return valid JSON.");
 }
 
+function buildStudioCompletionSuggestionPrompt(options: {
+	text: string;
+	selectionStart: number;
+	selectionEnd: number;
+	language?: string;
+	label?: string;
+	path?: string;
+}): string {
+	const text = String(options.text || "");
+	const start = Math.max(0, Math.min(Math.floor(options.selectionStart || 0), text.length));
+	const end = Math.max(start, Math.min(Math.floor(options.selectionEnd || start), text.length));
+	const prefix = text.slice(Math.max(0, start - STUDIO_COMPLETION_PREFIX_CHARS), start);
+	const selected = text.slice(start, end);
+	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();
+	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.",
+		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}`,
+		"",
+		"<prefix>",
+		prefix,
+		"</prefix>",
+		selected ? ["", "<selected>", selected, "</selected>"].join("\n") : "",
+		"",
+		"<suffix>",
+		suffix,
+		"</suffix>",
+	].filter((part) => part !== "").join("\n");
+}
+
+function cleanStudioCompletionSuggestion(text: string): string {
+	let value = String(text || "").replace(/\r\n/g, "\n");
+	value = value.replace(/^\s*(?:Here(?:'s| is) (?:the )?(?:completion|suggestion):|Completion:|Suggestion:)\s*/i, "");
+	return value;
+}
+
+async function runStudioCompletionSuggestion(ctx: StudioModelRequestContext, options: {
+	text: string;
+	selectionStart: number;
+	selectionEnd: number;
+	language?: string;
+	label?: string;
+	path?: string;
+}): Promise<string> {
+	const prompt = buildStudioCompletionSuggestionPrompt(options);
+	// Intentionally omit `reasoning`: pi-ai treats absent reasoning as off/disabled
+	// where supported. Passing "minimal" would still enable a reasoning path and slow completions.
+	const suggestion = cleanStudioCompletionSuggestion(await runStudioModelText(ctx, prompt, {
+		systemPrompt: "You are an inline autocomplete engine inside pi Studio. Return only text to insert at the cursor. Never explain. Never include Markdown fences unless literal fences are the intended insertion.",
+		maxTokens: 650,
+		timeoutMs: 60_000,
+		trim: false,
+	}));
+	if (!suggestion.trim()) throw new Error("Model returned an empty completion suggestion.");
+	return suggestion;
+}
+
 function inferStudioResponseKind(markdown: string): StudioRequestKind {
 	const lower = markdown.toLowerCase();
 	if (lower.includes("## critiques") && lower.includes("## document")) return "critique";
@@ -7486,6 +7685,24 @@ function parseIncomingMessage(data: RawData): IncomingStudioMessage | null {
 		};
 	}
 
+	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));
+		return {
+			type: "completion_suggestion_request",
+			requestId: msg.requestId,
+			text: msg.text,
+			selectionStart,
+			selectionEnd,
+			language: typeof msg.language === "string" ? msg.language : undefined,
+			label: typeof msg.label === "string" ? msg.label : undefined,
+			path: typeof msg.path === "string" ? msg.path : undefined,
+		};
+	}
+
 	if (msg.type === "quiz_generate_request" && typeof msg.requestId === "string" && typeof msg.sourceText === "string") {
 		const rawCount = typeof msg.questionCount === "number" && Number.isFinite(msg.questionCount) ? msg.questionCount : 5;
 		return {
@@ -8958,7 +9175,7 @@ function resolveRequestedStudioDocumentFromUrl(
 				label: requestedLabel || file.label,
 				source: "file",
 				path: file.resolvedPath,
-				resourceDir: requestedResourceDir || dirname(file.resolvedPath),
+				resourceDir: requestedResourceDir || fallback?.resourceDir || studioCwd,
 			};
 		}
 	}
@@ -9314,7 +9531,7 @@ ${cssVarsBlock}
       <button id="saveAsBtn" type="button" title="Save editor content to a new file path. Cmd/Ctrl+S falls back here when no direct save path is available.">Save editor as…</button>
       <button id="saveOverBtn" type="button" title="Overwrite current file with editor content. Shortcut: Cmd/Ctrl+S.">Save editor</button>
       <button id="refreshFromDiskBtn" type="button" title="Reload the current file-backed document from disk.">Refresh from disk</button>
-      <button id="clearWorkspaceBtn" type="button" title="Clear the current editor draft in this browser tab. Saved files and responses are not changed.">Clear editor</button>
+      <button id="clearWorkspaceBtn" type="button" title="Clear editor text and reset this tab to a fresh blank draft. Saved files and responses are not changed.">Reset editor</button>
       <label class="file-label" title="Load a local file into editor text.">Load file content<input id="fileInput" type="file" accept=".md,.markdown,.mdx,.qmd,.js,.mjs,.cjs,.jsx,.ts,.mts,.cts,.tsx,.py,.pyw,.sh,.bash,.zsh,.json,.jsonc,.json5,.rs,.c,.h,.cpp,.cxx,.cc,.hpp,.hxx,.jl,.f90,.f95,.f03,.f,.for,.r,.R,.m,.tex,.latex,.diff,.patch,.java,.go,.rb,.swift,.html,.htm,.css,.xml,.yaml,.yml,.toml,.lua,.txt,.rst,.adoc" /></label>
       <button id="loadGitDiffBtn" type="button" title="Load the current git diff from the Studio context into the editor.">Load git diff</button>
       <button id="getEditorBtn" type="button" title="Load the current terminal editor draft into Studio.">Load from pi editor</button>
@@ -9363,8 +9580,9 @@ ${cssVarsBlock}
               </select>
             </div>
             <div class="source-actions-row">
-              <button id="copyDraftBtn" type="button" title="Copy the current editor text to the clipboard.">Copy text</button>
-              <button id="openCompanionBtn" type="button" title="Open a detached copy of the current editor text in a new editor-only Studio tab.">Open new editor</button>
+              <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="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>
             <div class="source-actions-row">
@@ -9446,6 +9664,16 @@ ${cssVarsBlock}
                 <button id="editorSelectionJumpBtn" type="button" class="editor-selection-action-btn" hidden title="Jump to the current editor selection in the preview.">Jump</button>
               </div>
             </div>
+            <div id="completionSuggestionPanel" class="completion-suggestion-panel" hidden>
+              <div class="completion-suggestion-header">
+                <strong>Suggested completion</strong>
+                <button id="completionSuggestionDismissBtn" type="button" title="Dismiss this suggestion">Dismiss</button>
+              </div>
+              <pre id="completionSuggestionText" class="completion-suggestion-text"></pre>
+              <div class="completion-suggestion-actions">
+                <button id="completionSuggestionInsertBtn" type="button" title="Insert this suggestion at the cursor or original selection. You can also press Tab while the editor is focused.">Insert suggestion (Tab)</button>
+              </div>
+            </div>
             <div id="sourcePreview" class="panel-scroll rendered-markdown" hidden><pre class="plain-markdown"></pre></div>
           </div>
           <aside id="outlineOverlay" class="outline-dock-wrap" hidden>
@@ -9508,6 +9736,7 @@ ${cssVarsBlock}
             <option value="preview" selected>Response (Preview)</option>
             <option value="editor-preview">Editor (Preview)</option>
             <option value="trace">Working</option>
+            <option value="files">Files</option>
             <option value="repl">REPL</option>
           </select>
         </div>
@@ -9613,7 +9842,9 @@ ${cssVarsBlock}
           <dl>
             <div><dt>Cmd/Ctrl+S</dt><dd>Save editor</dd></div>
             <div><dt>Cmd/Ctrl+Enter</dt><dd>Run editor text, or queue steering during an active run</dd></div>
-            <div><dt>Tab / Shift+Tab</dt><dd>Indent or unindent selected editor text</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>Shift+Tab</dt><dd>Unindent selected editor text</dd></div>
           </dl>
         </section>
         <section class="shortcuts-group">
@@ -11193,6 +11424,53 @@ export default function (pi: ExtensionAPI) {
 			return;
 		}
 
+		if (msg.type === "completion_suggestion_request") {
+			if (!isValidRequestId(msg.requestId)) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "Invalid request ID." });
+				return;
+			}
+			const ctx = latestModelRequestCtx ?? lastCommandCtx;
+			if (!ctx) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "No active pi model context is available for editor suggestions." });
+				return;
+			}
+			if (!msg.text.trim()) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: "Editor is empty." });
+				return;
+			}
+			if (msg.text.length > STUDIO_COMPLETION_MAX_TEXT_CHARS) {
+				sendToClient(client, { type: "completion_suggestion_error", requestId: msg.requestId, message: `Editor text is too large for suggestions (${STUDIO_COMPLETION_MAX_TEXT_CHARS} character limit).` });
+				return;
+			}
+			sendToClient(client, { type: "completion_suggestion_progress", requestId: msg.requestId, message: "Generating suggestion…" });
+			void (async () => {
+				try {
+					const suggestion = await runStudioCompletionSuggestion(ctx, {
+						text: msg.text,
+						selectionStart: msg.selectionStart,
+						selectionEnd: msg.selectionEnd,
+						language: msg.language,
+						label: msg.label,
+						path: msg.path,
+					});
+					sendToClient(client, {
+						type: "completion_suggestion_result",
+						requestId: msg.requestId,
+						suggestion,
+						selectionStart: msg.selectionStart,
+						selectionEnd: msg.selectionEnd,
+					});
+				} catch (error) {
+					sendToClient(client, {
+						type: "completion_suggestion_error",
+						requestId: msg.requestId,
+						message: `Suggestion failed: ${error instanceof Error ? error.message : String(error)}`,
+					});
+				}
+			})();
+			return;
+		}
+
 		if (msg.type === "quiz_generate_request") {
 			if (!isValidRequestId(msg.requestId)) {
 				sendToClient(client, { type: "error", requestId: msg.requestId, message: "Invalid request ID." });
@@ -12677,6 +12955,34 @@ export default function (pi: ExtensionAPI) {
 			return;
 		}
 
+		if (requestUrl.pathname === "/file-browser") {
+			const token = requestUrl.searchParams.get("token") ?? "";
+			if (token !== serverState.token) {
+				respondJson(res, 403, { ok: false, error: "Invalid or expired studio token. Re-run /studio." });
+				return;
+			}
+
+			const method = (req.method ?? "GET").toUpperCase();
+			if (method !== "GET" && method !== "HEAD") {
+				res.setHeader("Allow", "GET, HEAD");
+				respondJson(res, 405, { ok: false, error: "Method not allowed. Use GET." });
+				return;
+			}
+
+			try {
+				const listing = listStudioFileBrowserDirectory(
+					requestUrl.searchParams.get("dir") ?? undefined,
+					requestUrl.searchParams.get("sourcePath") ?? undefined,
+					requestUrl.searchParams.get("resourceDir") ?? undefined,
+					studioCwd,
+				);
+				respondJson(res, 200, { ok: true, ...listing, entries: method === "HEAD" ? [] : listing.entries });
+			} catch (error) {
+				respondJson(res, 404, { ok: false, error: `File browser unavailable: ${error instanceof Error ? error.message : String(error)}` });
+			}
+			return;
+		}
+
 		if (requestUrl.pathname === "/local-preview-link") {
 			const token = requestUrl.searchParams.get("token") ?? "";
 			if (token !== serverState.token) {
@@ -13377,7 +13683,7 @@ export default function (pi: ExtensionAPI) {
 			label: file.label,
 			source: "file",
 			path: file.resolvedPath,
-			resourceDir: dirname(file.resolvedPath),
+			resourceDir: ctx.cwd,
 		};
 	};
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-studio",
-  "version": "0.9.14",
+  "version": "0.9.16",
   "description": "Two-pane browser workspace for pi with prompt/response editing, annotations, critiques, active quiz, prompt/response history, live previews, and tmux-backed REPL/literate REPL workflows",
   "type": "module",
   "license": "MIT",