chore(cli): save screenshots to tmp dir when no path provided (#247)
* fix(cli): save screenshots to tmp dir when no path provided Instead of outputting base64 to stdout (which is not useful for most CLI use cases), screenshots without a path now save to ~/.agent-browser/tmp/screenshots/ with a generated filename and return the path. This makes the behavior more ergonomic for AI agents and CLI users alike. * cleanup * cleanup * just revert the cargo.lock version for now * refactor: extract getAppDir() from getSocketDir() * docs: improve screenshot help text consistency
This commit is contained in:
+15
-6
@@ -1,5 +1,8 @@
|
||||
import type { Page, Frame } from 'playwright-core';
|
||||
import { mkdirSync } from 'node:fs';
|
||||
import path from 'node:path';
|
||||
import type { BrowserManager, ScreencastFrame } from './browser.js';
|
||||
import { getAppDir } from './daemon.js';
|
||||
import type {
|
||||
Command,
|
||||
Response,
|
||||
@@ -561,13 +564,19 @@ async function handleScreenshot(
|
||||
}
|
||||
|
||||
try {
|
||||
if (command.path) {
|
||||
await target.screenshot({ ...options, path: command.path });
|
||||
return successResponse(command.id, { path: command.path });
|
||||
} else {
|
||||
const buffer = await target.screenshot(options);
|
||||
return successResponse(command.id, { base64: buffer.toString('base64') });
|
||||
let savePath = command.path;
|
||||
if (!savePath) {
|
||||
const ext = command.format === 'jpeg' ? 'jpg' : 'png';
|
||||
const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
|
||||
const random = Math.random().toString(36).substring(2, 8);
|
||||
const filename = `screenshot-${timestamp}-${random}.${ext}`;
|
||||
const screenshotDir = path.join(getAppDir(), 'tmp', 'screenshots');
|
||||
mkdirSync(screenshotDir, { recursive: true });
|
||||
savePath = path.join(screenshotDir, filename);
|
||||
}
|
||||
|
||||
await target.screenshot({ ...options, path: savePath });
|
||||
return successResponse(command.id, { path: savePath });
|
||||
} catch (error) {
|
||||
if (command.selector) {
|
||||
throw toAIFriendlyError(error, command.selector);
|
||||
|
||||
+12
-9
@@ -51,27 +51,30 @@ function getPortForSession(session: string): number {
|
||||
* Get the base directory for socket/pid files.
|
||||
* Priority: AGENT_BROWSER_SOCKET_DIR > XDG_RUNTIME_DIR > ~/.agent-browser > tmpdir
|
||||
*/
|
||||
export function getSocketDir(): string {
|
||||
// 1. Explicit override
|
||||
if (process.env.AGENT_BROWSER_SOCKET_DIR) {
|
||||
return process.env.AGENT_BROWSER_SOCKET_DIR;
|
||||
}
|
||||
|
||||
// 2. XDG_RUNTIME_DIR (Linux standard)
|
||||
export function getAppDir(): string {
|
||||
// 1. XDG_RUNTIME_DIR (Linux standard)
|
||||
if (process.env.XDG_RUNTIME_DIR) {
|
||||
return path.join(process.env.XDG_RUNTIME_DIR, 'agent-browser');
|
||||
}
|
||||
|
||||
// 3. Home directory fallback (like Docker Desktop's ~/.docker/run/)
|
||||
// 2. Home directory fallback (like Docker Desktop's ~/.docker/run/)
|
||||
const homeDir = os.homedir();
|
||||
if (homeDir) {
|
||||
return path.join(homeDir, '.agent-browser');
|
||||
}
|
||||
|
||||
// 4. Last resort: temp dir
|
||||
// 3. Last resort: temp dir
|
||||
return path.join(os.tmpdir(), 'agent-browser');
|
||||
}
|
||||
|
||||
export function getSocketDir(): string {
|
||||
// Allow explicit override for socket directory
|
||||
if (process.env.AGENT_BROWSER_SOCKET_DIR) {
|
||||
return process.env.AGENT_BROWSER_SOCKET_DIR;
|
||||
}
|
||||
return getAppDir();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the socket path for the current session (Unix) or port (Windows)
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user