@hashgraph/solo/src/data/configuration/impl/layered-config.ts

An opinionated CLI tool to deploy and manage private Hedera Networks.

// SPDX-License-Identifier: Apache-2.0

import {type Config} from '../api/config.js';
import {type ClassConstructor} from '../../../business/utils/class-constructor.type.js';
import {type ConfigSource} from '../spi/config-source.js';
import {ReflectAssist} from '../../../business/utils/reflect-assist.js';
import {Comparators} from '../../../business/utils/comparators.js';

import {DuplicateConfigSourceError} from '../api/duplicate-config-source-error.js';
import {IllegalArgumentError} from '../../../business/errors/illegal-argument-error.js';

type ScalarMethod<T> = (key: string) => T;
type ObjectMethod<T> = (cls: ClassConstructor<T>, key?: string) => T;
type ObjectArrayMethod<T> = (cls: ClassConstructor<T>, key?: string) => T[];

export class LayeredConfig implements Config {
  private readonly _sources: ConfigSource[];

  public constructor(
    sources: ConfigSource[],
    public readonly mergeSourceValues: boolean = false,
  ) {
    if (sources) {
      sources.sort(Comparators.configSource);
    }

    this._sources = sources ?? [];
  }

  public get sources(): ConfigSource[] {
    return [...this._sources];
  }

  public addSource(source: ConfigSource): void {
    if (!source) {
      throw new IllegalArgumentError('source cannot be null or undefined');
    }

    if (this._sources.includes(source)) {
      throw new DuplicateConfigSourceError(source);
    }

    if (this._sources.some((s: ConfigSource): boolean => s.name === source.name && s.ordinal === source.ordinal)) {
      throw new DuplicateConfigSourceError(source);
    }

    this._sources.push(source);
    this._sources.sort(Comparators.configSource);
  }

  public asBoolean(key: string): boolean | null {
    return this.primitiveScalar<boolean>(this.asBoolean, key, true);
  }

  public asNumber(key: string): number | null {
    return this.primitiveScalar<number>(this.asNumber, key, 1);
  }

  public asObject<T>(cls: ClassConstructor<T>, key?: string): T {
    return this.objectScalar(this.asObject, cls, key);
  }

  public asObjectArray<T>(cls: ClassConstructor<T>, key?: string): Array<T> {
    return this.objectArray(this.asObjectArray, cls, key);
  }

  public asString(key: string): string | null {
    return this.primitiveScalar<string>(this.asString, key, 'string') as string;
  }

  public asStringArray(key: string): string[] | null {
    return this.primitiveScalar<string[]>(this.asStringArray, key, ['stringArray']);
  }

  public properties(): Map<string, string> {
    const finalMap: Map<string, string> = new Map<string, string>();

    for (const source of this.sources) {
      const sourceProperties: Map<string, string> = source.properties();
      for (const [key, value] of sourceProperties.entries()) {
        finalMap.set(key, value);
      }
    }

    return finalMap;
  }

  public propertyNames(): Set<string> {
    const finalSet: Set<string> = new Set<string>();

    for (const source of this.sources) {
      const sourcePropertyNames: Set<string> = source.propertyNames();
      for (const key of sourcePropertyNames) {
        finalSet.add(key);
      }
    }

    return finalSet;
  }

  public async refresh(): Promise<void> {
    for (const source of this.sources) {
      if (ReflectAssist.isRefreshable(source)) {
        await source.refresh();
      }
    }
  }

  private primitiveScalar<T>(method: ScalarMethod<T>, key: string, exampleInstance: unknown): T {
    let value: T = undefined;
    let scalarType: string = typeof exampleInstance;

    if (Array.isArray(exampleInstance) && exampleInstance && exampleInstance.length > 0) {
      scalarType = typeof exampleInstance[0];
    }

    switch (scalarType) {
      case 'boolean':
      case 'number':
      case 'string': {
        break;
      }
      default: {
        throw new IllegalArgumentError(`Unsupported scalar type: ${scalarType}`);
      }
    }

    for (const source of this.sources) {
      const currentValue: T = source[method.name](key);
      if (currentValue !== null && currentValue !== undefined) {
        value = currentValue;
      }
    }

    return value;
  }

  private objectScalar<T>(method: ObjectMethod<T>, cls: ClassConstructor<T>, key?: string): T {
    let value: T = undefined;

    for (const source of this.sources) {
      const currentValue: T = source[method.name](cls, key);
      if (currentValue !== null && currentValue !== undefined) {
        value = this.mergeSourceValues ? ReflectAssist.merge(value, currentValue) : currentValue;
      }
    }

    return value;
  }

  private objectArray<T>(method: ObjectArrayMethod<T>, cls: ClassConstructor<T>, key?: string): Array<T> {
    let value: Array<T> = undefined;

    for (const source of this.sources) {
      const currentValue: Array<T> = source[method.name](cls, key);
      if (currentValue !== null && currentValue !== undefined) {
        value = currentValue;
      }
    }

    return value;
  }
}