Iceshrimp.NET/Iceshrimp.Backend/Core/Services/NotificationService.cs

using System.Diagnostics.CodeAnalysis;
using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Helpers;
using Microsoft.EntityFrameworkCore;

namespace Iceshrimp.Backend.Core.Services;

public class NotificationService(
	[SuppressMessage("ReSharper", "SuggestBaseTypeForParameterInConstructor")]
	DatabaseContext db,
	EventService eventSvc
)
{
	public async Task GenerateMentionNotifications(Note note, IReadOnlyCollection<string> mentionedLocalUserIds)
	{
		if (mentionedLocalUserIds.Count == 0) return;

		var notifications = mentionedLocalUserIds
		                    .Where(p => p != note.UserId)
		                    .Select(p => new Notification
		                    {
			                    Id         = IdHelpers.GenerateSlowflakeId(),
			                    CreatedAt  = DateTime.UtcNow,
			                    Note       = note,
			                    NotifierId = note.UserId,
			                    NotifieeId = p,
			                    Type       = Notification.NotificationType.Mention
		                    })
		                    .ToList();

		await db.AddRangeAsync(notifications);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotifications(this, notifications);
	}

	public async Task GenerateReplyNotifications(Note note, IReadOnlyCollection<string> mentionedLocalUserIds)
	{
		if (note.Visibility != Note.NoteVisibility.Specified) return;
		if (note.VisibleUserIds.Count == 0) return;

		var users = mentionedLocalUserIds.Concat(note.VisibleUserIds).Distinct().Except(mentionedLocalUserIds).ToList();
		if (users.Count == 0) return;

		var notifications = users
		                    .Where(p => p != note.UserId)
		                    .Select(p => new Notification
		                    {
			                    Id         = IdHelpers.GenerateSlowflakeId(),
			                    CreatedAt  = DateTime.UtcNow,
			                    Note       = note,
			                    NotifierId = note.UserId,
			                    NotifieeId = p,
			                    Type       = Notification.NotificationType.Reply
		                    })
		                    .ToList();

		await db.AddRangeAsync(notifications);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotifications(this, notifications);
	}

	[SuppressMessage("ReSharper", "EntityFramework.UnsupportedServerSideFunctionCall",
	                 Justification = "Projectable functions are very much translatable")]
	public async Task GenerateEditNotifications(Note note)
	{
		var notifications = await db.Users
		                            .Where(p => p.Host == null && p != note.User && p.HasInteractedWith(note))
		                            .Select(p => new Notification
		                            {
			                            Id         = IdHelpers.GenerateSlowflakeId(DateTime.UtcNow),
			                            CreatedAt  = DateTime.UtcNow,
			                            Note       = note,
			                            NotifierId = note.UserId,
			                            Notifiee   = p,
			                            Type       = Notification.NotificationType.Edit
		                            })
		                            .ToListAsync();

		if (notifications.Count == 0)
			return;

		await db.AddRangeAsync(notifications);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotifications(this, notifications);
	}

	public async Task GenerateLikeNotification(Note note, User user)
	{
		if (note.UserHost != null) return;
		if (note.User == user) return;

		var notification = new Notification
		{
			Id        = IdHelpers.GenerateSlowflakeId(),
			CreatedAt = DateTime.UtcNow,
			Note      = note,
			Notifiee  = note.User,
			Notifier  = user,
			Type      = Notification.NotificationType.Like
		};

		await db.AddAsync(notification);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotification(this, notification);
	}

	public async Task GenerateFollowNotification(User follower, User followee)
	{
		if (followee.Host != null) return;

		var notification = new Notification
		{
			Id        = IdHelpers.GenerateSlowflakeId(),
			CreatedAt = DateTime.UtcNow,
			Notifiee  = followee,
			Notifier  = follower,
			Type      = Notification.NotificationType.Follow
		};

		await db.AddAsync(notification);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotification(this, notification);
	}

	public async Task GenerateFollowRequestReceivedNotification(FollowRequest followRequest)
	{
		if (followRequest.FolloweeHost != null) return;

		var notification = new Notification
		{
			Id            = IdHelpers.GenerateSlowflakeId(),
			CreatedAt     = DateTime.UtcNow,
			FollowRequest = followRequest,
			Notifier      = followRequest.Follower,
			Notifiee      = followRequest.Followee,
			Type          = Notification.NotificationType.FollowRequestReceived
		};

		await db.AddAsync(notification);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotification(this, notification);
	}

	public async Task GenerateFollowRequestAcceptedNotification(FollowRequest followRequest)
	{
		if (followRequest.FollowerHost != null) return;
		if (!followRequest.Followee.IsLocked) return;

		var notification = new Notification
		{
			Id            = IdHelpers.GenerateSlowflakeId(),
			CreatedAt     = DateTime.UtcNow,
			FollowRequest = followRequest,
			Notifier      = followRequest.Followee,
			Notifiee      = followRequest.Follower,
			Type          = Notification.NotificationType.FollowRequestAccepted
		};

		await db.AddAsync(notification);
		await db.SaveChangesAsync();
		eventSvc.RaiseNotification(this, notification);
	}
}