wangxuming/NianAIGC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add authenticated login and SSO protection

Browse Source
This commit is contained in:
inman
2026-05-29 15:54:13 +08:00
parent e36f28a668
commit 0648874801
50 changed files with 1853 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/api/generations/image/route.ts
View File

@@ -1,5 +1,6 @@
import { getGenerationJob, listGenerationJobs } from "@/lib/server/data-store";
import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
import { requireAppUser } from "@/lib/server/auth/current-user";
import { requestOrigin } from "@/lib/server/runtime";
import { submitImageJob } from "@/lib/server/generation-service";
import { assemblePrompt, type PromptAssemblyInput, type PromptMaterial } from "@/lib/prompt/assembler";
@@ -9,7 +10,8 @@ export const runtime = "nodejs";
export async function GET() {
try {
const jobs = (await listGenerationJobs()).filter((job) => job.capability !== "video.generate");
const user = await requireAppUser();
const jobs = (await listGenerationJobs(user.id)).filter((job) => job.capability !== "video.generate");
return jsonOk({ jobs });
} catch (error) {
return jsonError(error, 500);
@@ -18,6 +20,7 @@ export async function GET() {
export async function POST(request: Request) {
try {
const user = await requireAppUser();
const body = await readJsonBody<{
capability?: EnabledImageCapability;
prompt?: string;
@@ -41,6 +44,7 @@ export async function POST(request: Request) {
.filter((material) => material.type === "image")
.map((material) => material.url);
const job = await submitImageJob({
ownerId: user.id,
capability,
prompt: body.prompt || assembled?.prompt,
imageUrls: body.imageUrls || materialImages,