Add fs_wrapper::has_extension and get_files_extensions helper functions in run-make-support

Author: GuillaumeGomez

src/tools/run-make-support/src/fs_wrapper.rs

@@ -1,3 +1,4 @@
+use std::ffi::OsStr;
 use std::fs;
 use std::path::Path;
 
@@ -111,3 +112,9 @@ pub fn set_permissions<P: AsRef<Path>>(path: P, perm: fs::Permissions) {
         path.as_ref().display()
     ));
 }
+
+/// Checks if the given file path has the expected extension.
+#[track_caller]
+pub fn has_extension<P: AsRef<Path>, S: AsRef<OsStr>>(path: P, extension: S) -> bool {
+    path.as_ref().extension().is_some_and(|ext| ext == extension.as_ref())
+}

src/tools/run-make-support/src/lib.rs

@@ -15,7 +15,7 @@ pub mod rustc;
 pub mod rustdoc;
 
 use std::env;
-use std::ffi::OsString;
+use std::ffi::{OsStr, OsString};
 use std::fs;
 use std::io;
 use std::panic;
@@ -401,12 +401,26 @@ pub fn recursive_diff(dir1: impl AsRef<Path>, dir2: impl AsRef<Path>) {
     });
 }
 
-pub fn read_dir<F: FnMut(&Path)>(dir: impl AsRef<Path>, mut callback: F) {
+pub fn read_dir<F: FnMut(&Path), P: AsRef<Path>>(dir: P, mut callback: F) {
     for entry in fs_wrapper::read_dir(dir) {
         callback(&entry.unwrap().path());
     }
 }
 
+pub fn get_files_with_extension<P: AsRef<Path>, S: AsRef<OsStr>>(
+    dir: P,
+    extension: S,
+) -> Vec<PathBuf> {
+    let mut files = Vec::new();
+    let extension = extension.as_ref();
+    read_dir(dir, |path| {
+        if path.is_file() && fs_wrapper::has_extension(path, extension) {
+            files.push(path.to_path_buf());
+        }
+    });
+    files
+}
+
 /// Check that `actual` is equal to `expected`. Panic otherwise.
 #[track_caller]
 pub fn assert_equals<S1: AsRef<str>, S2: AsRef<str>>(actual: S1, expected: S2) {