From 8786e4c12b6f93bbbc1bf6e112a2ac038a6e24c7 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 11 Apr 2026 02:40:59 +0200
Subject: [PATCH] Add viewportGridVisible prop and effect to ViewportCanvas

---
 src/viewport-three/ViewportCanvas.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/viewport-three/ViewportCanvas.tsx b/src/viewport-three/ViewportCanvas.tsx
index 1efb105e..ea30c727 100644
--- a/src/viewport-three/ViewportCanvas.tsx
+++ b/src/viewport-three/ViewportCanvas.tsx
@@ -37,6 +37,7 @@ interface ViewportCanvasProps {
   whiteboxSelectionMode: WhiteboxSelectionMode;
   whiteboxSnapEnabled: boolean;
   whiteboxSnapStep: number;
+  viewportGridVisible: boolean;
   selection: EditorSelection;
   toolMode: ToolMode;
   toolPreview: ViewportToolPreview;
@@ -67,6 +68,7 @@ export function ViewportCanvas({
   whiteboxSelectionMode,
   whiteboxSnapEnabled,
   whiteboxSnapStep,
+  viewportGridVisible,
   selection,
   toolMode,
   toolPreview,
@@ -138,6 +140,10 @@ export function ViewportCanvas({
     hostRef.current?.setWhiteboxSnapSettings(whiteboxSnapEnabled, whiteboxSnapStep);
   }, [whiteboxSnapEnabled, whiteboxSnapStep]);
 
+  useEffect(() => {
+    hostRef.current?.setGridVisible(viewportGridVisible);
+  }, [viewportGridVisible]);
+
   useEffect(() => {
     hostRef.current?.setWhiteboxSelectionMode(whiteboxSelectionMode);
   }, [whiteboxSelectionMode]);
@@ -156,7 +162,7 @@ export function ViewportCanvas({
 
   useEffect(() => {
     hostRef.current?.setCameraState(cameraState);
-  }, [cameraState]);
+  }, [cameraState, sceneDocument, transformSession.kind]);
 
   useEffect(() => {
     hostRef.current?.setBrushSelectionChangeHandler(onSelectionChange);