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.Linq;
using System.Threading.Tasks; using System.Threading.Tasks;
using Dapper;
using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Http;
using PluralKit.Core; using PluralKit.Core;
@ -10,11 +13,11 @@ namespace PluralKit.API
{ {
public PKSystem CurrentSystem { get; set; } 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) public async Task InvokeAsync(HttpContext context, RequestDelegate next)
@ -22,7 +25,7 @@ namespace PluralKit.API
var token = context.Request.Headers["Authorization"].FirstOrDefault(); var token = context.Request.Headers["Authorization"].FirstOrDefault();
if (token != null) 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); await next.Invoke(context);