roles.ts

import type { MigrationOptions } from '../types';
import { escapeValue } from '../utils';
import type {
  AlterRole,
  CreateRole,
  DropRole,
  RenameRole,
  RoleOptions,
} from './rolesTypes';

export type { CreateRole, DropRole, AlterRole, RenameRole };

const formatRoleOptions = (roleOptions: RoleOptions = {}) => {
  const options: string[] = [];

  if (roleOptions.superuser !== undefined) {
    options.push(roleOptions.superuser ? 'SUPERUSER' : 'NOSUPERUSER');
  }

  if (roleOptions.createdb !== undefined) {
    options.push(roleOptions.createdb ? 'CREATEDB' : 'NOCREATEDB');
  }

  if (roleOptions.createrole !== undefined) {
    options.push(roleOptions.createrole ? 'CREATEROLE' : 'NOCREATEROLE');
  }

  if (roleOptions.inherit !== undefined) {
    options.push(roleOptions.inherit ? 'INHERIT' : 'NOINHERIT');
  }

  if (roleOptions.login !== undefined) {
    options.push(roleOptions.login ? 'LOGIN' : 'NOLOGIN');
  }

  if (roleOptions.replication !== undefined) {
    options.push(roleOptions.replication ? 'REPLICATION' : 'NOREPLICATION');
  }

  if (roleOptions.bypassrls !== undefined) {
    options.push(roleOptions.bypassrls ? 'BYPASSRLS' : 'NOBYPASSRLS');
  }

  if (roleOptions.limit) {
    options.push(`CONNECTION LIMIT ${Number(roleOptions.limit)}`);
  }

  if (roleOptions.password !== undefined) {
    const encrypted =
      roleOptions.encrypted === false ? 'UNENCRYPTED' : 'ENCRYPTED';
    options.push(`${encrypted} PASSWORD ${escapeValue(roleOptions.password)}`);
  }

  if (roleOptions.valid !== undefined) {
    const valid = roleOptions.valid
      ? escapeValue(roleOptions.valid)
      : "'infinity'";
    options.push(`VALID UNTIL ${valid}`);
  }

  if (roleOptions.inRole) {
    const inRole = Array.isArray(roleOptions.inRole)
      ? roleOptions.inRole.join(',')
      : roleOptions.inRole;
    options.push(`IN ROLE ${inRole}`);
  }

  if (roleOptions.role) {
    const role = Array.isArray(roleOptions.role)
      ? roleOptions.role.join(',')
      : roleOptions.role;
    options.push(`ROLE ${role}`);
  }

  if (roleOptions.admin) {
    const admin = Array.isArray(roleOptions.admin)
      ? roleOptions.admin.join(',')
      : roleOptions.admin;
    options.push(`ADMIN ${admin}`);
  }

  return options.join(' ');
};

export function dropRole(mOptions: MigrationOptions): DropRole {
  const _drop: DropRole = (roleName, options = {}) => {
    const { ifExists } = options;

    const ifExistsStr = ifExists ? ' IF EXISTS' : '';
    const roleNameStr = mOptions.literal(roleName);

    return `DROP ROLE${ifExistsStr} ${roleNameStr};`;
  };

  return _drop;
}

export function createRole(mOptions: MigrationOptions): CreateRole {
  const _create: CreateRole = (roleName, roleOptions = {}) => {
    const options = formatRoleOptions({
      ...roleOptions,
      superuser: roleOptions.superuser || false,
      createdb: roleOptions.createdb || false,
      createrole: roleOptions.createrole || false,
      inherit: roleOptions.inherit !== false,
      login: roleOptions.login || false,
      replication: roleOptions.replication || false,
    });
    const optionsStr = options ? ` WITH ${options}` : '';

    return `CREATE ROLE ${mOptions.literal(roleName)}${optionsStr};`;
  };

  _create.reverse = dropRole(mOptions);

  return _create;
}

export function alterRole(mOptions: MigrationOptions): AlterRole {
  const _alter: AlterRole = (roleName, roleOptions = {}) => {
    const options = formatRoleOptions(roleOptions);

    return options
      ? `ALTER ROLE ${mOptions.literal(roleName)} WITH ${options};`
      : '';
  };

  return _alter;
}

export function renameRole(mOptions: MigrationOptions): RenameRole {
  const _rename: RenameRole = (oldRoleName, newRoleName) => {
    const oldRoleNameStr = mOptions.literal(oldRoleName);
    const newRoleNameStr = mOptions.literal(newRoleName);

    return `ALTER ROLE ${oldRoleNameStr} RENAME TO ${newRoleNameStr};`;
  };

  _rename.reverse = (oldRoleName, newRoleName) =>
    _rename(newRoleName, oldRoleName);

  return _rename;
}