feat: 新增数据库

server/api/images/generate.post.ts

@@ -1,12 +1,21 @@
+import { randomUUID } from "node:crypto";
 import { consola } from "consola";
-
 import type {
   IImageGenerateData,
   IImageGenerateRequest
 } from "#shared/types/openai";
-import { isUnauthorizedError } from "~~/server/utils";
+import {
+  createRunningImageGeneration,
+  downloadImageAsBase64,
+  finishImageGenerationFailed,
+  finishImageGenerationSuccess,
+  getNewApiUserIdFromCookie,
+  isUnauthorizedError
+} from "~~/server/utils";
 
 export default defineEventHandler(async (event) => {
+  const requestId = randomUUID();
+  let stage = "read_body";
   const requestBody = await readBody<Partial<IImageGenerateRequest> | null>(
     event
   );
@@ -24,16 +33,96 @@ export default defineEventHandler(async (event) => {
     return createErrorResponse(400, "请输入图片描述");
   }
 
+  let record: { id: bigint; startedAt: Date } | null = null;
+
   try {
+    consola.info("[images.generate] 开始", {
+      requestId,
+      promptLength: prompt.length
+    });
+
+    stage = "read_user_id";
+    const userId = getNewApiUserIdFromCookie(event);
+    consola.info("[images.generate] 读取用户成功", {
+      requestId,
+      userId
+    });
+
+    stage = "create_running_record";
+    record = await createRunningImageGeneration(userId, prompt);
+    consola.info("[images.generate] 创建生图记录成功", {
+      requestId,
+      recordId: record.id.toString()
+    });
+
+    stage = "get_api_key";
     const apiKey = await getAiArtStudioTokenKey(event);
+    consola.info("[images.generate] 获取服务端 key 成功", {
+      requestId,
+      recordId: record.id.toString()
+    });
+
+    stage = "call_image_api";
     const result = await askImg({
       apiKey,
       prompt
     });
+    consola.info("[images.generate] 上游生图成功", {
+      requestId,
+      recordId: record.id.toString(),
+      hasImageUrl: Boolean(result.imageUrl)
+    });
 
-    consola.info("图片生成成功", { prompt, imageUrl: result.imageUrl });
-    return createSuccessResponse<IImageGenerateData>(result, "图片生成成功");
+    stage = "download_image_base64";
+    const downloadedImage = await downloadGeneratedImage(result.imageUrl);
+    consola.info("[images.generate] 图片归档下载完成", {
+      requestId,
+      recordId: record.id.toString(),
+      hasBase64: Boolean(downloadedImage.base64),
+      mimeType: downloadedImage.mimeType
+    });
+
+    stage = "finish_success_record";
+    await finishImageGenerationSuccess(record.id, record.startedAt, {
+      imageUrl: result.imageUrl,
+      imageBase64: downloadedImage.base64,
+      imageMimeType: downloadedImage.mimeType,
+      revisedPrompt: result.revisedPrompt,
+      upstreamResponse: result.upstreamResponse,
+      errorMessage: downloadedImage.errorMessage
+    });
+
+    consola.info("[images.generate] 完成", {
+      requestId,
+      recordId: record.id.toString()
+    });
+    return createSuccessResponse<IImageGenerateData>(
+      {
+        imageUrl: result.imageUrl,
+        revisedPrompt: result.revisedPrompt
+      },
+      "图片生成成功"
+    );
   } catch (error) {
+    consola.error("[images.generate] 失败", {
+      requestId,
+      stage,
+      recordId: record?.id.toString() ?? null,
+      error: toSafeLogError(error)
+    });
+
+    if (record) {
+      await finishImageGenerationFailed(record.id, record.startedAt, error).catch(
+        (recordError) => {
+          consola.error("[images.generate] 更新失败记录失败", {
+            requestId,
+            recordId: record?.id.toString() ?? null,
+            error: toSafeLogError(recordError)
+          });
+        }
+      );
+    }
+
     if (isUnauthorizedError(error)) {
       clearNewApiAuthCookies(event);
       return createErrorResponse(401, "未登录");
@@ -42,3 +131,53 @@ export default defineEventHandler(async (event) => {
     return createUpstreamErrorResponse(error, "图片生成失败");
   }
 });
+
+/** 图片 URL 仍然可用时，base64 保存失败不阻断本次生成结果 */
+const downloadGeneratedImage = async (
+  imageUrl: string
+): Promise<{
+  base64: string | null;
+  mimeType: string | null;
+  errorMessage: string | null;
+}> => {
+  try {
+    const image = await downloadImageAsBase64(imageUrl);
+    return {
+      base64: image.base64,
+      mimeType: image.mimeType,
+      errorMessage: null
+    };
+  } catch (error) {
+    consola.error("[images.generate] 保存生成图片 base64 失败", {
+      error: toSafeLogError(error)
+    });
+    return {
+      base64: null,
+      mimeType: null,
+      errorMessage: "图片生成成功，但 base64 保存失败"
+    };
+  }
+};
+
+const toSafeLogError = (error: unknown) => {
+  const maybeError = error as {
+    data?: unknown;
+    message?: string;
+    name?: string;
+    response?: {
+      _data?: unknown;
+      status?: number;
+    };
+    stack?: string;
+    status?: number;
+    statusCode?: number;
+  };
+
+  return {
+    name: maybeError.name,
+    message: maybeError.message,
+    status: maybeError.status ?? maybeError.statusCode ?? maybeError.response?.status,
+    data: maybeError.data ?? maybeError.response?._data ?? null,
+    stack: maybeError.stack
+  };
+};