@olierjs/fugue-cli 0.0.0-stage → 0.1.0

package/skills/plan-with-remote/scripts/storage.mjs

@@ -0,0 +1,74 @@
+import { randomUUID, createHash } from 'node:crypto';
+import { mkdir, readFile, writeFile, rename, open, unlink } from 'node:fs/promises';
+import { dirname, resolve } from 'node:path';
+
+export const digest = value => `sha256:${createHash('sha256').update(typeof value === 'string' ? value : JSON.stringify(value)).digest('hex')}`;
+export const readJson = async path => JSON.parse(await readFile(path, 'utf8'));
+
+export async function saveSession(path, state) {
+  const temp = `${path}.${randomUUID()}.tmp`;
+  await writeFile(temp, JSON.stringify(state, null, 2) + '\n', { mode: 0o600, flag: 'wx' });
+  await rename(temp, path);
+}
+
+export async function createSession(output, state) {
+  const directory = resolve(output);
+  await mkdir(directory, { recursive: true });
+  const path = resolve(directory, 'session.json');
+  await writeFile(path, JSON.stringify(state, null, 2) + '\n', { flag: 'wx', mode: 0o600 });
+  return path;
+}
+
+export async function loadSession(path) {
+  const state = await readJson(path);
+  if (state.version !== 1 || !state.requirementId || !state.baseUrl) throw new Error('任务文件无效。请使用脚本返回的 session 路径。');
+  return state;
+}
+
+export async function lockSession(path, work) {
+  const lock = `${path}.lock`;
+  try {
+    const handle = await open(lock, 'wx', 0o600);
+    await handle.writeFile(String(process.pid));
+    await handle.close();
+  } catch (error) {
+    if (error.code !== 'EEXIST') throw error;
+    const pid = Number(await readFile(lock, 'utf8'));
+    if (!Number.isSafeInteger(pid) || pid <= 0) throw new Error('任务锁无法识别，请确认没有其他命令运行后清理锁文件。');
+    try { process.kill(pid, 0); }
+    catch (cause) {
+      if (cause.code === 'ESRCH') { await unlink(lock); return lockSession(path, work); }
+      throw cause;
+    }
+    throw new Error('此任务已有命令运行，请等待该命令结束。');
+  }
+  try { return await work(); }
+  finally { await unlink(lock); }
+}
+
+export async function writeReview(sessionPath, state, outcome, documents) {
+  const directory = resolve(dirname(sessionPath), `review-${randomUUID()}`);
+  await mkdir(directory);
+  const files = {};
+  const hashes = {};
+  for (const [name, text] of Object.entries(documents)) {
+    const path = resolve(directory, name);
+    await writeFile(path, text, { flag: 'wx' });
+    files[name] = path;
+    hashes[path] = digest(text);
+  }
+  const path = resolve(directory, 'review.json');
+  await writeFile(path, JSON.stringify({ version: 1, session: resolve(sessionPath), requirementId: state.requirementId, baseUrl: state.baseUrl, outcome, hashes }, null, 2) + '\n', { flag: 'wx', mode: 0o600 });
+  return { review: path, documents: files };
+}
+
+export async function loadReview(path) {
+  const review = await readJson(path);
+  if (review.version !== 1 || !review.session || !review.outcome || !review.hashes) throw new Error('审阅凭据无效，请重新读取文档。');
+  const state = await loadSession(review.session);
+  if (review.requirementId !== state.requirementId || review.baseUrl !== state.baseUrl) throw new Error('审阅凭据与任务不匹配。');
+  for (const [file, hash] of Object.entries(review.hashes)) {
+    if (digest(await readFile(file, 'utf8')) !== hash) throw new Error('审阅文档已被修改，请重新读取正式文档后再作决定。');
+  }
+  return review;
+}