SyncPlayAccessHandler.cs 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. using System.Threading.Tasks;
  2. using Jellyfin.Api.Extensions;
  3. using Jellyfin.Data.Enums;
  4. using MediaBrowser.Controller.Library;
  5. using MediaBrowser.Controller.SyncPlay;
  6. using Microsoft.AspNetCore.Authorization;
  7. namespace Jellyfin.Api.Auth.SyncPlayAccessPolicy
  8. {
  9. /// <summary>
  10. /// Default authorization handler.
  11. /// </summary>
  12. public class SyncPlayAccessHandler : AuthorizationHandler<SyncPlayAccessRequirement>
  13. {
  14. private readonly ISyncPlayManager _syncPlayManager;
  15. private readonly IUserManager _userManager;
  16. /// <summary>
  17. /// Initializes a new instance of the <see cref="SyncPlayAccessHandler"/> class.
  18. /// </summary>
  19. /// <param name="syncPlayManager">Instance of the <see cref="ISyncPlayManager"/> interface.</param>
  20. /// <param name="userManager">Instance of the <see cref="IUserManager"/> interface.</param>
  21. public SyncPlayAccessHandler(
  22. ISyncPlayManager syncPlayManager,
  23. IUserManager userManager)
  24. {
  25. _syncPlayManager = syncPlayManager;
  26. _userManager = userManager;
  27. }
  28. /// <inheritdoc />
  29. protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, SyncPlayAccessRequirement requirement)
  30. {
  31. var userId = context.User.GetUserId();
  32. var user = _userManager.GetUserById(userId);
  33. if (requirement.RequiredAccess == SyncPlayAccessRequirementType.HasAccess)
  34. {
  35. if (user.SyncPlayAccess is SyncPlayUserAccessType.CreateAndJoinGroups or SyncPlayUserAccessType.JoinGroups
  36. || _syncPlayManager.IsUserActive(userId))
  37. {
  38. context.Succeed(requirement);
  39. }
  40. }
  41. else if (requirement.RequiredAccess == SyncPlayAccessRequirementType.CreateGroup)
  42. {
  43. if (user.SyncPlayAccess == SyncPlayUserAccessType.CreateAndJoinGroups)
  44. {
  45. context.Succeed(requirement);
  46. }
  47. }
  48. else if (requirement.RequiredAccess == SyncPlayAccessRequirementType.JoinGroup)
  49. {
  50. if (user.SyncPlayAccess == SyncPlayUserAccessType.CreateAndJoinGroups
  51. || user.SyncPlayAccess == SyncPlayUserAccessType.JoinGroups)
  52. {
  53. context.Succeed(requirement);
  54. }
  55. }
  56. else if (requirement.RequiredAccess == SyncPlayAccessRequirementType.IsInGroup)
  57. {
  58. if (_syncPlayManager.IsUserActive(userId))
  59. {
  60. context.Succeed(requirement);
  61. }
  62. }
  63. return Task.CompletedTask;
  64. }
  65. }
  66. }