feat: enhance existing validation scripts with better error messages

- Improve check-env.js with color-coded output and detailed solutions

- Enhance check-db.js with comprehensive troubleshooting steps

- Add variable descriptions and examples

- Include PostgreSQL installation and connection guidance
This commit is contained in:
Ayush3603 2025-11-10 22:11:56 +05:30
parent ddc005625d
commit 06422fb65f
2 changed files with 202 additions and 34 deletions

View file

@ -3,16 +3,70 @@ import 'dotenv/config';
import { execSync } from 'node:child_process'; import { execSync } from 'node:child_process';
import chalk from 'chalk'; import chalk from 'chalk';
import semver from 'semver'; import semver from 'semver';
// eslint-disable-next-line import/no-unresolved
import { PrismaClient } from '../generated/prisma/client.js'; import { PrismaClient } from '../generated/prisma/client.js';
import { PrismaPg } from '@prisma/adapter-pg'; import { PrismaPg } from '@prisma/adapter-pg';
const MIN_VERSION = '9.4.0'; const MIN_VERSION = '9.4.0';
if (process.env.SKIP_DB_CHECK) { if (process.env.SKIP_DB_CHECK) {
console.log('Skipping database check.'); console.log(chalk.yellow('⚠️ Skipping database check (SKIP_DB_CHECK is set).\n'));
process.exit(0); process.exit(0);
} }
console.log(chalk.bold.cyan('\n🔍 Checking Database Configuration...\n'));
function success(msg) {
console.log(chalk.greenBright(`${msg}`));
}
function error(msg, solution = null, documentation = null) {
console.log(chalk.redBright(`${msg}`));
if (solution) {
console.log(chalk.yellow(`\n💡 Solution:\n${solution}`));
}
if (documentation) {
console.log(chalk.blue(`\n📖 Documentation: ${documentation}`));
}
}
async function checkEnv() {
if (!process.env.DATABASE_URL) {
const solution = ` 1. Create a .env file in the project root
2. Add DATABASE_URL with your PostgreSQL connection string
3. Format: postgresql://username:password@host:port/database
4. Example: postgresql://umami:mypassword@localhost:5432/umami`;
error('DATABASE_URL is not defined.', solution, 'See .env.example for template');
throw new Error('DATABASE_URL is not defined.');
}
// Validate URL format before proceeding
try {
const url = new URL(process.env.DATABASE_URL);
if (url.protocol !== 'postgresql:') {
const solution = ` DATABASE_URL must use PostgreSQL protocol
Format: postgresql://username:password@host:port/database
Example: postgresql://umami:mypassword@localhost:5432/umami`;
error('DATABASE_URL must be a PostgreSQL connection string.', solution);
throw new Error('Invalid DATABASE_URL protocol');
}
success('DATABASE_URL is defined and format is valid.');
} catch (e) {
if (e.message === 'Invalid DATABASE_URL protocol') {
throw e;
}
const solution = ` DATABASE_URL format is invalid
Format: postgresql://username:password@host:port/database
Example: postgresql://umami:mypassword@localhost:5432/umami`;
error('DATABASE_URL format is invalid.', solution);
throw new Error('Invalid DATABASE_URL format');
}
}
async function checkConnection() {
const url = new URL(process.env.DATABASE_URL); const url = new URL(process.env.DATABASE_URL);
const adapter = new PrismaPg( const adapter = new PrismaPg(
@ -22,65 +76,126 @@ const adapter = new PrismaPg(
const prisma = new PrismaClient({ adapter }); const prisma = new PrismaClient({ adapter });
function success(msg) {
console.log(chalk.greenBright(`${msg}`));
}
function error(msg) {
console.log(chalk.redBright(`${msg}`));
}
async function checkEnv() {
if (!process.env.DATABASE_URL) {
throw new Error('DATABASE_URL is not defined.');
} else {
success('DATABASE_URL is defined.');
}
}
async function checkConnection() {
try { try {
await prisma.$connect(); await prisma.$connect();
success('Database connection successful.'); success('Database connection successful.');
await prisma.$disconnect();
} catch (e) { } catch (e) {
throw new Error('Unable to connect to the database: ' + e.message); const solution = ` Common causes and solutions:
1. PostgreSQL is not running
- Start PostgreSQL service
- Check: pg_ctl status or systemctl status postgresql
2. Wrong credentials
- Verify username and password in DATABASE_URL
- Test connection: psql -U username -d database -h host
3. Database doesn't exist
- Create database: createdb umami
- Or use psql: CREATE DATABASE umami;
4. Connection refused
- Check PostgreSQL is listening on the correct port
- Verify firewall settings
- Check pg_hba.conf for access permissions`;
error(`Unable to connect to the database: ${e.message}`, solution);
throw new Error('Database connection failed');
} }
} }
async function checkDatabaseVersion() { async function checkDatabaseVersion() {
const url = new URL(process.env.DATABASE_URL);
const adapter = new PrismaPg(
{ connectionString: url.toString() },
{ schema: url.searchParams.get('schema') },
);
const prisma = new PrismaClient({ adapter });
try {
const query = await prisma.$queryRaw`select version() as version`; const query = await prisma.$queryRaw`select version() as version`;
const version = semver.valid(semver.coerce(query[0].version)); const version = semver.valid(semver.coerce(query[0].version));
if (semver.lt(version, MIN_VERSION)) { if (semver.lt(version, MIN_VERSION)) {
throw new Error( const solution = ` Your PostgreSQL version (${version}) is below the minimum required (${MIN_VERSION})
`Database version is not compatible. Please upgrade to ${MIN_VERSION} or greater.`,
Upgrade options:
1. Using package manager (Ubuntu/Debian):
sudo apt-get update
sudo apt-get install postgresql-14
2. Using package manager (macOS with Homebrew):
brew upgrade postgresql
3. Download from official site:
https://www.postgresql.org/download/`;
error(
`Database version ${version} is not compatible. Minimum required: ${MIN_VERSION}`,
solution,
'https://www.postgresql.org/download/',
); );
await prisma.$disconnect();
throw new Error('Database version incompatible');
} }
success('Database version check successful.'); success(`Database version check successful (PostgreSQL ${version}).`);
await prisma.$disconnect();
} catch (e) {
if (e.message === 'Database version incompatible') {
throw e;
}
error(`Unable to check database version: ${e.message}`);
throw e;
}
} }
async function applyMigration() { async function applyMigration() {
if (!process.env.SKIP_DB_MIGRATION) { if (!process.env.SKIP_DB_MIGRATION) {
try {
console.log(chalk.cyan('\nApplying database migrations...\n'));
console.log(execSync('prisma migrate deploy').toString()); console.log(execSync('prisma migrate deploy').toString());
success('Database is up to date.'); success('Database is up to date.');
} catch {
const solution = ` Migration failed. Try these steps:
1. Ensure database is accessible
2. Check migration files in prisma/migrations
3. Reset database if needed: prisma migrate reset
4. Manually apply migrations: prisma migrate deploy`;
error('Failed to apply database migrations.', solution);
throw new Error('Migration failed');
}
} else {
console.log(chalk.yellow('⚠️ Skipping database migrations (SKIP_DB_MIGRATION is set).\n'));
} }
} }
(async () => { (async () => {
let err = false; let err = false;
for (const fn of [checkEnv, checkConnection, checkDatabaseVersion, applyMigration]) { const checks = [
{ name: 'Environment', fn: checkEnv },
{ name: 'Connection', fn: checkConnection },
{ name: 'Version', fn: checkDatabaseVersion },
{ name: 'Migration', fn: applyMigration },
];
for (const check of checks) {
try { try {
await fn(); await check.fn();
} catch (e) { } catch {
error(e.message);
err = true; err = true;
} finally { break; // Stop on first error
}
}
if (err) { if (err) {
console.log(chalk.red.bold('\n❌ Database check failed!\n'));
console.log(chalk.cyan('Please fix the errors above and try again.\n'));
process.exit(1); process.exit(1);
} } else {
} console.log(chalk.green.bold('\n✅ All database checks passed!\n'));
} }
})(); })();

View file

@ -1,5 +1,32 @@
/* eslint-disable no-console */ /* eslint-disable no-console */
import 'dotenv/config'; import 'dotenv/config';
import chalk from 'chalk';
/**
* Variable descriptions and examples
*/
const variableInfo = {
DATABASE_URL: {
description: 'PostgreSQL database connection string',
example: 'postgresql://username:password@localhost:5432/umami',
required: true,
},
CLOUD_URL: {
description: 'Umami Cloud URL',
example: 'https://cloud.umami.is',
required: false,
},
CLICKHOUSE_URL: {
description: 'ClickHouse database URL (required when CLOUD_URL is set)',
example: 'https://clickhouse.example.com',
required: false,
},
REDIS_URL: {
description: 'Redis connection URL (required when CLOUD_URL is set)',
example: 'redis://localhost:6379',
required: false,
},
};
function checkMissing(vars) { function checkMissing(vars) {
const missing = vars.reduce((arr, key) => { const missing = vars.reduce((arr, key) => {
@ -10,14 +37,37 @@ function checkMissing(vars) {
}, []); }, []);
if (missing.length) { if (missing.length) {
console.log(`The following environment variables are not defined:`); console.log(chalk.red.bold('\n❌ Environment Configuration Error\n'));
console.log(chalk.yellow('The following environment variables are not defined:\n'));
for (const item of missing) { for (const item of missing) {
console.log(' - ', item); const info = variableInfo[item] || {};
console.log(chalk.red(`${item}`));
if (info.description) {
console.log(chalk.gray(` Description: ${info.description}`));
} }
if (info.example) {
console.log(chalk.cyan(` Example: ${info.example}`));
}
console.log('');
}
console.log(chalk.yellow.bold('💡 Solution:\n'));
console.log(" 1. Create a .env file in the project root if it doesn't exist");
console.log(' 2. Copy the template from .env.example:');
console.log(chalk.cyan(' cp .env.example .env'));
console.log(' 3. Add the missing variables to your .env file\n');
console.log(chalk.blue('📖 For more information, see .env.example or SETUP.md\n'));
process.exit(1); process.exit(1);
} }
} }
// Check required variables based on configuration
if (!process.env.SKIP_DB_CHECK && !process.env.DATABASE_TYPE) { if (!process.env.SKIP_DB_CHECK && !process.env.DATABASE_TYPE) {
checkMissing(['DATABASE_URL']); checkMissing(['DATABASE_URL']);
} }
@ -25,3 +75,6 @@ if (!process.env.SKIP_DB_CHECK && !process.env.DATABASE_TYPE) {
if (process.env.CLOUD_URL) { if (process.env.CLOUD_URL) {
checkMissing(['CLOUD_URL', 'CLICKHOUSE_URL', 'REDIS_URL']); checkMissing(['CLOUD_URL', 'CLICKHOUSE_URL', 'REDIS_URL']);
} }
// Success message
console.log(chalk.green('✓ Environment variables validated successfully\n'));