ai-zero-token 2.0.1 → 2.0.3

package/dist/core/store/github-image-bed-store.js

@@ -0,0 +1,52 @@
+#!/usr/bin/env node
+import fs from "node:fs/promises";
+import path from "node:path";
+import { ensureStateMigrated, getStateDir } from "./state-paths.js";
+function createEmptyStore() {
+  return {
+    version: 1,
+    github: {
+      token: ""
+    }
+  };
+}
+async function loadGithubImageBedStore() {
+  try {
+    await ensureStateMigrated();
+    const raw = await fs.readFile(getGithubImageBedStorePath(), "utf8");
+    const parsed = JSON.parse(raw);
+    return {
+      version: 1,
+      github: {
+        token: typeof parsed.github?.token === "string" ? parsed.github.token.trim() : ""
+      }
+    };
+  } catch {
+    return createEmptyStore();
+  }
+}
+async function saveGithubImageBedStore(store) {
+  await ensureStateMigrated();
+  await fs.mkdir(getStateDir(), { recursive: true });
+  await fs.writeFile(getGithubImageBedStorePath(), `${JSON.stringify(store, null, 2)}
+`, "utf8");
+}
+async function updateGithubImageBedToken(token) {
+  const store = await loadGithubImageBedStore();
+  store.github.token = token.trim();
+  await saveGithubImageBedStore(store);
+  return store;
+}
+async function clearGithubImageBedStore() {
+  await saveGithubImageBedStore(createEmptyStore());
+}
+function getGithubImageBedStorePath() {
+  return path.join(getStateDir(), "github-image-bed.json");
+}
+export {
+  clearGithubImageBedStore,
+  getGithubImageBedStorePath,
+  loadGithubImageBedStore,
+  saveGithubImageBedStore,
+  updateGithubImageBedToken
+};

package/dist/core/store/profile-store.js

@@ -6,6 +6,7 @@ import {
   getStorePath
 } from "./state-paths.js";
 const PROFILE_CLAIM_PATH = "https://api.openai.com/profile";
+let storeMutationQueue = Promise.resolve();
 function createEmptyStore() {
   return {
     version: 1,
@@ -73,36 +74,50 @@ async function saveStore(store) {
   await fs.writeFile(getStorePath(), `${JSON.stringify(store, null, 2)}
 `, "utf8");
 }
+async function withStoreMutation(operation) {
+  const run = storeMutationQueue.then(operation, operation);
+  storeMutationQueue = run.then(
+    () => void 0,
+    () => void 0
+  );
+  return run;
+}
 async function saveProfile(profile) {
-  const store = await loadStore();
-  store.profiles[profile.profileId] = profile;
-  store.activeProfileId = profile.profileId;
-  await saveStore(store);
+  await withStoreMutation(async () => {
+    const store = await loadStore();
+    store.profiles[profile.profileId] = profile;
+    store.activeProfileId = profile.profileId;
+    await saveStore(store);
+  });
 }
 async function updateProfile(profileId, updater) {
-  const store = await loadStore();
-  const profile = store.profiles[profileId];
-  if (!profile) {
-    return null;
-  }
-  const updated = updater(profile);
-  store.profiles[profileId] = updated;
-  await saveStore(store);
-  return updated;
+  return withStoreMutation(async () => {
+    const store = await loadStore();
+    const profile = store.profiles[profileId];
+    if (!profile) {
+      return null;
+    }
+    const updated = updater(profile);
+    store.profiles[profileId] = updated;
+    await saveStore(store);
+    return updated;
+  });
 }
 async function listProfiles() {
   const store = await loadStore();
   return Object.values(store.profiles);
 }
 async function setActiveProfile(profileId) {
-  const store = await loadStore();
-  const profile = store.profiles[profileId];
-  if (!profile) {
-    return null;
-  }
-  store.activeProfileId = profileId;
-  await saveStore(store);
-  return profile;
+  return withStoreMutation(async () => {
+    const store = await loadStore();
+    const profile = store.profiles[profileId];
+    if (!profile) {
+      return null;
+    }
+    store.activeProfileId = profileId;
+    await saveStore(store);
+    return profile;
+  });
 }
 async function getActiveProfile() {
   const store = await loadStore();
@@ -114,19 +129,44 @@ async function getActiveProfile() {
   return first ?? null;
 }
 async function removeProfile(profileId) {
-  const store = await loadStore();
-  if (!store.profiles[profileId]) {
-    return null;
-  }
-  delete store.profiles[profileId];
-  if (store.activeProfileId === profileId) {
-    store.activeProfileId = Object.keys(store.profiles)[0];
-  }
-  await saveStore(store);
-  return store.activeProfileId ? store.profiles[store.activeProfileId] ?? null : null;
+  return withStoreMutation(async () => {
+    const store = await loadStore();
+    if (!store.profiles[profileId]) {
+      return null;
+    }
+    delete store.profiles[profileId];
+    if (store.activeProfileId === profileId) {
+      store.activeProfileId = Object.keys(store.profiles)[0];
+    }
+    await saveStore(store);
+    return store.activeProfileId ? store.profiles[store.activeProfileId] ?? null : null;
+  });
+}
+async function removeProfiles(profileIds) {
+  return withStoreMutation(async () => {
+    const idSet = new Set(profileIds.map((id) => id.trim()).filter(Boolean));
+    if (idSet.size === 0) {
+      return 0;
+    }
+    const store = await loadStore();
+    let removed = 0;
+    for (const profileId of idSet) {
+      if (store.profiles[profileId]) {
+        delete store.profiles[profileId];
+        removed += 1;
+      }
+    }
+    if (store.activeProfileId && !store.profiles[store.activeProfileId]) {
+      store.activeProfileId = Object.keys(store.profiles)[0];
+    }
+    await saveStore(store);
+    return removed;
+  });
 }
 async function clearStore() {
-  await fs.rm(getStateDir(), { recursive: true, force: true });
+  await withStoreMutation(async () => {
+    await fs.rm(getStateDir(), { recursive: true, force: true });
+  });
 }
 export {
   clearStore,
@@ -136,6 +176,7 @@ export {
   listProfiles,
   loadStore,
   removeProfile,
+  removeProfiles,
   saveProfile,
   saveStore,
   setActiveProfile,

package/dist/core/store/settings-store.js

@@ -18,6 +18,9 @@ function createDefaultSettings() {
     autoSwitch: {
       enabled: false
     },
+    runtime: {
+      quotaSyncConcurrency: 16
+    },
     server: {
       host: "0.0.0.0",
       port: 8787
@@ -42,6 +45,9 @@ async function loadSettings() {
       autoSwitch: {
         enabled: parsed.autoSwitch?.enabled ?? defaults.autoSwitch.enabled
       },
+      runtime: {
+        quotaSyncConcurrency: normalizeQuotaSyncConcurrency(parsed.runtime?.quotaSyncConcurrency, defaults.runtime.quotaSyncConcurrency)
+      },
       server: {
         host: parsed.server?.host ?? defaults.server.host,
         port: parsed.server?.port ?? defaults.server.port
@@ -51,6 +57,13 @@ async function loadSettings() {
     return createDefaultSettings();
   }
 }
+function normalizeQuotaSyncConcurrency(value, fallback = 16) {
+  const parsed = typeof value === "number" ? value : typeof value === "string" ? Number.parseInt(value, 10) : fallback;
+  if (!Number.isFinite(parsed)) {
+    return fallback;
+  }
+  return Math.min(32, Math.max(1, Math.trunc(parsed)));
+}
 async function saveSettings(settings) {
   await ensureStateMigrated();
   await fs.mkdir(getStateDir(), { recursive: true });
@@ -61,5 +74,6 @@ export {
   createDefaultSettings,
   getSettingsPath,
   loadSettings,
+  normalizeQuotaSyncConcurrency,
   saveSettings
 };

package/dist/desktop/main.js

@@ -1,13 +1,18 @@
 #!/usr/bin/env node
 import { app as electronApp, BrowserWindow, dialog, shell } from "electron";
+import { readFileSync } from "node:fs";
 import path from "node:path";
 import { fileURLToPath } from "node:url";
 import { startServer } from "../server/index.js";
 let gatewayServer = null;
 let mainWindow = null;
 let isQuitting = false;
+let isRestarting = false;
+let currentGatewayUrl = null;
+let currentAdminUrl = null;
 const desktopDir = path.dirname(fileURLToPath(import.meta.url));
 const appIconPath = path.resolve(desktopDir, "../../build/icon.png");
+const startupPageUrl = buildStartupPageUrl("\u6B63\u5728\u542F\u52A8\u672C\u5730\u7F51\u5173");
 electronApp.setName("AI Zero Token");
 function createBrowserUrl(host, port) {
   if (host === "0.0.0.0" || host === "::") {
@@ -28,11 +33,143 @@ function resolveAdminUrl(gatewayUrl) {
   const devUrl = process.env.AZT_ADMIN_UI_DEV_URL?.trim();
   return devUrl || gatewayUrl;
 }
+function resolvePreferredGatewayParams() {
+  const devUrl = process.env.AZT_DEV_GATEWAY_URL?.trim();
+  if (!devUrl) {
+    return void 0;
+  }
+  try {
+    const parsed = new URL(devUrl);
+    const host = parsed.hostname || void 0;
+    const port = Number.parseInt(parsed.port, 10);
+    return {
+      host,
+      port: Number.isFinite(port) ? port : void 0
+    };
+  } catch {
+    return void 0;
+  }
+}
+function isAllowedAppUrl(targetUrl) {
+  return Boolean(currentAdminUrl && isGatewayUrl(targetUrl, currentAdminUrl) || currentGatewayUrl && isGatewayUrl(targetUrl, currentGatewayUrl));
+}
+async function restartGateway() {
+  if (isRestarting) {
+    return;
+  }
+  isRestarting = true;
+  try {
+    if (mainWindow && !mainWindow.isDestroyed()) {
+      await mainWindow.loadURL(buildStartupPageUrl("\u6B63\u5728\u91CD\u542F\u672C\u5730\u7F51\u5173"));
+    }
+    await closeGatewayServer().catch((error) => {
+      console.error("[desktop:gateway:restart-close]", error);
+    });
+    const server = await ensureGatewayServer();
+    const gatewayUrl = createBrowserUrl(server.host, server.port);
+    const adminUrl = resolveAdminUrl(gatewayUrl);
+    currentGatewayUrl = gatewayUrl;
+    currentAdminUrl = adminUrl;
+    if (mainWindow && !mainWindow.isDestroyed()) {
+      await mainWindow.loadURL(adminUrl);
+    }
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    console.error("[desktop:gateway:restart]", error);
+    if (mainWindow && !mainWindow.isDestroyed()) {
+      await mainWindow.loadURL(buildStartupPageUrl(`\u7F51\u5173\u91CD\u542F\u5931\u8D25\uFF1A${message}`)).catch(() => void 0);
+    }
+    dialog.showErrorBox("AI Zero Token \u7F51\u5173\u91CD\u542F\u5931\u8D25", message);
+  } finally {
+    isRestarting = false;
+  }
+}
+function buildStartupPageUrl(subtitle) {
+  const iconUrl = `data:image/png;base64,${readFileSync(appIconPath).toString("base64")}`;
+  const html = `<!doctype html>
+<html lang="zh-CN">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <style>
+      :root { color-scheme: dark; }
+      html, body {
+        width: 100%;
+        height: 100%;
+        margin: 0;
+        background: #050816;
+        color: #f8fafc;
+        font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", sans-serif;
+      }
+      body {
+        display: grid;
+        place-items: center;
+      }
+      .wrap {
+        display: grid;
+        gap: 18px;
+        justify-items: center;
+      }
+      .mark {
+        width: 96px;
+        height: 96px;
+        border-radius: 24px;
+        box-shadow: 0 18px 60px rgba(0, 0, 0, 0.35);
+      }
+      .title {
+        font-size: 22px;
+        font-weight: 700;
+        letter-spacing: 0;
+      }
+      .sub {
+        font-size: 14px;
+        color: rgba(248, 250, 252, 0.66);
+      }
+      .bar {
+        width: 220px;
+        height: 4px;
+        border-radius: 999px;
+        background: rgba(255,255,255,0.08);
+        overflow: hidden;
+      }
+      .bar::after {
+        content: "";
+        display: block;
+        width: 45%;
+        height: 100%;
+        border-radius: inherit;
+        background: linear-gradient(90deg, #93c5fd 0%, #66f0c0 55%, #f97316 100%);
+        animation: load 1.2s ease-in-out infinite;
+      }
+      @keyframes load {
+        0% { transform: translateX(-30%); }
+        50% { transform: translateX(80%); }
+        100% { transform: translateX(-30%); }
+      }
+    </style>
+  </head>
+  <body>
+    <div class="wrap">
+      <img class="mark" src="${iconUrl}" alt="" />
+      <div class="title">AI Zero Token</div>
+      <div class="sub">${escapeHtml(subtitle)}</div>
+      <div class="bar" aria-hidden="true"></div>
+    </div>
+  </body>
+</html>`;
+  return `data:text/html;charset=utf-8,${encodeURIComponent(html)}`;
+}
+function escapeHtml(value) {
+  return value.replaceAll("&", "&amp;").replaceAll("<", "&lt;").replaceAll(">", "&gt;").replaceAll('"', "&quot;");
+}
 async function ensureGatewayServer() {
   if (gatewayServer) {
     return gatewayServer;
   }
-  gatewayServer = await startServer();
+  gatewayServer = await startServer({
+    ...resolvePreferredGatewayParams(),
+    onRestart: restartGateway
+  });
   const adminUrl = createBrowserUrl(gatewayServer.host, gatewayServer.port);
   console.log("AI Zero Token desktop gateway started.");
   console.log(`admin: ${adminUrl}`);
@@ -41,9 +178,6 @@ async function ensureGatewayServer() {
   return gatewayServer;
 }
 async function createMainWindow() {
-  const server = await ensureGatewayServer();
-  const gatewayUrl = createBrowserUrl(server.host, server.port);
-  const adminUrl = resolveAdminUrl(gatewayUrl);
   mainWindow = new BrowserWindow({
     width: 1440,
     height: 960,
@@ -51,7 +185,8 @@ async function createMainWindow() {
     minHeight: 720,
     title: "AI Zero Token",
     icon: appIconPath,
-    backgroundColor: "#f8fafc",
+    backgroundColor: "#050816",
+    show: false,
     webPreferences: {
       contextIsolation: true,
       nodeIntegration: false,
@@ -63,7 +198,7 @@ async function createMainWindow() {
     return { action: "deny" };
   });
   mainWindow.webContents.on("will-navigate", (event, url) => {
-    if (isGatewayUrl(url, adminUrl) || isGatewayUrl(url, gatewayUrl)) {
+    if (isAllowedAppUrl(url)) {
       return;
     }
     event.preventDefault();
@@ -72,6 +207,23 @@ async function createMainWindow() {
   mainWindow.on("closed", () => {
     mainWindow = null;
   });
+  mainWindow.once("ready-to-show", () => {
+    if (mainWindow) {
+      mainWindow.show();
+    }
+  });
+  await mainWindow.loadURL(startupPageUrl);
+  if (mainWindow && !mainWindow.isVisible()) {
+    mainWindow.show();
+  }
+  const server = await ensureGatewayServer();
+  if (!mainWindow) {
+    return;
+  }
+  const gatewayUrl = createBrowserUrl(server.host, server.port);
+  const adminUrl = resolveAdminUrl(gatewayUrl);
+  currentGatewayUrl = gatewayUrl;
+  currentAdminUrl = adminUrl;
   await mainWindow.loadURL(adminUrl);
 }
 function focusMainWindow() {