From 2fe826686236f1cb6ed17219ee1d5498a3696159 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 18 Apr 2026 19:52:49 +0200
Subject: [PATCH] Support terrain selection in viewport host

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

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 1ea341d4..c484764c 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -7003,6 +7003,8 @@ export class ViewportHost {
         return `brushEdge:${selection.brushId}:${selection.edgeId}`;
       case "brushVertex":
         return `brushVertex:${selection.brushId}:${selection.vertexId}`;
+      case "terrains":
+        return selection.ids.length === 1 ? `terrain:${selection.ids[0]}` : null;
       case "paths":
         return selection.ids.length === 1 ? `path:${selection.ids[0]}` : null;
       case "pathPoint":
@@ -7033,6 +7035,14 @@ export class ViewportHost {
     const pathId = hit.object.userData.pathId;
     const pathPointId = hit.object.userData.pathPointId;
 
+    const terrainId = hit.object.userData.terrainId;
+    if (typeof terrainId === "string") {
+      return {
+        kind: "terrains",
+        ids: [terrainId]
+      };
+    }
+
     if (typeof pathId === "string" && typeof pathPointId === "string") {
       return {
         kind: "pathPoint",
@@ -7130,6 +7140,9 @@ export class ViewportHost {
           renderObjects.line,
           ...renderObjects.pointMeshes.map((pointMesh) => pointMesh.mesh)
         ]).flat(),
+        ...Array.from(this.terrainRenderObjects.values(), (renderObjects) =>
+          renderObjects.mesh
+        ),
         ...Array.from(this.modelRenderObjects.values()),
         ...this.getBrushPickableObjects()
       ],