feat(options): add minZoom, maxZoom, defaultZoom closes #273

src/container/GraphView/index.tsx

@@ -43,6 +43,9 @@ export interface GraphViewProps {
   onlyRenderVisibleNodes: boolean;
   isInteractive: boolean;
   selectNodesOnDrag: boolean;
+  minZoom: number;
+  maxZoom: number;
+  defaultZoom: number;
 }
 
 const GraphView = memo(
@@ -66,6 +69,9 @@ const GraphView = memo(
     onlyRenderVisibleNodes,
     isInteractive,
     selectNodesOnDrag,
+    minZoom,
+    maxZoom,
+    defaultZoom,
   }: GraphViewProps) => {
     const zoomPane = useRef<HTMLDivElement>(null);
     const rendererNode = useRef<HTMLDivElement>(null);
@@ -79,6 +85,9 @@ const GraphView = memo(
     const setOnConnect = useStoreActions((a) => a.setOnConnect);
     const setSnapGrid = useStoreActions((actions) => actions.setSnapGrid);
     const setInteractive = useStoreActions((actions) => actions.setInteractive);
+    const updateTransform = useStoreActions((actions) => actions.updateTransform);
+    const setMinMaxZoom = useStoreActions((actions) => actions.setMinMaxZoom);
+
     const selectionKeyPressed = useKeyPress(selectionKeyCode);
     const rendererClasses = classnames('react-flow__renderer', { 'is-interactive': isInteractive });
 
@@ -106,6 +115,10 @@ const GraphView = memo(
         setOnConnect(onConnect);
       }
 
+      if (defaultZoom !== 1) {
+        updateTransform({ x: 0, y: 0, k: defaultZoom });
+      }
+
       return () => {
         window.onresize = null;
       };
@@ -132,6 +145,10 @@ const GraphView = memo(
       setInteractive(isInteractive);
     }, [isInteractive]);
 
+    useEffect(() => {
+      setMinMaxZoom({ minZoom, maxZoom });
+    }, [minZoom, maxZoom]);
+
     useGlobalKeyHandler({ onElementsRemove, deleteKeyCode });
     useElementUpdater(elements);
 

src/container/ReactFlow/index.tsx

@@ -54,6 +54,9 @@ export interface ReactFlowProps extends Omit<HTMLAttributes<HTMLDivElement>, 'on
   onlyRenderVisibleNodes: boolean;
   isInteractive: boolean;
   selectNodesOnDrag: boolean;
+  minZoom: number;
+  maxZoom: number;
+  defaultZoom: number;
 }
 
 const ReactFlow = ({
@@ -80,6 +83,9 @@ const ReactFlow = ({
   onlyRenderVisibleNodes,
   isInteractive,
   selectNodesOnDrag,
+  minZoom,
+  maxZoom,
+  defaultZoom,
 }: ReactFlowProps) => {
   const nodeTypesParsed = useMemo(() => createNodeTypes(nodeTypes), []);
   const edgeTypesParsed = useMemo(() => createEdgeTypes(edgeTypes), []);
@@ -107,6 +113,9 @@ const ReactFlow = ({
           onlyRenderVisibleNodes={onlyRenderVisibleNodes}
           isInteractive={isInteractive}
           selectNodesOnDrag={selectNodesOnDrag}
+          minZoom={minZoom}
+          maxZoom={maxZoom}
+          defaultZoom={defaultZoom}
         />
         {onSelectionChange && <SelectionListener onSelectionChange={onSelectionChange} />}
         {children}
@@ -136,6 +145,9 @@ ReactFlow.defaultProps = {
   onlyRenderVisibleNodes: true,
   isInteractive: true,
   selectNodesOnDrag: true,
+  minZoom: 0.5,
+  maxZoom: 2,
+  defaultZoom: 1,
 };
 
 export default ReactFlow;

src/hooks/useD3Zoom.ts

@@ -10,10 +10,6 @@ interface UseD3ZoomParams {
   onMove?: () => void;
 }
 
-const d3ZoomInstance = zoom()
-  .scaleExtent([0.5, 2])
-  .filter(() => !event.button);
-
 export default ({ zoomPane, onMove, selectionKeyPressed }: UseD3ZoomParams): void => {
   const transform = useStoreState((s) => s.transform);
   const d3Selection = useStoreState((s) => s.d3Selection);
@@ -24,16 +20,21 @@ export default ({ zoomPane, onMove, selectionKeyPressed }: UseD3ZoomParams): voi
 
   useEffect(() => {
     if (zoomPane.current) {
-      const selection = select(zoomPane.current).call(d3ZoomInstance);
-      initD3({ zoom: d3ZoomInstance, selection });
+      const nextD3ZoomInstance = zoom();
+      const selection = select(zoomPane.current).call(nextD3ZoomInstance);
+      initD3({ zoom: nextD3ZoomInstance, selection });
     }
   }, []);
 
   useEffect(() => {
+    if (!d3Zoom) {
+      return;
+    }
+
     if (selectionKeyPressed) {
-      d3ZoomInstance.on('zoom', null);
+      d3Zoom.on('zoom', null);
     } else {
-      d3ZoomInstance.on('zoom', () => {
+      d3Zoom.on('zoom', () => {
         if (event.sourceEvent && event.sourceEvent.target !== zoomPane.current) {
           return;
         }
@@ -53,7 +54,7 @@ export default ({ zoomPane, onMove, selectionKeyPressed }: UseD3ZoomParams): voi
     }
 
     return () => {
-      d3ZoomInstance.on('zoom', null);
+      d3Zoom.on('zoom', null);
     };
-  }, [selectionKeyPressed]);
+  }, [selectionKeyPressed, d3Zoom]);
 };

src/store/index.ts

@@ -47,6 +47,11 @@ type D3Init = {
   selection: D3Selection<Element, unknown, null, undefined>;
 };
 
+type SetMinMaxZoom = {
+  minZoom: number;
+  maxZoom: number;
+};
+
 type SetSnapGrid = {
   snapToGrid: boolean;
   snapGrid: [number, number];
@@ -64,6 +69,8 @@ export interface StoreModel {
   d3Zoom: ZoomBehavior<Element, unknown> | null;
   d3Selection: D3Selection<Element, unknown, null, undefined> | null;
   d3Initialised: boolean;
+  minZoom: number;
+  maxZoom: number;
 
   nodesSelectionActive: boolean;
   selectionActive: boolean;
@@ -104,6 +111,8 @@ export interface StoreModel {
 
   initD3: Action<StoreModel, D3Init>;
 
+  setMinMaxZoom: Action<StoreModel, SetMinMaxZoom>;
+
   setSnapGrid: Action<StoreModel, SetSnapGrid>;
 
   setConnectionPosition: Action<StoreModel, XYPosition>;
@@ -129,6 +138,8 @@ const storeModel: StoreModel = {
   d3Zoom: null,
   d3Selection: null,
   d3Initialised: false,
+  minZoom: 0.5,
+  maxZoom: 2,
 
   nodesSelectionActive: false,
   selectionActive: false,
@@ -326,10 +337,22 @@ const storeModel: StoreModel = {
 
   initD3: action((state, { zoom, selection }) => {
     state.d3Zoom = zoom;
+
+    state.d3Zoom.scaleExtent([state.minZoom, state.maxZoom]);
+
     state.d3Selection = selection;
     state.d3Initialised = true;
   }),
 
+  setMinMaxZoom: action((state, { minZoom, maxZoom }) => {
+    state.minZoom = minZoom;
+    state.maxZoom = maxZoom;
+
+    if (state.d3Zoom) {
+      state.d3Zoom.scaleExtent([minZoom, maxZoom]);
+    }
+  }),
+
   setConnectionPosition: action((state, position) => {
     state.connectionPosition = position;
   }),