From 853c0d8a5af27370556fed8d7877fd9b1d0ac5cf Mon Sep 17 00:00:00 2001
From: matcool <26722564+matcool@users.noreply.github.com>
Date: Tue, 9 Jul 2024 12:59:56 -0300
Subject: [PATCH] fix header include for free functions

---
 src/builder/shared.rs | 5 -----
 src/builder/traits.rs | 4 ++--
 src/config.rs         | 1 +
 3 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/builder/shared.rs b/src/builder/shared.rs
index ab659c9..18d98e6 100644
--- a/src/builder/shared.rs
+++ b/src/builder/shared.rs
@@ -203,11 +203,6 @@ pub fn fmt_field(field: &Entity, builder: &Builder) -> Html {
 }
 
 pub fn fmt_fun_decl(fun: &Entity, builder: &Builder) -> Html {
-    if fun.get_name().is_some_and(|x| x.contains("add")) {
-        dbg!(&fun);
-        dbg!(fun.get_arguments());
-        dbg!(fun.get_template_arguments());
-    }
     HtmlElement::new("details")
         .with_class("entity-desc")
         .with_attr_opt("id", member_fun_link(fun))
diff --git a/src/builder/traits.rs b/src/builder/traits.rs
index 8aa1336..e7f309d 100644
--- a/src/builder/traits.rs
+++ b/src/builder/traits.rs
@@ -60,8 +60,8 @@ impl<'e> EntityMethods<'e> for Entity<'e> {
     }
 
     fn definition_file(&self) -> Option<PathBuf> {
-        self.get_definition()?
-            .get_location()?
+        self.get_definition()
+            .map_or_else(|| self.get_location(), |d| d.get_location())?
             .get_file_location()
             .file?
             .get_path()
diff --git a/src/config.rs b/src/config.rs
index e2d29a5..216811f 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -62,6 +62,7 @@ macro_rules! default_scripts {
     };
 }
 
+#[derive(Debug)]
 pub struct Source {
     pub name: String,
     pub dir: UrlPath,