fix: fix puml format issue

core_model/src/url_format.rs

@@ -47,8 +47,8 @@ pub fn json_filename(text: &str) -> String {
     json_filename_suffix(text, None)
 }
 
-pub fn uml_filename(text: &str) -> String {
-    filename_suffix(text, None, Some("uml"))
+pub fn puml_filename(text: &str) -> String {
+    filename_suffix(text, None, Some("puml"))
 }
 
 pub fn uri_to_path(url: &str) -> PathBuf {

plugins/coco_struct_analysis/src/lib.rs

@@ -102,7 +102,7 @@ mod tests {
         let classes: Vec<ClassInfo> = serde_json::from_str(&code).unwrap();
         assert_eq!(9, classes.len());
 
-        let output_dir = base_dir.join("source.uml");
+        let output_dir = base_dir.join("source.puml");
         let _file = File::open(output_dir).unwrap();
     }
 }

plugins/coco_struct_analysis/src/plantuml_render.rs

@@ -21,7 +21,7 @@ impl PlantUmlRender {
             }
             content = format!("{}{}", content, methods.join(""));
 
-            rendered.push(format!("class {{\n{}}}", content));
+            rendered.push(format!("class {} {{\n{}}}", clazz.name, content));
         }
 
         let dep: Vec<String> = vec![];
@@ -53,7 +53,7 @@ mod tests {
         classes.push(demo);
 
         let str = PlantUmlRender::render(&classes);
-        assert_eq!("@startuml\n\nclass {\n}\n\n@enduml", str);
+        assert_eq!("@startuml\n\nclass Demo {\n}\n\n@enduml", str);
     }
 
     #[test]
@@ -71,7 +71,7 @@ mod tests {
 
         let str = PlantUmlRender::render(&classes);
         assert_eq!(
-            "@startuml\n\nclass {\n  -demo\n  -method()\n}\n\n@enduml",
+            "@startuml\n\nclass Demo {\n  -demo\n  -method()\n}\n\n@enduml",
             str
         );
     }

plugins/coco_struct_analysis/src/struct_analysis.rs

@@ -37,7 +37,7 @@ pub fn execute(config: CocoConfig) {
         let classes = run_ctags(&opt, &files);
 
         let result = serde_json::to_string_pretty(&classes).unwrap();
-        write_to_uml_file(url_str, &classes);
+        write_to_puml_file(url_str, &classes);
         write_to_json_file(url_str, &result);
     }
 }
@@ -57,8 +57,8 @@ fn write_to_json_file(url_str: &str, result: &String) {
     fs::write(output_file, result).expect("cannot write file");
 }
 
-fn write_to_uml_file(url_str: &str, classes: &Vec<ClassInfo>) {
-    let file_name = url_format::uml_filename(url_str);
+fn write_to_puml_file(url_str: &str, classes: &Vec<ClassInfo>) {
+    let file_name = url_format::puml_filename(url_str);
     let output_file = Settings::struct_analysis().join(file_name);
     let result = PlantUmlRender::render(classes);
     fs::write(output_file, result).expect("cannot write file");