certd/packages/ui/certd-server/src/config/loader.ts

import path from "path";
import * as _ from "lodash-es";
import yaml from "js-yaml";
import fs from "fs";
import { logger } from "@certd/basic";

function parseEnv(defaultConfig: any) {
  const config = {};
  for (const key in process.env) {
    let keyName = key;
    if (!keyName.startsWith("certd_")) {
      continue;
    }
    keyName = keyName.replace("certd_", "");
    const configKey = keyName.replaceAll("_", ".");
    const oldValue = _.get(defaultConfig, configKey);
    let value: any = process.env[key];
    if (typeof oldValue === "boolean") {
      value = value === "true";
    } else if (Number.isInteger(oldValue)) {
      value = parseInt(value, 10);
    } else if (typeof oldValue === "number") {
      value = parseFloat(value);
    }
    _.set(config, configKey, value);
  }
  return config;
}

export function load(config, env = "") {
  // Get document, or throw exception on error
  logger.info("load config", env);
  const yamlPath = path.join(process.cwd(), `.env.${env}.yaml`);
  if (fs.existsSync(yamlPath)) {
    const doc = yaml.load(fs.readFileSync(yamlPath, "utf8"));
    return _.merge(doc, parseEnv(config));
  }
  return parseEnv(config);
}

export function mergeConfig(config: any, envType: string) {
  _.merge(config, load(config, envType));
  const keys = _.get(config, "auth.jwt.secret");
  if (keys) {
    config.keys = keys;
  }
  return config;
}

export function loadDotEnv() {
  const envStr = fs.readFileSync(".env").toString();
  envStr.split("\n").forEach(line => {
    const [key, value] = line.trim().split("=");
    const oldValue = process.env[key];
    if (!oldValue) {
      process.env[key] = value;
    }
  });
}