feat: add organisations (#1820)

packages/lib/server-only/team/get-teams.ts

@@ -1,53 +1,48 @@
-import type { z } from 'zod';
+import { z } from 'zod';
 
 import { prisma } from '@documenso/prisma';
-import { SubscriptionSchema } from '@documenso/prisma/generated/zod/modelSchema/SubscriptionSchema';
-import { TeamMemberSchema } from '@documenso/prisma/generated/zod/modelSchema/TeamMemberSchema';
+import { TeamMemberRole } from '@documenso/prisma/generated/types';
 import { TeamSchema } from '@documenso/prisma/generated/zod/modelSchema/TeamSchema';
 
+import { buildTeamWhereQuery, getHighestTeamRoleInGroup } from '../../utils/teams';
+
 export type GetTeamsOptions = {
   userId: number;
+
+  /**
+   * If teamId is undefined we get all teams the user belongs to.
+   */
+  teamId?: number;
 };
 
 export const ZGetTeamsResponseSchema = TeamSchema.extend({
-  currentTeamMember: TeamMemberSchema.pick({
-    role: true,
-  }),
-  subscription: SubscriptionSchema.pick({
-    status: true,
-  }).nullable(),
+  teamRole: z.nativeEnum(TeamMemberRole),
 }).array();
 
 export type TGetTeamsResponse = z.infer<typeof ZGetTeamsResponseSchema>;
 
-export const getTeams = async ({ userId }: GetTeamsOptions): Promise<TGetTeamsResponse> => {
+export const getTeams = async ({ userId, teamId }: GetTeamsOptions) => {
   const teams = await prisma.team.findMany({
-    where: {
-      members: {
-        some: {
-          userId,
-        },
-      },
-    },
+    where: buildTeamWhereQuery({ teamId, userId }),
     include: {
-      subscription: {
-        select: {
-          status: true,
-        },
-      },
-      members: {
+      teamGroups: {
         where: {
-          userId,
-        },
-        select: {
-          role: true,
+          organisationGroup: {
+            organisationGroupMembers: {
+              some: {
+                organisationMember: {
+                  userId,
+                },
+              },
+            },
+          },
         },
       },
     },
   });
 
-  return teams.map(({ members, ...team }) => ({
+  return teams.map((team) => ({
     ...team,
-    currentTeamMember: members[0],
+    currentTeamRole: getHighestTeamRoleInGroup(team.teamGroups),
   }));
 };