All files / modules/72-templates-library/components/AbstractTemplate TemplateFactory.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
 
 
 
 
 
 
 
 
 
11x
 
 
 
 
 
 
11x
 
 
 
44x
 
 
 
 
 
 
 
 
 
 
22x
 
 
 
21x
 
 
 
39x
 
 
 
/*
 * Copyright 2021 Harness Inc. All rights reserved.
 * Use of this source code is governed by the PolyForm Shield 1.0.0 license
 * that can be found in the licenses directory at the root of this repository, also available at
 * https://polyformproject.org/wp-content/uploads/2020/06/PolyForm-Shield-1.0.0.txt.
 */
 
import type { Color } from '@harness/design-system'
import type { Template } from '@templates-library/components/AbstractTemplate/Template'
 
export class TemplateFactory {
  /**
   * Couples the factory with the templates it generates
   */
  protected templateBank: Map<string, Template<unknown>>
 
  constructor() {
    this.templateBank = new Map()
  }
 
  registerTemplate<T>(template: Template<T>): void {
    this.templateBank.set(template.getType(), template as Template<unknown>)
  }
 
  deregisterStep(type: string): void {
    const deletedTemplate = this.templateBank.get(type)
    if (deletedTemplate) {
      this.templateBank.delete(type)
    }
  }
 
  getTemplate<T>(type: string): Template<T> | undefined {
    return this.templateBank.get(type) as Template<T>
  }
 
  getTemplateName(type: string): string | undefined {
    return this.templateBank.get(type)?.getName()
  }
 
  getTemplateColor(type: string): Color | undefined {
    return this.templateBank.get(type)?.getColor()
  }
}