using LiteCharms.Features.Abstractions;
using LiteCharms.Features.Quartz.Abstractions;
using static LiteCharms.Features.Extensions.Timezones;

namespace LiteCharms.Features.Quartz;

public class JobOrchestrator(ISchedulerFactory schedulerFactory) : IJobOrchestrator
{
    public async Task SendAsync<TNotification>(TNotification notification, CancellationToken cancellationToken = default)
        where TNotification : IEvent
    {
        var chainedJobGroup = "onetime-jobs";

        var scheduler = await schedulerFactory.GetScheduler(cancellationToken);
        var jobKey = new JobKey($"{notification.Name.ToLower()}-{notification.CorrelationId.ToLower()}", chainedJobGroup);
        var triggerKey = new TriggerKey($"{jobKey.Name}-trigger", chainedJobGroup);

        var job = JobBuilder.Create<MediatorJob<TNotification>>()
            .WithIdentity(jobKey)
            .WithDescription($"Correlation ID: {notification.CorrelationId}")
            .UsingJobData(new JobDataMap { ["Payload"] = JsonSerializer.Serialize(notification) })
            .DisallowConcurrentExecution()
            .Build();

        var trigger = global::Quartz.TriggerBuilder.Create()
            .WithIdentity(triggerKey)
            .StartNow()
            .Build();

        await scheduler.ScheduleJob(job, new List<ITrigger> { trigger }.AsReadOnly(), replace: true, cancellationToken);
    }

    public async Task ScheduleAsync<TNotification>(TNotification notification, string cronExpression, CancellationToken cancellationToken = default)
        where TNotification : IEvent
    {
        var chainedJobGroup = "scheduled-jobs";

        var scheduler = await schedulerFactory.GetScheduler(cancellationToken);
        var jobKey = new JobKey($"{notification.Name.ToLower()}", chainedJobGroup);
        var triggerKey = new TriggerKey($"{jobKey.Name}-trigger", chainedJobGroup);

        var job = JobBuilder.Create<MediatorJob<TNotification>>()
            .WithIdentity(jobKey)
            .WithDescription($"Correlation ID: {notification.CorrelationId}")
            .UsingJobData(new JobDataMap { ["Payload"] = JsonSerializer.Serialize(notification) })
            .DisallowConcurrentExecution()
            .StoreDurably()
            .Build();

        var now = SouthAfricanTimeZone.UtcNow();

        var trigger = global::Quartz.TriggerBuilder.Create()
            .WithIdentity(triggerKey)
            .WithDescription($"Scheduled via Main Job at {now:g}")
            .WithCronSchedule(cronExpression, cron => cron.InTimeZone(SouthAfricanTimeZone)
                .WithMisfireHandlingInstructionFireAndProceed())
            .StartAt(now)
            .Build();

        await scheduler.AddJob(job, replace: true, cancellationToken);

        if (await scheduler.CheckExists(triggerKey, cancellationToken))
            await scheduler.RescheduleJob(triggerKey, trigger, cancellationToken);
        else
            await scheduler.ScheduleJob(job, new List<ITrigger> { trigger }.AsReadOnly(), replace: true, cancellationToken);
    }
}