⚡ (typebotLink) Better typebot link with merge option

Closes #675
2023-08-24 07:48:30 +02:00
parent 0acede92ef
commit ee3b94c35d
59 changed files with 1147 additions and 696 deletions
--- a/packages/schemas/features/chat/sessionState.ts
+++ b/packages/schemas/features/chat/sessionState.ts
@ -0,0 +1,125 @@
+import { z } from 'zod'
+import { answerSchema } from '../answer'
+import { resultSchema } from '../result'
+import { typebotInSessionStateSchema, dynamicThemeSchema } from './shared'
+
+const answerInSessionStateSchema = answerSchema.pick({
+  content: true,
+  blockId: true,
+  variableId: true,
+})
+
+const answerInSessionStateSchemaV2 = z.object({
+  key: z.string(),
+  value: z.string(),
+})
+
+export type AnswerInSessionState = z.infer<typeof answerInSessionStateSchemaV2>
+
+const resultInSessionStateSchema = resultSchema
+  .pick({
+    variables: true,
+  })
+  .merge(
+    z.object({
+      answers: z.array(answerInSessionStateSchema),
+      id: z.string().optional(),
+    })
+  )
+
+const sessionStateSchemaV1 = z.object({
+  typebot: typebotInSessionStateSchema,
+  dynamicTheme: dynamicThemeSchema.optional(),
+  linkedTypebots: z.object({
+    typebots: z.array(typebotInSessionStateSchema),
+    queue: z.array(z.object({ edgeId: z.string(), typebotId: z.string() })),
+  }),
+  currentTypebotId: z.string(),
+  result: resultInSessionStateSchema,
+  currentBlock: z
+    .object({
+      blockId: z.string(),
+      groupId: z.string(),
+    })
+    .optional(),
+  isStreamEnabled: z.boolean().optional(),
+})
+
+const sessionStateSchemaV2 = z.object({
+  version: z.literal('2'),
+  typebotsQueue: z.array(
+    z.object({
+      edgeIdToTriggerWhenDone: z.string().optional(),
+      isMergingWithParent: z.boolean().optional(),
+      resultId: z.string().optional(),
+      answers: z.array(answerInSessionStateSchemaV2),
+      typebot: typebotInSessionStateSchema,
+    })
+  ),
+  dynamicTheme: dynamicThemeSchema.optional(),
+  currentBlock: z
+    .object({
+      blockId: z.string(),
+      groupId: z.string(),
+    })
+    .optional(),
+  isStreamEnabled: z.boolean().optional(),
+})
+
+export type SessionState = z.infer<typeof sessionStateSchemaV2>
+
+export const sessionStateSchema = sessionStateSchemaV1
+  .or(sessionStateSchemaV2)
+  .transform((state): SessionState => {
+    if ('version' in state) return state
+    return {
+      version: '2',
+      typebotsQueue: [
+        {
+          typebot: state.typebot,
+          resultId: state.result.id,
+          answers: state.result.answers.map((answer) => ({
+            key:
+              (answer.variableId
+                ? state.typebot.variables.find(
+                    (variable) => variable.id === answer.variableId
+                  )?.name
+                : state.typebot.groups.find((group) =>
+                    group.blocks.find((block) => block.id === answer.blockId)
+                  )?.title) ?? '',
+            value: answer.content,
+          })),
+          isMergingWithParent: true,
+          edgeIdToTriggerWhenDone:
+            state.linkedTypebots.queue.length > 0
+              ? state.linkedTypebots.queue[0].edgeId
+              : undefined,
+        },
+        ...state.linkedTypebots.typebots.map(
+          (typebot, index) =>
+            ({
+              typebot,
+              resultId: state.result.id,
+              answers: state.result.answers.map((answer) => ({
+                key:
+                  (answer.variableId
+                    ? state.typebot.variables.find(
+                        (variable) => variable.id === answer.variableId
+                      )?.name
+                    : state.typebot.groups.find((group) =>
+                        group.blocks.find(
+                          (block) => block.id === answer.blockId
+                        )
+                      )?.title) ?? '',
+                value: answer.content,
+              })),
+              edgeIdToTriggerWhenDone: state.linkedTypebots.queue.at(index + 1)
+                ?.edgeId,
+            } satisfies SessionState['typebotsQueue'][number])
+        ),
+      ],
+      dynamicTheme: state.dynamicTheme,
+      currentBlock: state.currentBlock,
+      isStreamEnabled: state.isStreamEnabled,
+    }
+  })