From 66fbd5dc0bb84facca07ce6df90546c66a242666 Mon Sep 17 00:00:00 2001
From: Ben Smith <binjimin@gmail.com>
Date: Thu, 30 Jan 2025 16:10:24 -0800
Subject: [PATCH] Add im.treenode() overload where you can pass flags

---
 modules/imgui.lobster | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/modules/imgui.lobster b/modules/imgui.lobster
index 99af1873..3df1ee06 100644
--- a/modules/imgui.lobster
+++ b/modules/imgui.lobster
@@ -239,6 +239,13 @@ def treenode(label, body):
         treenode_end()
     return open
 
+def treenode(label, flags, body):
+    let open = treenode_start(label, flags)
+    if open:
+        body()
+        treenode_end()
+    return open
+
 def treenode_open(label, body):
     let open = treenode_start(label, treenode_default_open)
     if open: