@likec4/diagram 1.42.1 → 1.44.0

package/dist/base/BaseXYFlow.js

@@ -1,15 +1,16 @@
 import { jsxs, jsx } from 'react/jsx-runtime';
 import { cx } from '@likec4/styles/css';
 import { useMantineColorScheme } from '@mantine/core';
-import { useCallbackRef } from '@mantine/hooks';
 import { ReactFlow, useStore } from '@xyflow/react';
+import { useMemo } from 'react';
+import { useCallbackRef } from '../hooks/useCallbackRef.js';
 import { useUpdateEffect } from '../hooks/useUpdateEffect.js';
 import { useIsZoomTooSmall, useXYStoreApi } from '../hooks/useXYFlow.js';
 import { roundDpr } from '../utils/roundDpr.js';
 import { stopPropagation } from '../utils/xyflow.js';
 import { Background } from './Background.js';
+import { Base } from './Base.js';
 import { MinZoom, MaxZoom } from './const.js';
-import { Base } from './types.js';
 
 function BaseXYFlow({
   nodes,
@@ -35,6 +36,12 @@ function BaseXYFlow({
   onEdgeMouseLeave,
   ...props
 }) {
+  const fitViewOptions = useMemo(() => ({
+    minZoom: MinZoom,
+    maxZoom: 1,
+    padding: fitViewPadding,
+    includeHiddenNodes: false
+  }), [fitViewPadding]);
   const isBgWithPattern = background !== "transparent" && background !== "solid";
   const isZoomTooSmall = useIsZoomTooSmall();
   const xystore = useXYStoreApi();
@@ -64,12 +71,7 @@ function BaseXYFlow({
       maxZoom: zoomable ? MaxZoom : 1,
       minZoom: zoomable ? MinZoom : 1,
       fitView,
-      fitViewOptions: {
-        minZoom: MinZoom,
-        maxZoom: 1,
-        padding: fitViewPadding,
-        includeHiddenNodes: false
-      },
+      fitViewOptions,
       preventScrolling: zoomable || pannable,
       defaultMarkerColor: "var(--xy-edge-stroke)",
       noDragClassName: "nodrag",
@@ -99,57 +101,49 @@ function BaseXYFlow({
         }
         onMoveEnd?.(event, { x: roundedX, y: roundedY, zoom });
       }),
-      onNodeMouseEnter: useCallbackRef((_event, node) => {
+      onNodeMouseEnter: useCallbackRef((event, node) => {
         if (onNodeMouseEnter) {
-          onNodeMouseEnter(_event, node);
+          onNodeMouseEnter(event, node);
           return;
         }
-        if (!node.data.hovered) {
-          onNodesChange([{
-            id: node.id,
-            type: "replace",
-            item: Base.setHovered(node, true)
-          }]);
-        }
+        onNodesChange([{
+          id: node.id,
+          type: "replace",
+          item: Base.setHovered(node, true)
+        }]);
       }),
-      onNodeMouseLeave: useCallbackRef((_event, node) => {
+      onNodeMouseLeave: useCallbackRef((event, node) => {
         if (onNodeMouseLeave) {
-          onNodeMouseLeave(_event, node);
+          onNodeMouseLeave(event, node);
           return;
         }
-        if (node.data.hovered) {
-          onNodesChange([{
-            id: node.id,
-            type: "replace",
-            item: Base.setHovered(node, false)
-          }]);
-        }
+        onNodesChange([{
+          id: node.id,
+          type: "replace",
+          item: Base.setHovered(node, false)
+        }]);
       }),
-      onEdgeMouseEnter: useCallbackRef((_event, edge) => {
+      onEdgeMouseEnter: useCallbackRef((event, edge) => {
         if (onEdgeMouseEnter) {
-          onEdgeMouseEnter(_event, edge);
+          onEdgeMouseEnter(event, edge);
           return;
         }
-        if (!edge.data.hovered) {
-          onEdgesChange([{
-            id: edge.id,
-            type: "replace",
-            item: Base.setHovered(edge, true)
-          }]);
-        }
+        onEdgesChange([{
+          id: edge.id,
+          type: "replace",
+          item: Base.setHovered(edge, true)
+        }]);
       }),
-      onEdgeMouseLeave: useCallbackRef((_event, edge) => {
+      onEdgeMouseLeave: useCallbackRef((event, edge) => {
         if (onEdgeMouseLeave) {
-          onEdgeMouseLeave(_event, edge);
+          onEdgeMouseLeave(event, edge);
           return;
         }
-        if (edge.data.hovered) {
-          onEdgesChange([{
-            id: edge.id,
-            type: "replace",
-            item: Base.setHovered(edge, false)
-          }]);
-        }
+        onEdgesChange([{
+          id: edge.id,
+          type: "replace",
+          item: Base.setHovered(edge, false)
+        }]);
       }),
       onNodeDoubleClick: stopPropagation,
       onEdgeDoubleClick: stopPropagation,

package/dist/base/const.d.ts

@@ -9,7 +9,7 @@ export declare const MaxZoom = 3;
 export declare const FitViewPaddings: {
     default: "16px";
     withControls: {
-        top: "50px";
+        top: "58px";
         left: "16px";
         right: "16px";
         bottom: "16px";

package/dist/base/const.js

@@ -10,7 +10,7 @@ const MaxZoom = 3;
 const FitViewPaddings = {
   default: "16px",
   withControls: {
-    top: "50px",
+    top: "58px",
     left: "16px",
     right: "16px",
     bottom: "16px"

package/dist/base/index.d.ts

@@ -1,3 +1,7 @@
-export { BaseXYFlow, type BaseXYFlowProps } from './BaseXYFlow';
-export * from './types';
+export { Background, type XYBackground, type XYBackgroundProps, type XYBackgroundVariant, } from './Background';
+export { BaseXYFlow, type BaseXYFlowProps, } from './BaseXYFlow';
+export * from './Base';
 export * from './const';
+export * from './types';
+export { updateEdges } from './updateEdges';
+export { updateNodes } from './updateNodes';

package/dist/base/types.d.ts

@@ -1,5 +1,6 @@
 import { Edge, EdgeProps, Node, NodeProps } from '@xyflow/react';
 import { OptionalKeysOf, SetRequired, Simplify } from 'type-fest';
+import { Base } from './Base';
 export type BaseNodeData = {
     /**
      * Whether the cursor is hovering over the node
@@ -41,33 +42,6 @@ export type BaseNodePropsWithData<Data extends Record<string, unknown>> = BaseNo
 export interface BaseEdgeProps<E extends BaseEdge = BaseEdge> extends EdgeProps<E> {
 }
 export type BaseEdgePropsWithData<Data extends Record<string, unknown>> = BaseEdgeProps<BaseEdge<Data>>;
-type WithDimmed = {
-    data: {
-        dimmed?: Base.Dimmed;
-    };
-};
-type WithHovered = {
-    data: {
-        hovered?: boolean;
-    };
-};
-declare function setDimmed<T extends WithDimmed>(value: T, dimmed: 'immediate' | boolean): T;
-declare function setDimmed(dimmed: 'immediate' | boolean): <T extends WithDimmed>(value: T) => T;
-declare function setHovered<T extends WithHovered>(value: T, hovered: boolean): T;
-declare function setHovered(hovered: boolean): <T extends WithHovered>(value: T) => T;
-type WithData<D> = {
-    data: D;
-};
-declare function setData<E extends WithData<any>>(value: E, state: Partial<E['data']>): E;
-declare function setData<E extends WithData<any>>(state: Partial<E['data']>): (value: E) => E;
-export declare const Base: {
-    setDimmed: typeof setDimmed;
-    setHovered: typeof setHovered;
-    setData: typeof setData;
-};
-export declare namespace Base {
-    type Dimmed = 'immediate' | boolean;
-}
 /**
  * To cooperate with `exactOptionalPropertyTypes` in `tsconfig.json`
  *
@@ -90,4 +64,3 @@ export type NonOptional<T extends object, Keys extends OptionalKeysOf<T> = Optio
 } & {
     [P in Keys]-?: T[P] | undefined;
 }>;
-export {};

package/dist/base/updateEdges.js

@@ -4,14 +4,15 @@ import { isDefined, omit } from 'remeda';
 function _update(current, update) {
   return update.map((next) => {
     const existing = current.find((n) => n.id === next.id);
-    if (existing && deepEqual(existing.type, next.type)) {
-      if (deepEqual(existing.hidden ?? false, next.hidden ?? false) && deepEqual(existing.source, next.source) && deepEqual(existing.sourceHandle ?? null, next.sourceHandle ?? null) && deepEqual(existing.target, next.target) && deepEqual(existing.targetHandle ?? null, next.targetHandle ?? null) && deepEqual(existing.zIndex ?? 0, next.zIndex ?? 0) && deepEqual(existing.data, next.data)) {
+    if (existing && existing.type === next.type && existing.source === next.source && existing.target === next.target) {
+      const isSameData = deepEqual(existing.data, next.data);
+      if (isSameData && deepEqual(existing.hidden ?? false, next.hidden ?? false) && deepEqual(existing.selected ?? false, next.selected ?? false) && deepEqual(existing.animated ?? false, next.animated ?? false) && deepEqual(existing.sourceHandle ?? null, next.sourceHandle ?? null) && deepEqual(existing.targetHandle ?? null, next.targetHandle ?? null) && deepEqual(existing.zIndex ?? 0, next.zIndex ?? 0) && deepEqual(existing.label ?? null, next.label ?? null)) {
         return existing;
       }
       return {
         ...omit(existing, ["hidden", "zIndex"]),
         ...next,
-        data: next.data
+        data: isSameData ? existing.data : next.data
       };
     }
     return next;

package/dist/base/updateNodes.js

@@ -6,8 +6,9 @@ function _update(current, updated) {
   return updated.map((update) => {
     const existing = current.find((n) => n.id === update.id);
     if (existing && deepEqual(existing.type, update.type)) {
+      const isSameData = deepEqual(existing.data, update.data);
       const { width: existingWidth, height: existingHeight } = getNodeDimensions(existing);
-      if (deepEqual(existingWidth, update.initialWidth) && deepEqual(existingHeight, update.initialHeight) && deepEqual(existing.parentId ?? null, update.parentId ?? null) && deepEqual(existing.hidden ?? false, update.hidden ?? false) && deepEqual(existing.zIndex ?? 0, update.zIndex ?? 0) && deepEqual(existing.position, update.position) && deepEqual(existing.data, update.data)) {
+      if (isSameData && deepEqual(existingWidth, update.initialWidth) && deepEqual(existingHeight, update.initialHeight) && deepEqual(existing.parentId ?? null, update.parentId ?? null) && deepEqual(existing.hidden ?? false, update.hidden ?? false) && deepEqual(existing.zIndex ?? 0, update.zIndex ?? 0) && deepEqual(existing.position, update.position)) {
         return existing;
       }
       return {
@@ -16,7 +17,7 @@ function _update(current, updated) {
         // Force dimensions from update
         width: update.initialWidth,
         height: update.initialHeight,
-        data: update.data
+        data: isSameData ? existing.data : update.data
       };
     }
     return update;

package/dist/base-primitives/compound/CompoundActionButton.js

@@ -6,7 +6,7 @@ import { useDebouncedValue } from '@mantine/hooks';
 import * as m from 'motion/react-m';
 import { stopPropagation } from '../../utils/xyflow.js';
 import { compoundActionBtn } from './actionbtns.css.js';
-import IconZoomScan from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.1.1/node_modules/@tabler/icons-react/dist/esm/icons/IconZoomScan.js';
+import IconZoomScan from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.2.0/node_modules/@tabler/icons-react/dist/esm/icons/IconZoomScan.js';
 
 function CompoundActionButton({
   data: {

package/dist/base-primitives/compound/CompoundDetailsButton.js

@@ -6,7 +6,7 @@ import { useDebouncedValue } from '@mantine/hooks';
 import * as m from 'motion/react-m';
 import { stopPropagation } from '../../utils/xyflow.js';
 import { compoundActionBtn } from './actionbtns.css.js';
-import IconId from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.1.1/node_modules/@tabler/icons-react/dist/esm/icons/IconId.js';
+import IconId from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.2.0/node_modules/@tabler/icons-react/dist/esm/icons/IconId.js';
 
 function CompoundDetailsButton({
   data: {

package/dist/base-primitives/edge/EdgeActionButton.js

@@ -3,7 +3,7 @@ import { cx } from '@likec4/styles/css';
 import { edgeActionBtn } from '@likec4/styles/recipes';
 import { ActionIcon } from '@mantine/core';
 import { stopPropagation } from '../../utils/xyflow.js';
-import IconZoomScan from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.1.1/node_modules/@tabler/icons-react/dist/esm/icons/IconZoomScan.js';
+import IconZoomScan from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.2.0/node_modules/@tabler/icons-react/dist/esm/icons/IconZoomScan.js';
 
 function EdgeActionButton({ icon, onClick }) {
   return /* @__PURE__ */ jsx(

package/dist/base-primitives/edge/EdgeContainer.d.ts

@@ -7,5 +7,5 @@ type EdgeContainerProps = PropsWithChildren<BaseEdgePropsWithData<Data> & {
     component?: 'g' | 'svg' | undefined;
     className?: string | undefined;
 }>;
-export declare function EdgeContainer({ className, component, data: { color, hovered: isHovered, active: isActive, dimmed: isDimmed, ...data }, children, style, }: EdgeContainerProps): import("react/jsx-runtime").JSX.Element;
+export declare function EdgeContainer({ className, component, selected, data: { color, hovered: isHovered, active: isActive, dimmed: isDimmed, ...data }, children, style, }: EdgeContainerProps): import("react/jsx-runtime").JSX.Element;
 export {};

package/dist/base-primitives/edge/EdgeContainer.js

@@ -6,6 +6,7 @@ import { edgeContainer } from './edge.css.js';
 function EdgeContainer({
   className,
   component = "g",
+  selected = false,
   data: {
     color = "gray",
     hovered: isHovered = false,
@@ -27,6 +28,9 @@ function EdgeContainer({
     "data-edge-active": isActive,
     "data-edge-animated": isActive,
     "data-likec4-hovered": isHovered,
+    ...selected && {
+      "data-likec4-selected": selected
+    },
     ...isDimmed !== false && {
       "data-likec4-dimmed": isDimmed
     }

package/dist/base-primitives/edge/EdgeLabel.d.ts

@@ -7,6 +7,7 @@ type Data = UndefinedOnPartialDeep<Pick<DiagramEdge, 'label' | 'technology'>>;
 type EdgeLabelProps = HTMLMotionProps<'div'> & {
     children?: React.ReactNode;
     edgeProps: BaseEdgePropsWithData<Data>;
+    pointerEvents?: 'all' | 'none';
 };
 export declare const EdgeLabel: ForwardRefExoticComponent<Omit<EdgeLabelProps, "ref"> & RefAttributes<HTMLDivElement>>;
 export {};

package/dist/base-primitives/edge/EdgeLabel.js

@@ -18,6 +18,7 @@ const EdgeLabel = forwardRef(({
     selected = false,
     selectable = false
   },
+  pointerEvents = "all",
   className,
   style: _style,
   // omit
@@ -26,6 +27,7 @@ const EdgeLabel = forwardRef(({
 }, ref) => {
   const stepNum = isStepEdgeId(id) ? extractStep(id) : null;
   const classes = edgeLabel({
+    pointerEvents,
     isStepEdge: stepNum !== null,
     cursor: selectable || stepNum !== null ? "pointer" : "default"
   });

package/dist/base-primitives/edge/EdgeLabelContainer.d.ts

@@ -14,5 +14,5 @@ type EdgeLabelContainerProps = HTMLAttributes<HTMLDivElement> & {
     } | undefined;
     edgeProps: BaseEdgePropsWithData<Data>;
 };
-export declare function EdgeLabelContainer({ edgeProps: { id, data: { hovered: isHovered, active: isActive, dimmed: isDimmed, labelBBox, color, }, }, labelPosition: labelXY, className, style, children, ...rest }: EdgeLabelContainerProps): import("react/jsx-runtime").JSX.Element | null;
+export declare function EdgeLabelContainer({ edgeProps: { id, selected, data: { hovered: isHovered, active: isActive, dimmed: isDimmed, labelBBox, color, }, }, labelPosition: labelXY, className, style, children, ...rest }: EdgeLabelContainerProps): import("react/jsx-runtime").JSX.Element | null;
 export {};

package/dist/base-primitives/edge/EdgeLabelContainer.js

@@ -13,6 +13,7 @@ const toCssVarValue = (value) => {
 function EdgeLabelContainer({
   edgeProps: {
     id,
+    selected = false,
     data: {
       hovered: isHovered = false,
       active: isActive = false,
@@ -53,6 +54,9 @@ function EdgeLabelContainer({
       "data-likec4-color": color,
       "data-edge-active": isActive,
       "data-edge-animated": isActive,
+      ...selected !== false && {
+        "data-likec4-selected": selected
+      },
       ...isDimmed !== false && {
         "data-likec4-dimmed": isDimmed
       },

package/dist/base-primitives/edge/EdgePath.d.ts

@@ -6,6 +6,10 @@ type Data = UndefinedOnPartialDeep<Pick<DiagramEdge, 'line' | 'dir' | 'tail' | '
 type EdgePathProps = {
     edgeProps: BaseEdgePropsWithData<Data>;
     svgPath: string;
+    /**
+     * If true, the edge is being dragged (used to disable animations)
+     */
+    isDragging?: boolean;
     strokeWidth?: number;
     onEdgePointerDown?: PointerEventHandler<SVGGElement> | undefined;
 };

package/dist/base-primitives/edge/EdgePath.js

@@ -17,6 +17,8 @@ const EdgePath = forwardRef(({
     style,
     interactionWidth
   },
+  isDragging = false,
+  // omit
   onEdgePointerDown,
   strokeWidth,
   svgPath
@@ -37,16 +39,16 @@ const EdgePath = forwardRef(({
     strokeDasharray = "8,10";
   }
   return /* @__PURE__ */ jsxs(Fragment, { children: [
-    selectable && /* @__PURE__ */ jsx(
+    selectable && !isDragging && /* @__PURE__ */ jsx(
       "path",
       {
         className: cx(
           "react-flow__edge-interaction",
-          "hide-on-reduced-graphics",
           css({
             fill: "none"
           })
         ),
+        onPointerDown: onEdgePointerDown,
         d: svgPath,
         style: {
           strokeWidth: interactionWidth ?? 10,
@@ -60,7 +62,7 @@ const EdgePath = forwardRef(({
         MarkerStart && /* @__PURE__ */ jsx(MarkerStart, { id: "start" + id }),
         MarkerEnd && /* @__PURE__ */ jsx(MarkerEnd, { id: "end" + id })
       ] }),
-      /* @__PURE__ */ jsx(
+      !isDragging && /* @__PURE__ */ jsx(
         "path",
         {
           className: cx(
@@ -71,7 +73,8 @@ const EdgePath = forwardRef(({
           d: svgPath,
           style,
           strokeLinecap: "round"
-        }
+        },
+        "edge-path-bg"
       ),
       /* @__PURE__ */ jsx(
         "path",
@@ -89,7 +92,8 @@ const EdgePath = forwardRef(({
           strokeDasharray,
           markerStart: MarkerStart ? `url(#start${id})` : void 0,
           markerEnd: MarkerEnd ? `url(#end${id})` : void 0
-        }
+        },
+        "edge-path"
       )
     ] })
   ] });

package/dist/base-primitives/element/ElementActionButtons.d.ts

@@ -1,7 +1,8 @@
-import { BaseNodeData } from '../../base/types';
 type ElementActionButtonsProps = {
     selected?: boolean;
-    data: BaseNodeData;
+    data: {
+        hovered?: boolean;
+    };
     buttons: ElementActionButtons.Item[];
 };
 /**

package/dist/base-primitives/element/ElementActionButtons.js

@@ -7,7 +7,7 @@ import { ActionIcon } from '@mantine/core';
 import { useId } from '@mantine/hooks';
 import * as m from 'motion/react-m';
 import { stopPropagation } from '../../utils/xyflow.js';
-import IconBolt from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.1.1/node_modules/@tabler/icons-react/dist/esm/icons/IconBolt.js';
+import IconBolt from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.2.0/node_modules/@tabler/icons-react/dist/esm/icons/IconBolt.js';
 
 const container = hstack({
   position: "absolute",

package/dist/base-primitives/element/ElementDetailsButton.js

@@ -3,10 +3,9 @@ import { cx, css } from '@likec4/styles/css';
 import { Box } from '@likec4/styles/jsx';
 import { actionBtn } from '@likec4/styles/recipes';
 import { ActionIcon } from '@mantine/core';
-import { useId } from '@mantine/hooks';
 import * as m from 'motion/react-m';
 import { stopPropagation } from '../../utils/xyflow.js';
-import IconId from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.1.1/node_modules/@tabler/icons-react/dist/esm/icons/IconId.js';
+import IconId from '../../node_modules/.pnpm/@tabler_icons-react@3.35.0_react@19.2.0/node_modules/@tabler/icons-react/dist/esm/icons/IconId.js';
 
 const container = css({
   position: "absolute",
@@ -38,7 +37,6 @@ function ElementDetailsButton({
   icon,
   onClick
 }) {
-  const id = useId();
   return /* @__PURE__ */ jsx(Box, { className: cx(container, "details-button"), children: /* @__PURE__ */ jsx(
     ActionIcon,
     {
@@ -64,8 +62,7 @@ function ElementDetailsButton({
       onClick,
       onDoubleClick: stopPropagation,
       children: icon ?? /* @__PURE__ */ jsx(IconId, { stroke: 1.8, style: { width: "75%" } })
-    },
-    id
+    }
   ) });
 }
 

package/dist/base-primitives/element/ElementNodeContainer.css.js

@@ -2,42 +2,30 @@ import { css } from '@likec4/styles/css';
 
 const container = css({
   position: "relative",
-  width: "100%",
-  height: "100%",
+  width: "full",
+  height: "full",
   padding: "0",
   margin: "0",
   display: "flex",
   alignItems: "center",
   justifyContent: "center",
   pointerEvents: "none",
-  _before: {
-    content: '" "',
-    position: "absolute",
-    top: "[calc(100% - 4px)]",
-    left: "0",
-    width: "100%",
-    height: "24px",
-    background: "[transparent]",
-    pointerEvents: "all"
-  },
   _focusVisible: {
     outline: "none"
   },
-  // _whenDimmed: {
-  //   opacity: 0.25,
-  //   transition: `opacity 400ms {easings.inOut}, filter 500ms {easings.inOut}`,
-  //   transitionDelay: '50ms',
-  //   grayscale: 0.85,
-  //   blur: '2px',
-  //   // filter: `grayscale(0.85) ${fallbackVar(vars.safariAnimationHook, 'blur(2px)')}`,
-  // },
-  // _whenDimmedImmediate: {
-  //   opacity: 0.25,
-  //   grayscale: 0.85,
-  //   blur: '2px',
-  //   // transition: `opacity 100ms ${easings.inOut}, filter 100ms ${easings.inOut}`,
-  //   // filter: `grayscale(0.85) ${fallbackVar(vars.safariAnimationHook, 'blur(2px)')}`,
-  // },
+  _whenSelectable: {
+    pointerEvents: "all",
+    _before: {
+      content: '" "',
+      position: "absolute",
+      top: "[calc(100% - 4px)]",
+      left: "0",
+      width: "full",
+      height: "24px",
+      background: "transparent",
+      pointerEvents: "all"
+    }
+  },
   _reduceGraphicsOnPan: {
     _before: {
       display: "none"

package/dist/base-primitives/element/ElementTags.js

@@ -9,7 +9,8 @@ import { NodeToolbar, Position } from '@xyflow/react';
 import { deepEqual } from 'fast-equals';
 import { forwardRef, memo, useEffect, useCallback } from 'react';
 import { useTagSpecification } from '../../context/TagStylesContext.js';
-import { useDiagram } from '../../hooks/useDiagram.js';
+import '@xstate/react';
+import { useDiagram } from '../../hooks/safeContext.js';
 import { useCurrentZoom } from '../../hooks/useXYFlow.js';
 import { stopPropagation } from '../../utils/xyflow.js';
 

package/dist/base-primitives/memoEdge.js

@@ -4,9 +4,11 @@ import { memo } from 'react';
 const isSame = (a, b) => {
   return Math.abs(a - b) < 2.5;
 };
-const edgePropsEqual = (prev, next) => prev.id === next.id && deepEqual(prev.selected ?? false, next.selected ?? false) && deepEqual(prev.animated ?? false, next.animated ?? false) && deepEqual(prev.source, next.source) && deepEqual(prev.sourceHandleId ?? null, next.sourceHandleId ?? null) && deepEqual(prev.sourcePosition, next.sourcePosition) && deepEqual(prev.target, next.target) && deepEqual(prev.targetHandleId ?? null, next.targetHandleId ?? null) && deepEqual(prev.targetPosition, next.targetPosition) && isSame(prev.sourceX, next.sourceX) && isSame(prev.sourceY, next.sourceY) && isSame(prev.targetX, next.targetX) && isSame(prev.targetY, next.targetY) && deepEqual(prev.data, next.data);
+const edgePropsEqual = (prev, next) => prev.id === next.id && deepEqual(prev.selected ?? false, next.selected ?? false) && deepEqual(prev.animated ?? false, next.animated ?? false) && deepEqual(prev.source, next.source) && isSame(prev.sourceX, next.sourceX) && isSame(prev.sourceY, next.sourceY) && deepEqual(prev.sourceHandleId ?? null, next.sourceHandleId ?? null) && deepEqual(prev.sourcePosition, next.sourcePosition) && deepEqual(prev.target, next.target) && isSame(prev.targetY, next.targetY) && isSame(prev.targetX, next.targetX) && deepEqual(prev.targetHandleId ?? null, next.targetHandleId ?? null) && deepEqual(prev.targetPosition, next.targetPosition) && deepEqual(prev.data, next.data);
 function memoEdge(Edge) {
-  return memo(Edge, edgePropsEqual);
+  const Memo = memo(Edge, edgePropsEqual);
+  Memo.displayName = `MemoEdge(${Edge.displayName || Edge.name})`;
+  return Memo;
 }
 
 export { memoEdge };

package/dist/base-primitives/memoNode.d.ts

@@ -1,3 +1,3 @@
 import { FunctionComponent } from 'react';
 import { BaseNodeProps } from '../base/types';
-export declare function memoNode<P extends BaseNodeProps>(Node: FunctionComponent<P>): FunctionComponent<P>;
+export declare function memoNode<P extends BaseNodeProps>(Node: FunctionComponent<P>, displayName?: string): FunctionComponent<P>;

package/dist/base-primitives/memoNode.js

@@ -2,10 +2,10 @@ import { deepEqual } from 'fast-equals';
 import { memo } from 'react';
 
 function nodePropsEqual(prev, next) {
-  return prev.id === next.id && deepEqual(prev.selected ?? false, next.selected ?? false) && deepEqual(prev.dragging ?? false, next.dragging ?? false) && deepEqual(prev.width ?? 0, next.width ?? 0) && deepEqual(prev.height ?? 0, next.height ?? 0) && deepEqual(prev.zIndex ?? 0, next.zIndex ?? 0) && deepEqual(prev.data, next.data);
+  return prev.id === next.id && deepEqual(prev.type, next.type) && deepEqual(prev.selected ?? false, next.selected ?? false) && deepEqual(prev.dragging ?? false, next.dragging ?? false) && deepEqual(prev.width ?? 0, next.width ?? 0) && deepEqual(prev.height ?? 0, next.height ?? 0) && deepEqual(prev.zIndex ?? 0, next.zIndex ?? 0) && deepEqual(prev.data, next.data);
 }
 const isMemoized = Symbol.for("isMemoized");
-function memoNode(Node) {
+function memoNode(Node, displayName = "Node") {
   if (Node.hasOwnProperty(isMemoized)) {
     return Node;
   }
@@ -13,7 +13,7 @@ function memoNode(Node) {
     Node,
     nodePropsEqual
   );
-  NodeComponent.displayName = "Node";
+  NodeComponent.displayName = displayName;
   Object.defineProperty(NodeComponent, isMemoized, {
     enumerable: false,
     writable: false,