FirstTimeSetupHandler.cs 1.7 KB

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