PluralKit/PluralKit.API/Controllers/PKControllerBase.cs

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}$");

    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 Task<PKMember?> ResolveMember(string memberRef)
    {
        if (Guid.TryParse(memberRef, out var guid))
            return _repo.GetMemberByGuid(guid);

        if (_shortIdRegex.IsMatch(memberRef))
            return _repo.GetMemberByHid(memberRef);

        return Task.FromResult<PKMember?>(null);
    }

    protected Task<PKGroup?> ResolveGroup(string groupRef)
    {
        if (Guid.TryParse(groupRef, out var guid))
            return _repo.GetGroupByGuid(guid);

        if (_shortIdRegex.IsMatch(groupRef))
            return _repo.GetGroupByHid(groupRef);

        return Task.FromResult<PKGroup?>(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;
    }
}