chore: add groupsByUserId query options factory

Parkreiner · Parkreiner · commit 52cb1bffc2f3 · 2023-10-17T19:58:53.000Z
diff --git a/site/src/api/queries/groups.ts b/site/src/api/queries/groups.ts
@@ -1,4 +1,4 @@
-import { QueryClient } from "react-query";
+import { QueryClient, UseQueryOptions } from "react-query";
 import * as API from "api/api";
 import { checkAuthorization } from "api/api";
 import {
@@ -25,6 +25,44 @@ export const group = (groupId: string) => {
   };
 };
 
+export type GroupsByUserId = Readonly<Map<string, readonly Group[]>>;
+
+export function groupsByUserId(organizationId: string) {
+  return {
+    ...groups(organizationId),
+    select: (allGroups) => {
+      const userIdMapper = new Map<string, Group[]>();
+
+      for (const group of allGroups) {
+        for (const user of group.members) {
+          let groupsForUser = userIdMapper.get(user.id);
+          if (groupsForUser === undefined) {
+            groupsForUser = [];
+            userIdMapper.set(user.id, groupsForUser);
+          }
+
+          groupsForUser.push(group);
+        }
+      }
+
+      for (const groupsList of userIdMapper.values()) {
+        groupsList.sort((g1, g2) => {
+          const key =
+            g1.display_name && g2.display_name ? "display_name" : "name";
+
+          if (g1[key] === g2[key]) {
+            return 0;
+          }
+
+          return g1[key] < g2[key] ? -1 : 1;
+        });
+      }
+
+      return userIdMapper;
+    },
+  } satisfies UseQueryOptions<Group[], unknown, GroupsByUserId>;
+}
+
 export const groupPermissions = (groupId: string) => {
   return {
     queryKey: [...getGroupQueryKey(groupId), "permissions"],
