import { Hono } from "hono";
import { volumeService } from "../volumes/volume.service";
import { getVolumePath } from "../volumes/helpers";
import { eq } from "drizzle-orm";
import { db } from "../../db/db";
import { volumesTable } from "../../db/schema";

export const driverController = new Hono()
	.post("/VolumeDriver.Capabilities", (c) => {
		return c.json({
			Capabilities: {
				Scope: "global",
			},
		});
	})
	.post("/Plugin.Activate", (c) => {
		return c.json({
			Implements: ["VolumeDriver"],
		});
	})
	.post("/VolumeDriver.Create", (_) => {
		throw new Error("Volume creation is not supported via the driver");
	})
	.post("/VolumeDriver.Remove", (c) => {
		return c.json({
			Err: "",
		});
	})
	.post("/VolumeDriver.Mount", async (c) => {
		const body = await c.req.json();

		if (!body.Name) {
			return c.json({ Err: "Volume name is required" }, 400);
		}

		const shortId = body.Name.replace(/^zb-/, "");

		const volume = await db.query.volumesTable.findFirst({
			where: eq(volumesTable.shortId, shortId),
		});

		if (!volume) {
			return c.json({ Err: `Volume with shortId ${shortId} not found` }, 404);
		}

		return c.json({
			Mountpoint: getVolumePath(volume),
		});
	})
	.post("/VolumeDriver.Unmount", (c) => {
		return c.json({
			Err: "",
		});
	})
	.post("/VolumeDriver.Path", async (c) => {
		const body = await c.req.json();

		if (!body.Name) {
			return c.json({ Err: "Volume name is required" }, 400);
		}

		const shortId = body.Name.replace(/^zb-/, "");

		const volume = await db.query.volumesTable.findFirst({
			where: eq(volumesTable.shortId, shortId),
		});

		if (!volume) {
			return c.json({ Err: `Volume with shortId ${shortId} not found` }, 404);
		}

		return c.json({
			Mountpoint: getVolumePath(volume),
		});
	})
	.post("/VolumeDriver.Get", async (c) => {
		const body = await c.req.json();

		if (!body.Name) {
			return c.json({ Err: "Volume name is required" }, 400);
		}

		const shortId = body.Name.replace(/^zb-/, "");

		const volume = await db.query.volumesTable.findFirst({
			where: eq(volumesTable.shortId, shortId),
		});

		if (!volume) {
			return c.json({ Err: `Volume with shortId ${shortId} not found` }, 404);
		}

		return c.json({
			Volume: {
				Name: `zb-${volume.shortId}`,
				Mountpoint: getVolumePath(volume),
				Status: {},
			},
			Err: "",
		});
	})
	.post("/VolumeDriver.List", async (c) => {
		const volumes = await volumeService.listVolumes();

		const res = volumes.map((volume) => ({
			Name: `zb-${volume.shortId}`,
			Mountpoint: getVolumePath(volume),
			Status: {},
		}));

		return c.json({
			Volumes: res,
		});
	});