Walk proto dir instead of manually specifying needed files

Author: graysonguarino

Cargo.lock

@@ -9,6 +9,7 @@ prost = "0.13"
 
 [build-dependencies]
 tonic-build = "0.12"
+walkdir = "2.5"
 
 [lib]
 path = "lib.rs"

lib/proto/Cargo.toml

@@ -1,4 +1,5 @@
-use std::path::Path;
+use std::path::{Path, PathBuf};
+use walkdir::WalkDir;
 
 fn main() -> Result<(), Box<dyn std::error::Error>> {
     let workspace_dir = Path::new(env!("CARGO_WORKSPACE_DIR"));
@@ -8,13 +9,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let proto_file_descriptor_set_path = Path::new("generated/bin/descriptor.bin");
     let generated_include_file = Path::new("include.rs");
 
-    let v1_protos = [
-        proto_tuxtape_dir.join("server/database/v1/database.proto"),
-        proto_tuxtape_dir.join("server/registrar/v1/registrar.proto"),
-        proto_tuxtape_dir.join("server/fleet_client/v1/fleet_client.proto"),
-        proto_tuxtape_dir.join("kernel_builder/v1/kernel_builder.proto"),
-        proto_tuxtape_dir.join("patch_builder/v1/patch_builder.proto"),
-    ];
+    let v1_protos = fetch_proto_dirs(&proto_tuxtape_dir, 1, &["common"]);
 
     let mut config = tonic_build::Config::default();
     config.out_dir(proto_src_out_dir);
@@ -29,3 +24,32 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     Ok(())
 }
+
+fn fetch_proto_dirs(
+    proto_package_dir: &Path,
+    proto_version: usize,
+    ignore_modules: &[&str],
+) -> Vec<PathBuf> {
+    let version_str = format!("v{proto_version}");
+    WalkDir::new(proto_package_dir)
+        .into_iter()
+        // Remove all files that don't have read permission
+        .filter_map(|e| e.ok())
+        .filter(|entry| entry.path().extension().is_some_and(|ext| ext == "proto"))
+        .filter(|entry| {
+            entry
+                .path()
+                .components()
+                .filter_map(|comp| comp.as_os_str().to_str())
+                .any(|comp| comp == version_str)
+        })
+        .filter(|entry| {
+            !entry
+                .path()
+                .components()
+                .filter_map(|comp| comp.as_os_str().to_str())
+                .any(|comp| ignore_modules.contains(&comp))
+        })
+        .map(|entry| entry.path().to_path_buf())
+        .collect()
+}