using Quartz;

internal static class MonitoringCheckJobs
{
    public const string JobGroup = "monitoring-checks";
    public const string TriggerGroup = "monitoring-check-triggers";

    public static JobKey JobKeyFor(int monitorId) => new($"check-{monitorId}", JobGroup);
}

internal sealed class MonitoringCheckScheduler(ISchedulerFactory schedulerFactory, ScreeningRepository repository)
{
    public Task TriggerAsync(int monitorId, CancellationToken cancellationToken = default) =>
        ScheduleCheckAsync(monitorId, force: true, cancellationToken);

    public async Task<int> TriggerAllAsync(CancellationToken cancellationToken = default)
    {
        var monitors = await repository.GetNonArchivedMonitoringsAsync(cancellationToken);
        foreach (var monitor in monitors)
        {
            await ScheduleCheckAsync(monitor.Id, force: true, cancellationToken);
        }

        return monitors.Count;
    }

    public async Task ScheduleRunningChecksAsync(CancellationToken cancellationToken = default)
    {
        var monitors = await repository.GetActiveMonitoringsAsync(cancellationToken);
        foreach (var monitor in monitors)
        {
            await ScheduleCheckAsync(monitor.Id, force: false, cancellationToken);
        }
    }

    private async Task ScheduleCheckAsync(int monitorId, bool force, CancellationToken cancellationToken)
    {
        var scheduler = await schedulerFactory.GetScheduler(cancellationToken);
        var jobKey = MonitoringCheckJobs.JobKeyFor(monitorId);

        if (!await scheduler.CheckExists(jobKey, cancellationToken))
        {
            await scheduler.AddJob(
                JobBuilder
                    .Create<CheckMonitoringJob>()
                    .WithIdentity(jobKey)
                    .UsingJobData("monitorId", monitorId)
                    .StoreDurably()
                    .RequestRecovery()
                    .Build(),
                replace: true,
                cancellationToken
            );
        }

        var trigger = TriggerBuilder
            .Create()
            .ForJob(jobKey)
            .WithIdentity(new TriggerKey($"{monitorId}-{Guid.NewGuid():N}", MonitoringCheckJobs.TriggerGroup))
            .UsingJobData("force", force)
            .StartNow()
            .Build();

        await scheduler.ScheduleJob(trigger, cancellationToken);
    }
}