minecodex 2.0.3 → 2.0.5

package/features/images/src/backend/local-image-source.mjs

@@ -0,0 +1,138 @@
+import os from "node:os";
+import {createHash,randomUUID} from "node:crypto";
+import {readFile,writeFile,mkdir,rename} from "node:fs/promises";
+import path from "node:path";
+import { ImageLibrary } from "../image-library.mjs";
+import { ThreadMetadataIndex } from "../thread-metadata.mjs";
+
+export class LocalImageSource {
+  constructor({
+    dataDir,
+    sourceRoot = process.env.CODEX_IMAGEGEN_ROOT ?? path.join(os.homedir(), ".codex/generated_images"),
+    sessionRoots,
+    library,
+  } = {}) {
+    const root = path.join(os.homedir(), ".codex");
+    this.library = library ?? new ImageLibrary({
+      sourceRoot,
+      dataDir,
+      sessionRoots: sessionRoots ?? [
+        process.env.CODEX_SESSION_ROOT ?? path.join(root, "sessions"),
+        path.join(root, "archived_sessions"),
+      ],
+      threadMetadata: new ThreadMetadataIndex({
+        stateDatabasePath: process.env.CODEX_STATE_DATABASE ?? path.join(root, "state_5.sqlite"),
+        globalStatePath: process.env.CODEX_GLOBAL_STATE ?? path.join(root, ".codex-global-state.json"),
+        sessionIndexPath: process.env.CODEX_SESSION_INDEX ?? path.join(root, "session_index.jsonl"),
+      }),
+    });
+    this.dataDir = dataDir;
+    this.hiddenIds = new Set();
+    this.hiddenWrites = Promise.resolve();
+    this.closed = false;
+  }
+
+  async start() {
+    if (this.closed) throw new Error("Local image source is closed");
+    if (!this.starting) {
+      this.starting = (async () => {
+        await this.library.start();
+        if (this.dataDir) {
+          try {this.hiddenIds = new Set(JSON.parse(await readFile(path.join(this.dataDir,"hidden-image-ids.json"),"utf8")));}
+          catch(error) {if(error.code!=="ENOENT") throw error;}
+        }
+        if (this.closed) throw new Error("Local image source is closed");
+      })();
+    }
+    await this.starting;
+  }
+
+  image(image) {
+    return {
+      id: image.id,
+      threadId: image.threadId,
+      imageGenerationItemId: image.imageGenerationItemId,
+      sourcePath: image.sourcePath,
+      name: path.basename(image.sourcePath),
+      width: image.width,
+      height: image.height,
+      threadTitle: image.threadTitle,
+    };
+  }
+
+  async action(payload = {}) {
+    const {operation, offset = 0, id, query = ""} = payload;
+    if (!["list", "prompt", "reference", "register", "upload-file", "begin-upload", "hide"].includes(operation)) throw new Error("Unknown local image source action");
+    if (typeof query !== "string" || query.length > 256) throw new Error("Invalid image query");
+    if (operation === "list" && (!Number.isInteger(offset) || offset < 0 || offset > 100000)) {
+      throw new Error("Invalid image offset");
+    }
+    if (operation !== "list" && operation !== "hide" && !/^[a-f0-9]{24}$/.test(id ?? "")) throw new Error("Invalid image ID");
+    if (operation === "hide" && (!Array.isArray(payload.ids) || !payload.ids.length || payload.ids.length>1000 || payload.ids.some(value=>typeof value!=="string" || !/^[a-f0-9]{24}$/.test(value)))) throw new Error("Invalid image selection");
+    await this.start();
+    if (this.closed) throw new Error("Local image source is closed");
+    if (operation === "hide") {
+      const ids=[...new Set(payload.ids)];
+      if (!this.dataDir || ids.some(value=>!this.library.get(value))) throw new Error("Local image selection is unavailable");
+      const task=this.hiddenWrites.then(async()=>{
+        const hidden=new Set([...this.hiddenIds,...ids]);
+        await mkdir(this.dataDir,{recursive:true});
+        const file=path.join(this.dataDir,"hidden-image-ids.json"), temporary=`${file}.${randomUUID()}.tmp`;
+        await writeFile(temporary,JSON.stringify([...hidden]),{mode:0o600});
+        await rename(temporary,file);
+        this.hiddenIds=hidden;
+        return {hiddenIds:ids};
+      });
+      this.hiddenWrites=task.catch(()=>{});
+      return task;
+    }
+    if (["reference", "register", "upload-file", "begin-upload"].includes(operation)) {
+      const {accountId,userId,libraryFileId,fingerprint} = payload;
+      if (![accountId,userId].every(value => typeof value === "string" && /^[\w-]{1,128}$/.test(value))) throw new Error("Invalid Library account");
+      if (!this.dataDir) throw new Error("Local image references require a data directory");
+      const image = this.library.get(id);
+      if (!image) throw new Error("Local image is unavailable");
+      // 只读取索引中的图片；内容变化后不沿用旧云端身份。
+      const bytes = await readFile(image.sourcePath);
+      const digest = createHash("sha256").update(bytes).digest("hex");
+      const key = createHash("sha256").update(JSON.stringify([accountId,userId,id,digest])).digest("hex");
+      const directory = path.join(this.dataDir,"cloud-image-references");
+      const file = path.join(directory,`${key}.json`);
+      if (operation === "begin-upload") {
+        if (fingerprint !== digest) throw new Error("Local image content changed");
+        await mkdir(directory,{recursive:true});
+        try {await writeFile(file,JSON.stringify({status:"uploading"}),{flag:"wx",mode:0o600});}
+        catch(error) {if(error.code==="EEXIST") return {started:false};throw error;}
+        return {started:true};
+      }
+      if (operation === "register") {
+        if (fingerprint !== digest || !/^libfile_[\w-]{1,128}$/.test(libraryFileId ?? "")) throw new Error("Invalid Library image reference");
+        await mkdir(directory,{recursive:true});
+        const temporary = `${file}.${randomUUID()}.tmp`;
+        await writeFile(temporary,JSON.stringify({libraryFileId}),{mode:0o600});
+        await rename(temporary,file);
+        return {libraryFileId,fingerprint:digest};
+      }
+      if (operation === "upload-file") return {fingerprint:digest,name:path.basename(image.sourcePath),base64:bytes.toString("base64")};
+      let reference;
+      try {reference=JSON.parse(await readFile(file,"utf8"));} catch(error) {if(error.code!=="ENOENT") throw error;}
+      return {libraryFileId:reference?.libraryFileId ?? null,status:reference?.status ?? "ready",fingerprint:digest};
+    }
+    if (operation === "prompt") {
+      const image = await this.library.hydratePrompt(id);
+      return image?.prompt?.trim() ? {text: image.prompt, kind: "generation"} : null;
+    }
+    const images=this.hiddenIds.size ? this.library.list({query}).filter(image=>!this.hiddenIds.has(image.id)) : null;
+    return {
+      images:(images ? images.slice(offset,offset+36) : this.library.list({limit:36,offset,query})).map(image=>this.image(image)),
+      total:images ? images.length : this.library.count({query}),
+    };
+  }
+
+  async close() {
+    this.closed = true;
+    await this.starting?.catch(() => {});
+    await this.hiddenWrites;
+    this.library.close();
+  }
+}

package/features/images/src/frontend/cloud-image-actions.mjs

@@ -0,0 +1,39 @@
+// 工作流只管理云端身份与操作顺序；上传协议和原生组件由 Adapter 提供。
+export function createCloudImageActions({reference,upload,fileNode,favoriteStatus,setFavorite,assertAccount}) {
+  const pending = new Map();
+  function registered(image) {
+    if (!pending.has(image.id)) {
+      const task = (async () => {
+        assertAccount();
+        const cached = await reference(image,"reference");
+        if (cached.libraryFileId) return {entry:{node:await fileNode(cached.libraryFileId)},created:false};
+        const unconfirmed = () => Object.assign(new Error("Library upload is pending or could not be confirmed"),{code:"UPLOAD_UNCONFIRMED"});
+        if (cached.status === "uploading") throw unconfirmed();
+        const file = await reference(image,"upload-file");
+        assertAccount();
+        if (!(await reference(image,"begin-upload",{fingerprint:file.fingerprint})).started) throw unconfirmed();
+        let libraryFileId;
+        try {libraryFileId = await upload(file);} catch(error) {throw Object.assign(unconfirmed(),{cause:error});}
+        if (!libraryFileId) throw new Error("Native upload did not return a Library file");
+        // 先保存身份；收藏失败或取消文件夹选择后仍可复用上传结果。
+        await reference(image,"register",{libraryFileId,fingerprint:file.fingerprint});
+        assertAccount();
+        return {entry:{node:await fileNode(libraryFileId)},created:true};
+      })();
+      pending.set(image.id,task);
+      const release = () => {if (pending.get(image.id) === task) pending.delete(image.id);};
+      void task.then(release,release);
+    }
+    return pending.get(image.id);
+  }
+  return {
+    registered,
+    async favorite(image) {
+      const {entry} = await registered(image);
+      const favoriteId = await favoriteStatus(entry.node.id);
+      assertAccount();
+      if (!await setFavorite(entry.node.id,favoriteId,!favoriteId)) throw new Error("Library favorite could not be changed");
+      return !favoriteId;
+    },
+  };
+}

package/features/images/src/frontend/local-image-source.mjs

@@ -0,0 +1,194 @@
+// 只编排来源与加载状态；图库、标签、菜单及详情由原生 Adapter 承载。
+export function installLocalImageSource(config, signal, sources, native) {
+  let mounted;
+  let revision = 0;
+  let pendingPrompt;
+  const text = (en, zh) => (document.documentElement.lang || navigator.language).startsWith("zh") ? zh : en;
+
+  function requestPrompt(image) {
+    pendingPrompt = image.id;
+    document.dispatchEvent(new CustomEvent("minecodex:image-prompt", {detail:{imageId:image.id}}));
+  }
+
+  function unmount() {
+    revision++;
+    if (!mounted) return;
+    const record = mounted;
+    mounted = null;
+    record.observer?.disconnect();
+    record.view?.dispose();
+    record.nav?.dispose();
+    record.entry.tabs.hidden = record.tabsHidden;
+    record.entry.content.hidden = record.contentHidden;
+    record.entry.search?.removeEventListener("input", record.onSearch, true);
+    record.entry.search?.removeEventListener("keydown", record.onSearchKey, true);
+    if (record.entry.search) record.entry.search.value = record.searchValue;
+    record.navigation.remove();
+    record.area.remove();
+  }
+
+  function error(record) {
+    record.message.textContent = text("Could not load local images.", "本地图片加载失败。");
+    record.retry.hidden = false;
+  }
+
+  async function fetchPage(record, reset = false) {
+    if (record.fetching && !reset) return;
+    const current = ++revision;
+    record.fetching = true;
+    record.retry.hidden = true;
+    if (reset) {
+      record.images = [];
+      record.offset = 0;
+      record.total = 0;
+      record.view.render({images:[], loading:true});
+    }
+    try {
+      const page = await config.requestHostAction("local-image-source", {
+        operation:"list", offset:record.offset, query:record.query,
+      });
+      if (signal.aborted || mounted !== record || revision !== current || !record.active) return;
+      record.images.push(...page.images);
+      record.offset += page.images.length;
+      record.total = page.total;
+      record.view.render({images:record.images, loading:false});
+      record.message.textContent = page.total ? "" : text("No local images found.", "没有找到本地图片。");
+      record.sentinel.hidden = record.offset >= record.total;
+    } catch {
+      if (mounted === record && revision === current) {
+        record.view.render({images:record.images, loading:false});
+        error(record);
+      }
+    } finally {
+      if (mounted === record && revision === current) record.fetching = false;
+    }
+  }
+
+  function activate(record, key) {
+    revision++;
+    record.view?.clearSelection();
+    record.active = key === "imagegen";
+    record.entry.content.hidden = record.active || record.contentHidden;
+    record.area.hidden = !record.active;
+    if (record.entry.search) record.entry.search.value = record.active ? record.query : record.searchValue;
+    record.nav.render(key);
+    if (record.active) void fetchPage(record, true);
+  }
+
+  async function scan() {
+    if (signal.aborted) return;
+    const entry = sources.read();
+    if (mounted && mounted.entry.tabs === entry?.tabs) {
+      if (!mounted.active) mounted.nav?.render(entry.selectedKey);
+      return;
+    }
+    unmount();
+    if (!entry) return;
+    const navigation = document.createElement("div");
+    navigation.dataset.minecodexImageNavigation = "true";
+    entry.tabs.parentElement.append(navigation);
+    const area = document.createElement("section");
+    area.className = entry.content.className;
+    area.id = "minecodex-image-gen-content";
+    area.hidden = true;
+    const container = document.createElement("div");
+    const message = document.createElement("p");
+    message.className = "text-sm text-secondary";
+    const retry = document.createElement("button");
+    retry.type = "button";
+    retry.className = "rounded-full bg-text/5 px-4 py-2 text-sm text-default";
+    retry.textContent = text("Try again", "重试");
+    retry.hidden = true;
+    const sentinel = document.createElement("div");
+    sentinel.style.height = "1px";
+    sentinel.hidden = true;
+    sentinel.setAttribute("aria-hidden", "true");
+    area.append(container, message, retry, sentinel);
+    entry.content.after(area);
+    const record = {
+      entry, navigation, area, message, retry, sentinel,
+      tabsHidden:entry.tabs.hidden, contentHidden:entry.content.hidden,
+      searchValue:entry.search?.value ?? "", images:[], offset:0, total:0,
+      query:"", fetching:false, active:false,
+    };
+    mounted = record;
+    try {
+      const results = await Promise.allSettled([
+        native.mountNavigation(navigation, entry.tabs, key => activate(record, key)),
+        native.mount(container, entry.tabs, {
+          onOpen:(image, open) => open(image), onPrompt:requestPrompt,
+          onHidden:ids=>{
+            const hidden=new Set(ids),removed=record.images.filter(image=>hidden.has(image.id)).length;
+            record.images=record.images.filter(image=>!hidden.has(image.id));
+            record.offset=Math.max(0,record.offset-removed);record.total=Math.max(0,record.total-removed);
+            record.sentinel.hidden=record.offset>=record.total;
+          },
+          onRetry:error => {
+            console.error("MineCodex native Images rendering",error);
+            if (mounted !== record) return;
+            unmount();
+            config.codex.report("images.local-source", false, "Native image rendering failed");
+          },
+        }),
+      ]);
+      // 任一原生能力失败时也释放已建立的另一棵 root。
+      if (results.some(result => result.status === "rejected")) {
+        for (const result of results) if (result.status === "fulfilled") result.value.dispose();
+        throw results.find(result => result.status === "rejected").reason;
+      }
+      const [nav, view] = results.map(result => result.value);
+      if (signal.aborted || mounted !== record) {
+        nav.dispose();
+        view.dispose();
+        return;
+      }
+      record.nav = nav;
+      record.view = view;
+      entry.tabs.hidden = true;
+      nav.render(entry.selectedKey);
+      retry.onclick = () => fetchPage(record, true);
+      record.observer = new IntersectionObserver(entries => {
+        if (entries.some(entry => entry.isIntersecting) && record.active && !record.fetching && record.offset < record.total) {
+          void fetchPage(record);
+        }
+      }, {rootMargin:"400px"});
+      record.observer.observe(sentinel);
+      record.onSearch = event => {
+        if (!record.active) return;
+        event.stopImmediatePropagation();
+        record.query = entry.search.value.slice(0, 256);
+        void fetchPage(record, true);
+      };
+      record.onSearchKey = event => {
+        if (record.active && event.key === "Enter") {
+          event.preventDefault();
+          event.stopImmediatePropagation();
+        }
+      };
+      entry.search?.addEventListener("input", record.onSearch, true);
+      entry.search?.addEventListener("keydown", record.onSearchKey, true);
+    } catch {
+      if (mounted === record) {
+        unmount();
+        config.codex.report("images.local-source", false, "Native image capability unavailable");
+      }
+    }
+  }
+
+  let scheduled = false;
+  const observer = new MutationObserver(() => {
+    if (scheduled) return;
+    scheduled = true;
+    queueMicrotask(() => {scheduled = false; void scan();});
+  });
+  observer.observe(document.body, {childList:true, subtree:true, attributes:true, attributeFilter:["aria-selected"]});
+  void scan();
+  const cleanup = () => {observer.disconnect(); unmount(); pendingPrompt = null;};
+  signal.addEventListener("abort", cleanup, {once:true});
+  return {
+    cleanup,
+    readViewer:header => native.readViewer(header),
+    pendingPrompt:() => pendingPrompt,
+    consumePrompt:() => {pendingPrompt = null;},
+  };
+}