♻️ Remove @typebot.io/schemas from @typebot.io/lib
This commit is contained in:
@ -1,92 +0,0 @@
|
||||
import { Prisma, PrismaClient } from '@typebot.io/prisma'
|
||||
import { Block, Typebot } from '@typebot.io/schemas'
|
||||
import { deleteFilesFromBucket } from '../../s3/deleteFilesFromBucket'
|
||||
import { InputBlockType } from '@typebot.io/schemas/features/blocks/inputs/constants'
|
||||
|
||||
type ArchiveResultsProps = {
|
||||
typebot: Pick<Typebot, 'groups'>
|
||||
resultsFilter?: Omit<Prisma.ResultWhereInput, 'typebotId'> & {
|
||||
typebotId: string
|
||||
}
|
||||
}
|
||||
|
||||
export const archiveResults =
|
||||
(prisma: PrismaClient) =>
|
||||
async ({ typebot, resultsFilter }: ArchiveResultsProps) => {
|
||||
const batchSize = 100
|
||||
const fileUploadBlockIds = typebot.groups
|
||||
.flatMap<Block>((group) => group.blocks)
|
||||
.filter((block) => block.type === InputBlockType.FILE)
|
||||
.map((block) => block.id)
|
||||
|
||||
let currentTotalResults = 0
|
||||
|
||||
const resultsCount = await prisma.result.count({
|
||||
where: {
|
||||
...resultsFilter,
|
||||
OR: [{ isArchived: false }, { isArchived: null }],
|
||||
},
|
||||
})
|
||||
|
||||
if (resultsCount === 0) return { success: true }
|
||||
|
||||
let progress = 0
|
||||
|
||||
do {
|
||||
progress += batchSize
|
||||
console.log(`Archiving ${progress} / ${resultsCount} results...`)
|
||||
const resultsToDelete = await prisma.result.findMany({
|
||||
where: {
|
||||
...resultsFilter,
|
||||
OR: [{ isArchived: false }, { isArchived: null }],
|
||||
},
|
||||
select: {
|
||||
id: true,
|
||||
},
|
||||
take: batchSize,
|
||||
})
|
||||
|
||||
if (resultsToDelete.length === 0) break
|
||||
|
||||
currentTotalResults = resultsToDelete.length
|
||||
|
||||
const resultIds = resultsToDelete.map((result) => result.id)
|
||||
|
||||
if (fileUploadBlockIds.length > 0) {
|
||||
const filesToDelete = await prisma.answer.findMany({
|
||||
where: {
|
||||
resultId: { in: resultIds },
|
||||
blockId: { in: fileUploadBlockIds },
|
||||
},
|
||||
})
|
||||
if (filesToDelete.length > 0)
|
||||
await deleteFilesFromBucket({
|
||||
urls: filesToDelete.flatMap((a) => a.content.split(', ')),
|
||||
})
|
||||
}
|
||||
|
||||
await prisma.$transaction([
|
||||
prisma.log.deleteMany({
|
||||
where: {
|
||||
resultId: { in: resultIds },
|
||||
},
|
||||
}),
|
||||
prisma.answer.deleteMany({
|
||||
where: {
|
||||
resultId: { in: resultIds },
|
||||
},
|
||||
}),
|
||||
prisma.result.updateMany({
|
||||
where: {
|
||||
id: { in: resultIds },
|
||||
},
|
||||
data: {
|
||||
isArchived: true,
|
||||
variables: [],
|
||||
},
|
||||
}),
|
||||
])
|
||||
} while (currentTotalResults >= batchSize)
|
||||
|
||||
return { success: true }
|
||||
}
|
Reference in New Issue
Block a user