| | | 1 | | // SPDX-FileCopyrightText: 2026 Alper Çelik <[email protected]> |
| | | 2 | | // |
| | | 3 | | // SPDX-License-Identifier: AGPL-3.0-or-later |
| | | 4 | | |
| | | 5 | | using System.Security.Claims; |
| | | 6 | | using Api.Auth.Models; |
| | | 7 | | using Microsoft.AspNetCore.Authorization; |
| | | 8 | | using Microsoft.Extensions.Options; |
| | | 9 | | |
| | | 10 | | namespace Api.Auth.Handlers; |
| | | 11 | | |
| | | 12 | | public class PermissionCheckRequirement(UserPermissionBits permissionBits) |
| | | 13 | | : IAuthorizationRequirement |
| | | 14 | | { |
| | | 15 | | public UserPermissionBits PermissionBits { get; } = permissionBits; |
| | | 16 | | } |
| | | 17 | | |
| | | 18 | | public class PermissionPolicyProvider(IOptions<AuthorizationOptions> options) |
| | | 19 | | : DefaultAuthorizationPolicyProvider(options) |
| | | 20 | | { |
| | | 21 | | public const string POLICY_PREFIX = "PermissionBits_"; |
| | | 22 | | public const string RemoteServicePolicyName = "RemoteService"; |
| | | 23 | | |
| | | 24 | | public override Task<AuthorizationPolicy?> GetPolicyAsync(string policyName) |
| | | 25 | | { |
| | | 26 | | if ( |
| | | 27 | | policyName.StartsWith(POLICY_PREFIX) |
| | | 28 | | && long.TryParse(policyName[POLICY_PREFIX.Length..], out long permissionBits) |
| | | 29 | | ) |
| | | 30 | | { |
| | | 31 | | return Task.FromResult<AuthorizationPolicy?>( |
| | | 32 | | new AuthorizationPolicyBuilder() |
| | | 33 | | .AddRequirements( |
| | | 34 | | new PermissionCheckRequirement((UserPermissionBits)permissionBits) |
| | | 35 | | ) |
| | | 36 | | .Build() |
| | | 37 | | ); |
| | | 38 | | } |
| | | 39 | | |
| | | 40 | | if (policyName == RemoteServicePolicyName) |
| | | 41 | | { |
| | | 42 | | return Task.FromResult<AuthorizationPolicy?>( |
| | | 43 | | new AuthorizationPolicyBuilder() |
| | | 44 | | .RequireClaim(RemoteServiceToken.ServiceIdentifierType) |
| | | 45 | | .Build() |
| | | 46 | | ); |
| | | 47 | | } |
| | | 48 | | |
| | | 49 | | return base.GetPolicyAsync(policyName); |
| | | 50 | | } |
| | | 51 | | } |
| | | 52 | | |
| | | 53 | | class PermissionCheckAuthorizationHandler |
| | | 54 | | : AuthorizationHandler<PermissionCheckRequirement> |
| | | 55 | | { |
| | | 56 | | protected override Task HandleRequirementAsync( |
| | | 57 | | AuthorizationHandlerContext context, |
| | | 58 | | PermissionCheckRequirement requirement |
| | | 59 | | ) |
| | 1 | 60 | | { |
| | 1 | 61 | | var userPermissionBitsString = context.User.FindFirstValue( |
| | 1 | 62 | | UserTokenEF.PermissionBitsType |
| | 1 | 63 | | ); |
| | | 64 | | |
| | 1 | 65 | | if (userPermissionBitsString is null) |
| | 1 | 66 | | return Task.CompletedTask; |
| | | 67 | | |
| | 1 | 68 | | if (!long.TryParse(userPermissionBitsString, out long userPermissionBitsLong)) |
| | 0 | 69 | | return Task.CompletedTask; |
| | | 70 | | |
| | 1 | 71 | | var userPermissionBits = (UserPermissionBits)userPermissionBitsLong; |
| | | 72 | | |
| | 1 | 73 | | if ( |
| | 1 | 74 | | (requirement.PermissionBits & userPermissionBits) |
| | 1 | 75 | | == requirement.PermissionBits |
| | 1 | 76 | | ) |
| | 1 | 77 | | { |
| | 1 | 78 | | context.Succeed(requirement); |
| | 1 | 79 | | } |
| | 1 | 80 | | return Task.CompletedTask; |
| | 1 | 81 | | } |
| | | 82 | | } |
| | | 83 | | |
| | | 84 | | class PermissionCheckAuthorizeAttribute |
| | | 85 | | : HotChocolate.Authorization.AuthorizeAttribute, |
| | | 86 | | IAuthorizeData |
| | | 87 | | { |
| | | 88 | | public PermissionCheckAuthorizeAttribute(UserPermissionBits permissionBits) |
| | | 89 | | { |
| | | 90 | | Policy = $"{PermissionPolicyProvider.POLICY_PREFIX}{(long)permissionBits}"; |
| | | 91 | | } |
| | | 92 | | |
| | | 93 | | public string? AuthenticationSchemes { get; set; } |
| | | 94 | | string? IAuthorizeData.Roles { get; set; } |
| | | 95 | | } |