fix(viewer): 🐛 Custom metadata in viewer

apps/viewer/playwright/services/database.ts

@@ -0,0 +1,186 @@
+import {
+  Block,
+  CredentialsType,
+  defaultSettings,
+  defaultTheme,
+  PublicBlock,
+  PublicTypebot,
+  Step,
+  Typebot,
+} from 'models'
+import { PrismaClient, User } from 'db'
+import { readFileSync } from 'fs'
+import { encrypt } from 'utils'
+
+const prisma = new PrismaClient()
+
+export const teardownDatabase = async () => {
+  const ownerFilter = {
+    where: { ownerId: { in: ['freeUser', 'proUser'] } },
+  }
+  await prisma.user.deleteMany({
+    where: { id: { in: ['freeUser', 'proUser'] } },
+  })
+  await prisma.credentials.deleteMany(ownerFilter)
+  return prisma.typebot.deleteMany(ownerFilter)
+}
+
+export const setupDatabase = async () => {
+  await createUsers()
+  return createCredentials()
+}
+
+export const createUsers = () =>
+  prisma.user.createMany({
+    data: [
+      { id: 'freeUser', email: 'free-user@email.com', name: 'Free user' },
+      { id: 'proUser', email: 'pro-user@email.com', name: 'Pro user' },
+    ],
+  })
+
+export const getSignedInUser = (email: string) =>
+  prisma.user.findFirst({ where: { email } })
+
+export const createTypebots = async (partialTypebots: Partial<Typebot>[]) => {
+  await prisma.typebot.createMany({
+    data: partialTypebots.map(parseTestTypebot) as any[],
+  })
+  return prisma.publicTypebot.createMany({
+    data: partialTypebots.map((t) =>
+      parseTypebotToPublicTypebot(t.id + '-published', parseTestTypebot(t))
+    ) as any[],
+  })
+}
+
+const createCredentials = () => {
+  const { encryptedData, iv } = encrypt({
+    expiry_date: 1642441058842,
+    access_token:
+      'ya29.A0ARrdaM--PV_87ebjywDJpXKb77NBFJl16meVUapYdfNv6W6ZzqqC47fNaPaRjbDbOIIcp6f49cMaX5ndK9TAFnKwlVqz3nrK9nLKqgyDIhYsIq47smcAIZkK56SWPx3X3DwAFqRu2UPojpd2upWwo-3uJrod',
+    // This token is linked to a test Google account (typebot.test.user@gmail.com)
+    refresh_token:
+      '1//039xWRt8YaYa3CgYIARAAGAMSNwF-L9Iru9FyuTrDSa7lkSceggPho83kJt2J29G69iEhT1C6XV1vmo6bQS9puL_R2t8FIwR3gek',
+  })
+  return prisma.credentials.createMany({
+    data: [
+      {
+        name: 'test2@gmail.com',
+        ownerId: 'proUser',
+        type: CredentialsType.GOOGLE_SHEETS,
+        data: encryptedData,
+        iv,
+      },
+    ],
+  })
+}
+
+export const updateUser = (data: Partial<User>) =>
+  prisma.user.update({
+    data,
+    where: {
+      id: 'proUser',
+    },
+  })
+
+const parseTypebotToPublicTypebot = (
+  id: string,
+  typebot: Typebot
+): PublicTypebot => ({
+  id,
+  name: typebot.name,
+  blocks: parseBlocksToPublicBlocks(typebot.blocks),
+  typebotId: typebot.id,
+  theme: typebot.theme,
+  settings: typebot.settings,
+  publicId: typebot.publicId,
+  variables: typebot.variables,
+  edges: typebot.edges,
+  customDomain: null,
+})
+
+const parseBlocksToPublicBlocks = (blocks: Block[]): PublicBlock[] =>
+  blocks.map((b) => ({
+    ...b,
+    steps: b.steps.map((s) =>
+      'webhook' in s ? { ...s, webhook: s.webhook.id } : s
+    ),
+  }))
+
+const parseTestTypebot = (partialTypebot: Partial<Typebot>): Typebot => ({
+  id: partialTypebot.id ?? 'typebot',
+  folderId: null,
+  name: 'My typebot',
+  ownerId: 'proUser',
+  theme: defaultTheme,
+  settings: defaultSettings,
+  createdAt: new Date(),
+  publicId: partialTypebot.id + '-public',
+  publishedTypebotId: null,
+  updatedAt: new Date(),
+  customDomain: null,
+  variables: [{ id: 'var1', name: 'var1' }],
+  ...partialTypebot,
+  edges: [
+    {
+      id: 'edge1',
+      from: { blockId: 'block0', stepId: 'step0' },
+      to: { blockId: 'block1' },
+    },
+  ],
+  blocks: [
+    {
+      id: 'block0',
+      title: 'Block #0',
+      steps: [
+        {
+          id: 'step0',
+          type: 'start',
+          blockId: 'block0',
+          label: 'Start',
+          outgoingEdgeId: 'edge1',
+        },
+      ],
+      graphCoordinates: { x: 0, y: 0 },
+    },
+    ...(partialTypebot.blocks ?? []),
+  ],
+})
+
+export const parseDefaultBlockWithStep = (
+  step: Partial<Step>
+): Pick<Typebot, 'blocks'> => ({
+  blocks: [
+    {
+      graphCoordinates: { x: 200, y: 200 },
+      id: 'block1',
+      steps: [
+        {
+          id: 'step1',
+          blockId: 'block1',
+          ...step,
+        } as Step,
+      ],
+      title: 'Block #1',
+    },
+  ],
+})
+
+export const importTypebotInDatabase = async (
+  path: string,
+  updates?: Partial<Typebot>
+) => {
+  const typebot: any = {
+    ...JSON.parse(readFileSync(path).toString()),
+    ...updates,
+    ownerId: 'proUser',
+  }
+  await prisma.typebot.create({
+    data: typebot,
+  })
+  return prisma.publicTypebot.create({
+    data: parseTypebotToPublicTypebot(
+      updates?.id ? `${updates?.id}-public` : 'publicBot',
+      typebot
+    ),
+  })
+}