@stacknext/cli 0.6.1 → 0.7.0

package/dist/collector/server.d.ts

@@ -1,4 +1,4 @@
-import { TelemetryStore, type HealthResponse, type NextOtelVerboseStatus, type StackNextConfig } from '@stacknext/core';
+import { TelemetryStore, type HealthResponse, type NextOtelVerboseStatus, type StackNextConfig, type SnapshotSummary } from '@stacknext/core';
 export interface StartCollectorOptions {
     port: number;
     token: string;
@@ -21,6 +21,8 @@ export interface CollectorHandle {
     token: string;
     store: TelemetryStore;
     health: () => HealthResponse;
+    /** Saves the current sample before shutdown when snapshots.autoOnExit is on. */
+    saveSnapshotOnExit: () => Promise<SnapshotSummary | null>;
     close: () => Promise<void>;
 }
 export declare function resolveDashboardDir(): Promise<string>;

package/dist/collector/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/collector/server.ts"],"names":[],"mappings":"AAOA,OAAO,EAKL,cAAc,EAkBd,KAAK,cAAc,EACnB,KAAK,qBAAqB,EAC1B,KAAK,eAAe,EAYrB,MAAM,iBAAiB,CAAA;AAoCxB,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,MAAM,CAAA;IACnB,MAAM,EAAE,eAAe,CAAA;IACvB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAA;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAA;IAC/B,QAAQ,CAAC,EAAE,OAAO,CAAA;IAClB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,gFAAgF;IAChF,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,oFAAoF;IACpF,eAAe,CAAC,EAAE,qBAAqB,CAAA;CACxC;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,KAAK,EAAE,MAAM,CAAA;IACb,KAAK,EAAE,cAAc,CAAA;IACrB,MAAM,EAAE,MAAM,cAAc,CAAA;IAC5B,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAA;CAC3B;AAoSD,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,MAAM,CAAC,CA2B3D;AAoBD,wBAAsB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,eAAe,CAAC,CA+gC7F"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/collector/server.ts"],"names":[],"mappings":"AAOA,OAAO,EAKL,cAAc,EAkBd,KAAK,cAAc,EACnB,KAAK,qBAAqB,EAC1B,KAAK,eAAe,EAepB,KAAK,eAAe,EACrB,MAAM,iBAAiB,CAAA;AA8DxB,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,MAAM,CAAA;IACnB,MAAM,EAAE,eAAe,CAAA;IACvB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAA;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAA;IAC/B,QAAQ,CAAC,EAAE,OAAO,CAAA;IAClB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,gFAAgF;IAChF,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,oFAAoF;IACpF,eAAe,CAAC,EAAE,qBAAqB,CAAA;CACxC;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,KAAK,EAAE,MAAM,CAAA;IACb,KAAK,EAAE,cAAc,CAAA;IACrB,MAAM,EAAE,MAAM,cAAc,CAAA;IAC5B,gFAAgF;IAChF,kBAAkB,EAAE,MAAM,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAAA;IACzD,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAA;CAC3B;AAySD,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,MAAM,CAAC,CA2B3D;AAoBD,wBAAsB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,eAAe,CAAC,CAm6C7F"}

package/dist/collector/server.js

@@ -6,6 +6,7 @@ import { randomBytes, createHash, timingSafeEqual } from 'node:crypto';
 import { fileURLToPath } from 'node:url';
 import { createRequire } from 'node:module';
 import { DIAGNOSTIC_CATEGORIES, DIAGNOSTIC_RULE_IDS, DIAGNOSTIC_SEVERITIES, DIAGNOSTIC_STATUSES, TelemetryStore, computeCoverage, diagnosticReference, ingestEnvelopeSchema, nextOtelVerboseStatus, } from '@stacknext/core';
+import { SnapshotStorageError, createSnapshotPayload, createSnapshotView, importSnapshot, listSnapshots, loadSnapshotPayload, removeSnapshot, rotateAutomaticSnapshots, saveSnapshot, snapshotExport, snapshotPath, snapshotSummary, isSnapshotId, } from '@stacknext/core/snapshots';
 import { diagnosticOverview as buildDiagnosticOverview, generateDiagnosticReport, getDiagnosticDetail, listDiagnosticFindings, } from '@stacknext/diagnostics';
 import { completeRunRequestSchema, createRunRequestSchema, listAllBaselines, listAllComparisons, listAllRuns, loadBaseline, loadComparison, loadRun, RegressionStorageError, submitSampleRequestSchema, } from '@stacknext/regression';
 import { TestRunRegistry } from './test-runs.js';
@@ -19,6 +20,9 @@ const CONNECTION_RETENTION_MS = 10 * 60 * 1_000;
 const MAX_SSE_CLIENTS = 50;
 const SSE_HEARTBEAT_MS = 15_000;
 const MAX_SESSIONS = 100;
+const AUTO_ERROR_COOLDOWN_MS = 60_000;
+const MAX_SNAPSHOT_BODY_BYTES = 64 * 1024;
+const MAX_SNAPSHOT_CACHE_ENTRIES = 3;
 function securityHeaders() {
     return {
         'x-content-type-options': 'nosniff',
@@ -139,6 +143,9 @@ function filtersFromSearch(searchParams) {
         ...(queryNumber(searchParams.get('minDuration')) !== undefined
             ? { minDuration: queryNumber(searchParams.get('minDuration')) }
             : {}),
+        ...(queryBoolean(searchParams.get('httpOnly')) !== undefined
+            ? { httpOnly: queryBoolean(searchParams.get('httpOnly')) }
+            : {}),
         ...(sort === 'recent' || sort === 'oldest' || sort === 'slowest' ? { sort } : {}),
     };
 }
@@ -254,6 +261,8 @@ function contentTypeFor(file) {
             return 'text/css; charset=utf-8';
         case '.json':
             return 'application/json; charset=utf-8';
+        case '.webmanifest':
+            return 'application/manifest+json';
         case '.svg':
             return 'image/svg+xml';
         case '.png':
@@ -394,6 +403,12 @@ export async function startCollector(options) {
                 errors: summary.errorCount,
                 databaseQueries: summary.databaseQueryCount,
             },
+            sample: {
+                lastReceivedAt: summary.lastReceivedAt,
+                firstEventTime: summary.firstEventTime,
+                lastEventTime: summary.lastEventTime,
+                retentionMs: store.retentionMs,
+            },
             store: storeStats,
             server: {
                 ...serverStats,
@@ -430,6 +445,7 @@ export async function startCollector(options) {
                     projectName: options.projectName,
                     collectorVersion: options.version ?? VERSION,
                     config: options.config.diagnostics,
+                    slowRequestMs: options.config.slowRequestMs,
                     generatedAt: now,
                     connected: connectionList.some((connection) => connection.connected),
                     connectedInstances: connectionList.filter((connection) => connection.connected).length,
@@ -451,6 +467,298 @@ export async function startCollector(options) {
         };
     };
     const getDiagnosticReport = () => getDiagnosticState().report;
+    const snapshotCache = new Map();
+    let autoSaveQueue = Promise.resolve();
+    let lastAutoErrorSnapshotAt = 0;
+    const projectDir = options.projectDir;
+    const snapshotMaxBytes = options.config.snapshots.maxBytes;
+    const snapshotHasData = (payload) => payload.spans.length > 0 ||
+        payload.errors.length > 0 ||
+        payload.databaseQueries.length > 0;
+    const saveCurrentSnapshot = async (source, saveOptions = {}) => {
+        if (!projectDir) {
+            throw new SnapshotStorageError('invalid', 'Snapshot storage needs a project directory. Start StackNext with "stacknext dev" from the app directory.');
+        }
+        const payload = createSnapshotPayload({
+            store,
+            source,
+            projectName: options.projectName,
+            collectorVersion: options.version ?? VERSION,
+            config: options.config,
+            slowRequestMs: options.config.slowRequestMs,
+            ...(saveOptions.label ? { label: saveOptions.label } : {}),
+        });
+        if (!saveOptions.allowEmpty && !snapshotHasData(payload))
+            return null;
+        const saved = await saveSnapshot(projectDir, payload, { maxBytes: snapshotMaxBytes });
+        if (source === 'exit' || source === 'error') {
+            await rotateAutomaticSnapshots(projectDir, options.config.snapshots.autoKeep).catch((error) => {
+                log(`could not rotate automatic snapshots: ${error instanceof Error ? error.message : String(error)}`);
+            });
+        }
+        return saved;
+    };
+    const enqueueAutoErrorSnapshot = () => {
+        if (!options.config.snapshots.autoOnError)
+            return;
+        const now = Date.now();
+        if (now - lastAutoErrorSnapshotAt < AUTO_ERROR_COOLDOWN_MS)
+            return;
+        lastAutoErrorSnapshotAt = now;
+        autoSaveQueue = autoSaveQueue
+            .then(async () => {
+            const saved = await saveCurrentSnapshot('error');
+            if (saved)
+                log(`automatic error snapshot saved (${saved.id})`);
+        })
+            .catch((error) => {
+            log(`automatic error snapshot failed: ${error instanceof Error ? error.message : String(error)}`);
+        });
+    };
+    const snapshotView = async (id) => {
+        if (!projectDir) {
+            throw new SnapshotStorageError('not-found', 'Snapshot storage is not available.');
+        }
+        if (!isSnapshotId(id)) {
+            throw new SnapshotStorageError('invalid', `Invalid snapshot id "${id}".`);
+        }
+        const file = snapshotPath(projectDir, id);
+        let info;
+        try {
+            info = await stat(file);
+        }
+        catch (error) {
+            if (error.code === 'ENOENT') {
+                throw new SnapshotStorageError('not-found', 'Snapshot not found.');
+            }
+            throw error;
+        }
+        const cached = snapshotCache.get(id);
+        if (cached && cached.mtimeMs === info.mtimeMs && cached.ctimeMs === info.ctimeMs) {
+            snapshotCache.delete(id);
+            snapshotCache.set(id, cached);
+            return cached;
+        }
+        const payload = await loadSnapshotPayload(projectDir, id, { maxBytes: snapshotMaxBytes });
+        const entry = {
+            ctimeMs: info.ctimeMs,
+            mtimeMs: info.mtimeMs,
+            bytes: info.size,
+            payload,
+            store: createSnapshotView(payload),
+        };
+        snapshotCache.set(id, entry);
+        while (snapshotCache.size > MAX_SNAPSHOT_CACHE_ENTRIES) {
+            const oldest = snapshotCache.keys().next().value;
+            if (oldest === undefined)
+                break;
+            snapshotCache.delete(oldest);
+        }
+        return entry;
+    };
+    const sendSnapshotError = (response, error) => {
+        if (error instanceof SnapshotStorageError) {
+            if (error.code === 'not-found') {
+                sendError(response, 404, error.message);
+            }
+            else if (error.code === 'too-large') {
+                sendError(response, 413, error.message);
+            }
+            else if (error.code === 'exists') {
+                sendError(response, 409, error.message);
+            }
+            else {
+                sendError(response, 422, error.message);
+            }
+            return;
+        }
+        log(`snapshot operation failed: ${error instanceof Error ? error.message : String(error)}`);
+        sendError(response, 500, 'Snapshot operation failed');
+    };
+    const handleSnapshotWrite = async (request, response, url) => {
+        if (!isReadAuthorized(request)) {
+            serverStats.rejectedRequests += 1;
+            sendError(response, 401, 'Unauthorized');
+            return;
+        }
+        if (request.method === 'POST' && url.pathname === '/api/snapshots/import') {
+            if (!projectDir) {
+                sendSnapshotError(response, new SnapshotStorageError('invalid', 'Snapshot storage needs a project directory. Start StackNext with "stacknext dev" from the app directory.'));
+                return;
+            }
+            const contentType = String(request.headers['content-type'] ?? '');
+            if (!contentType.toLowerCase().includes('application/json')) {
+                sendError(response, 415, 'Content-Type must be application/json');
+                return;
+            }
+            const body = await readBody(request, snapshotMaxBytes);
+            if (!body.ok) {
+                sendError(response, 413, `Snapshot import exceeds ${snapshotMaxBytes} bytes.`);
+                return;
+            }
+            try {
+                const snapshot = await importSnapshot(projectDir, body.body, {
+                    maxBytes: snapshotMaxBytes,
+                });
+                sendJson(response, 201, { snapshot });
+            }
+            catch (error) {
+                sendSnapshotError(response, error);
+            }
+            return;
+        }
+        if (request.method === 'DELETE') {
+            const id = decodeURIComponent(url.pathname.replace(/\/$/, '').slice('/api/snapshots/'.length));
+            if (!isSnapshotId(id)) {
+                sendError(response, 400, 'Invalid snapshot id');
+                return;
+            }
+            if (!projectDir) {
+                sendError(response, 404, 'Snapshot not found');
+                return;
+            }
+            try {
+                const removed = await removeSnapshot(projectDir, id);
+                snapshotCache.delete(id);
+                if (!removed) {
+                    sendError(response, 404, 'Snapshot not found');
+                    return;
+                }
+                sendJson(response, 200, { deleted: true, id });
+            }
+            catch (error) {
+                sendSnapshotError(response, error);
+            }
+            return;
+        }
+        const contentType = String(request.headers['content-type'] ?? '');
+        if (!contentType.toLowerCase().includes('application/json')) {
+            sendError(response, 415, 'Content-Type must be application/json');
+            return;
+        }
+        const body = await readBody(request, MAX_SNAPSHOT_BODY_BYTES);
+        if (!body.ok) {
+            sendError(response, 413, `Snapshot request exceeds ${MAX_SNAPSHOT_BODY_BYTES} bytes`);
+            return;
+        }
+        let label;
+        if (body.body.trim() !== '') {
+            let parsed;
+            try {
+                parsed = JSON.parse(body.body);
+            }
+            catch {
+                sendError(response, 400, 'Malformed JSON payload');
+                return;
+            }
+            if (parsed && typeof parsed === 'object' && 'label' in parsed) {
+                const value = parsed.label;
+                if (value !== undefined && value !== null) {
+                    if (typeof value !== 'string' || value.trim().length === 0 || value.length > 200) {
+                        sendError(response, 422, 'Snapshot label must be 1-200 characters');
+                        return;
+                    }
+                    label = value.trim();
+                }
+            }
+        }
+        try {
+            const saved = await saveCurrentSnapshot('manual', {
+                ...(label ? { label } : {}),
+            });
+            if (!saved) {
+                sendError(response, 409, 'No telemetry in the current sample to save');
+                return;
+            }
+            sendJson(response, 201, { snapshot: saved });
+        }
+        catch (error) {
+            sendSnapshotError(response, error);
+        }
+    };
+    const handleSnapshotApi = async (request, response, url, id, suffix) => {
+        try {
+            if (suffix === '' || suffix === '/') {
+                const entry = await snapshotView(id);
+                const detail = {
+                    snapshot: snapshotSummary(entry.payload, entry.bytes),
+                    config: entry.payload.config,
+                    limitations: entry.payload.limitations,
+                    traces: entry.store.listTraces({ limit: 20 }, entry.payload.config.slowRequestMs).items,
+                    errors: entry.store.listErrors({ limit: 20 }).items,
+                    databaseQueries: entry.store.listDatabaseQueries({ limit: 20 }).items,
+                };
+                sendJson(response, 200, detail);
+                return;
+            }
+            if (suffix === '/export') {
+                const entry = await snapshotView(id);
+                sendJson(response, 200, snapshotExport(entry.payload), {
+                    'content-disposition': `attachment; filename="stacknext-${entry.payload.projectName.replace(/[^a-z0-9_-]+/gi, '-')}-${id}.json"`,
+                });
+                return;
+            }
+            if (suffix === '/traces') {
+                const entry = await snapshotView(id);
+                sendJson(response, 200, entry.store.listTraces(filtersFromSearch(url.searchParams), entry.payload.config.slowRequestMs));
+                return;
+            }
+            if (suffix.startsWith('/traces/')) {
+                const traceId = decodeURIComponent(suffix.slice('/traces/'.length));
+                if (!/^[0-9a-f]{32}$/i.test(traceId)) {
+                    sendError(response, 400, 'Invalid trace id');
+                    return;
+                }
+                const entry = await snapshotView(id);
+                const detail = entry.store.getTrace(traceId, entry.payload.config.slowRequestMs);
+                if (!detail) {
+                    sendError(response, 404, 'Trace not found in snapshot');
+                    return;
+                }
+                sendJson(response, 200, detail);
+                return;
+            }
+            if (suffix === '/errors') {
+                const entry = await snapshotView(id);
+                const result = entry.store.listErrors(errorFiltersFromSearch(url.searchParams));
+                const payload = {
+                    items: result.items,
+                    total: result.total,
+                    limit: result.limit,
+                    offset: result.offset,
+                    groups: result.groups,
+                };
+                sendJson(response, 200, payload);
+                return;
+            }
+            if (suffix.startsWith('/errors/')) {
+                const errorId = decodeURIComponent(suffix.slice('/errors/'.length));
+                if (!errorId || errorId.length > 128) {
+                    sendError(response, 400, 'Invalid error id');
+                    return;
+                }
+                const entry = await snapshotView(id);
+                const detail = entry.store.getError(errorId, options.config.slowRequestMs);
+                if (!detail) {
+                    sendError(response, 404, 'Error not found in snapshot');
+                    return;
+                }
+                const payload = detail;
+                sendJson(response, 200, payload);
+                return;
+            }
+            if (suffix === '/database') {
+                const entry = await snapshotView(id);
+                const result = entry.store.listDatabaseQueries(databaseFiltersFromSearch(url.searchParams));
+                sendJson(response, 200, result);
+                return;
+            }
+            sendError(response, 404, 'Not found');
+        }
+        catch (error) {
+            sendSnapshotError(response, error);
+        }
+    };
     const testRuns = new TestRunRegistry({
         slowRequestMs: options.config.slowRequestMs,
         isConnected: () => getConnections(Date.now()).some((connection) => connection.connected),
@@ -681,6 +989,8 @@ export async function startCollector(options) {
                 time: now,
             });
         }
+        if (envelope.errors.length > 0)
+            enqueueAutoErrorSnapshot();
         sendJson(response, 202, { ok: true, accepted, serverTime: now }, { 'x-stacknext-sequence': String(sequence) });
     };
     const handleTestRunWrite = async (request, response, url) => {
@@ -1017,20 +1327,6 @@ export async function startCollector(options) {
             sendJson(response, 200, store.listRequests(filtersFromSearch(url.searchParams), options.config.slowRequestMs));
             return;
         }
-        if (pathname.startsWith('/api/requests/')) {
-            const traceId = pathname.slice('/api/requests/'.length);
-            if (!/^[0-9a-f]{32}$/i.test(traceId)) {
-                sendError(response, 400, 'Invalid trace id');
-                return;
-            }
-            const detail = store.getRequest(traceId, options.config.slowRequestMs);
-            if (!detail) {
-                sendError(response, 404, 'Request not found');
-                return;
-            }
-            sendJson(response, 200, traceWithDiagnostics(detail));
-            return;
-        }
         if (pathname === '/api/traces') {
             const result = store.listTraces(filtersFromSearch(url.searchParams), options.config.slowRequestMs);
             sendJson(response, 200, result);
@@ -1082,6 +1378,45 @@ export async function startCollector(options) {
             sendJson(response, 200, result);
             return;
         }
+        if (pathname === '/api/snapshots') {
+            if (!projectDir) {
+                const empty = {
+                    items: [],
+                    total: 0,
+                    limit: 50,
+                    offset: 0,
+                    invalid: 0,
+                };
+                sendJson(response, 200, empty);
+                return;
+            }
+            try {
+                const page = await listSnapshots(projectDir, {
+                    ...(queryNumber(url.searchParams.get('limit')) !== undefined
+                        ? { limit: queryNumber(url.searchParams.get('limit')) }
+                        : {}),
+                    ...(queryNumber(url.searchParams.get('offset')) !== undefined
+                        ? { offset: queryNumber(url.searchParams.get('offset')) }
+                        : {}),
+                });
+                const payload = page;
+                sendJson(response, 200, payload);
+            }
+            catch (error) {
+                sendSnapshotError(response, error);
+            }
+            return;
+        }
+        if (pathname.startsWith('/api/snapshots/')) {
+            const match = /^\/api\/snapshots\/([^/]+)(\/.*)?$/.exec(pathname);
+            const id = match?.[1] ? decodeURIComponent(match[1]) : '';
+            if (!isSnapshotId(id)) {
+                sendError(response, 400, 'Invalid snapshot id');
+                return;
+            }
+            await handleSnapshotApi(request, response, url, id, match?.[2] ?? '');
+            return;
+        }
         if (pathname === '/api/export') {
             const data = store.exportAll();
             const payload = {
@@ -1146,12 +1481,31 @@ export async function startCollector(options) {
             if (url.pathname.startsWith('/api/')) {
                 const testRunWrite = url.pathname === '/api/test-runs' ||
                     url.pathname.startsWith('/api/test-runs/');
+                const snapshotWrite = url.pathname === '/api/snapshots' ||
+                    url.pathname.startsWith('/api/snapshots/');
                 if (request.method === 'POST' && testRunWrite) {
                     await handleTestRunWrite(request, response, url);
                     return;
                 }
+                if (request.method === 'POST' && url.pathname === '/api/snapshots/import') {
+                    await handleSnapshotWrite(request, response, url);
+                    return;
+                }
+                if (request.method === 'POST' &&
+                    (url.pathname === '/api/snapshots' || url.pathname === '/api/snapshots/')) {
+                    await handleSnapshotWrite(request, response, url);
+                    return;
+                }
+                if (request.method === 'DELETE' && /^\/api\/snapshots\/[^/]+\/?$/.test(url.pathname)) {
+                    await handleSnapshotWrite(request, response, url);
+                    return;
+                }
                 if (request.method !== 'GET' && request.method !== 'HEAD') {
-                    response.setHeader('allow', testRunWrite ? 'GET, HEAD, POST' : 'GET, HEAD');
+                    response.setHeader('allow', testRunWrite
+                        ? 'GET, HEAD, POST'
+                        : snapshotWrite
+                            ? 'GET, HEAD, POST, DELETE'
+                            : 'GET, HEAD');
                     sendError(response, 405, 'Method not allowed');
                     return;
                 }
@@ -1214,6 +1568,21 @@ export async function startCollector(options) {
     }
     port = actualPort;
     dashboardUrl = `http://${host === '127.0.0.1' ? '127.0.0.1' : host}:${port}`;
+    const saveSnapshotOnExit = async () => {
+        if (!options.config.snapshots.autoOnExit)
+            return null;
+        await autoSaveQueue;
+        try {
+            const saved = await saveCurrentSnapshot('exit');
+            if (saved)
+                log(`exit snapshot saved (${saved.id})`);
+            return saved;
+        }
+        catch (error) {
+            log(`exit snapshot failed: ${error instanceof Error ? error.message : String(error)}`);
+            return null;
+        }
+    };
     let closing = false;
     const close = async () => {
         if (closing)
@@ -1252,6 +1621,7 @@ export async function startCollector(options) {
         token: options.token,
         store,
         health: getHealth,
+        saveSnapshotOnExit,
         close,
     };
 }