import { docker } from "@/server/constants";
import { createTRPCRouter, protectedProcedure } from "../trpc";
import { getPublicIpWithFallback } from "@/server/wss/terminal";
import type { DockerNode } from "../services/cluster";
import { z } from "zod";
import { TRPCError } from "@trpc/server";
import { execAsync } from "@/server/utils/process/execAsync";

export const clusterRouter = createTRPCRouter({
	getNodes: protectedProcedure.query(async () => {
		const workers: DockerNode[] = await docker.listNodes();

		return workers;
	}),
	removeWorker: protectedProcedure
		.input(
			z.object({
				nodeId: z.string(),
			}),
		)
		.mutation(async ({ input }) => {
			try {
				await execAsync(
					`docker node update --availability drain ${input.nodeId}`,
				);
				await execAsync(`docker node rm ${input.nodeId} --force`);
				return true;
			} catch (error) {
				throw new TRPCError({
					code: "INTERNAL_SERVER_ERROR",
					message: "Error to remove the node",
					cause: error,
				});
			}
		}),
	addWorker: protectedProcedure.query(async ({ input }) => {
		const result = await docker.swarmInspect();
		return `docker swarm join --token ${
			result.JoinTokens.Worker
		} ${await getPublicIpWithFallback()}:2377`;
	}),
	addManager: protectedProcedure.query(async ({ input }) => {
		const result = await docker.swarmInspect();
		return `docker swarm join --token ${
			result.JoinTokens.Manager
		} ${await getPublicIpWithFallback()}:2377`;
	}),
});