From b02b6512fbcb56c91fa404ab80f6d2cb8869314f Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Mon, 13 Apr 2026 17:23:17 +0200
Subject: [PATCH] Add method to tag entity group with ID and kind

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

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 183c3bc9..ad4898fe 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -3669,6 +3669,26 @@ export class ViewportHost {
     group.add(mesh);
   }
 
+  private tagEntityGroup(
+    group: Group,
+    entityId: string,
+    entityKind: EntityInstance["kind"]
+  ): Mesh[] {
+    const meshes: Mesh[] = [];
+
+    group.traverse((object) => {
+      if (!(object instanceof Mesh)) {
+        return;
+      }
+
+      object.userData.entityId = entityId;
+      object.userData.entityKind = entityKind;
+      meshes.push(object);
+    });
+
+    return meshes;
+  }
+
   private createPointLightGizmoRenderObjects(
     entityId: string,
     position: Vec3,