|
| 1 | +import { $ } from "bun"; |
1 | 2 | import chalk from "chalk"; |
2 | | -import { drizzle } from "drizzle-orm/bun-sql"; |
3 | | -import { reset } from "drizzle-seed"; |
4 | | -import * as articlesSchema from "@/articles/articles.schema"; |
5 | | -import * as commentsSchema from "@/comments/comments.schema"; |
6 | | -import { env } from "@/core/env"; |
7 | | -import * as profilesSchema from "@/profiles/profiles.schema"; |
8 | | -import * as tagsSchema from "@/tags/tags.schema"; |
9 | | -import * as usersSchema from "@/users/users.schema"; |
10 | | - |
11 | | -const schema = { |
12 | | - ...usersSchema, |
13 | | - ...profilesSchema, |
14 | | - ...tagsSchema, |
15 | | - ...articlesSchema, |
16 | | - ...commentsSchema, |
17 | | -}; |
| 3 | +import { sql } from "drizzle-orm"; |
| 4 | +import { db } from "@/core/database"; |
18 | 5 |
|
19 | 6 | console.info(chalk.gray("Resetting database")); |
20 | | -// See: https://github.com/drizzle-team/drizzle-orm/issues/3599 |
21 | | -await reset(drizzle(env.DATABASE_URL), schema); |
| 7 | +const query = sql` |
| 8 | + -- Delete all tables |
| 9 | + DO $$ DECLARE |
| 10 | + r RECORD; |
| 11 | + BEGIN |
| 12 | + FOR r IN (SELECT tablename FROM pg_tables WHERE schemaname = current_schema()) LOOP |
| 13 | + EXECUTE 'DROP TABLE IF EXISTS ' || quote_ident(r.tablename) || ' CASCADE'; |
| 14 | + END LOOP; |
| 15 | + END $$; |
| 16 | +
|
| 17 | + -- Delete enums |
| 18 | + DO $$ DECLARE |
| 19 | + r RECORD; |
| 20 | + BEGIN |
| 21 | + FOR r IN (select t.typname as enum_name |
| 22 | + from pg_type t |
| 23 | + join pg_enum e on t.oid = e.enumtypid |
| 24 | + join pg_catalog.pg_namespace n ON n.oid = t.typnamespace |
| 25 | + where n.nspname = current_schema()) LOOP |
| 26 | + EXECUTE 'DROP TYPE IF EXISTS ' || quote_ident(r.enum_name); |
| 27 | + END LOOP; |
| 28 | + END $$; |
| 29 | + `; |
| 30 | + |
| 31 | +await db.execute(query); |
| 32 | + |
| 33 | +// Push the schema to the database |
| 34 | +await $`bun run db:push`.quiet(); |
22 | 35 | console.info(`[${chalk.green("✓")}] Database reset complete`); |
0 commit comments