🧑‍💻 (chat) Introduce startChat and continueChat endpoints

Closes #1030

apps/viewer/src/features/chat/api/legacy/sendMessageV2.ts

@@ -0,0 +1,184 @@
+import { publicProcedure } from '@/helpers/server/trpc'
+import { TRPCError } from '@trpc/server'
+import { getSession } from '@typebot.io/bot-engine/queries/getSession'
+import { startSession } from '@typebot.io/bot-engine/startSession'
+import { saveStateToDatabase } from '@typebot.io/bot-engine/saveStateToDatabase'
+import { restartSession } from '@typebot.io/bot-engine/queries/restartSession'
+import { continueBotFlow } from '@typebot.io/bot-engine/continueBotFlow'
+import { parseDynamicTheme } from '@typebot.io/bot-engine/parseDynamicTheme'
+import { isDefined } from '@typebot.io/lib/utils'
+import {
+  chatReplySchema,
+  sendMessageInputSchema,
+} from '@typebot.io/schemas/features/chat/legacy/schema'
+
+export const sendMessageV2 = publicProcedure
+  .meta({
+    openapi: {
+      method: 'POST',
+      path: '/v2/sendMessage',
+      summary: 'Send a message',
+      description:
+        'To initiate a chat, do not provide a `sessionId` nor a `message`.\n\nContinue the conversation by providing the `sessionId` and the `message` that should answer the previous question.\n\nSet the `isPreview` option to `true` to chat with the non-published version of the typebot.',
+      tags: ['Deprecated'],
+      deprecated: true,
+    },
+  })
+  .input(sendMessageInputSchema)
+  .output(chatReplySchema)
+  .mutation(
+    async ({
+      input: { sessionId, message, startParams, clientLogs },
+      ctx: { user },
+    }) => {
+      const session = sessionId ? await getSession(sessionId) : null
+
+      const isSessionExpired =
+        session &&
+        isDefined(session.state.expiryTimeout) &&
+        session.updatedAt.getTime() + session.state.expiryTimeout < Date.now()
+
+      if (isSessionExpired)
+        throw new TRPCError({
+          code: 'NOT_FOUND',
+          message: 'Session expired. You need to start a new session.',
+        })
+
+      if (!session) {
+        if (!startParams)
+          throw new TRPCError({
+            code: 'BAD_REQUEST',
+            message: 'Missing startParams',
+          })
+        const {
+          typebot,
+          messages,
+          input,
+          resultId,
+          dynamicTheme,
+          logs,
+          clientSideActions,
+          newSessionState,
+          visitedEdges,
+        } = await startSession({
+          version: 2,
+          startParams:
+            startParams.isPreview || typeof startParams.typebot !== 'string'
+              ? {
+                  type: 'preview',
+                  isOnlyRegistering: startParams.isOnlyRegistering ?? false,
+                  isStreamEnabled: startParams.isStreamEnabled,
+                  startFrom:
+                    'startGroupId' in startParams && startParams.startGroupId
+                      ? {
+                          type: 'group',
+                          groupId: startParams.startGroupId,
+                        }
+                      : 'startEventId' in startParams &&
+                        startParams.startEventId
+                      ? {
+                          type: 'event',
+                          eventId: startParams.startEventId,
+                        }
+                      : undefined,
+                  typebotId:
+                    typeof startParams.typebot === 'string'
+                      ? startParams.typebot
+                      : startParams.typebot.id,
+                  typebot:
+                    typeof startParams.typebot === 'string'
+                      ? undefined
+                      : startParams.typebot,
+                  message,
+                  userId: parseUserId(user?.id),
+                }
+              : {
+                  type: 'live',
+                  isOnlyRegistering: startParams.isOnlyRegistering ?? false,
+                  isStreamEnabled: startParams.isStreamEnabled,
+                  publicId: startParams.typebot,
+                  prefilledVariables: startParams.prefilledVariables,
+                  resultId: startParams.resultId,
+                  message,
+                },
+          message,
+        })
+
+        const allLogs = clientLogs ? [...(logs ?? []), ...clientLogs] : logs
+
+        const session = startParams?.isOnlyRegistering
+          ? await restartSession({
+              state: newSessionState,
+            })
+          : await saveStateToDatabase({
+              session: {
+                state: newSessionState,
+              },
+              input,
+              logs: allLogs,
+              clientSideActions,
+              visitedEdges,
+            })
+
+        return {
+          sessionId: session.id,
+          typebot: typebot
+            ? {
+                id: typebot.id,
+                theme: typebot.theme,
+                settings: typebot.settings,
+              }
+            : undefined,
+          messages,
+          input,
+          resultId,
+          dynamicTheme,
+          logs,
+          clientSideActions,
+        }
+      } else {
+        const {
+          messages,
+          input,
+          clientSideActions,
+          newSessionState,
+          logs,
+          lastMessageNewFormat,
+          visitedEdges,
+        } = await continueBotFlow(message, { version: 2, state: session.state })
+
+        const allLogs = clientLogs ? [...(logs ?? []), ...clientLogs] : logs
+
+        if (newSessionState)
+          await saveStateToDatabase({
+            session: {
+              id: session.id,
+              state: newSessionState,
+            },
+            input,
+            logs: allLogs,
+            clientSideActions,
+            visitedEdges,
+          })
+
+        return {
+          messages,
+          input,
+          clientSideActions,
+          dynamicTheme: parseDynamicTheme(newSessionState),
+          logs,
+          lastMessageNewFormat,
+        }
+      }
+    }
+  )
+
+const parseUserId = (userId?: string): string => {
+  if (!userId)
+    throw new TRPCError({
+      code: 'UNAUTHORIZED',
+      message: 'You need to be authenticated to perform this action',
+    })
+
+  return userId
+}