Files
vue-flow/docs/examples/custom-node/App.vue
2024-03-18 10:21:07 +01:00

83 lines
1.7 KiB
Vue

<script setup>
import { ref, toRef } from 'vue'
import { MiniMap } from '@vue-flow/minimap'
import { Position, VueFlow } from '@vue-flow/core'
import ColorSelectorNode from './ColorSelectorNode.vue'
import OutputNode from './OutputNode.vue'
import { presets } from './presets.js'
const nodes = ref([
{
id: '1',
type: 'color-selector',
data: { color: presets.ayame },
position: { x: 0, y: 50 },
},
{
id: '2',
type: 'output',
position: { x: 350, y: 114 },
targetPosition: Position.Left,
},
])
const colorSelectorData = toRef(() => nodes.value[0].data)
const edges = ref([
{
id: 'e1a-2',
source: '1',
sourceHandle: 'a',
target: '2',
animated: true,
style: () => ({
stroke: colorSelectorData.value?.color,
}),
},
])
// minimap stroke color functions
function nodeStroke(n) {
switch (n.type) {
case 'input':
return '#0041d0'
case 'color-selector':
return presets.sumi
case 'output':
return '#ff0072'
default:
return '#eee'
}
}
function nodeColor(n) {
if (n.type === 'color-selector') {
return colorSelectorData.value?.color
}
return '#fff'
}
</script>
<template>
<VueFlow
v-model:nodes="nodes"
:edges="edges"
class="custom-node-flow"
:class="[colorSelectorData?.isGradient ? 'animated-bg-gradient' : '']"
:style="{ backgroundColor: colorSelectorData?.color }"
:default-viewport="{ zoom: 1.5 }"
fit-view-on-init
>
<template #node-color-selector="{ id }">
<ColorSelectorNode :id="id" />
</template>
<template #node-output>
<OutputNode />
</template>
<MiniMap :node-stroke-color="nodeStroke" :node-color="nodeColor" />
</VueFlow>
</template>