config.js

/*
 * Copyright (c) 2021 RethinkDNS and its authors.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/**
 * Configuration file for node runtime
 * TODO: Remove all side-effects and use a constructor?
 * This module has side effects, sequentially setting up the environment.
 */
import { atob, btoa } from "buffer";
import { fetch, Headers, Request, Response } from "undici";
import { getTLSfromEnv } from "./util.js";
import * as blocklists from "./blocklists.js";
import Log from "../log.js";
import * as system from "../../system.js";
import { services } from "../svc.js";
import EnvManager from "../env.js";
import * as swap from "../linux/swap.js";

(async (main) => {
  system.when("prepare").then(prep);
  system.when("steady").then(up);
})();

async function prep() {
  // if this file execs... assume we're on nodejs.
  const isProd = process.env.NODE_ENV === "production";
  const onFly = process.env.CLOUD_PLATFORM === "fly";
  const profiling = process.env.PROFILE_DNS_RESOLVES === "true";

  let devutils = null;
  let dotenv = null;

  // dev utilities
  if (!isProd) {
    devutils = await import("./util-dev.js");
    dotenv = await import("dotenv");
  }

  /** Environment Variables */
  // Load env variables from .env file to process.env (if file exists)
  // NOTE: this won't overwrite existing
  if (dotenv) {
    dotenv.config();
    console.log("loading local .env");
  }

  globalThis.envManager = new EnvManager();

  /** Logger */
  globalThis.log = new Log({
    level: envManager.get("LOG_LEVEL"),
    levelize: isProd || profiling, // levelize if prod or profiling
    withTimestamps: true, // always log timestamps on node
  });

  /** TLS crt and key */
  // Raw TLS CERT and KEY are stored (base64) in an env var for fly deploys
  // (fly deploys are dev/prod nodejs deploys where env TLS_CN or TLS_ is set).
  // Otherwise, retrieve KEY and CERT from the filesystem (this is the case
  // for local non-prod nodejs deploys with self-signed certs).
  const _TLS_CRT_AND_KEY =
    eval(`process.env.TLS_${process.env.TLS_CN}`) || process.env.TLS_;

  if (isProd || _TLS_CRT_AND_KEY) {
    const [tlsKey, tlsCrt] = getTLSfromEnv(_TLS_CRT_AND_KEY);
    envManager.set("TLS_KEY", tlsKey);
    envManager.set("TLS_CRT", tlsCrt);
    log.i("env (fly) tls setup");
  } else {
    const [tlsKey, tlsCrt] = devutils.getTLSfromFile(
      envManager.get("TLS_KEY_PATH"),
      envManager.get("TLS_CRT_PATH")
    );
    envManager.set("TLS_KEY", tlsKey);
    envManager.set("TLS_CRT", tlsCrt);
    log.i("dev (local) tls setup");
  }

  /** Polyfills */
  if (!globalThis.fetch) {
    globalThis.fetch = isProd ? fetch : devutils.fetchPlus;
    globalThis.Headers = Headers;
    globalThis.Request = Request;
    globalThis.Response = Response;
    log.i("polyfill fetch web api");
  }

  if (!globalThis.atob || !globalThis.btoa) {
    globalThis.atob = atob;
    globalThis.btoa = btoa;
    log.i("polyfill atob / btoa");
  }

  /** Swap on Fly */
  if (onFly) {
    const ok = swap.mkswap();
    log.i("mkswap done?", ok);
  }

  /** signal ready */
  system.pub("ready");
}

async function up() {
  if (!services.ready) {
    log.e("services not yet ready yet and there is a sig-up!?");
    return;
  }

  const bw = services.blocklistWrapper;
  if (bw != null && !bw.disabled()) {
    await blocklists.setup(bw);
  } else {
    log.w("Config", "blocklists unavailable / disabled");
  }

  // signal all system are-a go
  system.pub("go");
}