using System; using System.Net; using System.Text.RegularExpressions; using System.Threading.Tasks; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.DependencyInjection; using NodaTime; using PluralKit.Core; namespace PluralKit.API { public class PKControllerBase: ControllerBase { private readonly Guid _requestId = Guid.NewGuid(); private readonly Regex _shortIdRegex = new Regex("^[a-z]{5}$"); private readonly Regex _snowflakeRegex = new Regex("^[0-9]{17,19}$"); protected readonly ApiConfig _config; protected readonly IDatabase _db; protected readonly ModelRepository _repo; public PKControllerBase(IServiceProvider svc) { _config = svc.GetRequiredService(); _db = svc.GetRequiredService(); _repo = svc.GetRequiredService(); } protected Task ResolveSystem(string systemRef) { if (systemRef == "@me") { HttpContext.Items.TryGetValue("SystemId", out var systemId); if (systemId == null) throw APIErrors.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 null; } protected Task ResolveMember(string memberRef) { if (Guid.TryParse(memberRef, out var guid)) return _repo.GetMemberByGuid(guid); if (_shortIdRegex.IsMatch(memberRef)) return _repo.GetMemberByHid(memberRef); return null; } protected Task ResolveGroup(string groupRef) { if (Guid.TryParse(groupRef, out var guid)) return _repo.GetGroupByGuid(guid); if (_shortIdRegex.IsMatch(groupRef)) return _repo.GetGroupByHid(groupRef); return null; } public 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; } public 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; } public 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; } } }