ai-zero-token 2.0.2 → 2.0.4

package/dist/core/services/config-service.js

@@ -3,8 +3,34 @@ import { getPreferredCodexModel, hasCodexModel } from "../models/openai-codex-mo
 import {
   createDefaultSettings,
   loadSettings,
+  normalizeQuotaSyncConcurrency,
   saveSettings
 } from "../store/settings-store.js";
+function normalizeNetworkProxy(settings, params) {
+  const requestedUrl = params.url?.trim() ?? "";
+  const url = requestedUrl || (!params.enabled ? settings.networkProxy.url : "");
+  const noProxy = params.noProxy?.trim() || settings.networkProxy.noProxy || "localhost,127.0.0.1,::1";
+  if (params.enabled) {
+    if (!url) {
+      throw new Error("\u542F\u7528\u4EE3\u7406\u65F6\u5FC5\u987B\u586B\u5199\u4EE3\u7406\u5730\u5740\u3002");
+    }
+    let parsed;
+    try {
+      parsed = new URL(url);
+    } catch {
+      throw new Error("\u4EE3\u7406\u5730\u5740\u683C\u5F0F\u9519\u8BEF\uFF0C\u8BF7\u586B\u5199\u5B8C\u6574\u7684\u4EE3\u7406 URL\u3002");
+    }
+    const supportedProtocols = /* @__PURE__ */ new Set(["http:", "https:", "socks4:", "socks4a:", "socks5:", "socks5h:"]);
+    if (!supportedProtocols.has(parsed.protocol)) {
+      throw new Error("\u4EE3\u7406\u5730\u5740\u4EC5\u652F\u6301 http\u3001https\u3001socks4\u3001socks4a\u3001socks5 \u6216 socks5h\u3002");
+    }
+  }
+  return {
+    enabled: params.enabled,
+    url,
+    noProxy
+  };
+}
 class ConfigService {
   async getSettings() {
     return this.ensureSettings();
@@ -43,31 +69,9 @@ class ConfigService {
   }
   async setNetworkProxy(params) {
     const settings = await this.getSettings();
-    const requestedUrl = params.url?.trim() ?? "";
-    const url = requestedUrl || (!params.enabled ? settings.networkProxy.url : "");
-    const noProxy = params.noProxy?.trim() || settings.networkProxy.noProxy || "localhost,127.0.0.1,::1";
-    if (params.enabled) {
-      if (!url) {
-        throw new Error("\u542F\u7528\u4EE3\u7406\u65F6\u5FC5\u987B\u586B\u5199\u4EE3\u7406\u5730\u5740\u3002");
-      }
-      let parsed;
-      try {
-        parsed = new URL(url);
-      } catch {
-        throw new Error("\u4EE3\u7406\u5730\u5740\u683C\u5F0F\u9519\u8BEF\uFF0C\u8BF7\u586B\u5199\u5B8C\u6574\u7684\u4EE3\u7406 URL\u3002");
-      }
-      const supportedProtocols = /* @__PURE__ */ new Set(["http:", "https:", "socks4:", "socks4a:", "socks5:", "socks5h:"]);
-      if (!supportedProtocols.has(parsed.protocol)) {
-        throw new Error("\u4EE3\u7406\u5730\u5740\u4EC5\u652F\u6301 http\u3001https\u3001socks4\u3001socks4a\u3001socks5 \u6216 socks5h\u3002");
-      }
-    }
     const next = {
       ...settings,
-      networkProxy: {
-        enabled: params.enabled,
-        url,
-        noProxy
-      }
+      networkProxy: normalizeNetworkProxy(settings, params)
     };
     await saveSettings(next);
     return next;
@@ -83,6 +87,18 @@ class ConfigService {
     await saveSettings(next);
     return next;
   }
+  async setRuntimeConfig(params) {
+    const settings = await this.getSettings();
+    const next = {
+      ...settings,
+      runtime: {
+        ...settings.runtime,
+        quotaSyncConcurrency: normalizeQuotaSyncConcurrency(params.quotaSyncConcurrency, settings.runtime.quotaSyncConcurrency)
+      }
+    };
+    await saveSettings(next);
+    return next;
+  }
   async getServerConfig() {
     const settings = await this.getSettings();
     return settings.server;
@@ -99,6 +115,54 @@ class ConfigService {
     await saveSettings(next);
     return next;
   }
+  async updateSettings(params) {
+    const settings = await this.getSettings();
+    let next = { ...settings };
+    if (params.defaultModel) {
+      if (!await hasCodexModel(params.defaultModel)) {
+        throw new Error(`\u5F53\u524D\u7F51\u5173\u672A\u627E\u5230\u53EF\u7528\u6A21\u578B: ${params.defaultModel}`);
+      }
+      next = {
+        ...next,
+        defaultProvider: "openai-codex",
+        defaultModel: params.defaultModel
+      };
+    }
+    if (params.networkProxy) {
+      next = {
+        ...next,
+        networkProxy: normalizeNetworkProxy(next, params.networkProxy)
+      };
+    }
+    if (params.autoSwitch) {
+      next = {
+        ...next,
+        autoSwitch: {
+          enabled: params.autoSwitch.enabled
+        }
+      };
+    }
+    if (params.runtime) {
+      next = {
+        ...next,
+        runtime: {
+          ...next.runtime,
+          quotaSyncConcurrency: normalizeQuotaSyncConcurrency(params.runtime.quotaSyncConcurrency, next.runtime.quotaSyncConcurrency)
+        }
+      };
+    }
+    if (params.server) {
+      next = {
+        ...next,
+        server: {
+          ...next.server,
+          port: params.server.port
+        }
+      };
+    }
+    await saveSettings(next);
+    return next;
+  }
   async resetSettings() {
     const defaults = createDefaultSettings();
     await saveSettings(defaults);

package/dist/core/services/version-service.js

@@ -56,7 +56,7 @@ class VersionService {
     const manifest = await readPackageManifest();
     const registryUrl = `https://registry.npmjs.org/${encodeURIComponent(manifest.name)}/latest`;
     try {
-      const latestVersion = await this.fetchNpmLatestVersion(registryUrl);
+      const latestVersion = process.env.AZT_FORCE_LATEST_VERSION || await this.fetchNpmLatestVersion(registryUrl);
       const needsUpdate = compareSemver(manifest.version, latestVersion) < 0;
       return {
         packageName: manifest.name,

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
 };