All files / modules/70-pipeline/components/Diagram/node/GroupNode GroupNodeModel.ts

30.77% Statements 4/13
0% Branches 0/3
0% Functions 0/3
30.77% Lines 4/13

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 57 58 59 60 61 62 63 64 65 66 67 68 69 70                    207x 207x   207x                         207x                                                                                      
/*
 * 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 { DeserializeEvent } from '@projectstorm/react-canvas-core'
import type { IconName } from '@wings-software/uicore'
import type { IconProps } from '@wings-software/uicore/dist/icons/Icon'
import { DiagramType, PortName } from '../../Constants'
import { DefaultNodeModelOptions, DefaultNodeModel } from '../DefaultNodeModel'
import type { DefaultLinkModelGenerics } from '../../link/DefaultLinkModel'
import { DefaultPortModel } from '../../port/DefaultPortModel'
 
export interface GroupNodeModelOptions
  extends Omit<DefaultNodeModelOptions, 'icon | iconProps | secondaryIconStyle | secondaryIconProps | secondaryIcon '> {
  icons: Array<IconName>
  parallelNodes: Array<string>
  iconPropsAr?: Array<IconProps>
}
 
export interface GroupNodeModelGenerics extends DefaultLinkModelGenerics {
  OPTIONS: GroupNodeModelOptions
}
 
export class GroupNodeModel extends DefaultNodeModel<GroupNodeModelGenerics> {
  constructor(name: string)
  constructor(options?: GroupNodeModelOptions)
  constructor(options: any = {}) {
    if (typeof options === 'string') {
      options = {
        name: options
      }
    }
    super({
      type: DiagramType.GroupNode,
      icons: [],
      iconPropsAr: [],
      ...options
    })
    this.addPort(
      new DefaultPortModel({
        in: true,
        name: PortName.In
      })
    )
    this.addPort(
      new DefaultPortModel({
        in: false,
        name: PortName.Out
      })
    )
  }
 
  deserialize(event: DeserializeEvent<this>): void {
    super.deserialize(event)
    this.options.icons = [...event.data.icons]
    this.options.iconPropsAr = [...event.data.iconPropsAr]
  }
 
  serialize(): any {
    return {
      ...super.serialize(),
      icons: this.options.icons,
      iconPropsAr: this.options.iconPropsAr
    }
  }
}