37 lines
1.3 KiB
TypeScript
37 lines
1.3 KiB
TypeScript
import type { PostgresAdapter } from '@payloadcms/db-postgres/types'
|
|
import type { Payload } from 'payload'
|
|
|
|
import { sql } from 'drizzle-orm'
|
|
|
|
import { isMongoose } from './isMongoose.js'
|
|
|
|
export async function resetDB(_payload: Payload, collectionSlugs: string[]) {
|
|
if (isMongoose(_payload) && 'collections' in _payload.db && collectionSlugs.length > 0) {
|
|
await _payload.db.collections[collectionSlugs[0]].db.dropDatabase()
|
|
} else {
|
|
const db: PostgresAdapter = _payload.db as unknown as PostgresAdapter
|
|
|
|
// Alternative to: await db.drizzle.execute(sql`drop schema public cascade; create schema public;`)
|
|
|
|
// Deleting the schema causes issues when restoring the database from a snapshot later on. That's why we only delete the table data here,
|
|
// To avoid having to re-create any table schemas / indexes / whatever
|
|
const schema = db.drizzle._.schema
|
|
if (!schema) {
|
|
return
|
|
}
|
|
const queries = Object.values(schema).map((table: any) => {
|
|
return sql.raw(`DELETE FROM ${db.schemaName ? db.schemaName + '.' : ''}${table.dbName}`)
|
|
})
|
|
|
|
await db.drizzle.transaction(async (trx) => {
|
|
await Promise.all(
|
|
queries.map(async (query) => {
|
|
if (query) {
|
|
await trx.execute(query)
|
|
}
|
|
}),
|
|
)
|
|
})
|
|
}
|
|
}
|