cal-heatmap/src/TemplateCollection.ts

Cal-Heatmap is a javascript module to create calendar heatmap to visualize time series data

import castArray from 'lodash-es/castArray';

import DefaultTemplates from './templates/index';
import type Options from './options/Options';
import type { Template, TemplateResult } from './index';
import type DateHelper from './helpers/DateHelper';

export default class TemplateCollection {
  dateHelper: DateHelper;

  options: Options;

  settings: Map<string, TemplateResult>;

  // Whether the default templates has been initiated
  initiated: boolean;

  constructor(dateHelper: DateHelper, options: Options) {
    this.settings = new Map();
    this.dateHelper = dateHelper;
    this.options = options;
    this.initiated = false;
  }

  get(subDomainType: string): TemplateResult {
    return this.settings.get(subDomainType)!;
  }

  has(subDomainType: string): boolean {
    return this.settings.has(subDomainType);
  }

  init() {
    if (!this.initiated) {
      this.initiated = true;
      this.add(DefaultTemplates);
    }
  }

  add(templates: Template | Template[]) {
    this.init();

    const tplWithParent: string[] = [];
    castArray(templates).forEach((f) => {
      const template = f(this.dateHelper, this.options.options);
      this.settings.set(template.name, template);

      if (template.hasOwnProperty('parent')) {
        tplWithParent.push(template.name);
      }
    });

    tplWithParent.forEach((name) => {
      const parentTemplate = this.settings.get(
        this.settings.get(name)!.parent!,
      );

      if (!parentTemplate) {
        return;
      }

      this.settings.set(name, {
        ...parentTemplate,
        ...this.settings.get(name),
      });
    });
  }
}