84 lines
2.2 KiB
TypeScript
84 lines
2.2 KiB
TypeScript
import { TRPCError } from '@trpc/server'
|
|
import {
|
|
ContinueChatResponse,
|
|
SessionState,
|
|
StartFrom,
|
|
} from '@typebot.io/schemas'
|
|
import { executeGroup } from './executeGroup'
|
|
import { getNextGroup } from './getNextGroup'
|
|
import { VisitedEdge } from '@typebot.io/prisma'
|
|
|
|
type Props = {
|
|
version: 1 | 2
|
|
state: SessionState
|
|
startFrom?: StartFrom
|
|
startTime?: number
|
|
}
|
|
|
|
export const startBotFlow = async ({
|
|
version,
|
|
state,
|
|
startFrom,
|
|
startTime,
|
|
}: Props): Promise<
|
|
ContinueChatResponse & {
|
|
newSessionState: SessionState
|
|
visitedEdges: VisitedEdge[]
|
|
}
|
|
> => {
|
|
let newSessionState = state
|
|
const visitedEdges: VisitedEdge[] = []
|
|
if (startFrom?.type === 'group') {
|
|
const group = state.typebotsQueue[0].typebot.groups.find(
|
|
(group) => group.id === startFrom.groupId
|
|
)
|
|
if (!group)
|
|
throw new TRPCError({
|
|
code: 'BAD_REQUEST',
|
|
message: "Start group doesn't exist",
|
|
})
|
|
return executeGroup(group, {
|
|
version,
|
|
state: newSessionState,
|
|
visitedEdges,
|
|
startTime,
|
|
})
|
|
}
|
|
const firstEdgeId = getFirstEdgeId({
|
|
state: newSessionState,
|
|
startEventId: startFrom?.type === 'event' ? startFrom.eventId : undefined,
|
|
})
|
|
if (!firstEdgeId) return { messages: [], newSessionState, visitedEdges: [] }
|
|
const nextGroup = await getNextGroup(newSessionState)(firstEdgeId)
|
|
newSessionState = nextGroup.newSessionState
|
|
if (nextGroup.visitedEdge) visitedEdges.push(nextGroup.visitedEdge)
|
|
if (!nextGroup.group) return { messages: [], newSessionState, visitedEdges }
|
|
return executeGroup(nextGroup.group, {
|
|
version,
|
|
state: newSessionState,
|
|
visitedEdges,
|
|
startTime,
|
|
})
|
|
}
|
|
|
|
const getFirstEdgeId = ({
|
|
state,
|
|
startEventId,
|
|
}: {
|
|
state: SessionState
|
|
startEventId: string | undefined
|
|
}) => {
|
|
const { typebot } = state.typebotsQueue[0]
|
|
if (startEventId) {
|
|
const event = typebot.events?.find((e) => e.id === startEventId)
|
|
if (!event)
|
|
throw new TRPCError({
|
|
code: 'BAD_REQUEST',
|
|
message: "Start event doesn't exist",
|
|
})
|
|
return event.outgoingEdgeId
|
|
}
|
|
if (typebot.version === '6') return typebot.events[0].outgoingEdgeId
|
|
return typebot.groups[0].blocks[0].outgoingEdgeId
|
|
}
|