ironmount/apps/server/src/modules/volumes/volume.controller.ts

import { Hono } from "hono";
import { validator } from "hono-openapi/arktype";
import {
	createVolumeBody,
	createVolumeDto,
	deleteVolumeDto,
	getVolumeDto,
	type ListVolumesResponseDto,
	listVolumesDto,
	testConnectionBody,
	testConnectionDto,
	updateVolumeBody,
	updateVolumeDto,
	mountVolumeDto,
	unmountVolumeDto,
	type GetVolumeResponseDto,
} from "./volume.dto";
import { volumeService } from "./volume.service";

export const volumeController = new Hono()
	.get("/", listVolumesDto, async (c) => {
		const volumes = await volumeService.listVolumes();

		const response = {
			volumes: volumes.map((volume) => ({
				...volume,
				updatedAt: volume.updatedAt.getTime(),
				createdAt: volume.createdAt.getTime(),
				lastHealthCheck: volume.lastHealthCheck.getTime(),
			})),
		} satisfies ListVolumesResponseDto;

		return c.json(response, 200);
	})
	.post("/", createVolumeDto, validator("json", createVolumeBody), async (c) => {
		const body = c.req.valid("json");
		const res = await volumeService.createVolume(body.name, body.config);

		return c.json({ message: "Volume created", volume: res.volume }, 201);
	})
	.post("/test-connection", testConnectionDto, validator("json", testConnectionBody), async (c) => {
		const body = c.req.valid("json");
		const result = await volumeService.testConnection(body.config);

		return c.json(result, 200);
	})
	.delete("/:name", deleteVolumeDto, async (c) => {
		const { name } = c.req.param();
		await volumeService.deleteVolume(name);

		return c.json({ message: "Volume deleted" }, 200);
	})
	.get("/:name", getVolumeDto, async (c) => {
		const { name } = c.req.param();
		const res = await volumeService.getVolume(name);

		const response = {
			...res,
			volume: {
				...res.volume,
				createdAt: res.volume.createdAt.getTime(),
				updatedAt: res.volume.updatedAt.getTime(),
				lastHealthCheck: res.volume.lastHealthCheck.getTime(),
			},
		} satisfies GetVolumeResponseDto;

		return c.json(response, 200);
	})
	.put("/:name", updateVolumeDto, validator("json", updateVolumeBody), async (c) => {
		const { name } = c.req.param();
		const body = c.req.valid("json");
		const res = await volumeService.updateVolume(name, body.config);

		const response = {
			message: "Volume updated",
			volume: {
				name: res.volume.name,
				path: res.volume.path,
				type: res.volume.type,
				createdAt: res.volume.createdAt.getTime(),
				updatedAt: res.volume.updatedAt.getTime(),
				config: res.volume.config,
			},
		};

		return c.json(response, 200);
	})
	.post("/:name/mount", mountVolumeDto, async (c) => {
		const { name } = c.req.param();
		const { error, status } = await volumeService.mountVolume(name);

		return c.json({ error, status }, error ? 500 : 200);
	})
	.post("/:name/unmount", unmountVolumeDto, async (c) => {
		const { name } = c.req.param();
		const { error, status } = await volumeService.unmountVolume(name);

		return c.json({ error, status }, error ? 500 : 200);
	});