using Iceshrimp.Backend.Controllers.Attributes;
using Iceshrimp.Backend.Controllers.Mastodon.Attributes;
using Iceshrimp.Backend.Controllers.Mastodon.Renderers;
using Iceshrimp.Backend.Controllers.Mastodon.Schemas;
using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Extensions;
using Iceshrimp.Backend.Core.Middleware;
using Microsoft.AspNetCore.Cors;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.RateLimiting;
using static Iceshrimp.Backend.Core.Database.Tables.Notification;

namespace Iceshrimp.Backend.Controllers.Mastodon;

[MastodonApiController]
[Route("/api/v1/notifications")]
[Authenticate]
[EnableCors("mastodon")]
[EnableRateLimiting("sliding")]
[Produces("application/json")]
public class NotificationController(DatabaseContext db, NotificationRenderer notificationRenderer) : Controller
{
	[HttpGet]
	[Authorize("read:notifications")]
	[LinkPagination(40, 80)]
	[Produces("application/json")]
	[ProducesResponseType(StatusCodes.Status200OK, Type = typeof(List<NotificationEntity>))]
	[ProducesResponseType(StatusCodes.Status404NotFound, Type = typeof(MastodonErrorResponse))]
	public async Task<IActionResult> GetNotifications(MastodonPaginationQuery query)
	{
		var user = HttpContext.GetUserOrFail();
		var res = await db.Notifications
		                  .IncludeCommonProperties()
		                  .Where(p => p.Notifiee == user)
		                  .Where(p => p.Type == NotificationType.Follow ||
		                              p.Type == NotificationType.Mention ||
		                              p.Type == NotificationType.Reply ||
		                              p.Type == NotificationType.Renote ||
		                              p.Type == NotificationType.Quote ||
		                              p.Type == NotificationType.Like ||
		                              p.Type == NotificationType.PollEnded ||
		                              p.Type == NotificationType.FollowRequestReceived || 
		                              p.Type == NotificationType.Edit)
		                  .EnsureNoteVisibilityFor(p => p.Note, user)
		                  .FilterBlocked(p => p.Notifier, user)
		                  .FilterBlocked(p => p.Note, user)
		                  .Paginate(query, ControllerContext)
		                  .RenderAllForMastodonAsync(notificationRenderer, user);

		//TODO: handle mutes
		//TODO: handle reply/renote visibility

		return Ok(res);
	}
}