github/Ombi
1
0
Fork 0
mirror of https://github.com/Ombi-app/Ombi.git synced 2025-08-22 06:13:22 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: apply default user settings to SSO users

Browse source
This commit is contained in:
Lea 2023-01-03 21:30:24 +01:00
commit 28c6129545
No known key found for this signature in database
GPG key ID: 1BAFFE8347019C42
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/Ombi/Controllers/V1/TokenController.cs
View file

@ -36,13 +36,15 @@ namespace Ombi.Controllers.V1
public class TokenController : ControllerBase public class TokenController : ControllerBase
{ {
public TokenController(OmbiUserManager um, ITokenRepository token, public TokenController(OmbiUserManager um, ITokenRepository token,
IPlexOAuthManager oAuthManager, ILogger<TokenController> logger, ISettingsService<AuthenticationSettings> auth) IPlexOAuthManager oAuthManager, ILogger<TokenController> logger, ISettingsService<AuthenticationSettings> auth,
ISettingsService<UserManagementSettings> userManagement)
{ {
_userManager = um; _userManager = um;
_token = token; _token = token;
_plexOAuthManager = oAuthManager; _plexOAuthManager = oAuthManager;
_log = logger; _log = logger;
_authSettings = auth; _authSettings = auth;
_userManagementSettings = userManagement;
} }
private readonly ITokenRepository _token; private readonly ITokenRepository _token;
@ -50,6 +52,7 @@ namespace Ombi.Controllers.V1
private readonly IPlexOAuthManager _plexOAuthManager; private readonly IPlexOAuthManager _plexOAuthManager;
private readonly ILogger<TokenController> _log; private readonly ILogger<TokenController> _log;
private readonly ISettingsService<AuthenticationSettings> _authSettings; private readonly ISettingsService<AuthenticationSettings> _authSettings;
private readonly ISettingsService<UserManagementSettings> _userManagementSettings;
/// <summary> /// <summary>
/// Gets the token. /// Gets the token.
@ -307,13 +310,21 @@ namespace Ombi.Controllers.V1
{ {
if (authSettings.HeaderAuthCreateUser) if (authSettings.HeaderAuthCreateUser)
{ {
var defaultSettings = await _userManagementSettings.GetSettingsAsync();
user = new OmbiUser { user = new OmbiUser {
UserName = username, UserName = username,
UserType = UserType.LocalUser, UserType = UserType.LocalUser,
StreamingCountry = "US", StreamingCountry = defaultSettings.DefaultStreamingCountry ?? "US",
MovieRequestLimit = defaultSettings.MovieRequestLimit,
MovieRequestLimitType = defaultSettings.MovieRequestLimitType,
EpisodeRequestLimit = defaultSettings.EpisodeRequestLimit,
EpisodeRequestLimitType = defaultSettings.EpisodeRequestLimitType,
MusicRequestLimit = defaultSettings.MusicRequestLimit,
MusicRequestLimitType = defaultSettings.MusicRequestLimitType,
}; };
await _userManager.CreateAsync(user); await _userManager.CreateAsync(user);
await _userManager.AddToRolesAsync(user, defaultSettings.DefaultRoles);
} }
else else
{ {