feat: update display for function name

unit-mesh · Aug 16, 2023 · 522a450 · 522a450
1 parent 8c58067
commit 522a450
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/counit-server/src/model/chapi_model.rs b/counit-server/src/model/chapi_model.rs
@@ -144,7 +144,7 @@ pub struct CodeFunction {
 }
 
 impl CodeFunction {
-    pub(crate) fn display(&self) -> String {
+    pub(crate) fn display(&self, class: &CodeDataStruct) -> String {
         let annotation = self.annotations.iter()
             .map(|it| {
                 let key_values = it.key_values.iter()
@@ -184,9 +184,12 @@ impl CodeFunction {
             .collect::<Vec<String>>()
             .join("\n");
 
+        let path = format!("path: {}", class.file_path);
+        let canonical_name = format!("canonical_name: {}.{}", class.package, class.node_name);
+
         format!(
-            "{}\n{}({}) -> {} {{\n{}\n}}",
-            annotation, self.name, params, return_type, body
+            "{}\n{}\n{}\n{}({}) -> {} {{\n{}\n}}",
+            path, canonical_name, annotation, self.name, params, return_type, body
         )
     }
 }

diff --git a/counit-server/src/server/archguard_api.rs b/counit-server/src/server/archguard_api.rs
@@ -124,7 +124,7 @@ pub async fn save_class_items(
                 let _ = &class.functions.iter().for_each(|method| {
                     tokio::task::block_in_place(|| {
                         Handle::current().block_on(async {
-                            let display_text = &method.display();
+                            let display_text = &method.display(class);
                             let origin_content = &method.content;
 
                             println!("class_items display_text {:?}", display_text);