From fa6a12b09d2acc2fa2f2045c1d382b937f51f788 Mon Sep 17 00:00:00 2001
From: Jordan Dominion <Cyberboss@users.noreply.github.com>
Date: Sun, 17 Aug 2025 17:30:55 -0400
Subject: [PATCH] Implement UserGroup data loaders

---
 .../GraphQL/Types/UserGroup.cs                | 41 ++++++++++++++++---
 .../GraphQL/Types/UserGroupsRepository.cs     |  8 ++--
 2 files changed, 41 insertions(+), 8 deletions(-)

diff --git a/src/Tgstation.Server.Host/GraphQL/Types/UserGroup.cs b/src/Tgstation.Server.Host/GraphQL/Types/UserGroup.cs
index 9dfb63459b..f97f4eefdb 100644
--- a/src/Tgstation.Server.Host/GraphQL/Types/UserGroup.cs
+++ b/src/Tgstation.Server.Host/GraphQL/Types/UserGroup.cs
@@ -1,8 +1,12 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 
+using GreenDonut;
+using GreenDonut.Data;
+
 using HotChocolate;
 using HotChocolate.CostAnalysis.Types;
 using HotChocolate.Data;
@@ -10,6 +14,8 @@ using HotChocolate.Types;
 using HotChocolate.Types.Relay;
 
 using Tgstation.Server.Host.Authority;
+using Tgstation.Server.Host.Authority.Core;
+using Tgstation.Server.Host.Extensions;
 using Tgstation.Server.Host.GraphQL.Transformers;
 
 namespace Tgstation.Server.Host.GraphQL.Types
@@ -25,21 +31,46 @@ namespace Tgstation.Server.Host.GraphQL.Types
 		/// </summary>
 		public required PermissionSet PermissionSet { get; set; }
 
+		/// <summary>
+		/// Implements the <see cref="IUserGroupsDataLoader"/>.
+		/// </summary>
+		/// <param name="ids">The <see cref="IReadOnlyList{T}"/> of <see cref="UserGroup"/> <see cref="Api.Models.EntityId.Id"/>s to load paired with if the system user should be allowed.</param>
+		/// <param name="userAuthority">The <see cref="IGraphQLAuthorityInvoker{TAuthority}"/> for the <see cref="IUserAuthority"/>.</param>
+		/// <param name="queryContext">The <see cref="QueryContext{TEntity}"/> for <see cref="User"/> mapped to an <see cref="AuthorityResponse{TResult}"/>.</param>
+		/// <param name="cancellationToken">The <see cref="CancellationToken"/> for the operation.</param>
+		/// <returns>A <see cref="ValueTask{TResult}"/> resulting in a <see cref="Dictionary{TKey, TValue}"/> of the requested <see cref="UserGroup"/> <see cref="AuthorityResponse{TResult}"/>s.</returns>
+		[DataLoader(AccessModifier = DataLoaderAccessModifier.PublicInterface)]
+		public static ValueTask<Dictionary<long, AuthorityResponse<UserGroup>>> GetUserGroups(
+			IReadOnlyList<long> ids,
+			IGraphQLAuthorityInvoker<IUserGroupAuthority> userAuthority,
+			QueryContext<AuthorityResponse<UserGroup>>? queryContext,
+			CancellationToken cancellationToken)
+		{
+			ArgumentNullException.ThrowIfNull(ids);
+			ArgumentNullException.ThrowIfNull(userAuthority);
+
+			return userAuthority.ExecuteDataLoader<Models.UserGroup, UserGroup, UserGroupTransformer>(
+				(authority, id) => authority.GetId<UserGroup>(id, cancellationToken),
+				ids,
+				queryContext);
+		}
+
 		/// <summary>
 		/// Node resolver for <see cref="User"/>s.
 		/// </summary>
 		/// <param name="id">The <see cref="Entity.Id"/> to lookup.</param>
-		/// <param name="userGroupAuthority">The <see cref="IGraphQLAuthorityInvoker{TAuthority}"/> for the <see cref="IUserGroupAuthority"/>.</param>
+		/// <param name="userGroupsDataLoader">The <see cref="IUserGroupsDataLoader"/> to use.</param>
+		/// <param name="queryContext">The <see cref="QueryContext{TEntity}"/> for the operation.</param>
 		/// <param name="cancellationToken">The <see cref="CancellationToken"/> for the operation.</param>
 		/// <returns>A <see cref="ValueTask"/> resulting in the queried <see cref="User"/>, if present.</returns>
 		public static ValueTask<UserGroup?> GetUserGroup(
 			long id,
-			[Service] IGraphQLAuthorityInvoker<IUserGroupAuthority> userGroupAuthority,
+			[Service] IUserGroupsDataLoader userGroupsDataLoader,
+			QueryContext<UserGroup>? queryContext,
 			CancellationToken cancellationToken)
 		{
-			ArgumentNullException.ThrowIfNull(userGroupAuthority);
-			return userGroupAuthority.InvokeTransformableAllowMissing<Models.UserGroup, UserGroup, UserGroupTransformer>(
-				authority => authority.GetId(id, false, cancellationToken));
+			ArgumentNullException.ThrowIfNull(userGroupsDataLoader);
+			return userGroupsDataLoader.LoadAuthorityResponse(queryContext, id, cancellationToken);
 		}
 
 		/// <summary>
diff --git a/src/Tgstation.Server.Host/GraphQL/Types/UserGroupsRepository.cs b/src/Tgstation.Server.Host/GraphQL/Types/UserGroupsRepository.cs
index 9837ac5f2e..eae602f30c 100644
--- a/src/Tgstation.Server.Host/GraphQL/Types/UserGroupsRepository.cs
+++ b/src/Tgstation.Server.Host/GraphQL/Types/UserGroupsRepository.cs
@@ -43,14 +43,16 @@ namespace Tgstation.Server.Host.GraphQL.Types
 		/// Gets a <see cref="UserGroup"/> by <see cref="Entity.Id"/>.
 		/// </summary>
 		/// <param name="id">The <see cref="Entity.Id"/> of the <see cref="User"/>.</param>
-		/// <param name="userGroupAuthority">The <see cref="IGraphQLAuthorityInvoker{TAuthority}"/> for the <see cref="IUserGroupAuthority"/>.</param>
+		/// <param name="userGroupsDataLoader">The <see cref="IUserGroupsDataLoader"/> to use.</param>
+		/// <param name="queryContext">The <see cref="QueryContext{TEntity}"/> for the operation.</param>
 		/// <param name="cancellationToken">The <see cref="CancellationToken"/> for the operation.</param>
 		/// <returns>The <see cref="UserGroup"/> represented by <paramref name="id"/>, if any.</returns>
 		public ValueTask<UserGroup?> ById(
 			[ID(nameof(UserGroup))] long id,
-			[Service] IGraphQLAuthorityInvoker<IUserGroupAuthority> userGroupAuthority,
+			[Service] IUserGroupsDataLoader userGroupsDataLoader,
+			QueryContext<UserGroup>? queryContext,
 			CancellationToken cancellationToken)
-			=> UserGroup.GetUserGroup(id, userGroupAuthority, cancellationToken);
+			=> UserGroup.GetUserGroup(id, userGroupsDataLoader, queryContext, cancellationToken);
 
 		/// <summary>
 		/// Queries all registered <see cref="UserGroup"/>s.