kjuulh/crunch
1
0
Fork 0
Code Issues 1 Pull Requests 4 Packages Projects Releases Wiki Activity

feat: with generated output

Browse Source 
Signed-off-by: kjuulh <contact@kjuulh.io>
This commit is contained in:
Kasper Juul Hermansen 2023-09-24 16:31:12 +02:00
parent 38f41db98c
commit fb20207593
Signed by: kjuulh
GPG Key ID: 9AA7BC13CE474394
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/crunch-cli/src/main.rs
View File

@ -64,6 +64,8 @@ async fn main() -> anyhow::Result<()> {
codegen
.generate_rust(&rel_schema_path, &rel_output_path)
.await?;
println!("success: generated crunch {}", &rel_output_path.display());
}
}
}

20
crates/crunch-codegen/src/lib.rs
View File

@ -14,11 +14,11 @@ impl Codegen {
pub async fn generate_rust(&self, input_path: &Path, output_path: &Path) -> anyhow::Result<()> {
let input_protos = self.discover_files(input_path, "proto")?;
let (input_proto_paths, input_dir) = self.copy_protos(input_protos, input_path).await?;
let (output_proto_paths, output_dir) = self
.generate_rust_from_proto(input_proto_paths, output_path, input_dir.path())
let (output_proto_paths, temp_output_dir) = self
.generate_rust_from_proto(input_proto_paths, input_dir.path())
.await?;
self.copy_rs(output_proto_paths, output_path, output_dir.path())
self.copy_rs(output_proto_paths, output_path, temp_output_dir.path())
.await?;
Ok(())
@ -29,13 +29,15 @@ impl Codegen {
for entry in WalkDir::new(input_path) {
let entry = entry?;
if let Some(extension) = entry.path().extension().and_then(|e| e.to_str()) {
proto_files.push(entry.into_path());
if let Some(ext) = entry.path().extension().and_then(|e| e.to_str()) {
if ext == extension {
proto_files.push(entry.into_path());
}
}
}
if proto_files.is_empty() {
anyhow::anyhow!(
anyhow::bail!(
"failed to find any protobuf files in: {}",
input_path.display()
);
@ -72,7 +74,6 @@ impl Codegen {
async fn generate_rust_from_proto(
&self,
input_proto_paths: Vec<PathBuf>,
output_path: &Path,
in_root_path: &Path,
) -> anyhow::Result<(Vec<PathBuf>, tempfile::TempDir)> {
let out_tempdir = tempfile::TempDir::new()?;
@ -149,9 +150,10 @@ impl Codegen {
for output_rs in &output_proto_paths {
let rel_proto_path = output_rs.strip_prefix(root_path).map_err(|e| {
anyhow!(
"output: {} does not match root_path: {}",
"output: {} does not match root_path: {}, error: {}",
output_rs.display(),
root_path.display()
root_path.display(),
e
)
})?;
let in_proto_path = output_path.join(rel_proto_path);