import {NextResponse} from "next/server";
import {and, eq} from "drizzle-orm";
import * as drizzleDb from "@/db";
import {db} from "@/db";
import {getDatabaseOrThrow, withAgentCheck} from "../../helpers";
import {isUuidv4} from "@/utils/verify-uuid";
import {eventEmitter} from "@/lib/event";
import {logger} from "@/lib/logger";

const log = logger.child({module: "api/agent/backup/upload/init"});

export type Body = {
    generatedId: string
    storageChannelId: string
    backupId: string
}
export const POST = withAgentCheck(async (request: Request, {params, agent}: {
    params: Promise<{ agentId: string }>,
    agent: any
}) => {
    try {
        const body: Body = await request.json();

        log.info({data: body}, "Body for backup upload init");

        const generatedId = body.generatedId;
        const storageChannelId = body.storageChannelId;
        const backupId = body.backupId;

        if (!generatedId || !isUuidv4(generatedId)) {
            return NextResponse.json(
                {error: "generatedId is not a valid UUID"},
                {status: 400}
            );
        }

        const database = await getDatabaseOrThrow(generatedId);

        const backup = await db.query.backup.findFirst({
            where: and(
                eq(drizzleDb.schemas.backup.id, backupId),
                eq(drizzleDb.schemas.backup.databaseId, database.id),
            ),
        });

        if (!backup) {
            return NextResponse.json(
                {error: "Unable to find the corresponding backup"},
                {status: 404}
            );
        }

        const [backupStorage] = await db
            .insert(drizzleDb.schemas.backupStorage)
            .values({
                backupId: backup.id,
                storageChannelId: storageChannelId,
                status: "pending",
            })
            .returning();

        eventEmitter.emit('modification', {update: true});

        return NextResponse.json(
            {
                message: "Backup storage successfully created",
                backupStorage: backupStorage
            },
            {status: 200}
        );
    } catch (error) {
        log.error({error: error}, "Error in POST for INIT backup");
        return NextResponse.json(
            {error: "Internal server error"},
            {status: 500}
        );
    }
});