import { parse } from "csv-parse/sync";
import fs from "fs/promises";
import { NextResponse } from "next/server";
import path from "path";
import { stringToBool } from "@utils/parsingUtils";
import { prisma } from "@utils/prisma";
import { getRandomNumber } from "@utils/maths";

const csvFilePath = path.resolve(process.cwd(), "public/artefacts.csv");

type CsvRow = {
	Type: string;
	Name: string;
	Description: string;
	WarehouseArea: string;
	EarthquakeCode: string;
	Required?: string;
	Price: string;
	PickedUp?: string;
	Picture: string;
};

export async function POST() {
	try {
		const fileContent = await fs.readFile(csvFilePath, "utf8");
		const records: CsvRow[] = parse(fileContent, {
			columns: true,
			skip_empty_lines: true,
		});

		const failedImports: { row: CsvRow; reason: string }[] = [];

		const artefacts = await Promise.all(
			records.map(async (row) => {
				const earthquake = await prisma.earthquake.findUnique({
					where: { code: row.EarthquakeCode },
				});

				const creators = await prisma.user.findMany({
					where: {
						role: { in: ["SCIENTIST", "ADMIN"] },
					},
				});
				const randomCreator = creators.length > 0 ? creators[getRandomNumber(0, creators.length - 1)] : null;

				if (!earthquake || !randomCreator) {
					failedImports.push({ row, reason: `Earthquake: ${earthquake}, RandomCreator: ${randomCreator}` });
					return undefined;
				}

				return {
					name: row.Name,
					description: row.Description,
					type: row.Type,
					warehouseArea: row.WarehouseArea,
					earthquakeId: earthquake.id,
					required: stringToBool(row.Required, true),
					shopPrice: row.Price && row.Price !== "" ? parseFloat(row.Price) : getRandomNumber(20, 500),
					pickedUp: stringToBool(row.PickedUp, false),
					creatorId: randomCreator.id,
					purchasedById: null,
					imageName: row.Picture,
				};
			})
		);

		const validArtefacts = artefacts.filter((artefact): artefact is NonNullable<typeof artefact> => artefact !== undefined);

		await prisma.artefact.createMany({
			data: validArtefacts,
		});

		if (failedImports.length > 0) {
			console.warn("Failed imports:", failedImports);
			await fs.writeFile(path.resolve(process.cwd(), "failed_imports_artefacts.json"), JSON.stringify(failedImports, null, 2));
		}

		return NextResponse.json({
			success: true,
			count: validArtefacts.length,
			failedCount: failedImports.length,
		});
	} catch (error: any) {
		console.error(error);
		return NextResponse.json({ success: false, error: error.message }, { status: 500 });
	}
}