using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Extensions;
using Iceshrimp.Backend.Core.Middleware;

namespace Iceshrimp.Backend.Controllers.Mastodon.Renderers;

public class NotificationRenderer(NoteRenderer noteRenderer, UserRenderer userRenderer)
{
	public async Task<NotificationEntity> RenderAsync(
		Notification notification, User user, List<AccountEntity>? accounts = null,
		IEnumerable<StatusEntity>? statuses = null
	)
	{
		var dbNotifier = notification.Notifier ?? throw new GracefulException("Notification has no notifier");

		var targetNote = notification.Type == Notification.NotificationType.Renote
			? notification.Note?.Renote
			: notification.Note;

		var note = targetNote != null
			? statuses?.FirstOrDefault(p => p.Id == targetNote.Id) ??
			  await noteRenderer.RenderAsync(targetNote, user, accounts)
			: null;

		var notifier = accounts?.FirstOrDefault(p => p.Id == dbNotifier.Id) ??
		               await userRenderer.RenderAsync(dbNotifier);

		var res = new NotificationEntity
		{
			Id        = notification.Id,
			Type      = NotificationEntity.EncodeType(notification.Type),
			Note      = note,
			Notifier  = notifier,
			CreatedAt = notification.CreatedAt.ToStringIso8601Like()
		};

		return res;
	}

	public async Task<IEnumerable<NotificationEntity>> RenderManyAsync(
		IEnumerable<Notification> notifications, User user
	)
	{
		var notificationList = notifications.ToList();

		var accounts = await noteRenderer.GetAccounts(notificationList.Where(p => p.Notifier != null)
		                                                              .Select(p => p.Notifier)
		                                                              .Concat(notificationList.Select(p => p.Notifiee))
		                                                              .Concat(notificationList
		                                                                      .Select(p => p.Note?.Renote?.User)
		                                                                      .Where(p => p != null))
		                                                              .Cast<User>()
		                                                              .DistinctBy(p => p.Id));

		var notes = await noteRenderer.RenderManyAsync(notificationList.Where(p => p.Note != null)
		                                                               .Select(p => p.Note)
		                                                               .Concat(notificationList
		                                                                       .Select(p => p.Note?.Renote)
		                                                                       .Where(p => p != null))
		                                                               .Cast<Note>()
		                                                               .DistinctBy(p => p.Id), user, accounts);

		return await notificationList
		             .Select(p => RenderAsync(p, user, accounts, notes))
		             .AwaitAllAsync();
	}
}