signal-desktop/ts/environment.ts

// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import { z } from 'zod';

import { makeEnumParser } from './util/enum';

// Many places rely on this enum being a string.
export enum Environment {
  Development = 'development',
  Production = 'production',
  Staging = 'staging',
  Test = 'test',
}

export const environmentSchema = z.nativeEnum(Environment);

let environment: undefined | Environment;

export function getEnvironment(): Environment {
  if (environment === undefined) {
    // This should never happen—we should always have initialized the environment by this
    //   point. It'd be nice to log here but the logger depends on the environment and we
    //   can't have circular dependencies.
    return Environment.Production;
  }
  return environment;
}

/**
 * Sets the current environment. Should be called early in a process's life, and can only
 * be called once.
 */
export function setEnvironment(env: Environment): void {
  if (environment !== undefined) {
    throw new Error('Environment has already been set');
  }
  environment = env;
}

export const parseEnvironment = makeEnumParser(
  Environment,
  Environment.Production
);

export const isTestEnvironment = (env: Environment): boolean =>
  env === Environment.Test;