mirror of
https://bitbucket.org/siakitem/my-pi.git
synced 2026-10-26 18:55:22 +00:00
feat(pi-hermes-memory): 导入上游 v0.9.10 源码快照
从官方 Git 仓库导入 chandra447/pi-hermes-memory 在 tag v0.9.10 (df624056a2e8c68bc89ba0a81e1c120856ec9665,2026-10-02)的源码快照。 本次仅导入源码,尚未接入组合包:package.json、加载顺序、依赖与权限策略 均未改动。 - 原样导入 src/、tests/、docs/、scripts/、README、CHANGELOG、PLAN、 AGENTS 与 LICENSE;该快照共 156 个上游文件。 - 剔除嵌套 .git、上游 .github 工作流、.ralph/、.claude/、node_modules、 覆盖率与构建产物。 - 新增中文 UPSTREAM.md,记录来源,以及启用前必须完成的待决事项: 在 pi-compaction 之前注册加载顺序、better-sqlite3 版本与根 allowScripts 对齐、strip-ansi 解析确认。 - 说明选择 tag 而非 main 的原因:导入时上游 main 已领先至 24fcd8ad26fc27badb8d7dbaf8034645de07b893(2026-10-11),选 tag 是为 对应一个已发布、可复现的版本。 尚未接线。启用前还需反转 tests/extension-load.test.ts 中"不得加载 Hermes"的断言,并同步 pi-tool-search 分组、权限策略与文档。 验证:快照 156 个上游文件全部导入,目录内仅多出 UPSTREAM.md 一份; 已确认与上游 main 内容存在差异,证明取的是 tag 而非 main。
This commit is contained in:
@@ -0,0 +1,101 @@
|
||||
import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
|
||||
|
||||
type MemoryContext = Pick<ExtensionContext, "cwd">;
|
||||
export type EnsureMemoryReady = (ctx: MemoryContext, signal?: AbortSignal) => Promise<void>;
|
||||
|
||||
/** Cancel one waiter without cancelling initialization shared with other users. */
|
||||
function waitForReady(promise: Promise<void>, signal?: AbortSignal): Promise<void> {
|
||||
if (!signal) return promise;
|
||||
return new Promise((resolve, reject) => {
|
||||
const abort = () => reject(signal.reason ?? new Error("Memory operation aborted"));
|
||||
signal.addEventListener("abort", abort, { once: true });
|
||||
if (signal.aborted) abort();
|
||||
promise.then(resolve, reject).finally(() => signal.removeEventListener("abort", abort));
|
||||
});
|
||||
}
|
||||
|
||||
/** Own initialization, per-use preparation and operations through shutdown. */
|
||||
export function createMemoryInitializer(
|
||||
initialize: (ctx?: MemoryContext) => Promise<void>,
|
||||
prepare?: (ctx: MemoryContext) => Promise<void>,
|
||||
) {
|
||||
let pending: Promise<void> | undefined;
|
||||
let initialized = false;
|
||||
let ready = false;
|
||||
let closed = false;
|
||||
const active = new Set<Promise<unknown>>();
|
||||
|
||||
function assertOpen(): void {
|
||||
if (closed) throw new Error("Memory session has shut down");
|
||||
}
|
||||
|
||||
function track<T>(work: Promise<T>): Promise<T> {
|
||||
active.add(work);
|
||||
void work.then(() => active.delete(work), () => active.delete(work));
|
||||
return work;
|
||||
}
|
||||
|
||||
async function ensure(ctx?: MemoryContext, signal?: AbortSignal): Promise<void> {
|
||||
assertOpen();
|
||||
signal?.throwIfAborted();
|
||||
if (!initialized) {
|
||||
pending ??= Promise.resolve().then(() => initialize(ctx)).then(() => {
|
||||
initialized = true;
|
||||
}).finally(() => {
|
||||
pending = undefined;
|
||||
});
|
||||
}
|
||||
// Track the underlying preparation even when an individual waiter cancels.
|
||||
const preparation = track((pending ?? Promise.resolve()).then(async () => {
|
||||
if (ctx) await prepare?.(ctx);
|
||||
ready = true;
|
||||
}));
|
||||
await waitForReady(preparation, signal);
|
||||
signal?.throwIfAborted();
|
||||
assertOpen();
|
||||
}
|
||||
|
||||
return {
|
||||
isReady: () => ready,
|
||||
ensure,
|
||||
run<T>(ctx: MemoryContext, work: () => Promise<T>, signal?: AbortSignal): Promise<T> {
|
||||
return track(ensure(ctx, signal).then(() => {
|
||||
assertOpen();
|
||||
signal?.throwIfAborted();
|
||||
return work();
|
||||
}));
|
||||
},
|
||||
async close(): Promise<void> {
|
||||
closed = true;
|
||||
// No new work is accepted. Join loads, project binds, backfill and callers
|
||||
// already executing before the owner checkpoints/closes its database.
|
||||
await Promise.allSettled([...active]);
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
/** Guard memory entry points without changing schemas, rendering or events. */
|
||||
export function withMemoryInitialization(
|
||||
pi: ExtensionAPI,
|
||||
initialization: Pick<ReturnType<typeof createMemoryInitializer>, "run">,
|
||||
): ExtensionAPI {
|
||||
return {
|
||||
...pi,
|
||||
registerTool(tool) {
|
||||
pi.registerTool({
|
||||
...tool,
|
||||
async execute(id, params, signal, onUpdate, ctx) {
|
||||
return initialization.run(ctx, () => tool.execute(id, params, signal, onUpdate, ctx), signal);
|
||||
},
|
||||
});
|
||||
},
|
||||
registerCommand(name, options) {
|
||||
pi.registerCommand(name, {
|
||||
...options,
|
||||
async handler(args, ctx) {
|
||||
return initialization.run(ctx, () => options.handler(args, ctx));
|
||||
},
|
||||
});
|
||||
},
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user