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) {