61 lines
1.9 KiB
C#
61 lines
1.9 KiB
C#
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<ApiConfig>();
|
|
_db = svc.GetRequiredService<IDatabase>();
|
|
_repo = svc.GetRequiredService<ModelRepository>();
|
|
}
|
|
|
|
protected Task<PKSystem?> ResolveSystem(string systemRef)
|
|
{
|
|
if (systemRef == "@me")
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null) return null;
|
|
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;
|
|
}
|
|
|
|
public LookupContext LookupContextFor(PKSystem target)
|
|
{
|
|
HttpContext.Items.TryGetValue("SystemId", out var systemId);
|
|
if (systemId == null) return LookupContext.ByNonOwner;
|
|
return target.Id == (SystemId)systemId ? LookupContext.ByOwner : LookupContext.ByNonOwner;
|
|
}
|
|
}
|
|
} |