ToolJet/server/scripts/drop-database.ts

import * as path from 'path';
import * as dotenv from 'dotenv';
import * as fs from 'fs';
import { exec } from 'child_process';
import * as Joi from 'joi';

function validateDatabaseConfig(): Joi.ValidationResult {
  const envVarsSchema = Joi.object()
    .keys({
      PG_HOST: Joi.string().required(),
      PG_PORT: Joi.number().positive().default(5432),
      PG_PASS: Joi.string().default(''),
      PG_USER: Joi.string().required(),
      PG_DB: Joi.string().required(),
    })
    .unknown();

  return envVarsSchema.validate(process.env);
}

function dropDatabaseFromFile(envPath: string): void {
  const result = dotenv.config({ path: envPath });

  if (result.error) {
    throw result.error;
  }

  dropDatabase();
}

function dropDatabase(): void {
  const { value: envVars, error } = validateDatabaseConfig();

  if (error) {
    throw new Error(`Config validation error: ${error.message}`);
  }

  exec('command -v dropdb', (err, _stdout, _stderr) => {
    if (err) {
      console.error(err);
      return;
    }

    const dropdb =
      `PGPASSWORD=${envVars.PG_PASS} dropdb ` +
      `-h ${envVars.PG_HOST} ` +
      `-p ${envVars.PG_PORT} ` +
      `-U ${envVars.PG_USER} ` +
      process.env.PG_DB;

    exec(dropdb, (err, _stdout, _stderr) => {
      if (err) {
        console.error(err);
        return;
      }

      console.log(`Dropped database ${envVars.PG_DB}`);
    });
  });
}

const nodeEnvPath = path.resolve(
  process.cwd(),
  process.env.NODE_ENV === 'test' ? '../.env.test' : '../.env',
);

const fallbackPath = path.resolve(process.cwd(), '../.env');

if (fs.existsSync(nodeEnvPath)) {
  dropDatabaseFromFile(nodeEnvPath);
} else if (fs.existsSync(fallbackPath)) {
  dropDatabaseFromFile(fallbackPath);
} else {
  console.log(
    `${nodeEnvPath} file not found to drop database\n` +
    'Inferring config from the environment',
  );
  dropDatabase();
}