[code gen]: generate the protobuf files in corresponding module

scripts/flowy-tool/src/proto/proto_gen.rs

@@ -30,7 +30,6 @@ impl ProtoGen {
 fn write_proto_files(crate_infos: &Vec<CrateProtoInfo>) {
     for crate_info in crate_infos {
         let dir = crate_info.inner.proto_file_output_dir();
-        remove_everything_in_dir(dir.as_str());
         crate_info.files.iter().for_each(|info| {
             let proto_file_path = format!("{}/{}.proto", dir, &info.file_name);
             save_content_to_file_with_diff_prompt(
@@ -76,36 +75,35 @@ fn write_flutter_protobuf_package_mod_file(
     crate_infos: &Vec<CrateProtoInfo>,
     package_info: &FlutterProtobufInfo,
 ) {
-    let mod_path = package_info.mod_file_path();
-    let _model_dir = package_info.model_dir();
-    match OpenOptions::new()
-        .create(true)
-        .write(true)
-        .append(false)
-        .truncate(true)
-        .open(&mod_path)
-    {
-        Ok(ref mut file) => {
-            let mut mod_file_content = String::new();
-            mod_file_content.push_str("// Auto-generated, do not edit \n");
+    let model_dir = package_info.model_dir();
+    for crate_info in crate_infos {
+        let mod_path = crate_info.flutter_mod_file(model_dir.as_str());
+        match OpenOptions::new()
+            .create(true)
+            .write(true)
+            .append(false)
+            .truncate(true)
+            .open(&mod_path)
+        {
+            Ok(ref mut file) => {
+                let mut mod_file_content = String::new();
+                mod_file_content.push_str("// Auto-generated, do not edit \n");
 
-            for crate_info in crate_infos {
-                let _mod_path = crate_info.inner.proto_model_mod_file();
                 walk_dir(
                     crate_info.inner.proto_file_output_dir().as_ref(),
                     |e| e.file_type().is_dir() == false,
                     |_, name| {
-                        let c = format!("export 'protobuf/{}.pb.dart';\n", &name);
+                        let c = format!("export './{}.pb.dart';\n", &name);
                         mod_file_content.push_str(c.as_ref());
                     },
                 );
-            }
 
-            file.write_all(mod_file_content.as_bytes()).unwrap();
-            file.flush().unwrap();
-        }
-        Err(err) => {
-            panic!("Failed to open file: {}", err);
+                file.write_all(mod_file_content.as_bytes()).unwrap();
+                file.flush().unwrap();
+            }
+            Err(err) => {
+                panic!("Failed to open file: {}", err);
+            }
         }
     }
 }
@@ -134,17 +132,21 @@ fn run_rust_protoc(crate_infos: &Vec<CrateProtoInfo>) {
 
 fn run_flutter_protoc(crate_infos: &Vec<CrateProtoInfo>, package_info: &FlutterProtobufInfo) {
     let model_dir = package_info.model_dir();
-    let removed_dir = format!("{}/", model_dir);
-    remove_everything_in_dir(removed_dir.as_str());
+    if !Path::new(&model_dir).exists() {
+        std::fs::create_dir_all(&model_dir).unwrap();
+    }
 
     for crate_info in crate_infos {
         let proto_path = crate_info.inner.proto_file_output_dir();
+        let crate_module_dir = crate_info.flutter_mod_dir(model_dir.as_str());
+        remove_everything_in_dir(crate_module_dir.as_str());
+
         walk_dir(
             proto_path.as_ref(),
             |e| is_proto_file(e),
             |proto_file, _| {
                 if cmd_lib::run_cmd! {
-                    protoc --dart_out=${model_dir} --proto_path=${proto_path} ${proto_file}
+                    protoc --dart_out=${crate_module_dir} --proto_path=${proto_path} ${proto_file}
                 }
                 .is_err()
                 {
@@ -156,16 +158,14 @@ fn run_flutter_protoc(crate_infos: &Vec<CrateProtoInfo>, package_info: &FlutterP
 }
 
 fn remove_everything_in_dir(dir: &str) {
-    if !Path::new(dir).exists() {
+    if Path::new(dir).exists() {
         if cmd_lib::run_cmd! {
             rm -rf ${dir}
-            mkdir ${dir}
         }
         .is_err()
         {
             panic!("Reset protobuf directory failed")
         };
-    } else {
-        std::fs::create_dir_all(dir).unwrap();
     }
+    std::fs::create_dir_all(dir).unwrap();
 }

scripts/flowy-tool/src/proto/proto_info.rs

@@ -36,6 +36,16 @@ pub use model::*;
             }
         }
     }
+
+    pub fn flutter_mod_dir(&self, root: &str) -> String {
+        let crate_module_dir = format!("{}/{}", root, self.inner.folder_name);
+        crate_module_dir
+    }
+
+    pub fn flutter_mod_file(&self, root: &str) -> String {
+        let crate_module_dir = format!("{}/{}/protobuf.dart", root, self.inner.folder_name);
+        crate_module_dir
+    }
 }
 
 #[derive(Clone, Debug)]

scripts/flowy-tool/src/util/file.rs

@@ -59,7 +59,7 @@ pub fn save_content_to_file_with_diff_prompt(content: &str, output_file: &str, _
             .open(output_file)
         {
             Ok(ref mut file) => file.write_all(content.as_bytes()).unwrap(),
-            Err(err) => panic!("Open or create file fail: {}", err),
+            Err(err) => panic!("Open or create to {} fail: {}", output_file, err),
         }
     }
 }