using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Extensions;
using Iceshrimp.Backend.Core.Services;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Caching.Distributed;
using ProtoBuf;
using StackExchange.Redis;

namespace Iceshrimp.Backend.Core.Queues;

public class DeliverQueue
{
	public static JobQueue<DeliverJob> Create(IConnectionMultiplexer redis, string prefix)
	{
		return new JobQueue<DeliverJob>("deliver", DeliverQueueProcessorDelegateAsync, 4, redis, prefix);
	}

	private static async Task DeliverQueueProcessorDelegateAsync(
		DeliverJob job, IServiceProvider scope,
		CancellationToken token
	)
	{
		var logger     = scope.GetRequiredService<ILogger<DeliverQueue>>();
		var httpClient = scope.GetRequiredService<HttpClient>();
		var httpRqSvc  = scope.GetRequiredService<HttpRequestService>();
		var cache      = scope.GetRequiredService<IDistributedCache>();
		var db         = scope.GetRequiredService<DatabaseContext>();
		var fedCtrl    = scope.GetRequiredService<ActivityPub.FederationControlService>();

		if (await fedCtrl.ShouldBlockAsync(job.InboxUrl, job.RecipientHost))
		{
			logger.LogDebug("Refusing to deliver activity to blocked instance ({uri})", job.InboxUrl);
			return;
		}

		logger.LogDebug("Delivering activity to: {uri}", job.InboxUrl);

		var key = await cache.FetchAsync($"userPrivateKey:{job.UserId}", TimeSpan.FromMinutes(60), async () =>
		{
			var keypair =
				await db.UserKeypairs.FirstOrDefaultAsync(p => p.UserId == job.UserId, token);
			return keypair?.PrivateKey ?? throw new Exception($"Failed to get keypair for user {job.UserId}");
		});

		var request =
			await httpRqSvc.PostSignedAsync(job.InboxUrl, job.Payload, job.ContentType, job.UserId, key);
		await httpClient.SendAsync(request, token);
	}
}

[ProtoContract]
public class DeliverJob : Job
{
	[ProtoMember(1)] public required string InboxUrl;
	[ProtoMember(2)] public required string Payload;
	[ProtoMember(3)] public required string ContentType;

	[ProtoMember(10)] public required string UserId;
	[ProtoMember(11)] public required string RecipientHost;
}