feat(debug): add custom "Code Generation" Comet debug metric

1 files changed, 8 insertions, 2 deletions

diff --git a/src/types/project.rs b/src/types/project.rs
index 145ab18..91b2bb1 100644
--- a/src/types/project.rs
+++ b/src/types/project.rs
@@ -83,7 +83,10 @@ impl Project {
     }
 
     pub fn app_code(&mut self) -> Result<String, Error> {
-        match self.element_tree {
+        use iced::debug;
+        let codegen = debug::time("Code Generation");
+
+        let result = match self.element_tree {
             Some(ref element_tree) => {
                 let (imports, view) = element_tree.codegen();
                 let theme = self.get_theme();
@@ -137,6 +140,9 @@ impl State {{
                 Ok(rustfmt.format_str(app_code)?)
             }
             None => Err("No element tree present".into()),
-        }
+        };
+
+        codegen.finish();
+        result
     }
 }