import { db, WebsiteProject } from "@/lib/db";
import { getSession } from "@/lib/auth";
import { NextRequest, NextResponse } from "next/server";
import { v4 as uuidv4 } from "uuid";

export async function POST(req: NextRequest) {
    try {
        const session = await getSession();
        if (!session) {
            return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
        }

        const body = await req.json();
        const { id, name, code, messages, previewImage, subdomain, clearSubdomain } = body;
        const userEmail = (session as any).email;

        if (!userEmail) {
            return NextResponse.json({ error: "User email not found in session" }, { status: 500 });
        }

        let project: WebsiteProject;

        if (subdomain) {
            const isAvailable = await db.checkSubdomainAvailability(subdomain, id);
            if (!isAvailable) {
                return NextResponse.json({ error: "Subdomain is already taken" }, { status: 400 });
            }
        }

        if (id) {
            // Update existing
            const existing = await db.getWebsite(id);
            if (existing && existing.userEmail !== userEmail) {
                return NextResponse.json({ error: "Forbidden" }, { status: 403 });
            }

            project = {
                id,
                userEmail,
                name: name || existing?.name || "Untitled Website",
                code,
                subdomain: clearSubdomain ? null : (subdomain || existing?.subdomain),
                messages: messages || existing?.messages || [],
                createdAt: existing?.createdAt || new Date().toISOString(),
                updatedAt: new Date().toISOString(),
                previewImage: previewImage || existing?.previewImage
            };
        } else {
            // Create new
            project = {
                id: uuidv4(),
                userEmail,
                name: name || "Untitled Website",
                code,
                subdomain,
                messages: messages || [],
                createdAt: new Date().toISOString(),
                updatedAt: new Date().toISOString(),
                previewImage
            };
        }

        await db.saveWebsite(project);
        return NextResponse.json({ success: true, project });

    } catch (error) {
        console.error("Save website error:", error);
        return NextResponse.json({ error: "Internal Server Error" }, { status: 500 });
    }
}

export async function GET(req: NextRequest) {
    try {
        const session = await getSession();
        if (!session) {
            return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
        }

        const userEmail = (session as any).email;

        if (!userEmail) {
            return NextResponse.json({ error: "User email not found in session" }, { status: 500 });
        }

        const projects = await db.listWebsites(userEmail);

        return NextResponse.json(projects);

    } catch (error) {
        return NextResponse.json({ error: "Internal Server Error" }, { status: 500 });
    }
}

export async function DELETE(req: NextRequest) {
    try {
        const session = await getSession();
        if (!session) {
            return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
        }

        const { searchParams } = new URL(req.url);
        const id = searchParams.get('id');

        if (!id) return NextResponse.json({ error: "Missing ID" }, { status: 400 });

        const userEmail = (session as any).email;

        if (!userEmail) {
            return NextResponse.json({ error: "User email not found in session" }, { status: 500 });
        }

        const existing = await db.getWebsite(id);
        if (existing && existing.userEmail !== userEmail) {
            return NextResponse.json({ error: "Forbidden" }, { status: 403 });
        }

        await db.deleteWebsite(id);
        return NextResponse.json({ success: true });

    } catch (error) {
        return NextResponse.json({ error: "Internal Server Error" }, { status: 500 });
    }
}