liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Migrate GetSystemByToken

Browse Source
This commit is contained in:
Ske 2020-06-13 19:44:01 +02:00
parent c927967c3b
commit 1d1b7b8716
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
PluralKit.API/Services/TokenAuthService.cs
View File

@ -1,5 +1,8 @@
using System.Linq;
using System.Threading.Tasks;
using Dapper;
using Microsoft.AspNetCore.Http;
using PluralKit.Core;
@ -10,11 +13,11 @@ namespace PluralKit.API
{
public PKSystem CurrentSystem { get; set; }
private IDataStore _data;
private readonly IDatabase _db;
public TokenAuthService(IDataStore data)
public TokenAuthService(IDatabase db)
{
_data = data;
_db = db;
}
public async Task InvokeAsync(HttpContext context, RequestDelegate next)
@ -22,7 +25,7 @@ namespace PluralKit.API
var token = context.Request.Headers["Authorization"].FirstOrDefault();
if (token != null)
{
CurrentSystem = await _data.GetSystemByToken(token);
CurrentSystem = await _db.Execute(c => c.QueryFirstOrDefaultAsync("select * from systems where token = @token", new { token }));
}
await next.Invoke(context);