2025-05-07 17:14:20 -04:00
|
|
|
import { Kysely, sql } from 'kysely';
|
|
|
|
|
|
|
|
|
|
export async function up(db: Kysely<any>): Promise<void> {
|
|
|
|
|
const { rows } = await sql<{ db: string }>`SELECT current_database() as db`.execute(db);
|
|
|
|
|
const databaseName = rows[0].db;
|
|
|
|
|
await sql.raw(`ALTER DATABASE "${databaseName}" SET search_path TO "$user", public, vectors`).execute(db);
|
2025-05-09 14:45:44 +01:00
|
|
|
const naturalearth_pkey = await sql<{ constraint_name: string }>`SELECT constraint_name
|
|
|
|
|
FROM information_schema.table_constraints
|
|
|
|
|
WHERE table_schema = 'public'
|
|
|
|
|
AND table_name = 'naturalearth_countries'
|
|
|
|
|
AND constraint_type = 'PRIMARY KEY';`.execute(db);
|
|
|
|
|
const naturalearth_pkey_name = naturalearth_pkey.rows[0]?.constraint_name;
|
|
|
|
|
if(naturalearth_pkey_name) {
|
|
|
|
|
await sql`ALTER TABLE "naturalearth_countries"
|
|
|
|
|
DROP CONSTRAINT ${sql.ref(naturalearth_pkey_name)};`.execute(db);
|
|
|
|
|
}
|
2025-05-07 17:14:20 -04:00
|
|
|
await sql`ALTER TABLE "naturalearth_countries" ADD CONSTRAINT "naturalearth_countries_pkey" PRIMARY KEY ("id") WITH (FILLFACTOR = 100);`.execute(db);
|
|
|
|
|
await sql`DROP INDEX IF EXISTS "IDX_02a43fd0b3c50fb6d7f0cb7282";`.execute(db);
|
|
|
|
|
await sql`DROP INDEX IF EXISTS "IDX_95ad7106dd7b484275443f580f";`.execute(db);
|
|
|
|
|
await sql`DROP INDEX IF EXISTS "IDX_7e077a8b70b3530138610ff5e0";`.execute(db);
|
|
|
|
|
await sql`DROP INDEX IF EXISTS "IDX_92e67dc508c705dd66c9461557";`.execute(db);
|
|
|
|
|
await sql`DROP INDEX IF EXISTS "IDX_6afb43681a21cf7815932bc38a";`.execute(db);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
export async function down(db: Kysely<any>): Promise<void> {
|
|
|
|
|
const { rows } = await sql<{ db: string }>`SELECT current_database() as db`.execute(db);
|
|
|
|
|
const databaseName = rows[0].db;
|
|
|
|
|
await sql.raw(`ALTER DATABASE "${databaseName}" RESET "search_path"`).execute(db);
|
|
|
|
|
}
|