using System; using System.Collections.Generic; using System.Linq; using System.Linq.Expressions; using System.Threading.Tasks; using Data.Abstractions; using Data.Models; using Microsoft.EntityFrameworkCore; using SharedLibraryCore.Configuration; using SharedLibraryCore.Dtos; using SharedLibraryCore.Interfaces; namespace SharedLibraryCore.Services { public class PenaltyService : IEntityService { private readonly ApplicationConfiguration _appConfig; private readonly IDatabaseContextFactory _contextFactory; public PenaltyService(IDatabaseContextFactory contextFactory, ApplicationConfiguration appConfig) { _contextFactory = contextFactory; _appConfig = appConfig; } public virtual async Task Create(EFPenalty newEntity) { await using var context = _contextFactory.CreateContext(); var penalty = new EFPenalty { Active = true, OffenderId = newEntity.Offender.ClientId, PunisherId = newEntity.Punisher.ClientId, LinkId = newEntity.Link.AliasLinkId, Type = newEntity.Type, Expires = newEntity.Expires, Offense = newEntity.Offense, When = DateTime.UtcNow, AutomatedOffense = newEntity.AutomatedOffense ?? newEntity.Punisher.AdministeredPenalties?.FirstOrDefault()?.AutomatedOffense, IsEvadedOffense = newEntity.IsEvadedOffense }; context.Penalties.Add(penalty); await context.SaveChangesAsync(); return newEntity; } public Task Delete(EFPenalty entity) { throw new NotImplementedException(); } public async Task> Find(Func expression) { throw await Task.FromResult(new Exception()); } public Task Get(int entityID) { throw new NotImplementedException(); } public Task GetUnique(long entityProperty) { throw new NotImplementedException(); } public Task Update(EFPenalty entity) { throw new NotImplementedException(); } public async Task> GetRecentPenalties(int count, int offset, EFPenalty.PenaltyType showOnly = EFPenalty.PenaltyType.Any, bool ignoreAutomated = true) { await using var context = _contextFactory.CreateContext(false); var iqPenalties = context.Penalties .Where(p => showOnly == EFPenalty.PenaltyType.Any ? p.Type != EFPenalty.PenaltyType.Any : p.Type == showOnly) .Where(_penalty => !ignoreAutomated || _penalty.PunisherId != 1) .OrderByDescending(p => p.When) .Skip(offset) .Take(count) .Select(_penalty => new PenaltyInfo { Id = _penalty.PenaltyId, Offense = _penalty.Offense, AutomatedOffense = _penalty.AutomatedOffense, OffenderId = _penalty.OffenderId, OffenderName = _penalty.Offender.CurrentAlias.Name, PunisherId = _penalty.PunisherId, PunisherName = _penalty.Punisher.CurrentAlias.Name, PunisherLevel = _penalty.Punisher.Level, PenaltyType = _penalty.Type, Expires = _penalty.Expires, TimePunished = _penalty.When, IsEvade = _penalty.IsEvadedOffense }); return await iqPenalties.ToListAsync(); } /// /// retrieves penalty information for meta service /// /// database id of the client /// how many items to retrieve /// not used /// retreive penalties older than this /// public async Task> GetClientPenaltyForMetaAsync(int clientId, int count, int offset, DateTime? startAt) { var linkedPenaltyType = Utilities.LinkedPenaltyTypes(); await using var context = _contextFactory.CreateContext(false); var linkId = await context.Clients.AsNoTracking() .Where(_penalty => _penalty.ClientId == clientId) .Select(_penalty => _penalty.AliasLinkId) .FirstOrDefaultAsync(); var iqPenalties = context.Penalties.AsNoTracking() .Where(_penalty => _penalty.OffenderId == clientId || _penalty.PunisherId == clientId || linkedPenaltyType.Contains(_penalty.Type) && _penalty.LinkId == linkId) .Where(_penalty => _penalty.When <= startAt) .OrderByDescending(_penalty => _penalty.When) .Skip(offset) .Take(count) .Select(_penalty => new PenaltyInfo { Id = _penalty.PenaltyId, Offense = _penalty.Offense, AutomatedOffense = _penalty.AutomatedOffense, OffenderId = _penalty.OffenderId, OffenderName = _penalty.Offender.CurrentAlias.Name, PunisherId = _penalty.PunisherId, PunisherName = _penalty.Punisher.CurrentAlias.Name, PunisherLevel = _penalty.Punisher.Level, PenaltyType = _penalty.Type, Expires = _penalty.Expires, TimePunished = _penalty.When, IsEvade = _penalty.IsEvadedOffense }); return await iqPenalties.Distinct().ToListAsync(); } public async Task> GetActivePenaltiesAsync(int linkId, int currentAliasId, int? ip = null, bool includePunisherName = false) { var now = DateTime.UtcNow; Expression> filter = p => new[] { EFPenalty.PenaltyType.TempBan, EFPenalty.PenaltyType.Ban, EFPenalty.PenaltyType.Flag }.Contains(p.Type) && p.Active && (p.Expires == null || p.Expires > now); await using var context = _contextFactory.CreateContext(false); IQueryable iqIpPenalties; if (_appConfig.EnableImplicitAccountLinking) { iqIpPenalties = context.Aliases .Where(a => a.IPAddress != null && a.IPAddress == ip) .SelectMany(a => a.Link.ReceivedPenalties) .Where(filter); } else { var usedIps = await context.Aliases.AsNoTracking() .Where(alias => (alias.LinkId == linkId || alias.AliasId == currentAliasId) && alias.IPAddress != null) .Select(alias => alias.IPAddress).ToListAsync(); var aliasedIds = await context.Aliases.AsNoTracking().Where(alias => usedIps.Contains(alias.IPAddress)) .Select(alias => alias.LinkId) .ToListAsync(); iqIpPenalties = context.Penalties.AsNoTracking() .Where(penalty => aliasedIds.Contains(penalty.LinkId) || penalty.LinkId == linkId) .Where(filter); } var activeIpPenalties = await iqIpPenalties.ToListAsync(); var activePenalties = activeIpPenalties.Distinct(); // this is a bit more performant in memory (ordering) return activePenalties.OrderByDescending(p => p.When).ToList(); } public virtual async Task RemoveActivePenalties(int aliasLinkId, int? ipAddress = null) { await using var context = _contextFactory.CreateContext(); var now = DateTime.UtcNow; var penaltiesByLink = context.Penalties .Where(p => p.LinkId == aliasLinkId) .Where(p => p.Expires > now || p.Expires == null); var penaltiesByIp = context.Penalties .Where(p => p.Offender.CurrentAlias.IPAddress != null && p.Offender.CurrentAlias.IPAddress == null) .Where(p => p.Expires > now || p.Expires == null); await penaltiesByLink.Union(penaltiesByIp).Distinct().ForEachAsync(p => { p.Active = false; p.Expires = now; }); await context.SaveChangesAsync(); } } }