ironmount/app/server/modules/backends/directory/directory-backend.ts

import * as fs from "node:fs/promises";
import * as npath from "node:path";
import { toMessage } from "../../../utils/errors";
import { logger } from "../../../utils/logger";
import type { VolumeBackend } from "../backend";
import { BACKEND_STATUS, type BackendConfig } from "~/schemas/volumes";

const mount = async (config: BackendConfig, _volumePath: string) => {
	if (config.backend !== "directory") {
		return { status: BACKEND_STATUS.error, error: "Invalid backend type" };
	}

	logger.info("Mounting directory volume from:", config.path);

	try {
		await fs.access(config.path);
		const stats = await fs.stat(config.path);

		if (!stats.isDirectory()) {
			return { status: BACKEND_STATUS.error, error: "Path is not a directory" };
		}

		return { status: BACKEND_STATUS.mounted };
	} catch (error) {
		logger.error("Failed to mount directory volume:", error);
		return { status: BACKEND_STATUS.error, error: toMessage(error) };
	}
};

const unmount = async () => {
	logger.info("Cannot unmount directory volume.");
	return { status: BACKEND_STATUS.unmounted };
};

const checkHealth = async (config: BackendConfig) => {
	if (config.backend !== "directory") {
		return { status: BACKEND_STATUS.error, error: "Invalid backend type" };
	}

	try {
		await fs.access(config.path);

		// Try to create a temporary file to ensure write access
		const tempFilePath = npath.join(config.path, `.healthcheck-${Date.now()}`);
		await fs.writeFile(tempFilePath, "healthcheck");
		await fs.unlink(tempFilePath);

		return { status: BACKEND_STATUS.mounted };
	} catch (error) {
		logger.error("Directory health check failed:", error);
		return { status: BACKEND_STATUS.error, error: toMessage(error) };
	}
};

const getVolumePath = (config: BackendConfig): string => {
	if (config.backend !== "directory") {
		throw new Error("Invalid backend type");
	}

	return config.path;
};

export const makeDirectoryBackend = (config: BackendConfig, volumePath: string): VolumeBackend => ({
	mount: () => mount(config, volumePath),
	unmount,
	checkHealth: () => checkHealth(config),
	getVolumePath: () => getVolumePath(config),
	getBackupPath: async () => getVolumePath(config),
});