SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

refine log in

Browse source
This commit is contained in:
Jicheng Lu 2024-05-20 11:35:45 -05:00
parent 7b8704680d
commit b35f0e6536
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Infrastructure/BotSharp.Abstraction/Users/IUserService.cs
View file

@ -6,6 +6,6 @@ public interface IUserService
{
Task<User> GetUser(string id);
Task<User> CreateUser(User user);
Task<Token> GetToken(string authorization);
Task<Token?> GetToken(string authorization);
Task<User> GetMyProfile();
}

8
src/Infrastructure/BotSharp.Core/Users/Services/UserService.cs
View file

@ -1,4 +1,3 @@
using BotSharp.Abstraction.Repositories;
using BotSharp.Abstraction.Users.Models;
using Microsoft.Extensions.Configuration;
using Microsoft.IdentityModel.Tokens;
@ -60,7 +59,7 @@ public class UserService : IUserService
return record;
}
public async Task<Token> GetToken(string authorization)
public async Task<Token?> GetToken(string authorization)
{
var base64 = Encoding.UTF8.GetString(Convert.FromBase64String(authorization));
var (id, password) = base64.SplitAsTuple(":");
@ -72,13 +71,14 @@ public class UserService : IUserService
record = db.GetUserByUserName(id);
}
User? user = null;
var hooks = _services.GetServices<IAuthenticationHook>();
if (record == null || record.Source != "internal")
{
// check 3rd party user
foreach (var hook in hooks)
{
var user = await hook.Authenticate(id, password);
user = await hook.Authenticate(id, password);
if (user == null)
{
continue;
@ -109,7 +109,7 @@ public class UserService : IUserService
}
}
if (record == null)
if ((!hooks.IsNullOrEmpty() && user == null) || record == null)
{
return default;
}