import { Prisma, PrismaClient } from "@prisma/client";
import { LogItem } from "./project_monitor";

type LogRecord = LogItem & {
	id: number;
};

class LogStore {
	constructor(private prisma: PrismaClient) {}

	async store(projectId: number, serviceName: string, workerId: string, logs: LogItem[]) {
		await this.prisma.log.createMany({
			data: logs.map((log) => ({
				projectId,
				serviceName,
				workerId,
				runId: log.runId,
				commit: log.commit ?? undefined,
				contents: log.contents,
				timestampMilli: log.timestampMilli,
			})),
		});
	}

	async get(
		projectId: number,
		serviceName: string,
		workerId: string,
		afterId?: number,
		numRecords?: number,
	): Promise<LogRecord[]> {
		const where: Prisma.LogWhereInput = { projectId, serviceName, workerId };
		if (afterId) {
			where.id = { gt: afterId };
		}
		const logs = await this.prisma.log.findMany({
			where,
			orderBy: { timestampMilli: "asc" },
			take: numRecords ?? 100,
		});
		return logs.map((log) => ({
			id: log.id,
			timestampMilli: Number(log.timestampMilli),
			contents: log.contents,
			runId: log.runId,
			commit: log.commit ?? undefined,
		}));
	}
}

export default LogStore;