66 lines
1.8 KiB
TypeScript
66 lines
1.8 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
|
|
import { CACHE_DIR } from './config.js';
|
|
import type { GroupStatus } from './group-queue.js';
|
|
import type { AgentType } from './types.js';
|
|
|
|
export interface StatusSnapshotEntry {
|
|
jid: string;
|
|
name: string;
|
|
folder: string;
|
|
agentType: AgentType;
|
|
status: GroupStatus['status'];
|
|
elapsedMs: number | null;
|
|
pendingMessages: boolean;
|
|
pendingTasks: number;
|
|
}
|
|
|
|
export interface StatusSnapshot {
|
|
agentType: AgentType;
|
|
assistantName: string;
|
|
updatedAt: string;
|
|
entries: StatusSnapshotEntry[];
|
|
}
|
|
|
|
const STATUS_SNAPSHOT_DIR = path.join(CACHE_DIR, 'status-dashboard');
|
|
|
|
export function writeStatusSnapshot(snapshot: StatusSnapshot): void {
|
|
fs.mkdirSync(STATUS_SNAPSHOT_DIR, { recursive: true });
|
|
const targetPath = path.join(
|
|
STATUS_SNAPSHOT_DIR,
|
|
`${snapshot.agentType}.json`,
|
|
);
|
|
const tempPath = `${targetPath}.tmp`;
|
|
fs.writeFileSync(tempPath, JSON.stringify(snapshot, null, 2));
|
|
fs.renameSync(tempPath, targetPath);
|
|
}
|
|
|
|
export function readStatusSnapshots(maxAgeMs: number): StatusSnapshot[] {
|
|
if (!fs.existsSync(STATUS_SNAPSHOT_DIR)) return [];
|
|
|
|
const now = Date.now();
|
|
const snapshots: StatusSnapshot[] = [];
|
|
|
|
for (const entry of fs.readdirSync(STATUS_SNAPSHOT_DIR)) {
|
|
if (!entry.endsWith('.json')) continue;
|
|
const snapshotPath = path.join(STATUS_SNAPSHOT_DIR, entry);
|
|
|
|
try {
|
|
const raw = fs.readFileSync(snapshotPath, 'utf-8');
|
|
const parsed = JSON.parse(raw) as StatusSnapshot;
|
|
if (!parsed.updatedAt || !parsed.agentType || !Array.isArray(parsed.entries))
|
|
continue;
|
|
|
|
const ageMs = now - new Date(parsed.updatedAt).getTime();
|
|
if (Number.isNaN(ageMs) || ageMs > maxAgeMs) continue;
|
|
|
|
snapshots.push(parsed);
|
|
} catch {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
return snapshots;
|
|
}
|