@launchql/core/esm/init/client.js

LaunchQL Package and Migration Tools

import { Logger } from '@launchql/logger';
import { readFileSync } from 'fs';
import { join } from 'path';
import { getPgPool } from 'pg-cache';
const log = new Logger('init');
export class LaunchQLInit {
    pool;
    pgConfig;
    constructor(config) {
        this.pgConfig = config;
        this.pool = getPgPool(this.pgConfig);
    }
    /**
     * Bootstrap standard roles (anonymous, authenticated, administrator)
     */
    async bootstrapRoles() {
        try {
            log.info('Bootstrapping LaunchQL roles...');
            const sqlPath = join(__dirname, 'sql', 'bootstrap-roles.sql');
            const sql = readFileSync(sqlPath, 'utf-8');
            await this.pool.query(sql);
            log.success('Successfully bootstrapped LaunchQL roles');
        }
        catch (error) {
            log.error('Failed to bootstrap roles:', error);
            throw error;
        }
    }
    /**
     * Bootstrap test roles (roles only, no users)
     */
    async bootstrapTestRoles() {
        try {
            log.warn('WARNING: This command creates test roles and should NEVER be run on a production database!');
            log.info('Bootstrapping LaunchQL test roles...');
            const sqlPath = join(__dirname, 'sql', 'bootstrap-test-roles.sql');
            const sql = readFileSync(sqlPath, 'utf-8');
            await this.pool.query(sql);
            log.success('Successfully bootstrapped LaunchQL test roles');
        }
        catch (error) {
            log.error('Failed to bootstrap test roles:', error);
            throw error;
        }
    }
    /**
     * Bootstrap database roles with custom username and password
     */
    async bootstrapDbRoles(username, password) {
        try {
            log.info(`Bootstrapping LaunchQL database roles for user: ${username}...`);
            const sql = `
BEGIN;
DO $do$
BEGIN
    IF NOT EXISTS (
        SELECT
        FROM
            pg_catalog.pg_roles
        WHERE
            rolname = '${username}') THEN
    CREATE ROLE ${username} LOGIN PASSWORD '${password}';
END IF;
END
$do$;

GRANT anonymous TO ${username};
GRANT authenticated TO ${username};
COMMIT;
      `;
            await this.pool.query(sql);
            log.success(`Successfully bootstrapped LaunchQL database roles for user: ${username}`);
        }
        catch (error) {
            log.error(`Failed to bootstrap database roles for user ${username}:`, error);
            throw error;
        }
    }
    /**
     * Remove database roles and revoke grants
     */
    async removeDbRoles(username) {
        try {
            log.info(`Removing LaunchQL database roles for user: ${username}...`);
            const sql = `
BEGIN;
DO $do$
BEGIN
    IF EXISTS (
        SELECT
        FROM
            pg_catalog.pg_roles
        WHERE
            rolname = '${username}') THEN
    REVOKE anonymous FROM ${username};
    REVOKE authenticated FROM ${username};
    DROP ROLE ${username};
END IF;
END
$do$;
COMMIT;
      `;
            await this.pool.query(sql);
            log.success(`Successfully removed LaunchQL database roles for user: ${username}`);
        }
        catch (error) {
            log.error(`Failed to remove database roles for user ${username}:`, error);
            throw error;
        }
    }
    /**
     * Close the database connection
     */
    async close() {
    }
}