FirstTimeSetupHandler.cs 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. using System.Threading.Tasks;
  2. using Jellyfin.Api.Constants;
  3. using Jellyfin.Api.Extensions;
  4. using MediaBrowser.Common.Configuration;
  5. using Microsoft.AspNetCore.Authorization;
  6. namespace Jellyfin.Api.Auth.FirstTimeSetupPolicy
  7. {
  8. /// <summary>
  9. /// Authorization handler for requiring first time setup or default privileges.
  10. /// </summary>
  11. public class FirstTimeSetupHandler : AuthorizationHandler<FirstTimeSetupRequirement>
  12. {
  13. private readonly IConfigurationManager _configurationManager;
  14. /// <summary>
  15. /// Initializes a new instance of the <see cref="FirstTimeSetupHandler" /> class.
  16. /// </summary>
  17. /// <param name="configurationManager">Instance of the <see cref="IConfigurationManager"/> interface.</param>
  18. public FirstTimeSetupHandler(IConfigurationManager configurationManager)
  19. {
  20. _configurationManager = configurationManager;
  21. }
  22. /// <inheritdoc />
  23. protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, FirstTimeSetupRequirement requirement)
  24. {
  25. // Succeed if the startup wizard / first time setup is not complete
  26. if (!_configurationManager.CommonConfiguration.IsStartupWizardCompleted)
  27. {
  28. context.Succeed(requirement);
  29. }
  30. // Succeed if user is admin
  31. else if (context.User.IsInRole(UserRoles.Administrator))
  32. {
  33. context.Succeed(requirement);
  34. }
  35. // Fail if admin is required and user is not admin
  36. else if (requirement.RequireAdmin)
  37. {
  38. context.Fail();
  39. }
  40. // Succeed if admin is not required and user is not guest
  41. else if (context.User.IsInRole(UserRoles.User))
  42. {
  43. context.Succeed(requirement);
  44. }
  45. // Any user-specific checks are handled in the DefaultAuthorizationHandler.
  46. return Task.CompletedTask;
  47. }
  48. }
  49. }