123 lines
4.2 KiB
C#
123 lines
4.2 KiB
C#
using System.Text.RegularExpressions;
|
|
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
using PluralKit.Core;
|
|
|
|
namespace PluralKit.API;
|
|
|
|
public class PKControllerBase: ControllerBase
|
|
{
|
|
private readonly Guid _requestId = Guid.NewGuid();
|
|
private readonly Regex _shortIdRegex = new("^[a-z]{5}$");
|
|
private readonly Regex _snowflakeRegex = new("^[0-9]{17,19}$");
|
|
|
|
private List<PKMember>? _memberLookupCache { get; set; }
|
|
private List<PKGroup>? _groupLookupCache { get; set; }
|
|
|
|
protected readonly ApiConfig _config;
|
|
protected readonly IDatabase _db;
|
|
protected readonly ModelRepository _repo;
|
|
protected readonly DispatchService _dispatch;
|
|
|
|
public PKControllerBase(IServiceProvider svc)
|
|
{
|
|
_config = svc.GetRequiredService<ApiConfig>();
|
|
_db = svc.GetRequiredService<IDatabase>();
|
|
_repo = svc.GetRequiredService<ModelRepository>();
|
|
_dispatch = svc.GetRequiredService<DispatchService>();
|
|
}
|
|
|
|
protected Task<PKSystem?> ResolveSystem(string systemRef)
|
|
{
|
|
if (systemRef == "@me")
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null)
|
|
throw Errors.GenericAuthError;
|
|
return _repo.GetSystem((SystemId)systemId);
|
|
}
|
|
|
|
if (Guid.TryParse(systemRef, out var guid))
|
|
return _repo.GetSystemByGuid(guid);
|
|
|
|
if (_snowflakeRegex.IsMatch(systemRef))
|
|
return _repo.GetSystemByAccount(ulong.Parse(systemRef));
|
|
|
|
if (_shortIdRegex.IsMatch(systemRef))
|
|
return _repo.GetSystemByHid(systemRef);
|
|
|
|
return Task.FromResult<PKSystem?>(null);
|
|
}
|
|
|
|
protected async Task<PKMember?> ResolveMember(string memberRef, bool cache = false)
|
|
{
|
|
if (cache)
|
|
{
|
|
if (_memberLookupCache == null)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null)
|
|
throw new Exception("Authenticated user must not be null to use lookup cache!");
|
|
|
|
_memberLookupCache = await _repo.GetSystemMembers((SystemId)systemId).ToListAsync();
|
|
}
|
|
|
|
return _memberLookupCache.FirstOrDefault(x => x.Hid == memberRef || x.Uuid.ToString() == memberRef);
|
|
}
|
|
|
|
if (Guid.TryParse(memberRef, out var guid))
|
|
return await _repo.GetMemberByGuid(guid);
|
|
|
|
if (_shortIdRegex.IsMatch(memberRef))
|
|
return await _repo.GetMemberByHid(memberRef);
|
|
|
|
return null;
|
|
}
|
|
|
|
protected async Task<PKGroup?> ResolveGroup(string groupRef, bool cache = false)
|
|
{
|
|
if (cache)
|
|
{
|
|
if (_groupLookupCache == null)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null)
|
|
throw new Exception("Authenticated user must not be null to use lookup cache!");
|
|
|
|
_groupLookupCache = await _repo.GetSystemGroups((SystemId)systemId).ToListAsync();
|
|
}
|
|
|
|
return _groupLookupCache.FirstOrDefault(x => x.Hid == groupRef || x.Uuid.ToString() == groupRef);
|
|
}
|
|
|
|
if (Guid.TryParse(groupRef, out var guid))
|
|
return await _repo.GetGroupByGuid(guid);
|
|
|
|
if (_shortIdRegex.IsMatch(groupRef))
|
|
return await _repo.GetGroupByHid(groupRef);
|
|
|
|
return null;
|
|
}
|
|
|
|
protected LookupContext ContextFor(PKSystem system)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null) return LookupContext.ByNonOwner;
|
|
return (SystemId)systemId == system.Id ? LookupContext.ByOwner : LookupContext.ByNonOwner;
|
|
}
|
|
|
|
protected LookupContext ContextFor(PKMember member)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null) return LookupContext.ByNonOwner;
|
|
return (SystemId)systemId == member.System ? LookupContext.ByOwner : LookupContext.ByNonOwner;
|
|
}
|
|
|
|
protected LookupContext ContextFor(PKGroup group)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null) return LookupContext.ByNonOwner;
|
|
return (SystemId)systemId == group.System ? LookupContext.ByOwner : LookupContext.ByNonOwner;
|
|
}
|
|
} |