149 lines
4.3 KiB
TypeScript
149 lines
4.3 KiB
TypeScript
import { resolveWorkspaceRoot } from "@/lib/workspace";
|
|
|
|
export const dynamic = "force-dynamic";
|
|
export const runtime = "nodejs";
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Singleton watcher: one chokidar instance shared across all SSE connections.
|
|
// Uses polling (no native fs.watch FDs) so it doesn't compete with Next.js's
|
|
// own dev watcher for the macOS per-process file-descriptor limit.
|
|
// ---------------------------------------------------------------------------
|
|
|
|
type Listener = (type: string, relPath: string) => void;
|
|
|
|
let listeners = new Set<Listener>();
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
let sharedWatcher: any = null;
|
|
let sharedRoot: string | null = null;
|
|
let _watcherReady = false;
|
|
|
|
async function ensureWatcher(root: string) {
|
|
if (sharedWatcher && sharedRoot === root) {return;}
|
|
|
|
// Root changed (e.g. profile switch) -- close the old watcher first.
|
|
if (sharedWatcher) {
|
|
await sharedWatcher.close();
|
|
sharedWatcher = null;
|
|
sharedRoot = null;
|
|
_watcherReady = false;
|
|
}
|
|
|
|
try {
|
|
const chokidar = await import("chokidar");
|
|
sharedRoot = root;
|
|
sharedWatcher = chokidar.watch(root, {
|
|
ignoreInitial: true,
|
|
usePolling: true,
|
|
interval: 1500,
|
|
binaryInterval: 3000,
|
|
ignored: [
|
|
/(^|[\\/])node_modules([\\/]|$)/,
|
|
/(^|[\\/])\.git([\\/]|$)/,
|
|
/(^|[\\/])\.next([\\/]|$)/,
|
|
/(^|[\\/])dist([\\/]|$)/,
|
|
/\.duckdb\.wal$/,
|
|
/\.duckdb\.tmp$/,
|
|
],
|
|
depth: 5,
|
|
});
|
|
|
|
sharedWatcher.on("all", (eventType: string, filePath: string) => {
|
|
const rel = filePath.startsWith(root)
|
|
? filePath.slice(root.length + 1)
|
|
: filePath;
|
|
for (const fn of listeners) {fn(eventType, rel);}
|
|
});
|
|
|
|
sharedWatcher.once("ready", () => {_watcherReady = true;});
|
|
|
|
sharedWatcher.on("error", () => {
|
|
// Swallow; polling mode shouldn't hit EMFILE but be safe.
|
|
});
|
|
} catch {
|
|
// chokidar unavailable -- listeners simply won't fire.
|
|
}
|
|
}
|
|
|
|
function stopWatcherIfIdle() {
|
|
if (listeners.size > 0 || !sharedWatcher) {return;}
|
|
sharedWatcher.close();
|
|
sharedWatcher = null;
|
|
sharedRoot = null;
|
|
_watcherReady = false;
|
|
}
|
|
|
|
/**
|
|
* GET /api/workspace/watch
|
|
*
|
|
* Server-Sent Events endpoint that watches the workspace for file changes.
|
|
* Falls back gracefully if chokidar is unavailable.
|
|
*/
|
|
export async function GET(req: Request) {
|
|
const root = resolveWorkspaceRoot();
|
|
if (!root) {
|
|
return new Response("Workspace not found", { status: 404 });
|
|
}
|
|
|
|
const encoder = new TextEncoder();
|
|
let closed = false;
|
|
let heartbeat: ReturnType<typeof setInterval> | null = null;
|
|
let debounceTimer: ReturnType<typeof setTimeout> | null = null;
|
|
|
|
const stream = new ReadableStream({
|
|
async start(controller) {
|
|
controller.enqueue(encoder.encode("event: connected\ndata: {}\n\n"));
|
|
|
|
const listener: Listener = (_type, _rel) => {
|
|
if (closed) {return;}
|
|
if (debounceTimer) {clearTimeout(debounceTimer);}
|
|
debounceTimer = setTimeout(() => {
|
|
if (closed) {return;}
|
|
try {
|
|
const data = JSON.stringify({ type: _type, path: _rel });
|
|
controller.enqueue(encoder.encode(`event: change\ndata: ${data}\n\n`));
|
|
} catch { /* stream closed */ }
|
|
}, 300);
|
|
};
|
|
|
|
heartbeat = setInterval(() => {
|
|
if (closed) {return;}
|
|
try {
|
|
controller.enqueue(encoder.encode(": heartbeat\n\n"));
|
|
} catch { /* closed */ }
|
|
}, 30_000);
|
|
|
|
function teardown() {
|
|
if (closed) {return;}
|
|
closed = true;
|
|
listeners.delete(listener);
|
|
if (heartbeat) {clearInterval(heartbeat);}
|
|
if (debounceTimer) {clearTimeout(debounceTimer);}
|
|
stopWatcherIfIdle();
|
|
}
|
|
|
|
req.signal.addEventListener("abort", teardown, { once: true });
|
|
|
|
listeners.add(listener);
|
|
await ensureWatcher(root);
|
|
|
|
if (!sharedWatcher) {
|
|
controller.enqueue(
|
|
encoder.encode("event: error\ndata: {\"error\":\"File watching unavailable\"}\n\n"),
|
|
);
|
|
}
|
|
},
|
|
cancel() {
|
|
closed = true;
|
|
},
|
|
});
|
|
|
|
return new Response(stream, {
|
|
headers: {
|
|
"Content-Type": "text/event-stream",
|
|
"Cache-Control": "no-cache, no-transform",
|
|
Connection: "keep-alive",
|
|
"X-Accel-Buffering": "no",
|
|
},
|
|
});
|
|
}
|