From 624438a0eaf51cb4138bec60dc217182d1d3da52 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Mon, 10 Oct 2022 18:54:22 +0200
Subject: [PATCH] feat(handles): allow setting connectable to `single` for
 single connection handles

---
 packages/core/src/components/Handle/Handle.vue | 13 +++++++++++--
 packages/core/src/types/handle.ts              |  2 +-
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/packages/core/src/components/Handle/Handle.vue b/packages/core/src/components/Handle/Handle.vue
index 8dfa8e4b..bd09ddfd 100644
--- a/packages/core/src/components/Handle/Handle.vue
+++ b/packages/core/src/components/Handle/Handle.vue
@@ -1,4 +1,5 @@
 <script lang="ts" setup>
+import { isString } from '@vueuse/core'
 import { useHandle, useNode, useVueFlow } from '../../composables'
 import type { Position } from '../../types'
 import { ConnectionMode } from '../../types'
@@ -9,12 +10,20 @@ const { type = 'source', position = 'top' as Position, connectable = true, id, i
 
 const { connectionStartHandle, connectionMode, vueFlowRef } = $(useVueFlow())
 
-const { id: nodeId, node, nodeEl } = useNode()
+const { id: nodeId, node, nodeEl, connectedEdges } = useNode()
 
 const handle = ref<HTMLDivElement>()
 
 const handleId = $computed(() => id ?? (connectionMode === ConnectionMode.Strict ? null : `${nodeId}__handle-${position}`))
 
+const isConnectable = computed(() => {
+  if (isString(connectable) && connectable === 'single') {
+    return !connectedEdges.value.some((edge) => edge[type] === nodeId && edge[`${type}Handle`] === handleId)
+  }
+
+  return connectable
+})
+
 const { onMouseDown, onClick } = useHandle()
 
 const onMouseDownHandler = (event: MouseEvent) => {
@@ -34,7 +43,7 @@ const getClasses = computed(() => {
     {
       source: type !== 'target',
       target: type === 'target',
-      connectable,
+      connectable: isConnectable,
       connecting:
         connectionStartHandle &&
         connectionStartHandle.nodeId === nodeId &&
diff --git a/packages/core/src/types/handle.ts b/packages/core/src/types/handle.ts
index 771ac756..325fd7b8 100644
--- a/packages/core/src/types/handle.ts
+++ b/packages/core/src/types/handle.ts
@@ -32,5 +32,5 @@ export interface HandleProps {
   /** A valid connection func {@link ValidConnectionFunc} */
   isValidConnection?: ValidConnectionFunc
   /** Enable/disable connecting to handle */
-  connectable?: boolean
+  connectable?: boolean | 'single'
 }