import { Job } from "../core/scheduler";
import { volumeService } from "../modules/volumes/volume.service";
import { logger } from "@zerobyte/core/node";
import { db } from "../db/db";
import { withContext } from "../core/request-context";

export class VolumeHealthCheckJob extends Job {
	async run() {
		logger.debug("Running health check for all volumes...");

		const volumes = await db.query.volumesTable.findMany({
			where: {
				OR: [{ status: "mounted" }, { status: "error" }],
			},
		});

		for (const volume of volumes) {
			try {
				await withContext({ organizationId: volume.organizationId }, async () => {
					await volumeService.ensureHealthyVolume(volume.shortId);
				});
			} catch (error) {
				logger.error(`Health check failed for volume ${volume.name}:`, error);
			}
		}

		return { done: true, timestamp: new Date() };
	}
}
