All files / modules/70-pipeline/components/Diagram/label DefaultLabelModel.tsx

75% Statements 6/8
66.67% Branches 2/3
75% Functions 3/4
75% Lines 6/8

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 45              207x   207x                   207x   1x               1x                 1x            
/*
 * Copyright 2020 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 { LabelModel, LabelModelGenerics, LabelModelOptions } from '@projectstorm/react-diagrams-core'
import type { DeserializeEvent } from '@projectstorm/react-canvas-core'
import { DiagramType } from '../Constants'
 
export interface DefaultLabelModelOptions extends LabelModelOptions {
  label?: string
}
 
export interface DefaultLabelModelGenerics extends LabelModelGenerics {
  OPTIONS: DefaultLabelModelOptions
}
 
export class DefaultLabelModel extends LabelModel<DefaultLabelModelGenerics> {
  constructor(options: DefaultLabelModelOptions = {}) {
    super({
      offsetY: options.offsetY == null ? -23 : options.offsetY,
      type: DiagramType.Default,
      ...options
    })
  }
 
  setLabel(label: string): void {
    this.options.label = label
  }
 
  deserialize(event: DeserializeEvent<this>): void {
    super.deserialize(event)
    this.options.label = event.data.label
  }
 
  serialize(): any {
    return {
      ...super.serialize(),
      label: this.options.label
    }
  }
}