All files / modules/70-pipeline/components/Diagram/port DefaultPortModel.ts

34.78% Statements 8/23
35.71% Branches 5/14
33.33% Functions 2/6
34.78% Lines 8/23

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 71 72 73 74 75 76 77 78 79 80 81 82 83 84              207x               207x 207x                     207x       328x             328x 328x                             554x                                                          
/*
 * 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 {
  LinkModel,
  PortModel,
  PortModelAlignment,
  PortModelGenerics,
  PortModelOptions
} from '@projectstorm/react-diagrams-core'
import type { AbstractModelFactory, DeserializeEvent } from '@projectstorm/react-canvas-core'
import { DefaultLinkModel } from '../link/DefaultLinkModel'
import { DiagramType } from '../Constants'
 
export interface DefaultPortModelOptions extends PortModelOptions {
  label?: string
  in?: boolean
}
 
export interface DefaultPortModelGenerics extends PortModelGenerics {
  OPTIONS: DefaultPortModelOptions
}
 
export class DefaultPortModel extends PortModel<DefaultPortModelGenerics> {
  constructor(isIn: boolean, name?: string, label?: string)
  constructor(options: DefaultPortModelOptions)
  constructor(options: DefaultPortModelOptions | boolean, name?: string, label?: string) {
    Iif (name) {
      options = {
        in: !!options,
        name: name,
        label: label
      }
    }
    options = options as DefaultPortModelOptions
    super({
      label: options.label || options.name,
      alignment: options.in ? PortModelAlignment.LEFT : PortModelAlignment.RIGHT,
      type: DiagramType.Default,
      ...options
    })
  }
 
  deserialize(event: DeserializeEvent<this>): void {
    super.deserialize(event)
    this.options.in = event.data.in
    this.options.label = event.data.label
  }
 
  serialize(): any {
    return {
      ...super.serialize(),
      in: this.options.in,
      label: this.options.label
    }
  }
 
  link<T extends LinkModel>(port: PortModel, factory?: AbstractModelFactory<T>): T {
    const link = this.createLinkModel(factory)
    link.setSourcePort(this)
    link.setTargetPort(port)
    return link as T
  }
 
  canLinkToPort(port: PortModel): boolean {
    if (port instanceof DefaultPortModel) {
      return this.options.in !== port.getOptions().in
    }
    return true
  }
 
  createLinkModel(factory?: AbstractModelFactory<LinkModel>): LinkModel {
    const link = super.createLinkModel()
    if (!link && factory) {
      return factory.generateModel({})
    }
    return link || new DefaultLinkModel()
  }
}