Genarrative/apps/ai-game-creator-shell/src/services/projectSnapshotWorkspace.ts

type WorkspaceInvoke = (
  command: string,
  args?: Record<string, unknown>,
) => Promise<unknown>;

/** 同一 WebView 的登记串行提交；旧请求完成后不能覆盖更新的工程状态。 */
export function createProjectSnapshotWorkspaceRegistration(
  invoke: WorkspaceInvoke,
) {
  let requestedPath: string | null | undefined;
  let revision = 0;
  let ownerRevision = 0;
  let queue = Promise.resolve();

  function setProject(projectPath: string | null) {
    if (requestedPath === projectPath) return queue;
    requestedPath = projectPath;
    const requestRevision = ++revision;
    queue = queue.then(async () => {
      try {
        await invoke('set_active_project_snapshot_workspace', { projectPath });
      } catch {
        // 失败不能被记作已登记；下一次相同路径登记仍可重新提交。
        if (revision === requestRevision) requestedPath = undefined;
        try {
          await invoke('append_application_log', {
            level: 'error',
            source: 'project-snapshot',
            message: 'project_snapshot.workspace.registration.failed',
          });
        } catch {
          // 原生桥本身不可用时也不阻断项目创作。
        }
      }
    });
    return queue;
  }

  function acquire() {
    const owner = ++ownerRevision;
    return () => {
      // StrictMode 的同轮卸载/重挂以及工作台替换不代表工程真的关闭。
      queueMicrotask(() => {
        if (owner === ownerRevision) void setProject(null);
      });
    };
  }

  return { setProject, acquire };
}

const registrations = new WeakMap<
  WorkspaceInvoke,
  ReturnType<typeof createProjectSnapshotWorkspaceRegistration>
>();

export function projectSnapshotWorkspaceRegistration(invoke: WorkspaceInvoke) {
  let registration = registrations.get(invoke);
  if (!registration) {
    registration = createProjectSnapshotWorkspaceRegistration(invoke);
    registrations.set(invoke, registration);
  }
  return registration;
}