✨ Introducing The Forge (#1072)

The Forge allows anyone to easily create their own Typebot Block.

Closes #380

packages/bot-engine/continueBotFlow.ts

@@ -12,7 +12,6 @@ import { getNextGroup } from './getNextGroup'
 import { validateEmail } from './blocks/inputs/email/validateEmail'
 import { formatPhoneNumber } from './blocks/inputs/phone/formatPhoneNumber'
 import { validateUrl } from './blocks/inputs/url/validateUrl'
-import { resumeChatCompletion } from './blocks/integrations/openai/resumeChatCompletion'
 import { resumeWebhookExecution } from './blocks/integrations/webhook/resumeWebhookExecution'
 import { upsertAnswer } from './queries/upsertAnswer'
 import { parseButtonsReply } from './blocks/inputs/buttons/parseButtonsReply'
@@ -21,8 +20,8 @@ import { validateNumber } from './blocks/inputs/number/validateNumber'
 import { parseDateReply } from './blocks/inputs/date/parseDateReply'
 import { validateRatingReply } from './blocks/inputs/rating/validateRatingReply'
 import { parsePictureChoicesReply } from './blocks/inputs/pictureChoice/parsePictureChoicesReply'
-import { parseVariables } from './variables/parseVariables'
-import { updateVariablesInSession } from './variables/updateVariablesInSession'
+import { parseVariables } from '@typebot.io/variables/parseVariables'
+import { updateVariablesInSession } from '@typebot.io/variables/updateVariablesInSession'
 import { startBotFlow } from './startBotFlow'
 import { TRPCError } from '@trpc/server'
 import { parseNumber } from './blocks/inputs/number/parseNumber'
@@ -37,6 +36,9 @@ import { defaultPictureChoiceOptions } from '@typebot.io/schemas/features/blocks
 import { defaultFileInputOptions } from '@typebot.io/schemas/features/blocks/inputs/file/constants'
 import { VisitedEdge } from '@typebot.io/prisma'
 import { getBlockById } from '@typebot.io/lib/getBlockById'
+import { ForgedBlock, forgedBlocks } from '@typebot.io/forge-schemas'
+import { enabledBlocks } from '@typebot.io/forge-repository'
+import { resumeChatCompletion } from './blocks/integrations/legacy/openai/resumeChatCompletion'
 
 type Params = {
   version: 1 | 2
@@ -80,14 +82,9 @@ export const continueBotFlow = async (
       }
       newSessionState = updateVariablesInSession(state)([newVariable])
     }
-  } else if (reply && block.type === IntegrationBlockType.WEBHOOK) {
-    const result = resumeWebhookExecution({
-      state,
-      block,
-      response: JSON.parse(reply),
-    })
-    if (result.newSessionState) newSessionState = result.newSessionState
-  } else if (
+  }
+  // Legacy
+  else if (
     block.type === IntegrationBlockType.OPEN_AI &&
     block.options?.task === 'Create chat completion'
   ) {
@@ -99,6 +96,58 @@ export const continueBotFlow = async (
       })(reply)
       newSessionState = result.newSessionState
     }
+  } else if (reply && block.type === IntegrationBlockType.WEBHOOK) {
+    const result = resumeWebhookExecution({
+      state,
+      block,
+      response: JSON.parse(reply),
+    })
+    if (result.newSessionState) newSessionState = result.newSessionState
+  } else if (
+    enabledBlocks.includes(block.type as (typeof enabledBlocks)[number])
+  ) {
+    if (reply) {
+      const options = (block as ForgedBlock).options
+      const action = forgedBlocks
+        .find((b) => b.id === block.type)
+        ?.actions.find((a) => a.name === options?.action)
+      if (action) {
+        if (action.run?.stream?.getStreamVariableId) {
+          firstBubbleWasStreamed = true
+          const variableToUpdate =
+            state.typebotsQueue[0].typebot.variables.find(
+              (v) => v.id === action?.run?.stream?.getStreamVariableId(options)
+            )
+          if (variableToUpdate)
+            newSessionState = updateVariablesInSession(state)([
+              {
+                ...variableToUpdate,
+                value: reply,
+              },
+            ])
+        }
+
+        if (
+          action.run?.web?.displayEmbedBubble?.waitForEvent?.getSaveVariableId
+        ) {
+          const variableToUpdate =
+            state.typebotsQueue[0].typebot.variables.find(
+              (v) =>
+                v.id ===
+                action?.run?.web?.displayEmbedBubble?.waitForEvent?.getSaveVariableId?.(
+                  options
+                )
+            )
+          if (variableToUpdate)
+            newSessionState = updateVariablesInSession(state)([
+              {
+                ...variableToUpdate,
+                value: reply,
+              },
+            ])
+        }
+      }
+    }
   }
 
   let formattedReply: string | undefined