jiawei/aiartstudio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 新增数据库

Browse Source
This commit is contained in:
Marcus
2026-04-25 21:52:18 +08:00
parent 223845b0ed
commit abd1b03529
23 changed files with 1890 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/api/auth/login.post.ts
+6 -1
View File
@@ -9,7 +9,8 @@ import {
extractCookieMetaFromSetCookie,
extractCookieValueFromSetCookie,
newApiFetchRaw,
setNewApiAuthCookies
setNewApiAuthCookies,
upsertUserSnapshot
} from "~~/server/utils";
interface INewApiWrappedLoginResponse {
@@ -92,6 +93,10 @@ export default defineEventHandler(async (event) => {
extractCookieMetaFromSetCookie(result.headers, "session")
);
upsertUserSnapshot(normalized.user).catch((error) => {
consola.error("/api/auth/login 保存用户快照失败", error);
});
return createSuccessResponse<IUserLoginData>(
normalized.user,
normalized.message
server/api/auth/me.get.ts
+4 -1
View File
@@ -3,7 +3,8 @@ import {
clearNewApiAuthCookies,
createErrorResponse,
createSuccessResponse,
newApiAuthedFetch
newApiAuthedFetch,
upsertUserSnapshot
} from "~~/server/utils";
interface INewApiWrappedUserResponse {
@@ -40,6 +41,8 @@ export default defineEventHandler(async (event) => {
return createErrorResponse(401, normalized.message);
}
upsertUserSnapshot(normalized.user).catch(() => {});
return createSuccessResponse<IUserMeData>(
normalized.user,
normalized.message
server/api/images/db-health.get.ts
+22
View File
@@ -0,0 +1,22 @@
import { consola } from "consola";
import { prisma } from "~~/server/utils";
export default defineEventHandler(async () => {
const startedAt = Date.now();
try {
await prisma.$queryRaw`SELECT 1`;
return createSuccessResponse(
{
ok: true,
elapsedMs: Date.now() - startedAt
},
"数据库连接正常"
);
} catch (error) {
consola.error("[images.db-health] 数据库连接失败", error);
return createUpstreamErrorResponse(error, "数据库连接失败");
}
});
server/api/images/generate.post.ts
+143 -4
View File
@@ -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
};
};
server/api/images/history.get.ts
+52
View File
@@ -0,0 +1,52 @@
import type { IImageHistoryListData } from "#shared/types/openai";
import {
clearNewApiAuthCookies,
getNewApiUserIdFromCookie,
isUnauthorizedError,
listImageGenerationHistory
} from "~~/server/utils";
const DEFAULT_PAGE = 1;
const DEFAULT_PAGE_SIZE = 10;
const MAX_PAGE_SIZE = 50;
export default defineEventHandler(async (event) => {
try {
const userId = getNewApiUserIdFromCookie(event);
const query = getQuery(event);
const page = normalizePositiveInt(query.page, DEFAULT_PAGE);
const pageSize = Math.min(
normalizePositiveInt(query.size, DEFAULT_PAGE_SIZE),
MAX_PAGE_SIZE
);
const result = await listImageGenerationHistory({
userId,
page,
pageSize
});
return createSuccessResponse<IImageHistoryListData>(
result,
"获取生图历史成功"
);
} catch (error) {
if (isUnauthorizedError(error)) {
clearNewApiAuthCookies(event);
return createErrorResponse(401, "未登录");
}
return createUpstreamErrorResponse(error, "获取生图历史失败");
}
});
const normalizePositiveInt = (
value: unknown,
fallbackValue: number
): number => {
const rawValue = Array.isArray(value) ? value[0] : value;
const numberValue = Number.parseInt(String(rawValue ?? ""), 10);
return Number.isInteger(numberValue) && numberValue > 0
? numberValue
: fallbackValue;
};
server/api/images/history/[id].delete.ts
+38
View File
@@ -0,0 +1,38 @@
import {
clearNewApiAuthCookies,
getNewApiUserIdFromCookie,
isUnauthorizedError,
softDeleteImageGeneration
} from "~~/server/utils";
export default defineEventHandler(async (event) => {
try {
const userId = getNewApiUserIdFromCookie(event);
const recordId = parseRecordId(getRouterParam(event, "id"));
const deleted = await softDeleteImageGeneration(userId, recordId);
if (!deleted) {
return createErrorResponse(404, "生图记录不存在");
}
return createSuccessResponse(null, "删除生图记录成功");
} catch (error) {
if (isUnauthorizedError(error)) {
clearNewApiAuthCookies(event);
return createErrorResponse(401, "未登录");
}
return createUpstreamErrorResponse(error, "删除生图记录失败");
}
});
const parseRecordId = (value: string | undefined): bigint => {
if (!value || !/^\d+$/.test(value)) {
throw createError({
statusCode: 400,
statusMessage: "生图记录 ID 不正确"
});
}
return BigInt(value);
};
server/api/images/history/[id].get.ts
+42
View File
@@ -0,0 +1,42 @@
import type { IImageHistoryDetail } from "#shared/types/openai";
import {
clearNewApiAuthCookies,
getImageGenerationDetail,
getNewApiUserIdFromCookie,
isUnauthorizedError
} from "~~/server/utils";
export default defineEventHandler(async (event) => {
try {
const userId = getNewApiUserIdFromCookie(event);
const recordId = parseRecordId(getRouterParam(event, "id"));
const result = await getImageGenerationDetail(userId, recordId);
if (!result) {
return createErrorResponse(404, "生图记录不存在");
}
return createSuccessResponse<IImageHistoryDetail>(
result,
"获取生图详情成功"
);
} catch (error) {
if (isUnauthorizedError(error)) {
clearNewApiAuthCookies(event);
return createErrorResponse(401, "未登录");
}
return createUpstreamErrorResponse(error, "获取生图详情失败");
}
});
const parseRecordId = (value: string | undefined): bigint => {
if (!value || !/^\d+$/.test(value)) {
throw createError({
statusCode: 400,
statusMessage: "生图记录 ID 不正确"
});
}
return BigInt(value);
};
server/api/images/stats.get.ts
+15
View File
@@ -0,0 +1,15 @@
import type { IImageGenerationStatsData } from "#shared/types/openai";
import { getGenerationStats } from "~~/server/utils";
export default defineEventHandler(async () => {
try {
const result = await getGenerationStats();
return createSuccessResponse<IImageGenerationStatsData>(
result,
"获取生图统计成功"
);
} catch (error) {
return createUpstreamErrorResponse(error, "获取生图统计失败");
}
});