import { withMethods } from "@lib/api-middleware/with-methods"
import { getHtmlFromFile } from "@lib/server/get-html-from-drift-file"
import { parseQueryParam } from "@lib/server/parse-query-param"
import { prisma } from "@lib/server/prisma"
import { NextApiRequest, NextApiResponse } from "next"

export default withMethods(
	["GET"],
	async (req: NextApiRequest, res: NextApiResponse) => {
		const query = req.query
		const fileId = parseQueryParam(query.fileId)
		const content = parseQueryParam(query.content)
		const title = parseQueryParam(query.title) || "Untitled"

		if (fileId && (content || title)) {
			return res.status(400).json({ error: "Too many arguments" })
		}

		if (fileId) {
			const file = await prisma.file.findUnique({
				where: {
					id: fileId
				}
			})

			if (!file) {
				return res.status(404).json({ error: "File not found" })
			}

			return res.json(file.html)
		} else {
			if (!content || !title) {
				return res.status(400).json({ error: "Missing arguments" })
			}

			const renderedHTML = await getHtmlFromFile({
				title,
				content
			})

			res.setHeader("Content-Type", "text/plain")
			res.status(200).write(renderedHTML)
			res.end()
			return
		}
	}
)