You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.6 KiB
TypeScript
49 lines
1.6 KiB
TypeScript
import * as dotenv from 'dotenv';
|
|
import { migrate as neonMigrate } from 'drizzle-orm/neon-serverless/migrator';
|
|
import { migrate as nodeMigrate } from 'drizzle-orm/node-postgres/migrator';
|
|
import { join } from 'node:path';
|
|
|
|
import { serverDB } from '../../src/database/server/core/db';
|
|
import { DB_FAIL_INIT_HINT, PGVECTOR_HINT } from './errorHint';
|
|
|
|
// Read the `.env` file if it exists, or a file specified by the
|
|
// dotenv_config_path parameter that's passed to Node.js
|
|
dotenv.config();
|
|
|
|
const migrationsFolder = join(__dirname, '../../src/database/migrations');
|
|
|
|
const runMigrations = async () => {
|
|
if (process.env.DATABASE_DRIVER === 'node') {
|
|
await nodeMigrate(serverDB, { migrationsFolder });
|
|
} else {
|
|
await neonMigrate(serverDB, { migrationsFolder });
|
|
}
|
|
|
|
console.log('✅ database migration pass.');
|
|
// eslint-disable-next-line unicorn/no-process-exit
|
|
process.exit(0);
|
|
};
|
|
|
|
let connectionString = process.env.DATABASE_URL;
|
|
|
|
// only migrate database if the connection string is available
|
|
if (connectionString) {
|
|
// eslint-disable-next-line unicorn/prefer-top-level-await
|
|
runMigrations().catch((err) => {
|
|
console.error('❌ Database migrate failed:', err);
|
|
|
|
const errMsg = err.message as string;
|
|
|
|
if (errMsg.includes('extension "vector" is not available')) {
|
|
console.info(PGVECTOR_HINT);
|
|
} else if (errMsg.includes(`Cannot read properties of undefined (reading 'migrate')`)) {
|
|
console.info(DB_FAIL_INIT_HINT);
|
|
}
|
|
|
|
// eslint-disable-next-line unicorn/no-process-exit
|
|
process.exit(1);
|
|
});
|
|
} else {
|
|
console.log('🟢 not find database env, migration skipped');
|
|
}
|