2
0
Files
bot/packages/migrations/migrateTypebotFromV5ToV6.ts
2024-03-15 16:32:29 +01:00

126 lines
3.4 KiB
TypeScript

import {
BlockV5,
BlockV6,
GoogleSheetsBlockV5,
GoogleSheetsBlockV6,
PublicTypebotV5,
PublicTypebotV6,
TypebotV5,
TypebotV6,
} from '@typebot.io/schemas'
import { IntegrationBlockType } from '@typebot.io/schemas/features/blocks/integrations/constants'
import { GoogleSheetsAction } from '@typebot.io/schemas/features/blocks/integrations/googleSheets/constants'
import { ComparisonOperators } from '@typebot.io/schemas/features/blocks/logic/condition/constants'
import { createId } from '@typebot.io/lib/createId'
import { EventType } from '@typebot.io/schemas/features/events/constants'
import { byId } from '@typebot.io/lib/utils'
export const migrateTypebotFromV5ToV6 = async (
typebot: TypebotV5 | PublicTypebotV5
): Promise<TypebotV6 | PublicTypebotV6> => {
const startGroup = typebot.groups.find((group) =>
group.blocks.some((b) => b.type === 'start')
)
if (!startGroup) throw new Error('Start group not found')
const startBlock = startGroup?.blocks.find((b) => b.type === 'start')
if (!startBlock) throw new Error('Start block not found')
const startOutgoingEdge = typebot.edges.find(byId(startBlock.outgoingEdgeId))
return {
...typebot,
groups: migrateGroups(
typebot.groups.filter((g) => g.blocks.some((b) => b.type !== 'start'))
),
version: '6',
events: [
{
id: startGroup.id,
type: EventType.START,
graphCoordinates: startGroup.graphCoordinates,
outgoingEdgeId: startBlock.outgoingEdgeId,
},
],
edges: startOutgoingEdge
? [
{
...startOutgoingEdge,
from: {
eventId: startGroup.id,
},
},
...typebot.edges.filter((e) => e.id !== startOutgoingEdge.id),
]
: typebot.edges,
}
}
const migrateGroups = (groups: TypebotV5['groups']): TypebotV6['groups'] =>
groups.map((group) => ({
...group,
blocks: migrateBlocksFromV1ToV2(group.blocks),
}))
const migrateBlocksFromV1ToV2 = (
blocks: TypebotV5['groups'][0]['blocks']
): BlockV6[] =>
(
blocks.filter((block) => block.type !== 'start') as Exclude<
BlockV5,
{ type: 'start' }
>[]
).map((block) => {
if (block.type === IntegrationBlockType.GOOGLE_SHEETS) {
return {
...block,
options: migrateGoogleSheetsOptions(block.options),
}
}
return block
})
const migrateGoogleSheetsOptions = (
options: GoogleSheetsBlockV5['options']
): GoogleSheetsBlockV6['options'] => {
if (!options) return
if (options.action === GoogleSheetsAction.GET) {
if (options.filter || !options.referenceCell) return options
return {
...options,
filter: {
comparisons: [
{
id: createId(),
column: options.referenceCell?.column,
comparisonOperator: ComparisonOperators.EQUAL,
value: options.referenceCell?.value,
},
],
},
}
}
if (options.action === GoogleSheetsAction.INSERT_ROW) {
return options
}
if (options.action === GoogleSheetsAction.UPDATE_ROW) {
if (options.filter || !options.referenceCell) return options
return {
...options,
filter: {
comparisons: [
{
id: createId(),
column: options.referenceCell?.column,
comparisonOperator: ComparisonOperators.EQUAL,
value: options.referenceCell?.value,
},
],
},
}
}
return options
}