import type { RepositoryConfig } from "~/schemas/restic";
import { normalizeAbsolutePath } from "~/utils/path";
import type { Readable } from "node:stream";
import { logger } from "~/server/utils/logger";
import { ResticError } from "~/server/utils/errors";
import { safeSpawn } from "~/server/utils/spawn";
import type { ResticDumpStream } from "../types";
import { addCommonArgs } from "../helpers/add-common-args";
import { buildEnv } from "../helpers/build-env";
import { buildRepoUrl } from "../helpers/build-repo-url";
import { cleanupTemporaryKeys } from "../helpers/cleanup-temporary-keys";

const normalizeDumpPath = (pathToDump?: string): string => {
	const trimmedPath = pathToDump?.trim();
	if (!trimmedPath) {
		return "/";
	}

	return normalizeAbsolutePath(trimmedPath);
};

export const dump = async (
	config: RepositoryConfig,
	snapshotRef: string,
	options: {
		organizationId: string;
		path?: string;
		archive?: false;
	},
): Promise<ResticDumpStream> => {
	const repoUrl = buildRepoUrl(config);
	const env = await buildEnv(config, options.organizationId);
	const pathToDump = normalizeDumpPath(options.path);

	const args: string[] = ["--repo", repoUrl, "dump", snapshotRef, pathToDump];

	if (options.archive !== false) {
		args.push("--archive", "tar");
	}

	addCommonArgs(args, env, config, { includeJson: false });

	logger.debug(`Executing: restic ${args.join(" ")}`);

	let didCleanup = false;
	const cleanup = async () => {
		if (didCleanup) {
			return;
		}

		didCleanup = true;
		await cleanupTemporaryKeys(env);
	};

	let stream: Readable | null = null;
	let abortController: AbortController | null = new AbortController();

	const maxStderrChars = 64 * 1024;
	let stderrTail = "";

	const completion = safeSpawn({
		command: "restic",
		args,
		env,
		signal: abortController.signal,
		stdoutMode: "raw",
		onSpawn: (child) => {
			stream = child.stdout;
		},
		onStderr: (line) => {
			const chunk = line.trim();
			if (chunk) {
				stderrTail += `${line}\n`;
				if (stderrTail.length > maxStderrChars) {
					stderrTail = stderrTail.slice(-maxStderrChars);
				}
			}
		},
	})
		.then((result) => {
			if (result.exitCode === 0) {
				return;
			}

			const stderr = stderrTail.trim() || result.error;
			logger.error(`Restic dump failed: ${stderr}`);
			throw new ResticError(result.exitCode, stderr);
		})
		.finally(async () => {
			abortController = null;
			await cleanup();
		});

	completion.catch(() => {});
	const completionPromise = new Promise<void>((resolve, reject) => completion.then(resolve, reject));

	if (!stream) {
		await cleanup();
		throw new Error("Failed to initialize restic dump stream");
	}

	return {
		stream,
		completion: completionPromise,
		abort: () => {
			if (abortController) {
				abortController.abort();
			}
		},
	};
};