All files / src/Program/options postgres.ts

65.71% Statements 23/35
25% Branches 2/8
25% Functions 2/8
74.07% Lines 20/27

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140  38x 38x   38x 38x 38x 38x 38x 38x 38x 38x 38x 38x 38x 38x 38x 38x                                                 38x       43x   38x   38x                                                                                                                                                                                  
import { Command } from 'commander';
import { addOptions, newOption } from './util';
import { existingFileValidator, integerValidator } from '../../util/validators';
 
export enum PostgresOptionDescriptions {
  ConnectionString = 'PostgreSQL Connection string',
  SrvServiceName = 'PostgreSQL SRV service name when using service discovery',
  Db = 'PostgreSQL database name',
  DbFile = 'PostgreSQL database name file path',
  User = 'PostgreSQL user',
  UserFile = 'PostgreSQL user file path',
  Password = 'PostgreSQL password',
  PasswordFile = 'PostgreSQL password file path',
  SslCaFile = 'PostgreSQL SSL CA file path',
  Host = 'PostgreSQL host',
  Port = 'PostgreSQL port',
  PoolMax = 'Maximum number of clients in the PostgreSQL pool',
  ServiceDiscoveryArgs = 'Postgres SRV service name, db, user and password'
}
 
export interface BasePosgresProgramOptions {
  postgresConnectionString?: string;
  postgresSrvServiceName?: string;
  postgresDb?: string;
  postgresDbFile?: string;
  postgresUser?: string;
  postgresUserFile?: string;
  postgresPassword?: string;
  postgresPasswordFile?: string;
  postgresHost?: string;
  postgresPoolMax?: number;
  postgresPort?: string;
  postgresSslCaFile?: string;
}
 
export type ConnectionNames = 'Asset' | 'DbSync' | 'Handle' | 'StakePool' | 'WalletApi' | '';
 
export type PosgresProgramOptions<
  Suffix extends ConnectionNames,
  Options extends keyof BasePosgresProgramOptions = keyof BasePosgresProgramOptions
> = { [k in Options as `${k}${Suffix}`]?: BasePosgresProgramOptions[k] };
 
export const getPostgresOption = <Suffix extends ConnectionNames, Options extends keyof BasePosgresProgramOptions>(
  suffix: Suffix,
  option: Options,
  args: PosgresProgramOptions<Suffix, Options> | undefined
) => args?.[`${option}${suffix}`] as BasePosgresProgramOptions[typeof option];
 
export const suffixType2Cli = (suffix: ConnectionNames) => suffix.replace(/[A-Z]/g, (_) => `-${_.toLowerCase()}`);
 
export const withPostgresOptions = (command: Command, suffixes: ConnectionNames[]) => {
  for (const suffix of suffixes) {
    const cliSuffix = suffix ? suffixType2Cli(suffix) : '';
    const dscSuffix = suffix ? suffix.replace(/[A-Z]/g, (_) => ` ${_.toLowerCase()}`) : '';
    const envSuffix = suffix ? suffix.replace(/[A-Z]/g, (_) => `_${_}`).replace(/[a-z]/g, (_) => _.toUpperCase()) : '';
 
    const descSuffix = ` for${dscSuffix}`;
 
    addOptions(command, [
      newOption(
        `--postgres-connection-string${cliSuffix} <postgresConnectionString${suffix}>`,
        PostgresOptionDescriptions.ConnectionString + descSuffix,
        `POSTGRES_CONNECTION_STRING${envSuffix}`
      )
        .conflicts(`postgresSrvServiceName${suffix}`)
        .conflicts(`postgresDb${suffix}`)
        .conflicts(`postgresDbFile${suffix}`)
        .conflicts(`postgresUser${suffix}`)
        .conflicts(`postgresUserFile${suffix}`)
        .conflicts(`postgresPassword${suffix}`)
        .conflicts(`postgresPasswordFile${suffix}`)
        .conflicts(`postgresHost${suffix}`)
        .conflicts(`postgresPort${suffix}`),
      newOption(
        `--postgres-srv-service-name${cliSuffix} <postgresSrvServiceName${suffix}>`,
        PostgresOptionDescriptions.SrvServiceName + descSuffix,
        `POSTGRES_SRV_SERVICE_NAME${envSuffix}`
      )
        .conflicts(`postgresHost${suffix}`)
        .conflicts(`postgresPort${suffix}`),
      newOption(
        `--postgres-db${cliSuffix} <postgresDb${suffix}>`,
        PostgresOptionDescriptions.Db + descSuffix,
        `POSTGRES_DB${envSuffix}`
      ).conflicts(`postgresDbFile${suffix}`),
      newOption(
        `--postgres-db-file${cliSuffix} <postgresDbFile${suffix}>`,
        PostgresOptionDescriptions.DbFile + descSuffix,
        `POSTGRES_DB_FILE${envSuffix}`,
        existingFileValidator
      ),
      newOption(
        `--postgres-user${cliSuffix} <postgresUser${suffix}>`,
        PostgresOptionDescriptions.User + descSuffix,
        `POSTGRES_USER${envSuffix}`
      ).conflicts(`postgresUserFile${suffix}`),
      newOption(
        `--postgres-user-file${cliSuffix} <postgresUserFile${suffix}>`,
        PostgresOptionDescriptions.UserFile + descSuffix,
        `POSTGRES_USER_FILE${envSuffix}`,
        existingFileValidator
      ),
      newOption(
        `--postgres-password${cliSuffix} <postgresPassword${suffix}>`,
        PostgresOptionDescriptions.Password + descSuffix,
        `POSTGRES_PASSWORD${envSuffix}`
      ).conflicts(`postgresPasswordFile${suffix}`),
      newOption(
        `--postgres-password-file${cliSuffix} <postgresPasswordFile${suffix}>`,
        PostgresOptionDescriptions.PasswordFile + descSuffix,
        `POSTGRES_PASSWORD_FILE${envSuffix}`,
        existingFileValidator
      ),
      newOption(
        `--postgres-host${cliSuffix} <postgresHost${suffix}>`,
        PostgresOptionDescriptions.Host + descSuffix,
        `POSTGRES_HOST${envSuffix}`
      ),
      newOption(
        `--postgres-pool-max${cliSuffix} <postgresPoolMax${suffix}>`,
        PostgresOptionDescriptions.PoolMax + descSuffix,
        `POSTGRES_POOL_MAX${envSuffix}`,
        integerValidator(PostgresOptionDescriptions.PoolMax + descSuffix)
      ),
      newOption(
        `--postgres-port${cliSuffix} <postgresPort${suffix}>`,
        PostgresOptionDescriptions.Port + descSuffix,
        `POSTGRES_PORT${envSuffix}`
      ),
      newOption(
        `--postgres-ssl-ca-file${cliSuffix} <postgresSslCaFile${suffix}>`,
        PostgresOptionDescriptions.SslCaFile + descSuffix,
        `POSTGRES_SSL_CA_FILE${envSuffix}`
      )
    ]);
  }
 
  return command;
};