feat: add local auth billing and usage management
This commit is contained in:
1 parent
f642b5e71f
commit
196fdde83f
119 files changed
+15695
-2650
No files matched your search
@@ -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) {
|
||||
|
||||
@@ -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);
|
||||
|
||||
Reference in new issue
Block a user