All files / modules/70-pipeline/components/Diagram/node/IconNode IconNodeModel.ts

83.33% Statements 10/12
33.33% Branches 1/3
66.67% Functions 2/3
83.33% Lines 10/12

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 46 47 48 49 50 51 52 53 54 55 56                207x 207x 207x               207x       3x 3x         3x 3x           3x                 4x                      
/*
 * 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 { IconName } from '@wings-software/uicore'
import { DefaultNodeModel, DefaultNodeModelOptions } from '../DefaultNodeModel'
import { DiagramType, PortName } from '../../Constants'
import { DefaultPortModel } from '../../port/DefaultPortModel'
 
export interface IconNodeModelOptions extends Omit<DefaultNodeModelOptions, 'name'> {
  name?: string
  icon?: IconName
  iconSize?: number
}
 
export class IconNodeModel extends DefaultNodeModel {
  name: string
 
  constructor(options: IconNodeModelOptions = {}) {
    const name = options.name || ''
    super({
      ...options,
      type: DiagramType.IconNode,
      name
    })
    this.name = name
    this.addPort(
      new DefaultPortModel({
        in: true,
        name: PortName.In
      })
    )
    this.addPort(
      new DefaultPortModel({
        in: false,
        name: PortName.Out
      })
    )
  }
 
  serialize(): any {
    return {
      ...super.serialize(),
      text: this.name
    }
  }
 
  deserialize(event: any): void {
    super.deserialize(event)
    this.name = event.data.text
  }
}