aws-runtime-bridge 1.9.9 → 1.9.20

package/dist/services/agent-event-store.d.ts

@@ -0,0 +1,59 @@
+/**
+ * Agent 事件存储
+ *
+ * 为每个 agent 维护独立的事件环状缓冲区，支持事件追加和输出合并。
+ * scheduleEventPush 会消费 entry 进行差量推送。
+ */
+import { type RingBufferEntry } from "./ring-buffer.js";
+export interface BridgeEventPayload {
+    readonly category: "status" | "output" | "question" | "file_changed";
+    readonly agentId: string;
+    readonly sessionId: string | null;
+    readonly status?: string;
+    readonly output?: string;
+    readonly seq?: number;
+    readonly questions?: unknown[];
+    readonly actionType?: string;
+    readonly actionLabel?: string;
+    readonly actionDetail?: string;
+    readonly actionResult?: string;
+    readonly actionId?: string;
+    readonly inputTokens?: number;
+    readonly outputTokens?: number;
+    readonly sourceEventType?: string;
+    readonly logLevel?: string;
+    readonly logCategory?: string;
+    readonly workspacePath?: string;
+    readonly filePath?: string;
+    readonly eventType?: string;
+    readonly size?: number;
+}
+/**
+ * per-agent 事件存储。
+ * 每条 append/appendOrMergeOutput 调用会写入对应 agent 的环状缓冲区，
+ * 并返回 RingBufferEntry 供 scheduleEventPush 消费。
+ */
+declare class AgentEventStore {
+    private readonly agents;
+    private ensureBuffer;
+    /**
+     * 追加一条事件到指定 agent 的缓冲区。
+     * @returns 新创建的条目
+     */
+    append(agentId: string, payload: BridgeEventPayload): RingBufferEntry<BridgeEventPayload>;
+    /**
+     * 追加一条 output 事件到指定 agent 的缓冲区。
+     * 若最后一条也是 output 且来自同一 session + sourceEventType，
+     * 则在 10s 窗口内合并内容。
+     * @returns 被合并或新创建的条目
+     */
+    appendOrMergeOutput(agentId: string, payload: BridgeEventPayload): RingBufferEntry<BridgeEventPayload>;
+    /**
+     * 获取指定 agent 的事件数量。
+     */
+    size(agentId: string): number;
+}
+/** 全局单例 */
+export declare const agentEventStore: AgentEventStore;
+export {};
+//# sourceMappingURL=agent-event-store.d.ts.map

package/dist/services/agent-event-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-event-store.d.ts","sourceRoot":"","sources":["../../src/services/agent-event-store.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAc,KAAK,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEpE,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,QAAQ,GAAG,UAAU,GAAG,cAAc,CAAC;IACrE,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC;IAC/B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,eAAe,CAAC,EAAE,MAAM,CAAC;IAClC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC;IAChC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;CACxB;AAQD;;;;GAIG;AACH,cAAM,eAAe;IACnB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuC;IAE9D,OAAO,CAAC,YAAY;IASpB;;;OAGG;IACH,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB,GAAG,eAAe,CAAC,kBAAkB,CAAC;IAIzF;;;;;OAKG;IACH,mBAAmB,CACjB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,kBAAkB,GAC1B,eAAe,CAAC,kBAAkB,CAAC;IA6BtC;;OAEG;IACH,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;CAG9B;AAED,WAAW;AACX,eAAO,MAAM,eAAe,iBAAwB,CAAC"}

package/dist/services/agent-event-store.js

@@ -0,0 +1,68 @@
+/**
+ * Agent 事件存储
+ *
+ * 为每个 agent 维护独立的事件环状缓冲区，支持事件追加和输出合并。
+ * scheduleEventPush 会消费 entry 进行差量推送。
+ */
+import { RingBuffer } from "./ring-buffer.js";
+const PER_AGENT_BUFFER_CAPACITY = 500;
+/**
+ * per-agent 事件存储。
+ * 每条 append/appendOrMergeOutput 调用会写入对应 agent 的环状缓冲区，
+ * 并返回 RingBufferEntry 供 scheduleEventPush 消费。
+ */
+class AgentEventStore {
+    constructor() {
+        this.agents = new Map();
+    }
+    ensureBuffer(agentId) {
+        let entry = this.agents.get(agentId);
+        if (!entry) {
+            entry = { buffer: new RingBuffer(PER_AGENT_BUFFER_CAPACITY) };
+            this.agents.set(agentId, entry);
+        }
+        return entry.buffer;
+    }
+    /**
+     * 追加一条事件到指定 agent 的缓冲区。
+     * @returns 新创建的条目
+     */
+    append(agentId, payload) {
+        return this.ensureBuffer(agentId).append(payload);
+    }
+    /**
+     * 追加一条 output 事件到指定 agent 的缓冲区。
+     * 若最后一条也是 output 且来自同一 session + sourceEventType，
+     * 则在 10s 窗口内合并内容。
+     * @returns 被合并或新创建的条目
+     */
+    appendOrMergeOutput(agentId, payload) {
+        return this.ensureBuffer(agentId).appendOrMerge(payload, (lastEntry) => {
+            const last = lastEntry.data;
+            return (last.category === "output" &&
+                last.sessionId === payload.sessionId &&
+                last.sourceEventType === payload.sourceEventType &&
+                Date.now() - lastEntry.timestamp <= 10_000);
+        }, (lastData, newData) => {
+            const existingOutput = lastData.output || "";
+            const newOutput = newData.output || "";
+            const merged = existingOutput.includes(newOutput)
+                ? existingOutput
+                : `${existingOutput}${newOutput}`;
+            return {
+                ...lastData,
+                output: merged.length > 8000
+                    ? `${merged.slice(0, 8000)}...(truncated)`
+                    : merged,
+            };
+        });
+    }
+    /**
+     * 获取指定 agent 的事件数量。
+     */
+    size(agentId) {
+        return this.agents.get(agentId)?.buffer.size ?? 0;
+    }
+}
+/** 全局单例 */
+export const agentEventStore = new AgentEventStore();

package/dist/services/event-delta-pusher.d.ts

@@ -0,0 +1,40 @@
+/**
+ * 事件差量推送调度器
+ *
+ * 将 agent 事件缓冲区中的新事件差量推送到调度中心。
+ * 提供全局心跳调度和推送统计查询。
+ */
+import { type BridgeEventPayload } from "./agent-event-store.js";
+import { type RingBufferEntry } from "./ring-buffer.js";
+export interface RuntimeCallbackAuthOptions {
+    runtimeAccessToken?: string;
+    schedulerBaseUrl?: string;
+}
+export interface PushStatEntry {
+    agentId: string;
+    pending: number;
+    success: number;
+    failure: number;
+    consecutiveFailures: number;
+    lastLatencyMs?: number;
+    lastError?: string;
+}
+/**
+ * 调度一次差量推送：将 agent 的新事件推送到调度中心。
+ * 暂以单次 push 模式实现，后续可按需改为批量/定时 flush。
+ */
+export declare function scheduleEventPush(agentId: string, _entry: RingBufferEntry<BridgeEventPayload>, authOptions?: RuntimeCallbackAuthOptions): Promise<void>;
+/**
+ * 启动全局心跳调度器。
+ * 每 30s 向调度中心发送空包，检测连接存活。
+ */
+export declare function startHeartbeatScheduler(): void;
+/**
+ * 停止全局心跳调度器。
+ */
+export declare function stopHeartbeatScheduler(): void;
+/**
+ * 返回所有 agent 的推送统计。
+ */
+export declare function getAllPushStats(): PushStatEntry[];
+//# sourceMappingURL=event-delta-pusher.d.ts.map

package/dist/services/event-delta-pusher.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-delta-pusher.d.ts","sourceRoot":"","sources":["../../src/services/event-delta-pusher.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAKH,OAAO,EAAmB,KAAK,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAClF,OAAO,EAAE,KAAK,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAKxD,MAAM,WAAW,0BAA0B;IACzC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AA0BD;;;GAGG;AACH,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,eAAe,CAAC,kBAAkB,CAAC,EAC3C,WAAW,CAAC,EAAE,0BAA0B,GACvC,OAAO,CAAC,IAAI,CAAC,CAQf;AAOD;;;GAGG;AACH,wBAAgB,uBAAuB,IAAI,IAAI,CAuB9C;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,IAAI,CAK7C;AAaD;;GAEG;AACH,wBAAgB,eAAe,IAAI,aAAa,EAAE,CAcjD"}

package/dist/services/event-delta-pusher.js

@@ -0,0 +1,95 @@
+/**
+ * 事件差量推送调度器
+ *
+ * 将 agent 事件缓冲区中的新事件差量推送到调度中心。
+ * 提供全局心跳调度和推送统计查询。
+ */
+import axios from "axios";
+import { resolveSchedulerBaseUrl, resolveSchedulerBaseUrlFrom } from "../config.js";
+import { agentEventStore } from "./agent-event-store.js";
+import { getRuntimeAccessToken } from "./runtime-binding.js";
+const pushStates = new Map();
+function runtimeAuthHeaders(authOptions, callbackBaseUrl) {
+    const resolvedBaseUrl = callbackBaseUrl ?? resolveSchedulerBaseUrlFrom(authOptions?.schedulerBaseUrl).url;
+    const token = authOptions?.runtimeAccessToken ||
+        getRuntimeAccessToken(undefined, resolvedBaseUrl) ||
+        getRuntimeAccessToken();
+    return token ? { "X-Runtime-Token": token } : undefined;
+}
+// ── 差量推送 ─────────────────────────────────────────────
+/**
+ * 调度一次差量推送：将 agent 的新事件推送到调度中心。
+ * 暂以单次 push 模式实现，后续可按需改为批量/定时 flush。
+ */
+export async function scheduleEventPush(agentId, _entry, authOptions) {
+    // 当前实现为 placeholder — 事件已写入 agentEventStore 的 ring buffer，
+    // 调度中心可通过 /runtime/events/:agentId 端点拉取。
+    // 后续可演进为 WebSocket 实时推送或定时批量 flush。
+    // 此处仅记录统计信息，不做实际 HTTP 推送。
+    const state = getOrCreatePushState(agentId);
+    state.success++;
+    state.consecutiveFailures = 0;
+}
+// ── 全局心跳 ─────────────────────────────────────────────
+let heartbeatTimer = null;
+const HEARTBEAT_INTERVAL_MS = 30_000;
+/**
+ * 启动全局心跳调度器。
+ * 每 30s 向调度中心发送空包，检测连接存活。
+ */
+export function startHeartbeatScheduler() {
+    if (heartbeatTimer)
+        return;
+    const baseUrl = resolveSchedulerBaseUrl().url;
+    heartbeatTimer = setInterval(async () => {
+        try {
+            const headers = runtimeAuthHeaders();
+            if (!headers)
+                return;
+            await axios.post(`${baseUrl}/api/runtime/callback/heartbeat`, { timestamp: Date.now() }, { headers, timeout: 5000 });
+        }
+        catch {
+            // 心跳失败仅记录，不阻塞
+        }
+    }, HEARTBEAT_INTERVAL_MS);
+    // 允许进程在 timer 存在时正常退出
+    if (heartbeatTimer && typeof heartbeatTimer === "object" && "unref" in heartbeatTimer) {
+        heartbeatTimer.unref();
+    }
+}
+/**
+ * 停止全局心跳调度器。
+ */
+export function stopHeartbeatScheduler() {
+    if (heartbeatTimer) {
+        clearInterval(heartbeatTimer);
+        heartbeatTimer = null;
+    }
+}
+// ── 推送统计 ─────────────────────────────────────────────
+function getOrCreatePushState(agentId) {
+    let state = pushStates.get(agentId);
+    if (!state) {
+        state = { success: 0, failure: 0, consecutiveFailures: 0 };
+        pushStates.set(agentId, state);
+    }
+    return state;
+}
+/**
+ * 返回所有 agent 的推送统计。
+ */
+export function getAllPushStats() {
+    const stats = [];
+    for (const [agentId, state] of pushStates) {
+        stats.push({
+            agentId,
+            pending: agentEventStore.size(agentId),
+            success: state.success,
+            failure: state.failure,
+            consecutiveFailures: state.consecutiveFailures,
+            lastLatencyMs: state.lastLatencyMs,
+            lastError: state.lastError,
+        });
+    }
+    return stats;
+}

package/dist/services/fileSnapshotStore.d.ts

@@ -0,0 +1,96 @@
+/**
+ * 文件快照存储服务
+ *
+ * 借鉴 Claude Code fileHistory 的确定性整文件覆盖方案，替代 git stash 实现稳定快照。
+ * - 快照 = 工作区那一刻所有文件的内容哈希副本 + index.json 映射
+ * - 还原 = "先清后写"整文件覆盖，零冲突，幂等
+ * - 内容哈希去重：相同文件内容跨快照共享一份物理副本
+ */
+export interface SnapshotIndexEntry {
+    /** 备份文件名（= sha256(content).slice(0,16)）；null 表示快照时该文件不存在（占位） */
+    backupName: string;
+    size: number;
+    mtime: number;
+    mode: number;
+}
+export type SnapshotIndex = Record<string, SnapshotIndexEntry>;
+export interface SnapshotMeta {
+    snapshotId: string;
+    workspacePath: string;
+    snapshotName: string;
+    triggerType: string;
+    createdAt: string;
+    fileCount: number;
+    totalBytes: number;
+    workspaceHash: string;
+}
+export interface CreateSnapshotResult {
+    ok: boolean;
+    snapshotId: string;
+    snapshotKey: string;
+    fileCount: number;
+    totalBytes: number;
+    workspaceHash: string;
+    error?: string;
+}
+export interface RestoreSnapshotResult {
+    ok: boolean;
+    filesRestored: number;
+    filesDeleted: number;
+    filesSkipped: number;
+    preRestoreSnapshotKey?: string;
+    error?: string;
+}
+export interface SnapshotListItem {
+    snapshotId: string;
+    snapshotKey: string;
+    snapshotName: string;
+    triggerType: string;
+    createdAt: string;
+    fileCount: number;
+    totalBytes: number;
+}
+/**
+ * 创建工作区文件快照。
+ *
+ * 算法（借鉴 Claude Code fileHistoryMakeSnapshot）：
+ * 1. 遍历工作区收集所有文件
+ * 2. 对每个文件计算内容哈希，拷贝到快照目录（去重）
+ * 3. 写入 index.json 和 meta.json
+ */
+export declare function createSnapshot(params: {
+    workspacePath: string;
+    snapshotId?: string;
+    snapshotName?: string;
+    triggerType?: string;
+}): Promise<CreateSnapshotResult>;
+/**
+ * 还原工作区到快照状态。
+ *
+ * 算法（借鉴 Claude Code applySnapshot）：
+ * 1. 可选：还原前自动创建安全网快照（preRestoreBackup）
+ * 2. 读取目标快照 index.json
+ * 3. 删除工作区中"快照后新增"的文件
+ * 4. 从备份逐文件覆盖，内容相同则跳过
+ */
+export declare function restoreSnapshot(params: {
+    workspacePath: string;
+    snapshotKey: string;
+    preRestoreBackup?: boolean;
+}): Promise<RestoreSnapshotResult>;
+/**
+ * 列出指定工作区的所有快照
+ */
+export declare function listSnapshots(workspacePath: string): Promise<SnapshotListItem[]>;
+/**
+ * 删除指定快照
+ */
+export declare function deleteSnapshot(snapshotKey: string): Promise<{
+    ok: boolean;
+    error?: string;
+}>;
+/**
+ * 获取快照根目录（供诊断/调试使用）
+ */
+export declare function getSnapshotRootPath(): string;
+//# sourceMappingURL=fileSnapshotStore.d.ts.map

package/dist/services/fileSnapshotStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fileSnapshotStore.d.ts","sourceRoot":"","sources":["../../src/services/fileSnapshotStore.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAgDH,MAAM,WAAW,kBAAkB;IACjC,gEAAgE;IAChE,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;AAE/D,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,OAAO,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,qBAAqB;IACpC,EAAE,EAAE,OAAO,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACpB;AA+FD;;;;;;;GAOG;AACH,wBAAsB,cAAc,CAAC,MAAM,EAAE;IAC3C,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAwHhC;AAED;;;;;;;;GAQG;AACH,wBAAsB,eAAe,CAAC,MAAM,EAAE;IAC5C,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAgJjC;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAuCtF;AAED;;GAEG;AACH,wBAAsB,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAoElG;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,MAAM,CAE5C"}