@musnows/scriverse 0.6.2 → 0.6.3

package/dist/app.js

@@ -21,7 +21,7 @@ import { applyImportFileHints, parseNovelText } from "./parser.js";
 import { attachmentPermissionModules, Store, versionedEntityTypes } from "./store.js";
 import { paginated, parsePagination } from "./pagination.js";
 import { normalizeUploadFileName } from "./utils.js";
-import { assertSafeAiEndpoint, createApiRateLimitMiddleware, createAuthenticationRateLimitMiddleware, createBasicAuthMiddleware, createSameOriginMiddleware, createSecurityHeadersMiddleware, createUploadRateLimitMiddleware, verifySetupToken } from "./security.js";
+import { assertSafeAiEndpoint, createApiRateLimitMiddleware, createAuthenticationRateLimitMiddleware, createBasicAuthMiddleware, createCaptchaRateLimitMiddleware, createExpensiveApiRateLimitMiddleware, createSameOriginMiddleware, createSecurityHeadersMiddleware, createUploadRateLimitMiddleware, enforceCaseInsensitiveRouting, normalizeApiPath, resolveTrustProxySetting, verifySetupToken } from "./security.js";
 import { ImageCaptchaService } from "./image-captcha.js";
 import { assertSafeImportedPlainText, decodeUtf8ImportedText } from "./import-security.js";
 import { InvalidRasterImageError, readRasterImageMetadata } from "./image-metadata.js";
@@ -379,6 +379,7 @@ const aiProcessStepSchema = z.discriminatedUnion("type", [
 ]);
 const workAiSettingsSchema = z.object({
     systemPrompt: z.string().max(100_000).optional(),
+    dailyTokenQuota: z.number().int().min(10_000).max(2_000_000_000).nullable().optional(),
     autoRunEnabled: z.boolean().optional(),
     autoRunConcurrency: z.number().int().min(1).max(8).optional(),
     autoRunBatchLimit: z.number().int().min(1).max(200).optional(),
@@ -386,6 +387,8 @@ const workAiSettingsSchema = z.object({
     autoRunFailureThreshold: z.number().int().min(1).max(10).optional(),
     bookSummaryContextPercent: z.number().int().min(1).max(90).optional(),
     contextCompactThreshold: z.number().int().min(50).max(90).optional(),
+    agentToolCallLimit: z.number().int().min(5).max(48).optional(),
+    agentToolCallGlobalMultiplier: z.number().int().min(1).max(6).optional(),
     agentTools: z.array(z.enum(["story_index", "read_chapters", "grep", "search_story_entities", "read_character_sections", "search_drafts"])).max(6).optional(),
     titleGenerationModelId: z.string().trim().max(200).optional()
 }).strict();
@@ -626,6 +629,88 @@ function redactAiCallContext(record, permissions) {
     result.contextScope = restricted ? { ...redactedScope, restricted: true } : redactedScope;
     return result;
 }
+const proseRestrictedPlaceholder = "（正文读取权限受限）";
+function redactContinuationGuard(record, permissions) {
+    if (permissions.prose !== "none")
+        return record;
+    return {
+        ...record,
+        issues: [],
+        contextRefs: {},
+        failure: null,
+        restricted: true
+    };
+}
+/** 无正文读取权限时移除建议中的原文、指令和检查证据，避免通过 AI 接口绕过 prose=none。 */
+function redactSuggestion(record, permissions) {
+    if (permissions.prose !== "none")
+        return record;
+    const guard = recordValue(record.guard);
+    return {
+        ...record,
+        instruction: proseRestrictedPlaceholder,
+        sourceText: "",
+        ...(guard ? { guard: redactContinuationGuard(guard, permissions) } : {}),
+        restricted: true
+    };
+}
+function redactAiConversationMessage(item, permissions) {
+    if (permissions.prose !== "none")
+        return item;
+    const message = recordValue(item);
+    if (!message)
+        return item;
+    return {
+        ...message,
+        content: proseRestrictedPlaceholder,
+        citations: [],
+        metadata: { restricted: true },
+        restricted: true
+    };
+}
+/** 无正文读取权限时隐藏对话预览与消息正文，避免历史对话泄露章节原文。 */
+function redactAiConversation(record, permissions) {
+    if (permissions.prose !== "none")
+        return record;
+    const result = {
+        ...record,
+        title: proseRestrictedPlaceholder
+    };
+    if (typeof result.preview === "string" && result.preview.length > 0) {
+        result.preview = proseRestrictedPlaceholder;
+    }
+    if (Array.isArray(result.messages)) {
+        result.messages = result.messages.map((item) => redactAiConversationMessage(item, permissions));
+    }
+    const messagesPage = recordValue(result.messagesPage);
+    if (messagesPage && Array.isArray(messagesPage.items)) {
+        result.messagesPage = {
+            ...messagesPage,
+            items: messagesPage.items.map((item) => redactAiConversationMessage(item, permissions))
+        };
+    }
+    return { ...result, restricted: true };
+}
+/** SSE 错误事件只暴露 AppError 的公开信息，避免透传内部异常 message。 */
+export function publicAiStreamError(error) {
+    if (error instanceof AppError) {
+        const details = error.details && typeof error.details === "object" && !Array.isArray(error.details)
+            ? error.details
+            : null;
+        return {
+            code: error.code,
+            message: error.message,
+            status: error.status,
+            ...(error.status < 500 && typeof details?.failure === "string" ? { failure: details.failure } : {}),
+            ...(typeof details?.callId === "string" ? { callId: details.callId } : {}),
+            ...(typeof details?.providerName === "string" ? { providerName: details.providerName } : {}),
+            ...(typeof details?.providerId === "string" ? { providerId: details.providerId } : {}),
+            ...(typeof details?.modelId === "string" ? { modelId: details.modelId } : {}),
+            ...(typeof details?.modelRecordId === "string" ? { modelRecordId: details.modelRecordId } : {})
+        };
+    }
+    return { code: "AI_STREAM_FAILED", message: "AI 流式调用失败" };
+}
 function redactMergeRecords(value, mapper) {
     const record = recordValue(value);
     if (!record)
@@ -652,6 +737,7 @@ export function createRuntime(options) {
         sameOriginEnforced: options.security?.enforceSameOrigin ?? true
     });
     const database = new Database(options.databasePath);
+    const bootId = randomUUID();
     const temporaryAttachmentRoot = options.databasePath === ":memory:" && !options.attachmentDirectory
         ? mkdtempSync(join(tmpdir(), "scriverse-attachments-"))
         : null;
@@ -722,6 +808,7 @@ export function createRuntime(options) {
         }, false, actor?.allowAdminAccess ?? false);
     });
     const app = express();
+    enforceCaseInsensitiveRouting(app);
     const upload = multer({
         storage: multer.memoryStorage(),
         limits: { fileSize: 30 * 1024 * 1024, files: 1, fields: 10, fieldSize: 64 * 1024, parts: 11, headerPairs: 100 }
@@ -742,13 +829,18 @@ export function createRuntime(options) {
         limits: { fileSize: 30 * 1024 * 1024, files: 1, fields: 4, fieldSize: 16 * 1024, parts: 5, headerPairs: 100 }
     });
     app.disable("x-powered-by");
-    if (options.security?.trustProxy !== undefined)
-        app.set("trust proxy", options.security.trustProxy);
+    const trustProxy = resolveTrustProxySetting(options.security?.trustProxy);
+    if (options.security?.trustProxy === true) {
+        logger.warn("security.trust_proxy.coerced", { from: true, to: 1 });
+    }
+    if (trustProxy !== undefined)
+        app.set("trust proxy", trustProxy);
     app.use(createRequestLoggingMiddleware());
     app.use(createSecurityHeadersMiddleware());
     app.get("/api/health", (_request, response) => {
         data(response, {
             status: "ok",
+            bootId,
             version: APP_VERSION,
             protocol: "openai-chat-completions",
             protocols: ["openai-chat-completions", "anthropic-messages"],
@@ -758,6 +850,7 @@ export function createRuntime(options) {
     if (options.security?.auth)
         app.use(createBasicAuthMiddleware(options.security.auth));
     app.use(createAuthenticationRateLimitMiddleware());
+    app.use(createCaptchaRateLimitMiddleware());
     app.use(createApiRateLimitMiddleware(options.security?.apiRateLimit, options.security?.apiRateWindowMs));
     if (options.security?.enforceSameOrigin ?? true)
         app.use(createSameOriginMiddleware());
@@ -769,12 +862,12 @@ export function createRuntime(options) {
         const setupTokenRequired = setupRequired && Boolean(options.security?.setupToken);
         const developmentUser = getDevelopmentUser();
         if (!session && developmentUser) {
-            data(response, { authenticated: true, user: developmentUser, csrfToken: null, setupRequired: false, setupTokenRequired: false, registrationOpen });
+            data(response, { authenticated: true, user: developmentUser, csrfToken: null, bootId, setupRequired: false, setupTokenRequired: false, registrationOpen });
             return;
         }
         data(response, session
-            ? { authenticated: true, user: session.user, csrfToken: session.csrfToken, setupRequired: false, setupTokenRequired: false, registrationOpen }
-            : { authenticated: false, user: null, csrfToken: null, setupRequired, setupTokenRequired, registrationOpen });
+            ? { authenticated: true, user: session.user, csrfToken: session.csrfToken, bootId, setupRequired: false, setupTokenRequired: false, registrationOpen }
+            : { authenticated: false, user: null, csrfToken: null, bootId, setupRequired, setupTokenRequired, registrationOpen });
     });
     app.get("/api/auth/captcha", (_request, response) => {
         data(response, captcha.create());
@@ -808,6 +901,7 @@ export function createRuntime(options) {
         resolveBypassUser: getDevelopmentUser
     }));
     app.use(createUploadRateLimitMiddleware());
+    app.use(createExpensiveApiRateLimitMiddleware());
     app.use(createCliApiScopeMiddleware(options.disableUserAuth));
     app.use(createWorkAuthorizationMiddleware(auth, options.disableUserAuth));
     app.get("/api/cli/session", (request, response) => {
@@ -916,6 +1010,8 @@ export function createRuntime(options) {
     app.patch("/api/users/:userId", (request, response) => {
         if (!request.authUser)
             throw new AppError(401, "AUTH_REQUIRED", "请先登录");
+        if (request.authUser.role !== "admin")
+            throw new AppError(403, "ADMIN_REQUIRED", "该操作仅限系统管理员");
         const updated = auth.updateUser(request.authUser, request.params.userId, parse(userUpdateSchema, request.body));
         store.audit(null, "user.updated", "user", updated.userId, { role: updated.role, status: updated.status });
         data(response, updated);
@@ -1780,7 +1876,8 @@ export function createRuntime(options) {
             page: request.query.page ?? "1",
             limit: request.query.limit ?? "20"
         }) ?? { page: 1, limit: 20, offset: 0 };
-        data(response, store.listAiConversationsPage(request.params.workId, pagination));
+        const permissions = requestPermissions(request, request.params.workId);
+        data(response, mapRecords(store.listAiConversationsPage(request.params.workId, pagination), (conversation) => (redactAiConversation(conversation, permissions))));
     });
     app.post("/api/works/:workId/ai-conversations", (request, response) => {
         const input = parse(z.object({ title: z.string().max(200).optional() }), request.body ?? {});
@@ -1788,11 +1885,17 @@ export function createRuntime(options) {
     });
     app.get("/api/ai-conversations/:conversationId", (request, response) => {
         const pagination = parsePagination(request.query);
-        data(response, pagination ? store.getAiConversationPage(request.params.conversationId, pagination) : store.getAiConversation(request.params.conversationId));
+        const conversation = pagination
+            ? store.getAiConversationPage(request.params.conversationId, pagination)
+            : store.getAiConversation(request.params.conversationId);
+        const permissions = requestPermissions(request, String(conversation.workId));
+        data(response, redactAiConversation(conversation, permissions));
     });
     app.post("/api/ai-conversations/:conversationId/fork", (request, response) => {
         const input = parse(z.object({ messageId: identifier, title: z.string().max(200).optional() }), request.body);
-        data(response, store.forkAiConversation(request.params.conversationId, input.messageId, input.title), 201);
+        const forked = store.forkAiConversation(request.params.conversationId, input.messageId, input.title);
+        const permissions = requestPermissions(request, String(forked.workId));
+        data(response, redactAiConversation(forked, permissions), 201);
     });
     app.post("/api/ai-conversations/:conversationId/messages", (request, response) => {
         const input = parse(z.object({
@@ -1809,7 +1912,10 @@ export function createRuntime(options) {
                 processSteps: z.array(aiProcessStepSchema).max(50).optional()
             }).optional()
         }), request.body);
-        data(response, store.addAiConversationMessage(request.params.conversationId, input), 201);
+        const message = store.addAiConversationMessage(request.params.conversationId, input);
+        const conversation = store.getAiConversationSummary(request.params.conversationId);
+        const permissions = requestPermissions(request, String(conversation.workId));
+        data(response, redactAiConversationMessage(message, permissions), 201);
     });
     app.post("/api/ai-conversations/:conversationId/context/prepare", async (request, response) => {
         const input = parse(z.object({
@@ -1891,7 +1997,10 @@ export function createRuntime(options) {
     app.get("/api/works/:workId/suggestions", (request, response) => {
         const status = typeof request.query.status === "string" ? request.query.status : undefined;
         const pagination = parsePagination(request.query);
-        data(response, pagination ? ai.listSuggestionsPage(request.params.workId, pagination, status) : ai.listSuggestions(request.params.workId, status));
+        const permissions = requestPermissions(request, request.params.workId);
+        data(response, pagination
+            ? mapRecords(ai.listSuggestionsPage(request.params.workId, pagination, status), (suggestion) => redactSuggestion(suggestion, permissions))
+            : ai.listSuggestions(request.params.workId, status).map((suggestion) => redactSuggestion(suggestion, permissions)));
     });
     app.post("/api/works/:workId/suggestions", async (request, response) => {
         const input = parse(z.object({
@@ -1907,14 +2016,14 @@ export function createRuntime(options) {
             if (store.getChapter(citation.chapterId).workId !== request.params.workId)
                 throw new AppError(400, "CITATION_WORK_MISMATCH", "引用章节不属于当前作品");
         }
-        data(response, await ai.createSuggestion({
+        data(response, redactSuggestion(await ai.createSuggestion({
             workId: request.params.workId,
             taskType: input.taskType,
             instruction: instructionWithCitations(input.instruction, citations),
             scope: input.scope,
             ...(input.modelId ? { modelId: input.modelId } : {}),
             ...(input.parameters ? { parameters: input.parameters } : {})
-        }), 201);
+        }), requestPermissions(request, request.params.workId)), 201);
     });
     app.post("/api/works/:workId/chat/stream", async (request, response) => {
         const input = parse(z.object({
@@ -1955,6 +2064,7 @@ export function createRuntime(options) {
                 throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
             }
             const conversationId = String(conversation.id);
+            const permissions = requestPermissions(request, request.params.workId);
             const prepared = await ai.prepareConversationContext({
                 conversationId,
                 workId: request.params.workId,
@@ -1965,10 +2075,10 @@ export function createRuntime(options) {
             });
             sendEvent("context", {
                 ...prepared,
-                conversation: {
+                conversation: redactAiConversation({
                     ...store.getAiConversationSummary(conversationId),
                     contextWarningPending: prepared.action === "warn"
-                }
+                }, permissions)
             });
             if (prepared.action === "warn")
                 return;
@@ -1981,7 +2091,7 @@ export function createRuntime(options) {
                 });
             const currentMessageId = input.currentMessageId ?? String(userMessage?.id ?? "");
             if (userMessage)
-                sendEvent("user_message", { message: userMessage });
+                sendEvent("user_message", { message: redactAiConversationMessage(userMessage, permissions) });
             const suggestion = await ai.createStreamingChat({
                 workId: request.params.workId,
                 instruction: instructionWithCitations(input.instruction, citations),
@@ -2019,20 +2129,11 @@ export function createRuntime(options) {
         }
         catch (error) {
             if (!controller.signal.aborted) {
-                const details = error instanceof AppError && error.details && typeof error.details === "object" && !Array.isArray(error.details)
-                    ? error.details
-                    : null;
-                sendEvent("error", {
-                    code: error instanceof AppError ? error.code : "AI_STREAM_FAILED",
-                    message: error instanceof Error ? error.message : "AI 流式调用失败",
-                    ...(error instanceof AppError ? { status: error.status } : {}),
-                    ...(typeof details?.failure === "string" ? { failure: details.failure } : {}),
-                    ...(typeof details?.callId === "string" ? { callId: details.callId } : {}),
-                    ...(typeof details?.providerName === "string" ? { providerName: details.providerName } : {}),
-                    ...(typeof details?.providerId === "string" ? { providerId: details.providerId } : {}),
-                    ...(typeof details?.modelId === "string" ? { modelId: details.modelId } : {}),
-                    ...(typeof details?.modelRecordId === "string" ? { modelRecordId: details.modelRecordId } : {})
+                logger.error("ai.stream.failed", {
+                    workId: request.params.workId,
+                    error: sanitizeError(error)
                 });
+                sendEvent("error", publicAiStreamError(error));
             }
         }
         finally {
@@ -2040,22 +2141,34 @@ export function createRuntime(options) {
                 response.end();
         }
     });
-    app.get("/api/suggestions/:suggestionId", (request, response) => data(response, ai.getSuggestion(request.params.suggestionId)));
+    app.get("/api/suggestions/:suggestionId", (request, response) => {
+        const suggestion = ai.getSuggestion(request.params.suggestionId);
+        const permissions = requestPermissions(request, String(suggestion.workId));
+        data(response, redactSuggestion(suggestion, permissions));
+    });
     app.get("/api/suggestions/:suggestionId/guards", (request, response) => {
         const pagination = parsePagination(request.query);
-        data(response, pagination
+        const suggestion = ai.getSuggestion(request.params.suggestionId);
+        const permissions = requestPermissions(request, String(suggestion.workId));
+        data(response, mapRecords(pagination
             ? store.listContinuationGuardsPage(request.params.suggestionId, pagination)
-            : store.listContinuationGuards(request.params.suggestionId));
+            : store.listContinuationGuards(request.params.suggestionId), (guard) => redactContinuationGuard(guard, permissions)));
     });
     app.post("/api/suggestions/:suggestionId/guard", async (request, response) => {
         const input = parse(z.object({ content: z.string().max(2_000_000).optional() }), request.body ?? {});
-        data(response, await ai.runSuggestionGuard(request.params.suggestionId, input.content), 201);
+        const suggestion = ai.getSuggestion(request.params.suggestionId);
+        const permissions = requestPermissions(request, String(suggestion.workId));
+        data(response, redactContinuationGuard(await ai.runSuggestionGuard(request.params.suggestionId, input.content), permissions), 201);
     });
     app.post("/api/suggestions/:suggestionId/accept", (request, response) => {
         const input = parse(z.object({ content: z.string().max(2_000_000).optional() }), request.body ?? {});
         data(response, ai.acceptSuggestion(request.params.suggestionId, input.content));
     });
-    app.post("/api/suggestions/:suggestionId/reject", (request, response) => data(response, ai.rejectSuggestion(request.params.suggestionId)));
+    app.post("/api/suggestions/:suggestionId/reject", (request, response) => {
+        const suggestion = ai.rejectSuggestion(request.params.suggestionId);
+        const permissions = requestPermissions(request, String(suggestion.workId));
+        data(response, redactSuggestion(suggestion, permissions));
+    });
     app.get("/api/works/:workId/ai-calls", (request, response) => {
         const pagination = parsePagination(request.query);
         const permissions = requestPermissions(request, request.params.workId);
@@ -2072,7 +2185,7 @@ export function createRuntime(options) {
         data(response, await ai.searchWork(request.params.workId, query.q, { type: query.type, limit: query.limit }));
     });
     app.get("/api/works/:workId/export", async (request, response) => {
-        const format = parse(z.enum(["json", "txt", "markdown"]), request.query.format ?? "json");
+        const format = parse(z.enum(["json", "txt", "markdown", "docx"]), request.query.format ?? "json");
         if (format === "json") {
             response.setHeader("Content-Disposition", `attachment; filename=novel-${request.params.workId}.json`);
             data(response, store.exportWork(request.params.workId));
@@ -2092,6 +2205,12 @@ export function createRuntime(options) {
             }), response);
             return;
         }
+        if (format === "docx") {
+            response.type("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
+            response.setHeader("Content-Disposition", `attachment; filename=novel-${request.params.workId}.docx`);
+            response.send(await store.exportDocx(request.params.workId));
+            return;
+        }
         response.type("text/plain");
         response.setHeader("Content-Disposition", `attachment; filename=novel-${request.params.workId}.txt`);
         response.send(store.exportText(request.params.workId, format));
@@ -2151,7 +2270,7 @@ export function createRuntime(options) {
             setHeaders: setStaticCacheControl
         }));
         app.get("/{*path}", (request, response, next) => {
-            if (request.path.startsWith("/api/"))
+            if (normalizeApiPath(request.path).startsWith("/api/"))
                 return next();
             sendIndexHtml(request, response);
         });