From 560f9817103036781b4d4c22e41f5a15c99f4008 Mon Sep 17 00:00:00 2001
From: Leo dev <klesti.selimay@icloud.com>
Date: Fri, 22 Nov 2024 06:29:39 +0100
Subject: [PATCH] tree search

---
 lib.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/lib.rs b/lib.rs
index 589a0a6..328024c 100644
--- a/lib.rs
+++ b/lib.rs
@@ -81,10 +81,13 @@ pub fn element(_args: TokenStream, input: TokenStream) -> TokenStream {
                 }
             }
             fn get_element_by_id(&mut self, id: &str) -> Option<&mut Element> {
+                if self.id == id {
+                    return Some(self);
+                }
                 if let Children::Children(children, _) = &mut self.children {
                     for elem in children {
-                        if elem.get_data().2 == id {
-                            return Some(elem);
+                        if let Some(e) elem.get_element_by_id(id) {
+                            return Some(e);
                         }
                     }
                 }