@atomist/sdm-pack-aspect/src/lib/analysis/offline/persist/pgClientFactory.ts

an Atomist SDM Extension Pack for visualizing drift across an organization

/*
 * Copyright © 2019 Atomist, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { Configuration } from "@atomist/automation-client";
import * as _ from "lodash";
import { Pool } from "pg";
import { ClientFactory } from "./pgUtils";

const PoolHolder: { pool: Pool } = { pool: undefined };

export function sdmConfigClientFactory(config: Configuration): ClientFactory {
    const usedConfig = {
        database: "org_viz",
        ...(_.get(config, "sdm.postgres") || {}),
    };
    if (!PoolHolder.pool) {
        PoolHolder.pool = new Pool(usedConfig);
    }
    return () => {
        return PoolHolder.pool.connect().catch(e => {
            throw new Error(`${ConnectionErrorHeading}
            Connection parameters: ${JSON.stringify(usedConfig, hidePassword)}
            Error message: ${e.message}`);
        });
    };
}

export const ConnectionErrorHeading = "Could not connect to Postgres.";

function hidePassword(key: any, value: any): any {
    if (key === "password") {
        return "*****";
    }
    return value;
}