123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
import { parseDsn } from "./utils.ts";
function getPgEnv(): IConnectionParams {  // this is dummy env object, if program  // was run with --allow-env permission then  // it's filled with actual values  let pgEnv: IConnectionParams = {};
  if (Deno.permissions().env) {    const env = Deno.env();
    pgEnv = {      database: env.PGDATABASE,      host: env.PGHOST,      port: env.PGPORT,      user: env.PGUSER,      password: env.PGPASSWORD,      application_name: env.PGAPPNAME    };  }
  return pgEnv;}
function selectFrom(sources: Object[], key: string): string | undefined {  for (const source of sources) {    if (source[key]) {      return source[key];    }  }
  return undefined;}
const DEFAULT_CONNECTION_PARAMS = {  host: "127.0.0.1",  port: "5432",  application_name: "deno_postgres"};
export interface IConnectionParams {  database?: string;  host?: string;  port?: string;  user?: string;  password?: string;  application_name?: string;}
class ConnectionParamsError extends Error {  constructor(message: string) {    super(message);    this.name = "ConnectionParamsError";  }}
export class ConnectionParams {  database: string;  host: string;  port: string;  user: string;  password?: string;  application_name: string;  // TODO: support other params
  constructor(config?: string | IConnectionParams) {    if (!config) {      config = {};    }
    const pgEnv = getPgEnv();
    if (typeof config === "string") {      const dsn = parseDsn(config);      if (dsn.driver !== "postgres") {        throw new Error(`Supplied DSN has invalid driver: ${dsn.driver}.`);      }      config = dsn;    }
    this.database = selectFrom([config, pgEnv], "database");    this.host = selectFrom([config, pgEnv, DEFAULT_CONNECTION_PARAMS], "host");    this.port = selectFrom([config, pgEnv, DEFAULT_CONNECTION_PARAMS], "port");    this.user = selectFrom([config, pgEnv], "user");    this.password = selectFrom([config, pgEnv], "password");    this.application_name = selectFrom(      [config, pgEnv, DEFAULT_CONNECTION_PARAMS],      "application_name"    );
    const missingParams: string[] = [];
    ["database", "user"].forEach(param => {      if (!this[param]) {        missingParams.push(param);      }    });
    if (missingParams.length) {      throw new ConnectionParamsError(        `Missing connection parameters: ${missingParams.join(          ", "        )}. Connection parameters can be read         from environment only if Deno is run with env permission (deno run --allow-env)`      );    }  }}