🛂 (billing) Add isPastDue field in workspace

Closes #1039
baptisteArno · Nov 21, 2023 · 55cb6e8 · 55cb6e8
1 parent 94886ca
commit 55cb6e8
Show file tree

Hide file tree

Showing 25 changed files with 339 additions and 70 deletions.
diff --git a/apps/builder/src/features/blocks/logic/typebotLink/api/getLinkedTypebots.ts b/apps/builder/src/features/blocks/logic/typebotLink/api/getLinkedTypebots.ts
@@ -56,7 +56,17 @@ export const getLinkedTypebots = authenticatedProcedure
         variables: true,
         name: true,
         createdAt: true,
-        workspaceId: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+              },
+            },
+          },
+        },
         collaborators: {
           select: {
             type: true,
@@ -97,7 +107,17 @@ export const getLinkedTypebots = authenticatedProcedure
           variables: true,
           name: true,
           createdAt: true,
-          workspaceId: true,
+          workspace: {
+            select: {
+              isQuarantined: true,
+              isPastDue: true,
+              members: {
+                select: {
+                  userId: true,
+                },
+              },
+            },
+          },
           collaborators: {
             select: {
               type: true,

diff --git a/apps/builder/src/features/collaboration/api/getCollaborators.ts b/apps/builder/src/features/collaboration/api/getCollaborators.ts
@@ -32,6 +32,17 @@ export const getCollaborators = authenticatedProcedure
       },
       include: {
         collaborators: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+              },
+            },
+          },
+        },
       },
     })
     if (

diff --git a/apps/builder/src/features/results/api/deleteResults.ts b/apps/builder/src/features/results/api/deleteResults.ts
@@ -36,8 +36,19 @@ export const deleteResults = authenticatedProcedure
         id: typebotId,
       },
       select: {
-        workspaceId: true,
         groups: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+                role: true,
+              },
+            },
+          },
+        },
         collaborators: {
           select: {
             userId: true,

diff --git a/apps/builder/src/features/results/api/getResult.ts b/apps/builder/src/features/results/api/getResult.ts
@@ -33,8 +33,18 @@ export const getResult = authenticatedProcedure
       },
       select: {
         id: true,
-        workspaceId: true,
         groups: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+              },
+            },
+          },
+        },
         collaborators: {
           select: {
             userId: true,

diff --git a/apps/builder/src/features/results/api/getResultLogs.ts b/apps/builder/src/features/results/api/getResultLogs.ts
@@ -28,8 +28,18 @@ export const getResultLogs = authenticatedProcedure
       },
       select: {
         id: true,
-        workspaceId: true,
         groups: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+              },
+            },
+          },
+        },
         collaborators: {
           select: {
             userId: true,

diff --git a/apps/builder/src/features/results/api/getResults.ts b/apps/builder/src/features/results/api/getResults.ts
@@ -44,14 +44,24 @@ export const getResults = authenticatedProcedure
       },
       select: {
         id: true,
-        workspaceId: true,
         groups: true,
         collaborators: {
           select: {
             userId: true,
             type: true,
           },
         },
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+              },
+            },
+          },
+        },
       },
     })
     if (!typebot || (await isReadTypebotForbidden(typebot, user)))

diff --git a/apps/builder/src/features/typebot/api/deleteTypebot.ts b/apps/builder/src/features/typebot/api/deleteTypebot.ts
@@ -33,8 +33,19 @@ export const deleteTypebot = authenticatedProcedure
       },
       select: {
         id: true,
-        workspaceId: true,
         groups: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+                role: true,
+              },
+            },
+          },
+        },
         collaborators: {
           select: {
             userId: true,

diff --git a/apps/builder/src/features/typebot/api/getPublishedTypebot.ts b/apps/builder/src/features/typebot/api/getPublishedTypebot.ts
@@ -52,6 +52,17 @@ export const getPublishedTypebot = authenticatedProcedure
         include: {
           collaborators: true,
           publishedTypebot: true,
+          workspace: {
+            select: {
+              isQuarantined: true,
+              isPastDue: true,
+              members: {
+                select: {
+                  userId: true,
+                },
+              },
+            },
+          },
         },
       })
       if (

diff --git a/apps/builder/src/features/typebot/api/getTypebot.ts b/apps/builder/src/features/typebot/api/getTypebot.ts
@@ -41,6 +41,17 @@ export const getTypebot = authenticatedProcedure
         },
         include: {
           collaborators: true,
+          workspace: {
+            select: {
+              isQuarantined: true,
+              isPastDue: true,
+              members: {
+                select: {
+                  userId: true,
+                },
+              },
+            },
+          },
         },
       })
       if (

diff --git a/apps/builder/src/features/typebot/api/publishTypebot.ts b/apps/builder/src/features/typebot/api/publishTypebot.ts
@@ -46,6 +46,14 @@ export const publishTypebot = authenticatedProcedure
         workspace: {
           select: {
             plan: true,
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+                role: true,
+              },
+            },
           },
         },
       },

diff --git a/apps/builder/src/features/typebot/api/unpublishTypebot.ts b/apps/builder/src/features/typebot/api/unpublishTypebot.ts
@@ -32,6 +32,18 @@ export const unpublishTypebot = authenticatedProcedure
       include: {
         collaborators: true,
         publishedTypebot: true,
+        workspace: {
+          select: {
+            isQuarantined: true,
+            isPastDue: true,
+            members: {
+              select: {
+                userId: true,
+                role: true,
+              },
+            },
+          },
+        },
       },
     })
     if (!existingTypebot?.publishedTypebot)

diff --git a/apps/builder/src/features/typebot/api/updateTypebot.ts b/apps/builder/src/features/typebot/api/updateTypebot.ts
@@ -79,7 +79,6 @@ export const updateTypebot = authenticatedProcedure
           id: true,
           customDomain: true,
           publicId: true,
-          workspaceId: true,
           collaborators: {
             select: {
               userId: true,
@@ -88,7 +87,16 @@ export const updateTypebot = authenticatedProcedure
           },
           workspace: {
             select: {
+              id: true,
               plan: true,
+              isQuarantined: true,
+              isPastDue: true,
+              members: {
+                select: {
+                  userId: true,
+                  role: true,
+                },
+              },
             },
           },
           updatedAt: true,
@@ -160,7 +168,7 @@ export const updateTypebot = authenticatedProcedure
           selectedThemeTemplateId: typebot.selectedThemeTemplateId,
           events: typebot.events ?? undefined,
           groups: typebot.groups
-            ? await sanitizeGroups(existingTypebot.workspaceId)(typebot.groups)
+            ? await sanitizeGroups(existingTypebot.workspace.id)(typebot.groups)
             : undefined,
           theme: typebot.theme ? typebot.theme : undefined,
           settings: typebot.settings

diff --git a/apps/builder/src/features/typebot/helpers/isReadTypebotForbidden.ts b/apps/builder/src/features/typebot/helpers/isReadTypebotForbidden.ts
@@ -1,26 +1,25 @@
-import prisma from '@typebot.io/lib/prisma'
 import { env } from '@typebot.io/env'
-import { CollaboratorsOnTypebots, User } from '@typebot.io/prisma'
-import { Typebot } from '@typebot.io/schemas'
+import {
+  CollaboratorsOnTypebots,
+  User,
+  Workspace,
+  MemberInWorkspace,
+} from '@typebot.io/prisma'
 
 export const isReadTypebotForbidden = async (
-  typebot: Pick<Typebot, 'workspaceId'> & {
+  typebot: {
     collaborators: Pick<CollaboratorsOnTypebots, 'userId'>[]
+  } & {
+    workspace: Pick<Workspace, 'isQuarantined' | 'isPastDue'> & {
+      members: Pick<MemberInWorkspace, 'userId'>[]
+    }
   },
   user: Pick<User, 'email' | 'id'>
-) => {
-  if (
-    env.ADMIN_EMAIL === user.email ||
-    typebot.collaborators.find(
+) =>
+  typebot.workspace.isQuarantined ||
+  typebot.workspace.isPastDue ||
+  (env.ADMIN_EMAIL !== user.email &&
+    !typebot.collaborators.some(
       (collaborator) => collaborator.userId === user.id
-    )
-  )
-    return false
-  const memberInWorkspace = await prisma.memberInWorkspace.findFirst({
-    where: {
-      workspaceId: typebot.workspaceId,
-      userId: user.id,
-    },
-  })
-  return memberInWorkspace === null
-}
+    ) &&
+    !typebot.workspace.members.some((member) => member.userId === user.id))
diff --git a/apps/builder/src/features/typebot/helpers/isWriteTypebotForbidden.ts b/apps/builder/src/features/typebot/helpers/isWriteTypebotForbidden.ts
@@ -1,29 +1,31 @@
-import prisma from '@typebot.io/lib/prisma'
 import {
   CollaborationType,
   CollaboratorsOnTypebots,
+  MemberInWorkspace,
   User,
+  Workspace,
 } from '@typebot.io/prisma'
-import { Typebot } from '@typebot.io/schemas'
-import { isNotDefined } from '@typebot.io/lib'
 
 export const isWriteTypebotForbidden = async (
-  typebot: Pick<Typebot, 'workspaceId'> & {
+  typebot: {
     collaborators: Pick<CollaboratorsOnTypebots, 'userId' | 'type'>[]
+  } & {
+    workspace: Pick<Workspace, 'isQuarantined' | 'isPastDue'> & {
+      members: Pick<MemberInWorkspace, 'userId' | 'role'>[]
+    }
   },
   user: Pick<User, 'id'>
 ) => {
-  if (
-    typebot.collaborators.find(
-      (collaborator) => collaborator.userId === user.id
-    )?.type === CollaborationType.WRITE
+  return (
+    typebot.workspace.isQuarantined ||
+    typebot.workspace.isPastDue ||
+    !(
+      typebot.collaborators.find(
+        (collaborator) => collaborator.userId === user.id
+      )?.type === CollaborationType.WRITE &&
+      typebot.workspace.members.some(
+        (m) => m.userId === user.id && m.role !== 'GUEST'
+      )
+    )
   )
-    return false
-  const memberInWorkspace = await prisma.memberInWorkspace.findFirst({
-    where: {
-      workspaceId: typebot.workspaceId,
-      userId: user.id,
-    },
-  })
-  return isNotDefined(memberInWorkspace) || memberInWorkspace.role === 'GUEST'
 }
diff --git a/apps/builder/src/features/upload/api/generateUploadUrl.ts b/apps/builder/src/features/upload/api/generateUploadUrl.ts
@@ -147,7 +147,19 @@ const parseFilePath = async ({
       id: input.typebotId,
     },
     select: {
-      workspaceId: true,
+      workspace: {
+        select: {
+          plan: true,
+          isQuarantined: true,
+          isPastDue: true,
+          members: {
+            select: {
+              userId: true,
+              role: true,
+            },
+          },
+        },
+      },
       collaborators: {
         select: {
           userId: true,