Flow/browser/vue-flow/src/components/flow/node.vue

<template>
  <g
    class="flow-node"
    :class="{
      'flow-node--dragging':dragging,
      'flow-node--selected': selected
    }"
    :status="status"
    @mousedown.stop.prevent="$emit('nodePointerDown',$event)"
    @contextmenu.capture.prevent="$emit('nodeRightClick', $event)"
    @dblclick="$emit('nodeDoubleClick',$event)"
  >

    <!-- shape -->
    <template>
      <svg
        v-if="style.shape == 'thing'"
        ref="body"
        viewBox="0 0 100 100"
        preserveAspectRatio="XMinYMin"
        class="flow-node__body"
        :class="{'flow-node__body--dragging':dragging}"
        v-bind="bodyProps"
      >
        <path d=" M 0 0 l 90 0 l 10 50 l -10 50 l -90 0 c 10 -20, 10 -80, 0 -100 Z " />
      </svg>
      <circle
        v-else-if="style.shape == 'circle'"
        ref="body"
        class="flow-node__body"
        :class="{'flow-node__body--dragging':dragging}"
        v-bind="bodyProps"
      />
      <rect
        v-else
        ref="body"
        class="flow-node__body"
        :class="{'flow-node__body--dragging':dragging}"
        v-bind="bodyProps"
      />
    </template>

    <!-- selection square -->
    <rect
      class="flow-node__selection"
      stroke-dasharray="7,3"
      :x="bodyProps.x-4"
      :y="bodyProps.y-4"
      :width="bodyProps.width+8"
      :height="bodyProps.height+8"
    />

    <!-- label -->
    <text
      ref="label"
      class="flow-node__label"
      v-bind="labelProps"
      text-anchor="middle">
      <tspan
        :key="'label-wrap' + i"
        x="0"
        dy="1em"
        v-for="(s,i) in labelWrap" >
        {{ s }}
      </tspan>
    </text>

    <!-- Sockets -->
    <!-- input -->
    <g
      v-for="(inp,i) in inputs"
      :key="'in'+i"
      :data-nodeid="id"
      :data-in="i"
      v-bind="inputProps(i)"
      @mousedown.stop.prevent="socketPointerDown($event, {in:i})"
      class="flow-node__socket flow-node__socket--inputs"
      :class="{'flow-node__socket--match': match.type == 'socket-in' && (inp.type == match.dtype || match.dtype == 'interface {}' || inp.type=='interface {}')}"
    >
      <circle r="5" />
      <!--<rect :x="-16-inputLabel(i).length*7 " :y="-10" :width="inputLabel(i).length*7" :height="20" fill="red" stroke-width="0"/>-->
      <rect
        v-bind="inputLabelBGProps(i)"
        class="flow-node__socket-detail--background"
      />
      <text
        ref="inputLabel"
        text-anchor="end"
        :x="-14"
        :y="4"
        class="flow-node__socket-detail"
      >{{ inputLabel(i) }}</text>

    </g>
    <!-- output -->
    <g
      v-if="output"
      v-bind="outputProps(0)"
      :data-nodeid="id"
      :data-out="0"
      :key="'out'+0"
      @mousedown.stop.prevent="socketPointerDown($event, {out:0})"
      class="flow-node__socket flow-node__socket--outputs"
      :class="{ 'flow-node__socket--match': match.type =='socket-out' && (output.type == match.dtype || match.dtype == 'interface {}' || output.type == 'interface {}'), }"
    >
      <circle r="5" />

      <rect
        v-bind="outputLabelBGProps(0)"
        class="flow-node__socket-detail--background"
      />
      <text
        ref="outputLabel"
        class="flow-node__socket-detail"
        :x="14"
        :y="4">
        {{ outputLabel(0) }}
      </text>
    </g>

    <!-- TRIGGER SOCKETS -->
    <rect
      class="flow-node__trigger flow-node__socket--trigger"
      :class="{'flow-node__trigger--match': match.type == 'trigger-out'}"
      :data-nodeid="id"
      data-dir="in"
      :x="-5"
      :y="-bodyProps.height/2-5"
      width="10"
      height="10"
      @mousedown.stop.prevent="triggerPointerDown($event, 'in')"
    />
    <rect
      class="flow-node__trigger flow-node__socket--trigger"
      :class="{'flow-node__trigger--match': match.type == 'trigger-in'}"
      :data-nodeid="id"
      data-dir="out"
      :x="-5"
      :y="bodyProps.height/2 -5"
      width="10"
      height="10"
      @mousedown.stop.prevent="triggerPointerDown($event, 'out')"
    />

    <flow-node-activity
      v-if="activity"
      :node-id="id"
      :transform="'translate('+bodyProps.width/2 +','+ -bodyProps.height/2 +')'"/>
  </g>
</template>

<script>
import FlowNodeActivity from './node-activity'

import utils from '@/utils/utils'
const shapeOpts = {
  'circle': {
    textWrap: 'any'
  },
  default: {
    textWrap: 'white-space'
  }
}
export default {
  name: 'FlowNode',
  components: {FlowNodeActivity},
  props: {
    'id': {type: String, required: true},
    'label': {type: String, default: ''},
    'inputs': {type: Array, default: () => []},
    'output': {type: Object, default: ''},
    'match': {type: Object, default: () => {}},
    'dragging': {type: Boolean, default: false},
    'selected': {type: Boolean, default: false},
    'activity': {type: Object, default: () => {}},
    'nodeStyle': {type: Object, default: () => {}}
  },
  data () {
    return {
      // maintain reference here?
      labelRect: {x: 0, y: 0, width: 0, height: 0}
      // bodyRect: {x: 0, y: 0, width: 0, height: 0}
    }
  },
  computed: {
    style () {
      return this.nodeStyle || {}
    },
    status () {
      return this.activity && (this.activity.status || '')
    },
    labelWrap () {
      let wrapThreshold = 8 // initial wrap threshold
      const opt = shapeOpts[this.style.shape] || shapeOpts.default
      return utils.textWrap(this.label, wrapThreshold, opt.textWrap)
    },
    labelProps () {
      return {
        x: 0,
        y: 0,
        fill: this.textColor,
        // transform: `translate(${-this.labelRect.width / 2},${-this.labelRect.height / 2})`
        transform: `translate(0,${-this.labelRect.height / 2})`
      }
    },
    bodyProps () {
      let width = this.labelRect.width + 46
      let height = Math.max(this.labelRect.height + 20, 60, this.inputs.length * 25)
      if (this.style.shape === 'circle') {
        width = height = Math.max(width, height)
      }

      const rect = {
        x: -width / 2,
        y: -height / 2,
        width: width,
        height: height,
        stroke: this.style.color || '#777',
        fill: this.style.color || '#777'
      }
      if (this.style.shape === 'circle') {
        rect.r = Math.max(width / 2, height / 2)
      }
      return rect
    },
    inputProps () {
      return (i) => {
        const {x, y} = this.inputPos(i)
        return {
          transform: `translate(${x} ${y})`
        }
      }
    },
    outputProps () {
      return (i) => {
        const {x, y} = this.outputPos(i)
        return {
          transform: `translate(${x} ${y})`,
          r: 5
        }
      }
    },
    inputLabel () {
      return (i) => {
        let input = ''
        if (this.inputs[i].name) {
          input += this.inputs[i].name + ':'
        }
        input += this.inputs[i].type
        return input
      }
    },
    inputLabelBGProps () {
      return (i) => {
        if (!this.$refs.inputLabel) {
          this.$nextTick(this.$forceUpdate)
          return {x: 0, y: 0, width: 0, height: 0}
        }

        let bbox = this.$refs.inputLabel[i].getBBox()

        return {x: bbox.x - 5, y: bbox.y - 2, width: bbox.width + 10, height: bbox.height + 4}
      }
    },
    outputLabel () {
      return (i) => {
        var output = ''
        if (this.output.name) {
          output += this.output.name + ':'
        }
        output += this.output.type

        return output
      }
    },
    outputLabelBGProps () {
      return (i) => {
        if (!this.$refs.outputLabel) {
          this.$nextTick(this.$forceUpdate)
          return {x: 0, y: 0, width: 0, height: 0}
        }

        let bbox = this.$refs.outputLabel.getBBox()

        return {x: bbox.x - 5, y: bbox.y - 2, width: bbox.width + 10, height: bbox.height + 4}
      }
    }

  },
  watch: {
    // the only thing now that affects geometry
    'label' () {
      this.$nextTick(() => {
        this.labelRect = this.$refs.label.getBBox()
      })
    }
  },
  mounted () {
    // After render
    this.$nextTick(() => { // after mount we reupdate with new values
      if (!this.$refs.label) return
      this.labelRect = this.$refs.label.getBBox()
      this.$forceUpdate()
    })
  },
  methods: {
    inputPos (i) {
      const ilen = this.inputs.length
      if (ilen === 0) return {}
      const d = this.bodyProps.height / (ilen * 2)
      return {
        x: this.bodyProps.x + 7,
        y: this.bodyProps.y + d + (i * 2 * d)
      }
    },

    outputPos (i) {
      const rect = this.bodyProps
      return {
        x: rect.x + rect.width - 7,
        y: 0
      }
    },

    socketPointerDown (ev, socket) {
      this.$emit('socketPointerDown', ev, socket)
    },
    triggerPointerDown (ev, dir) {
      this.$emit('triggerPointerDown', ev, dir)
    }

  }
}
</script>
<style>

.flow-view:not(.activity) .flow-node:hover,
.flow-node--dragging {
  cursor:move;
}

.flow-node__body {
  opacity:0.9;
  transition: all var(--transition-speed);
}

.flow-node[status=running] .flow-node__body{
  stroke: yellow !important;
}

.flow-node[status=error] .flow-node__body{
  stroke: red !important;
}

.flow-node[status=finished] .flow-node__body{
  stroke: green !important;
}

/* sockets */
.flow-node__socket {
  pointer-events: none;
  stroke-width:1;
  opacity:0;
  transition: all var(--transition-speed);
}

.flow-view:not(.activity) .flow-node__socket:hover {
  stroke-width:10;
  cursor:pointer;
}

.flow-node__socket-detail {
  opacity:1;
  stroke-width:0 !important;
  fill: #fff !default;
  transition: all var(--transition-speed);
}

.flow-node__socket-detail--background {
  stroke-width:0;
  transition:all var(--transition-speed);
}

.flow-node__socket--match {
  cursor:pointer;
  stroke-width:10;
}

.flow-linking .flow-node__socket {
  opacity:1;
  pointer-events: inherit;
}

.flow-node__socket--match {
  opacity:1;
  pointer-events: inherit;
}

/* triggers */
.flow-node__trigger {
  pointer-events:none;
  opacity:0;
  transition: all var(--transition-speed);
}

.flow-triggers .flow-node__trigger {
  pointer-events:inherit;
  opacity:1;
}

.flow-view:not(.activity) .flow-node__trigger:hover {
  stroke-width:10;
  cursor:pointer;
}

.flow-node__trigger--match {
  stroke-width:10;
  opacity:1;
  pointer-events: inherit;
}

/*
Override flow-node
for hidden
 */

.flow-node__label {
  stroke:none;
  pointer-events:none;
  user-select:none;
  fill:#333 !default;
}

.flow-node .flow-node__selection {
  opacity:0;
  stroke-width:2;
  stroke: var(--primary-lighter);
  pointer-events:none;
  transition: all var(--transition-speed);
}

.flow-node--selected .flow-node__selection {
  opacity:0.9;
}

</style>