From 03fd2b956849565e39e3d3ad32c36e831af1e3e8 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Tue, 14 Mar 2023 13:15:36 +0100
Subject: [PATCH] feat(utils): add getHandleBounds

---
 packages/utils/src/utils.ts | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/packages/utils/src/utils.ts b/packages/utils/src/utils.ts
index b2d6d802..47f568ed 100644
--- a/packages/utils/src/utils.ts
+++ b/packages/utils/src/utils.ts
@@ -7,6 +7,8 @@ import type {
   BaseNode,
   BaseEdge,
   NodeOrigin,
+  HandleElement,
+  Position,
 } from '@reactflow/system';
 import { getConnectedEdgesBase } from './graph';
 
@@ -193,3 +195,35 @@ export const getPositionWithOrigin = ({
     y: y - height * origin[1],
   };
 };
+
+export const getHandleBounds = (
+  selector: string,
+  nodeElement: HTMLDivElement,
+  zoom: number,
+  nodeOrigin: NodeOrigin = [0, 0]
+): HandleElement[] | null => {
+  const handles = nodeElement.querySelectorAll(selector);
+
+  if (!handles || !handles.length) {
+    return null;
+  }
+
+  const handlesArray = Array.from(handles) as HTMLDivElement[];
+  const nodeBounds = nodeElement.getBoundingClientRect();
+  const nodeOffset = {
+    x: nodeBounds.width * nodeOrigin[0],
+    y: nodeBounds.height * nodeOrigin[1],
+  };
+
+  return handlesArray.map((handle): HandleElement => {
+    const handleBounds = handle.getBoundingClientRect();
+
+    return {
+      id: handle.getAttribute('data-handleid'),
+      position: handle.getAttribute('data-handlepos') as unknown as Position,
+      x: (handleBounds.left - nodeBounds.left - nodeOffset.x) / zoom,
+      y: (handleBounds.top - nodeBounds.top - nodeOffset.y) / zoom,
+      ...getDimensions(handle),
+    };
+  });
+};