feat(client): persist digital employee snapshots

This commit is contained in:
baiyanyun
2026-06-05 16:27:15 +08:00
parent 15a3285ed5
commit 40151c3902
5 changed files with 251 additions and 63 deletions

View File

@@ -0,0 +1,83 @@
import { ipcMain } from "electron";
import type { HandlerContext } from "@shared/types/ipc";
import { FEATURES } from "@shared/featureFlags";
import { agentService } from "../services/engines/unifiedAgent";
import { mcpProxyManager } from "../services/packages/mcp";
import { getComputerServerStatus } from "../services/computerServer";
import { getGuiAgentServerStatus } from "../services/packages/guiAgentServer";
import { getWindowsMcpStatus } from "../services/packages/windowsMcp";
import { isWindows } from "../services/system/shellEnv";
import {
recordDigitalEmployeeSnapshot,
readDigitalEmployeeState,
type DigitalEmployeeServiceStatus,
type DigitalEmployeeSnapshot,
} from "../services/digitalEmployee/stateService";
export function registerDigitalEmployeeHandlers(ctx: HandlerContext): void {
ipcMain.handle("digitalEmployee:getSnapshot", async () => {
const snapshot = buildSnapshot(ctx);
const state = recordDigitalEmployeeSnapshot(snapshot);
return {
...snapshot,
state: {
updatedAt: state.updatedAt,
events: state.events,
},
};
});
ipcMain.handle("digitalEmployee:getState", async () => {
return readDigitalEmployeeState();
});
}
function buildSnapshot(ctx: HandlerContext): DigitalEmployeeSnapshot {
const sessions = agentService.listAllSessionsDetailed();
return {
generatedAt: new Date().toISOString(),
services: {
agent: {
running: agentService.isReady,
engineType: agentService.getEngineType(),
},
mcp: normalizeServiceStatus(mcpProxyManager.getStatus()),
computerServer: normalizeServiceStatus(getComputerServerStatus()),
lanproxy: normalizeServiceStatus(ctx.lanproxy.status()),
fileServer: normalizeServiceStatus(ctx.fileServer.status()),
guiServer: normalizeServiceStatus(getGuiStatus()),
},
sessions: {
total: sessions.length,
active: sessions.filter((session) =>
session.status === "active" || session.status === "pending",
).length,
items: sessions.slice(0, 10),
},
};
}
function normalizeServiceStatus(
value: object | null | undefined,
): DigitalEmployeeServiceStatus {
if (!value) return { running: false };
const record = value as Record<string, unknown>;
return {
running: record.running === true,
error: typeof record.error === "string" ? record.error : undefined,
engineType: typeof record.engineType === "string" ? record.engineType : undefined,
port: typeof record.port === "number" ? record.port : undefined,
serverCount: typeof record.serverCount === "number" ? record.serverCount : undefined,
serverNames: Array.isArray(record.serverNames)
? record.serverNames.filter((item): item is string => typeof item === "string")
: undefined,
pid: typeof record.pid === "number" ? record.pid : undefined,
};
}
function getGuiStatus(): object {
if (!FEATURES.ENABLE_GUI_AGENT_SERVER) {
return { running: false, error: "GUI Agent Server is not available" };
}
return isWindows() ? getWindowsMcpStatus() : getGuiAgentServerStatus();
}

View File

@@ -14,6 +14,7 @@ import { registerSandboxHandlers } from "./sandboxHandlers";
import { registerPerfHandlers } from "./perfHandlers";
import { registerGuiServerHandlers } from "./guiServerHandlers";
import { registerI18nHandlers } from "./i18nHandlers";
import { registerDigitalEmployeeHandlers } from "./digitalEmployeeHandlers";
import log from "electron-log";
export function registerAllHandlers(ctx: HandlerContext): void {
@@ -32,6 +33,7 @@ export function registerAllHandlers(ctx: HandlerContext): void {
registerPerfHandlers();
registerGuiServerHandlers();
registerI18nHandlers();
registerDigitalEmployeeHandlers(ctx);
log.info("IPC handlers registered");
}