From 0c323670a51b30c545e56cbfe60d1c0e0d5e92d6 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:13:26 +0200
Subject: [PATCH] Update tool state to manage foliage layer ID

---
 src/viewport-three/viewport-host.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 28281b7b..6106d28e 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -9560,7 +9560,11 @@ export class ViewportHost {
       settings: toolState,
       tool: toolState.tool,
       referenceHeight,
-      layerIndex: toolState.tool === "paint" ? toolState.layerIndex : null
+      layerIndex: toolState.tool === "paint" ? toolState.layerIndex : null,
+      foliageLayerId:
+        toolState.tool === "foliagePaint" || toolState.tool === "foliageErase"
+          ? toolState.foliageLayerId
+          : null
     });
   }