463 lines
18 KiB
C#
463 lines
18 KiB
C#
using Microsoft.EntityFrameworkCore;
|
|
using SharedLibraryCore.Database;
|
|
using SharedLibraryCore.Database.Models;
|
|
using SharedLibraryCore.Objects;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using static SharedLibraryCore.Database.Models.EFClient;
|
|
|
|
namespace SharedLibraryCore.Services
|
|
{
|
|
|
|
public class ClientService : Interfaces.IEntityService<EFClient>
|
|
{
|
|
public async Task<EFClient> Create(EFClient entity)
|
|
{
|
|
using (var context = new DatabaseContext())
|
|
{
|
|
var client = new EFClient()
|
|
{
|
|
Level = Permission.User,
|
|
FirstConnection = DateTime.UtcNow,
|
|
LastConnection = DateTime.UtcNow,
|
|
NetworkId = entity.NetworkId,
|
|
AliasLink = new EFAliasLink()
|
|
};
|
|
|
|
client.CurrentAlias = new Alias()
|
|
{
|
|
Name = entity.Name,
|
|
Link = client.AliasLink,
|
|
DateAdded = DateTime.UtcNow,
|
|
IPAddress = entity.IPAddress,
|
|
};
|
|
|
|
context.Clients.Add(client);
|
|
await context.SaveChangesAsync();
|
|
|
|
return client;
|
|
}
|
|
}
|
|
|
|
private async Task UpdateAlias(string name, int? ip, EFClient entity, DatabaseContext context)
|
|
{
|
|
// entity is the tracked db context item
|
|
// get all aliases by IP address and LinkId
|
|
var iqAliases = context.Aliases
|
|
.Include(a => a.Link)
|
|
// we only want alias that have the same IP address or share a link
|
|
.Where(_alias => _alias.IPAddress == ip || (_alias.LinkId == entity.AliasLinkId && _alias.Active));
|
|
|
|
#if DEBUG == true
|
|
var aliasSql = iqAliases.ToSql();
|
|
#endif
|
|
var aliases = await iqAliases.ToListAsync();
|
|
|
|
// see if they have a matching IP + Name but new NetworkId
|
|
var existingExactAlias = aliases.FirstOrDefault(a => a.Name == name && a.IPAddress == ip);
|
|
bool hasExactAliasMatch = existingExactAlias != null;
|
|
|
|
// if existing alias matches link them
|
|
var newAliasLink = existingExactAlias?.Link;
|
|
// if no exact matches find the first IP or LinkId that matches
|
|
newAliasLink = newAliasLink ?? aliases.FirstOrDefault()?.Link;
|
|
// if no matches are found, use our current one ( it will become permanent )
|
|
newAliasLink = newAliasLink ?? entity.AliasLink;
|
|
|
|
bool hasExistingAlias = aliases.Count > 0;
|
|
bool isAliasLinkUpdated = newAliasLink.AliasLinkId != entity.AliasLink.AliasLinkId;
|
|
|
|
// this happens when the link we found is different than the one we create before adding an IP
|
|
if (isAliasLinkUpdated)
|
|
{
|
|
entity.CurrentServer.Logger.WriteDebug($"found a link for {entity} so we are updating link from {entity.AliasLink.AliasLinkId} to {newAliasLink.AliasLinkId}");
|
|
|
|
var oldAliasLink = entity.AliasLink;
|
|
|
|
entity.AliasLink = newAliasLink;
|
|
entity.AliasLinkId = newAliasLink.AliasLinkId;
|
|
|
|
// update all previous aliases
|
|
await context.Aliases
|
|
.Where(_alias => _alias.LinkId == oldAliasLink.AliasLinkId)
|
|
.ForEachAsync(_alias => { _alias.LinkId = newAliasLink.AliasLinkId; _alias.Active = true; });
|
|
|
|
await context.SaveChangesAsync();
|
|
// we want to delete the now inactive alias
|
|
context.AliasLinks.Remove(oldAliasLink);
|
|
await context.SaveChangesAsync();
|
|
}
|
|
|
|
// the existing alias matches ip and name, so we can just ignore the temporary one
|
|
if (hasExactAliasMatch)
|
|
{
|
|
entity.CurrentServer.Logger.WriteDebug($"{entity} has exact alias match");
|
|
|
|
var oldAlias = entity.CurrentAlias;
|
|
entity.CurrentAliasId = existingExactAlias.AliasId;
|
|
entity.CurrentAlias = existingExactAlias;
|
|
await context.SaveChangesAsync();
|
|
|
|
// the alias is the same so we can just remove it
|
|
if (oldAlias.AliasId != existingExactAlias.AliasId)
|
|
{
|
|
context.Aliases.Remove(oldAlias);
|
|
await context.SaveChangesAsync();
|
|
}
|
|
}
|
|
|
|
// theres no exact match, but they've played before with the GUID or IP
|
|
else if (hasExistingAlias)
|
|
{
|
|
entity.CurrentServer.Logger.WriteDebug($"Connecting player is using a new alias {entity}");
|
|
|
|
// this happens when a temporary alias gets updated
|
|
if (entity.CurrentAlias.Name == name && entity.CurrentAlias.IPAddress == null)
|
|
{
|
|
entity.CurrentAlias.IPAddress = ip;
|
|
await context.SaveChangesAsync();
|
|
}
|
|
|
|
else
|
|
{
|
|
var newAlias = new EFAlias()
|
|
{
|
|
DateAdded = DateTime.UtcNow,
|
|
IPAddress = ip,
|
|
LinkId = newAliasLink.AliasLinkId,
|
|
Name = name,
|
|
Active = true,
|
|
};
|
|
|
|
entity.CurrentAlias = newAlias;
|
|
await context.SaveChangesAsync();
|
|
}
|
|
}
|
|
|
|
// no record of them playing
|
|
else
|
|
{
|
|
entity.AliasLink.Active = true;
|
|
entity.CurrentAlias.Active = true;
|
|
entity.CurrentAlias.IPAddress = ip;
|
|
entity.CurrentAlias.Name = name;
|
|
|
|
await context.SaveChangesAsync();
|
|
}
|
|
|
|
//var linkIds = aliases.Select(a => a.LinkId);
|
|
|
|
//if (linkIds.Count() > 0 &&
|
|
// aliases.Count(_alias => _alias.Name == name && _alias.IPAddress == ip) > 0)
|
|
//{
|
|
// var highestLevel = await context.Clients
|
|
// .Where(c => linkIds.Contains(c.AliasLinkId))
|
|
// .MaxAsync(c => c.Level);
|
|
|
|
// if (entity.Level != highestLevel)
|
|
// {
|
|
// entity.CurrentServer.Logger.WriteDebug($"{entity} updating user level");
|
|
// // todo: log level changes here
|
|
// context.Update(entity);
|
|
// entity.SetLevel(highestLevel, Utilities.IW4MAdminClient(entity.CurrentServer));
|
|
// await context.SaveChangesAsync();
|
|
// }
|
|
//}
|
|
}
|
|
|
|
/// <summary>
|
|
/// updates the permission level of the given target to the given permission level
|
|
/// </summary>
|
|
/// <param name="newPermission"></param>
|
|
/// <param name="temporalClient"></param>
|
|
/// <param name="origin"></param>
|
|
/// <param name="ctx"></param>
|
|
/// <returns></returns>
|
|
public async Task UpdateLevel(Permission newPermission, EFClient temporalClient, EFClient origin)
|
|
{
|
|
using (var ctx = new DatabaseContext())
|
|
{
|
|
var entity = await ctx.Clients
|
|
.Where(_client => _client.AliasLinkId == temporalClient.AliasLinkId)
|
|
.FirstAsync();
|
|
|
|
// if their permission level has been changed to level that needs to be updated on all accounts
|
|
if ((entity.Level != newPermission) &&
|
|
(newPermission == Permission.Banned ||
|
|
newPermission == Permission.Flagged ||
|
|
newPermission == Permission.User))
|
|
{
|
|
var changeSvc = new ChangeHistoryService();
|
|
|
|
// get all clients that have the same linkId
|
|
var iqMatchingClients = ctx.Clients
|
|
.Where(_client => _client.AliasLinkId == entity.AliasLinkId)
|
|
// make sure we don't select ourselves twice
|
|
.Where(_client => _client.ClientId != temporalClient.ClientId);
|
|
|
|
var matchingClients = await iqMatchingClients.ToListAsync();
|
|
|
|
// this updates the level for all the clients with the same LinkId
|
|
// only if their new level is flagged or banned
|
|
foreach (var client in matchingClients)
|
|
{
|
|
client.Level = newPermission;
|
|
// hack this saves our change to the change history log
|
|
await changeSvc.Add(new GameEvent()
|
|
{
|
|
Type = GameEvent.EventType.ChangePermission,
|
|
Extra = newPermission,
|
|
Origin = origin,
|
|
Target = client
|
|
}, ctx);
|
|
}
|
|
}
|
|
|
|
entity.Level = newPermission;
|
|
await ctx.SaveChangesAsync();
|
|
}
|
|
|
|
temporalClient.Level = newPermission;
|
|
}
|
|
|
|
public async Task<EFClient> Delete(EFClient entity)
|
|
{
|
|
using (var context = new DatabaseContext())
|
|
{
|
|
var client = context.Clients
|
|
.Single(e => e.ClientId == entity.ClientId);
|
|
entity.Active = false;
|
|
context.Entry(entity).State = EntityState.Modified;
|
|
await context.SaveChangesAsync();
|
|
return entity;
|
|
}
|
|
}
|
|
|
|
public async Task<IList<EFClient>> Find(Func<EFClient, bool> e)
|
|
{
|
|
return await Task.Run(() =>
|
|
{
|
|
using (var context = new DatabaseContext(true))
|
|
{
|
|
return context.Clients
|
|
.Include(c => c.CurrentAlias)
|
|
.Include(c => c.AliasLink.Children)
|
|
.Where(e).ToList();
|
|
}
|
|
});
|
|
}
|
|
|
|
public async Task<EFClient> Get(int entityID)
|
|
{
|
|
using (var context = new DatabaseContext(true))
|
|
{
|
|
var iqClient = from client in context.Clients
|
|
.Include(c => c.CurrentAlias)
|
|
.Include(c => c.AliasLink.Children)
|
|
.Include(c => c.Meta)
|
|
where client.ClientId == entityID
|
|
select new
|
|
{
|
|
Client = client,
|
|
LinkedAccounts = (from linkedClient in context.Clients
|
|
where client.AliasLinkId == linkedClient.AliasLinkId
|
|
select new
|
|
{
|
|
linkedClient.ClientId,
|
|
linkedClient.NetworkId
|
|
})
|
|
};
|
|
#if DEBUG == true
|
|
var clientSql = iqClient.ToSql();
|
|
#endif
|
|
var foundClient = await iqClient.FirstOrDefaultAsync();
|
|
|
|
if (foundClient == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
foundClient.Client.LinkedAccounts = new Dictionary<int, long>();
|
|
// todo: find out the best way to do this
|
|
// I'm doing this here because I don't know the best way to have multiple awaits in the query
|
|
foreach (var linked in foundClient.LinkedAccounts)
|
|
{
|
|
foundClient.Client.LinkedAccounts.Add(linked.ClientId, linked.NetworkId);
|
|
}
|
|
|
|
return foundClient.Client;
|
|
}
|
|
}
|
|
|
|
private static readonly Func<DatabaseContext, long, Task<EFClient>> _getUniqueQuery =
|
|
EF.CompileAsyncQuery((DatabaseContext context, long networkId) =>
|
|
context.Clients
|
|
.Include(c => c.CurrentAlias)
|
|
.Include(c => c.AliasLink.Children)
|
|
.Include(c => c.ReceivedPenalties)
|
|
.FirstOrDefault(c => c.NetworkId == networkId)
|
|
);
|
|
|
|
public async Task<EFClient> GetUnique(long entityAttribute)
|
|
{
|
|
using (var context = new DatabaseContext(true))
|
|
{
|
|
return await _getUniqueQuery(context, entityAttribute);
|
|
}
|
|
}
|
|
|
|
public async Task UpdateAlias(EFClient temporalClient)
|
|
{
|
|
using (var context = new DatabaseContext())
|
|
{
|
|
var entity = context.Clients
|
|
.Include(c => c.AliasLink)
|
|
.Include(c => c.CurrentAlias)
|
|
.First(e => e.ClientId == temporalClient.ClientId);
|
|
|
|
entity.CurrentServer = temporalClient.CurrentServer;
|
|
|
|
await UpdateAlias(temporalClient.Name, temporalClient.IPAddress, entity, context);
|
|
|
|
temporalClient.CurrentAlias = entity.CurrentAlias;
|
|
temporalClient.CurrentAliasId = entity.CurrentAliasId;
|
|
temporalClient.AliasLink = entity.AliasLink;
|
|
temporalClient.AliasLinkId = entity.AliasLinkId;
|
|
}
|
|
}
|
|
|
|
public async Task<EFClient> Update(EFClient temporalClient)
|
|
{
|
|
using (var context = new DatabaseContext())
|
|
{
|
|
// grab the context version of the entity
|
|
var entity = context.Clients
|
|
.First(client => client.ClientId == temporalClient.ClientId);
|
|
|
|
entity.LastConnection = temporalClient.LastConnection;
|
|
entity.Connections = temporalClient.Connections;
|
|
entity.FirstConnection = temporalClient.FirstConnection;
|
|
entity.Masked = temporalClient.Masked;
|
|
entity.TotalConnectionTime = temporalClient.TotalConnectionTime;
|
|
entity.Password = temporalClient.Password;
|
|
entity.PasswordSalt = temporalClient.PasswordSalt;
|
|
|
|
// update in database
|
|
await context.SaveChangesAsync();
|
|
|
|
// this is set so future updates don't trigger a new alias add
|
|
if (temporalClient.CurrentAlias.AliasId == 0)
|
|
{
|
|
temporalClient.CurrentAlias.AliasId = entity.CurrentAlias.AliasId;
|
|
}
|
|
|
|
return entity;
|
|
}
|
|
}
|
|
|
|
#region ServiceSpecific
|
|
public async Task<IList<EFClient>> GetOwners()
|
|
{
|
|
using (var context = new DatabaseContext())
|
|
{
|
|
return await context.Clients
|
|
.Where(c => c.Level == Permission.Owner)
|
|
.ToListAsync();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// retrieves the number of owners
|
|
/// (client level is owner)
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public async Task<int> GetOwnerCount()
|
|
{
|
|
using (var ctx = new DatabaseContext(true))
|
|
{
|
|
return await ctx.Clients.AsNoTracking()
|
|
.CountAsync(_client => _client.Level == Permission.Owner);
|
|
}
|
|
}
|
|
|
|
public async Task<List<EFClient>> GetPrivilegedClients()
|
|
{
|
|
using (var context = new DatabaseContext(disableTracking: true))
|
|
{
|
|
var iqClients = from client in context.Clients.AsNoTracking()
|
|
where client.Level >= Permission.Trusted
|
|
where client.Active
|
|
select new EFClient()
|
|
{
|
|
AliasLinkId = client.AliasLinkId,
|
|
CurrentAlias = client.CurrentAlias,
|
|
ClientId = client.ClientId,
|
|
Level = client.Level,
|
|
Password = client.Password,
|
|
PasswordSalt = client.PasswordSalt,
|
|
NetworkId = client.NetworkId,
|
|
LastConnection = client.LastConnection
|
|
};
|
|
|
|
#if DEBUG == true
|
|
var clientsSql = iqClients.ToSql();
|
|
#endif
|
|
|
|
return await iqClients.ToListAsync();
|
|
}
|
|
}
|
|
|
|
public async Task<IList<EFClient>> FindClientsByIdentifier(string identifier)
|
|
{
|
|
if (identifier.Length < 3)
|
|
{
|
|
return new List<EFClient>();
|
|
}
|
|
|
|
identifier = identifier.ToLower();
|
|
|
|
using (var context = new DatabaseContext(disableTracking: true))
|
|
{
|
|
long networkId = identifier.ConvertLong();
|
|
int? ipAddress = identifier.ConvertToIP();
|
|
|
|
var iqLinkIds = (from alias in context.Aliases
|
|
where (alias.IPAddress != null && alias.IPAddress == ipAddress) ||
|
|
alias.Name.ToLower().Contains(identifier)
|
|
select alias.LinkId).Distinct();
|
|
|
|
var linkIds = iqLinkIds.ToList();
|
|
|
|
var iqClients = context.Clients
|
|
.Where(c => linkIds.Contains(c.AliasLinkId) ||
|
|
networkId == c.NetworkId)
|
|
.Include(c => c.CurrentAlias)
|
|
.Include(c => c.AliasLink.Children);
|
|
|
|
#if DEBUG == true
|
|
var iqClientsSql = iqClients.ToSql();
|
|
#endif
|
|
|
|
return await iqClients.ToListAsync();
|
|
}
|
|
}
|
|
|
|
public async Task<int> GetTotalClientsAsync()
|
|
{
|
|
using (var context = new DatabaseContext(true))
|
|
{
|
|
return await context.Clients
|
|
.CountAsync();
|
|
}
|
|
}
|
|
|
|
public Task<EFClient> CreateProxy()
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
#endregion
|
|
}
|
|
}
|