github-mirrors/AppFlowy
1
0
Fork 0
mirror of https://github.com/AppFlowy-IO/AppFlowy.git synced 2025-04-24 14:47:13 -04:00
Code Issues Projects Releases Packages Wiki Activity

chore: rename functions

Browse source
This commit is contained in:
appflowy 2022-02-15 13:27:54 +08:00
parent aecf29ddca
commit 427d2c2e25
1 changed files with 29 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

41
shared-lib/lib-infra/src/pb.rs
View file

@ -11,18 +11,23 @@ use std::path::PathBuf;
use std::process::Command; use std::process::Command;
use walkdir::WalkDir; use walkdir::WalkDir;
pub fn gen_files(crate_name: &str, root: &str) { pub fn gen_files(crate_name: &str, proto_file_dir: &str) {
// 1. generate the proto files to proto_file_dir
#[cfg(feature = "proto_gen")] #[cfg(feature = "proto_gen")]
let _ = gen_protos(crate_name); let _ = gen_protos(crate_name);
let mut paths = vec![]; let mut paths = vec![];
let mut file_names = vec![]; let mut file_names = vec![];
for (path, file_name) in WalkDir::new(root).into_iter().filter_map(|e| e.ok()).map(|e| { for (path, file_name) in WalkDir::new(proto_file_dir)
let path = e.path().to_str().unwrap().to_string(); .into_iter()
let file_name = e.path().file_stem().unwrap().to_str().unwrap().to_string(); .filter_map(|e| e.ok())
(path, file_name) .map(|e| {
}) { let path = e.path().to_str().unwrap().to_string();
let file_name = e.path().file_stem().unwrap().to_str().unwrap().to_string();
(path, file_name)
})
{
if path.ends_with(".proto") { if path.ends_with(".proto") {
// https://stackoverflow.com/questions/49077147/how-can-i-force-build-rs-to-run-again-without-cleaning-my-whole-project // https://stackoverflow.com/questions/49077147/how-can-i-force-build-rs-to-run-again-without-cleaning-my-whole-project
println!("cargo:rerun-if-changed={}", path); println!("cargo:rerun-if-changed={}", path);
@ -31,22 +36,34 @@ pub fn gen_files(crate_name: &str, root: &str) {
} }
} }
println!("cargo:rerun-if-changed=build.rs"); println!("cargo:rerun-if-changed=build.rs");
let protoc_bin_path = protoc_bin_vendored::protoc_bin_path().unwrap();
let protoc_path = protoc_bin_vendored::protoc_bin_path().unwrap(); // 2. generate the protobuf files(Dart)
#[cfg(feature = "dart")] #[cfg(feature = "dart")]
gen_pb_for_dart(crate_name, root, &paths, &file_names, &protoc_path); generate_dart_protobuf_files(crate_name, proto_file_dir, &paths, &file_names, &protoc_bin_path);
// 3. generate the protobuf files(Rust)
generate_rust_protobuf_files(&protoc_bin_path, &paths, proto_file_dir);
}
fn generate_rust_protobuf_files(protoc_bin_path: &PathBuf, input_paths: &Vec<String>, proto_file_dir: &str) {
protoc_rust::Codegen::new() protoc_rust::Codegen::new()
.out_dir("./src/protobuf/model") .out_dir("./src/protobuf/model")
.protoc_path(protoc_path) .protoc_path(protoc_bin_path)
.inputs(&paths) .inputs(input_paths)
.include(root) .include(proto_file_dir)
.run() .run()
.expect("Running protoc failed."); .expect("Running protoc failed.");
} }
#[cfg(feature = "dart")] #[cfg(feature = "dart")]
fn gen_pb_for_dart(name: &str, root: &str, paths: &Vec<String>, file_names: &Vec<String>, proto_path: &PathBuf) { fn generate_dart_protobuf_files(
name: &str,
root: &str,
paths: &Vec<String>,
file_names: &Vec<String>,
proto_path: &PathBuf,
) {
if std::env::var("CARGO_MAKE_WORKING_DIRECTORY").is_err() { if std::env::var("CARGO_MAKE_WORKING_DIRECTORY").is_err() {
log::warn!("CARGO_MAKE_WORKING_DIRECTORY was not set, skip generate dart pb"); log::warn!("CARGO_MAKE_WORKING_DIRECTORY was not set, skip generate dart pb");
return; return;