feat: add local auth billing and usage management

app/api/generations/image/[id]/retry/route.ts

@@ -1,15 +1,18 @@
 import { jsonError, jsonOk } from "@/lib/server/api";
-import { requireAppUser } from "@/lib/server/auth/current-user";
+import { requireAppSession } from "@/lib/server/auth/current-user";
 import { requestOrigin } from "@/lib/server/runtime";
 import { retryImageJob } from "@/lib/server/generation-service";
+import { resolvePlatformUsageContext } from "@/lib/server/usage-context";
 
 export const runtime = "nodejs";
 
 export async function POST(request: Request, context: { params: Promise<{ id: string }> }) {
   try {
-    const user = await requireAppUser();
+    const session = await requireAppSession();
+    const user = session.user;
     const { id } = await context.params;
-    const job = await retryImageJob(id, requestOrigin(request), user.id);
+    const usageContext = await resolvePlatformUsageContext(session);
+    const job = await retryImageJob(id, requestOrigin(request), user.id, usageContext);
     return jsonOk({ job }, { status: 202 });
   } catch (error) {
     return jsonError(error);

app/api/generations/image/[id]/route.ts

@@ -1,7 +1,8 @@
-import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob } from "@/lib/server/data-store";
+import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob, updateGenerationJob } from "@/lib/server/data-store";
 import { jsonError, jsonOk } from "@/lib/server/api";
 import { requireAppUser } from "@/lib/server/auth/current-user";
 import { deleteStoredAsset } from "@/lib/server/storage";
+import { refundGenerationCharge } from "@/lib/server/billing-service";
 
 export const runtime = "nodejs";
 
@@ -31,6 +32,10 @@ export async function DELETE(_request: Request, context: { params: Promise<{ id:
       await deleteAsset(asset.id);
       deletedAssetIds.push(asset.id);
     }
+    const cancellableJob = ["queued", "running"].includes(job.status)
+      ? await updateGenerationJob(job.id, { status: "cancelled", completedAt: new Date().toISOString() })
+      : job;
+    await refundGenerationCharge(cancellableJob, "任务已删除");
     await deleteGenerationJob(id);
     return jsonOk({ ok: true, deletedJobId: id, deletedAssetIds });
   } catch (error) {

app/api/generations/image/route.ts

@@ -1,11 +1,12 @@
 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 { requireAppSession, 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";
 import type { EnabledImageCapability } from "@/lib/types";
 import type { ImageCreationEngine } from "@/lib/evolink/image-client";
+import { resolvePlatformUsageContext } from "@/lib/server/usage-context";
 
 export const runtime = "nodejs";
 
@@ -21,7 +22,9 @@ export async function GET() {
 
 export async function POST(request: Request) {
   try {
-    const user = await requireAppUser();
+    const session = await requireAppSession();
+    const user = session.user;
+    const usageContext = await resolvePlatformUsageContext(session);
     const body = await readJsonBody<{
       capability?: EnabledImageCapability;
       engine?: ImageCreationEngine;
@@ -58,7 +61,8 @@ export async function POST(request: Request) {
       min_ratio: asNumber(body.min_ratio),
       max_ratio: asNumber(body.max_ratio),
       force_single: Boolean(body.force_single),
-      quality: typeof body.quality === "string" ? body.quality : undefined
+      quality: typeof body.quality === "string" ? body.quality : undefined,
+      usageContext
     }, requestOrigin(request));
     return jsonOk({ job: await getGenerationJob(job.id) }, { status: 202 });
   } catch (error) {

app/api/generations/video/[id]/route.ts

@@ -1,7 +1,8 @@
-import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob } from "@/lib/server/data-store";
+import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob, updateGenerationJob } from "@/lib/server/data-store";
 import { jsonError, jsonOk } from "@/lib/server/api";
 import { requireAppUser } from "@/lib/server/auth/current-user";
 import { deleteStoredAsset } from "@/lib/server/storage";
+import { refundGenerationCharge } from "@/lib/server/billing-service";
 
 export const runtime = "nodejs";
 
@@ -31,6 +32,10 @@ export async function DELETE(_request: Request, context: { params: Promise<{ id:
       await deleteAsset(asset.id);
       deletedAssetIds.push(asset.id);
     }
+    const cancellableJob = ["queued", "running"].includes(job.status)
+      ? await updateGenerationJob(job.id, { status: "cancelled", completedAt: new Date().toISOString() })
+      : job;
+    await refundGenerationCharge(cancellableJob, "任务已删除");
     await deleteGenerationJob(id);
     return jsonOk({ ok: true, deletedJobId: id, deletedAssetIds });
   } catch (error) {

app/api/generations/video/route.ts

@@ -1,8 +1,9 @@
 import { listGenerationJobs } from "@/lib/server/data-store";
 import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
-import { requireAppUser } from "@/lib/server/auth/current-user";
+import { requireAppSession, requireAppUser } from "@/lib/server/auth/current-user";
 import { requestOrigin } from "@/lib/server/runtime";
 import { submitVideoJob, type SubmitVideoJobInput } from "@/lib/server/video-generation-service";
+import { resolvePlatformUsageContext } from "@/lib/server/usage-context";
 
 export const runtime = "nodejs";
 
@@ -18,9 +19,11 @@ export async function GET() {
 
 export async function POST(request: Request) {
   try {
-    const user = await requireAppUser();
+    const session = await requireAppSession();
+    const user = session.user;
     const body = await readJsonBody<SubmitVideoJobInput & Record<string, unknown>>(request);
-    const job = await submitVideoJob({ ...body, ownerId: user.id }, requestOrigin(request));
+    const usageContext = await resolvePlatformUsageContext(session);
+    const job = await submitVideoJob({ ...body, ownerId: user.id, usageContext }, requestOrigin(request));
     return jsonOk({ job }, { status: 202 });
   } catch (error) {
     return jsonError(error);