@musnows/scriverse 0.6.3 → 0.6.5

package/dist/google-vertex-auth.js

@@ -0,0 +1,155 @@
+import { createSign } from "node:crypto";
+import { AppError } from "./errors.js";
+export const GOOGLE_OAUTH_TOKEN_URL = "https://oauth2.googleapis.com/token";
+export const GOOGLE_CLOUD_PLATFORM_SCOPE = "https://www.googleapis.com/auth/cloud-platform";
+const TOKEN_EXPIRY_SKEW_MS = 60_000;
+const JWT_LIFETIME_SECONDS = 3_600;
+const GOOGLE_VERTEX_REGIONAL_HOST_PATTERN = /^[a-z0-9](?:[a-z0-9-]*[a-z0-9])?-aiplatform\.googleapis\.com$/u;
+export function isOfficialGoogleVertexBaseUrl(raw) {
+    try {
+        const url = new URL(raw);
+        const hostname = url.hostname.toLowerCase();
+        return url.protocol === "https:"
+            && !url.username
+            && !url.password
+            && (url.port === "" || url.port === "443")
+            && (hostname === "aiplatform.googleapis.com" || GOOGLE_VERTEX_REGIONAL_HOST_PATTERN.test(hostname));
+    }
+    catch {
+        return false;
+    }
+}
+export function assertOfficialGoogleVertexBaseUrl(raw) {
+    if (!isOfficialGoogleVertexBaseUrl(raw)) {
+        throw new AppError(400, "INVALID_VERTEX_BASE_URL", "Google Vertex 接口地址必须使用 aiplatform.googleapis.com 或官方区域 *-aiplatform.googleapis.com 域名");
+    }
+}
+function base64UrlEncode(value) {
+    const buffer = typeof value === "string" ? Buffer.from(value, "utf8") : value;
+    return buffer.toString("base64url");
+}
+function readNonEmptyString(record, key) {
+    const value = record[key];
+    if (typeof value !== "string" || !value.trim()) {
+        throw new AppError(400, "INVALID_SERVICE_ACCOUNT", `服务账号 JSON 缺少有效字段：${key}`);
+    }
+    return value.trim();
+}
+export function parseGoogleServiceAccount(raw) {
+    let parsed;
+    try {
+        parsed = JSON.parse(raw);
+    }
+    catch {
+        throw new AppError(400, "INVALID_SERVICE_ACCOUNT", "服务账号 JSON 无法解析");
+    }
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+        throw new AppError(400, "INVALID_SERVICE_ACCOUNT", "服务账号 JSON 必须是对象");
+    }
+    const record = parsed;
+    if (record.type !== "service_account") {
+        throw new AppError(400, "INVALID_SERVICE_ACCOUNT", "服务账号 JSON 的 type 必须为 service_account");
+    }
+    const clientEmail = readNonEmptyString(record, "client_email");
+    const privateKey = readNonEmptyString(record, "private_key");
+    const projectId = typeof record.project_id === "string" && record.project_id.trim()
+        ? record.project_id.trim()
+        : undefined;
+    return {
+        type: "service_account",
+        client_email: clientEmail,
+        private_key: privateKey,
+        ...(projectId ? { project_id: projectId } : {})
+    };
+}
+export function maskServiceAccountHint(account) {
+    return `sa:${account.client_email}`;
+}
+export function createGoogleServiceAccountJwt(account, nowSeconds = Math.floor(Date.now() / 1000)) {
+    const header = base64UrlEncode(JSON.stringify({ alg: "RS256", typ: "JWT" }));
+    const payload = base64UrlEncode(JSON.stringify({
+        iss: account.client_email,
+        sub: account.client_email,
+        aud: GOOGLE_OAUTH_TOKEN_URL,
+        iat: nowSeconds,
+        exp: nowSeconds + JWT_LIFETIME_SECONDS,
+        scope: GOOGLE_CLOUD_PLATFORM_SCOPE
+    }));
+    const unsigned = `${header}.${payload}`;
+    const signer = createSign("RSA-SHA256");
+    signer.update(unsigned);
+    signer.end();
+    let signature;
+    try {
+        signature = signer.sign(account.private_key);
+    }
+    catch {
+        throw new AppError(400, "INVALID_SERVICE_ACCOUNT", "服务账号私钥无法用于签名");
+    }
+    return `${unsigned}.${base64UrlEncode(signature)}`;
+}
+export async function exchangeGoogleServiceAccountToken(account, fetchToken, nowSeconds = Math.floor(Date.now() / 1000)) {
+    const jwt = createGoogleServiceAccountJwt(account, nowSeconds);
+    return fetchToken(jwt);
+}
+export class GoogleVertexTokenCache {
+    tokens = new Map();
+    clear(cacheKey) {
+        if (cacheKey)
+            this.tokens.delete(cacheKey);
+        else
+            this.tokens.clear();
+    }
+    async getAccessToken(cacheKey, account, fetchToken, nowMs = Date.now()) {
+        const cached = this.tokens.get(cacheKey);
+        if (cached && cached.expiresAtMs - TOKEN_EXPIRY_SKEW_MS > nowMs) {
+            return cached.accessToken;
+        }
+        const exchanged = await exchangeGoogleServiceAccountToken(account, fetchToken, Math.floor(nowMs / 1000));
+        const expiresInSeconds = Number.isFinite(exchanged.expiresInSeconds) && exchanged.expiresInSeconds > 0
+            ? exchanged.expiresInSeconds
+            : JWT_LIFETIME_SECONDS;
+        this.tokens.set(cacheKey, {
+            accessToken: exchanged.accessToken,
+            expiresAtMs: nowMs + expiresInSeconds * 1_000
+        });
+        return exchanged.accessToken;
+    }
+}
+export async function fetchGoogleOAuthAccessToken(jwt, outboundFetch) {
+    const response = await outboundFetch(GOOGLE_OAUTH_TOKEN_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/x-www-form-urlencoded",
+            Accept: "application/json"
+        },
+        body: new URLSearchParams({
+            grant_type: "urn:ietf:params:oauth:grant-type:jwt-bearer",
+            assertion: jwt
+        }).toString()
+    });
+    const body = await response.text();
+    if (!response.ok) {
+        throw new AppError(502, "VERTEX_TOKEN_EXCHANGE_FAILED", `Google OAuth 换票失败：HTTP ${response.status}`);
+    }
+    let payload;
+    try {
+        payload = JSON.parse(body);
+    }
+    catch {
+        throw new AppError(502, "VERTEX_TOKEN_EXCHANGE_FAILED", "Google OAuth 换票返回了无效 JSON");
+    }
+    if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
+        throw new AppError(502, "VERTEX_TOKEN_EXCHANGE_FAILED", "Google OAuth 换票返回了无效响应");
+    }
+    const record = payload;
+    const accessToken = typeof record.access_token === "string" ? record.access_token.trim() : "";
+    if (!accessToken) {
+        throw new AppError(502, "VERTEX_TOKEN_EXCHANGE_FAILED", "Google OAuth 换票响应缺少 access_token");
+    }
+    const expiresInSeconds = typeof record.expires_in === "number" && Number.isFinite(record.expires_in)
+        ? record.expires_in
+        : JWT_LIFETIME_SECONDS;
+    return { accessToken, expiresInSeconds };
+}
+//# sourceMappingURL=google-vertex-auth.js.map

package/dist/google-vertex-auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-vertex-auth.js","sourceRoot":"","sources":["../src/google-vertex-auth.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,MAAM,CAAC,MAAM,sBAAsB,GAAG,qCAAqC,CAAC;AAC5E,MAAM,CAAC,MAAM,2BAA2B,GAAG,gDAAgD,CAAC;AAE5F,MAAM,oBAAoB,GAAG,MAAM,CAAC;AACpC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AACnC,MAAM,mCAAmC,GAAG,gEAAgE,CAAC;AAc7G,MAAM,UAAU,6BAA6B,CAAC,GAAW;IACvD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QAC5C,OAAO,GAAG,CAAC,QAAQ,KAAK,QAAQ;eAC3B,CAAC,GAAG,CAAC,QAAQ;eACb,CAAC,GAAG,CAAC,QAAQ;eACb,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,CAAC;eACvC,CAAC,QAAQ,KAAK,2BAA2B,IAAI,mCAAmC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IACxG,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,GAAW;IAC3D,IAAI,CAAC,6BAA6B,CAAC,GAAG,CAAC,EAAE,CAAC;QACxC,MAAM,IAAI,QAAQ,CAChB,GAAG,EACH,yBAAyB,EACzB,uFAAuF,CACxF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAsB;IAC7C,MAAM,MAAM,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAC9E,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA+B,EAAE,GAAW;IACtE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;QAC/C,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,yBAAyB,EAAE,oBAAoB,GAAG,EAAE,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,GAAW;IACnD,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;IACtC,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,yBAAyB,EAAE,gBAAgB,CAAC,CAAC;IACvE,CAAC;IACD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,yBAAyB,EAAE,iBAAiB,CAAC,CAAC;IACxE,CAAC;IACD,MAAM,MAAM,GAAG,MAAiC,CAAC;IACjD,IAAI,MAAM,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QACtC,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,yBAAyB,EAAE,sCAAsC,CAAC,CAAC;IAC7F,CAAC;IACD,MAAM,WAAW,GAAG,kBAAkB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IAC7D,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE;QACjF,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE;QAC1B,CAAC,CAAC,SAAS,CAAC;IACd,OAAO;QACL,IAAI,EAAE,iBAAiB;QACvB,YAAY,EAAE,WAAW;QACzB,WAAW,EAAE,UAAU;QACvB,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAA6B;IAClE,OAAO,MAAM,OAAO,CAAC,YAAY,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,OAA6B,EAC7B,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IAE1C,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAC7E,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC;QAC7C,GAAG,EAAE,OAAO,CAAC,YAAY;QACzB,GAAG,EAAE,OAAO,CAAC,YAAY;QACzB,GAAG,EAAE,sBAAsB;QAC3B,GAAG,EAAE,UAAU;QACf,GAAG,EAAE,UAAU,GAAG,oBAAoB;QACtC,KAAK,EAAE,2BAA2B;KACnC,CAAC,CAAC,CAAC;IACJ,MAAM,QAAQ,GAAG,GAAG,MAAM,IAAI,OAAO,EAAE,CAAC;IACxC,MAAM,MAAM,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;IACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxB,MAAM,CAAC,GAAG,EAAE,CAAC;IACb,IAAI,SAAiB,CAAC;IACtB,IAAI,CAAC;QACH,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,yBAAyB,EAAE,cAAc,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,GAAG,QAAQ,IAAI,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;AACrD,CAAC;AAID,MAAM,CAAC,KAAK,UAAU,iCAAiC,CACrD,OAA6B,EAC7B,UAA8B,EAC9B,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IAE1C,MAAM,GAAG,GAAG,6BAA6B,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IAC/D,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,OAAO,sBAAsB;IAChB,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEzD,KAAK,CAAC,QAAiB;QACrB,IAAI,QAAQ;YAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;;YACtC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,QAAgB,EAChB,OAA6B,EAC7B,UAA8B,EAC9B,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE;QAElB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,MAAM,IAAI,MAAM,CAAC,WAAW,GAAG,oBAAoB,GAAG,KAAK,EAAE,CAAC;YAChE,OAAO,MAAM,CAAC,WAAW,CAAC;QAC5B,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,iCAAiC,CACvD,OAAO,EACP,UAAU,EACV,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,CACzB,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,gBAAgB,CAAC,IAAI,SAAS,CAAC,gBAAgB,GAAG,CAAC;YACpG,CAAC,CAAC,SAAS,CAAC,gBAAgB;YAC5B,CAAC,CAAC,oBAAoB,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE;YACxB,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,WAAW,EAAE,KAAK,GAAG,gBAAgB,GAAG,KAAK;SAC9C,CAAC,CAAC;QACH,OAAO,SAAS,CAAC,WAAW,CAAC;IAC/B,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,GAAW,EACX,aAAoE;IAEpE,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,sBAAsB,EAAE;QAC3D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,mCAAmC;YACnD,MAAM,EAAE,kBAAkB;SAC3B;QACD,IAAI,EAAE,IAAI,eAAe,CAAC;YACxB,UAAU,EAAE,6CAA6C;YACzD,SAAS,EAAE,GAAG;SACf,CAAC,CAAC,QAAQ,EAAE;KACd,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,8BAA8B,EAAE,0BAA0B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACvG,CAAC;IACD,IAAI,OAAgB,CAAC;IACrB,IAAI,CAAC;QACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;IACxC,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,8BAA8B,EAAE,2BAA2B,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,8BAA8B,EAAE,wBAAwB,CAAC,CAAC;IACpF,CAAC;IACD,MAAM,MAAM,GAAG,OAAkC,CAAC;IAClD,MAAM,WAAW,GAAG,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9F,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,8BAA8B,EAAE,kCAAkC,CAAC,CAAC;IAC9F,CAAC;IACD,MAAM,gBAAgB,GAAG,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC;QAClG,CAAC,CAAC,MAAM,CAAC,UAAU;QACnB,CAAC,CAAC,oBAAoB,CAAC;IACzB,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,CAAC;AAC3C,CAAC"}

package/dist/public/ai-mentions.js

@@ -12,6 +12,7 @@ export function findAiMention(value, cursor = value.length) {
 export function listAiMentionOptions(characters, settings, chapters, query = "", limit = 12) {
   const keyword = String(query).trim().toLocaleLowerCase("zh-CN");
   const groups = [
+    [{ kind: "context-settings", kindLabel: "能力", id: "include-setting-info", name: "注入上下文设定" }],
     characters.map((item) => ({ kind: "character", kindLabel: "角色", id: String(item.id), name: String(item.name) })),
     settings.map((item) => ({ kind: "setting", kindLabel: "设定", id: String(item.id), name: String(item.title) })),
     chapters.map((item) => ({
@@ -47,9 +48,22 @@ export function buildAiReferenceScope(references) {
   const chapterIds = [...new Set(references.filter((item) => item.kind === "chapter").map((item) => item.id))];
   const characterIds = [...new Set(references.filter((item) => item.kind === "character").map((item) => item.id))];
   const settingIds = [...new Set(references.filter((item) => item.kind === "setting").map((item) => item.id))];
+  const includeSettingInfo = references.some((item) => item.kind === "context-settings" && item.id === "include-setting-info");
   return {
     ...(chapterIds.length ? { chapterIds } : {}),
     ...(characterIds.length ? { characterIds } : {}),
-    ...(settingIds.length ? { settingIds } : {})
+    ...(settingIds.length ? { settingIds } : {}),
+    ...(includeSettingInfo ? { includeSettingInfo: true } : {})
   };
 }
+
+export function mergeAiReferenceScope(scope, references) {
+  const merged = { ...scope };
+  const referenceScope = buildAiReferenceScope(references);
+  for (const key of ["chapterIds", "characterIds", "settingIds"]) {
+    const values = [...new Set([...(Array.isArray(scope?.[key]) ? scope[key] : []), ...(referenceScope[key] ?? [])])];
+    if (values.length) merged[key] = values;
+  }
+  if (referenceScope.includeSettingInfo === true) merged.includeSettingInfo = true;
+  return merged;
+}