Add APIs for getting files in a project folder

1 files changed, 13 insertions, 4 deletions

diff --git a/rust/src/project/folder.rs b/rust/src/project/folder.rs
index 1ab33bc8..e14b84a8 100644
--- a/rust/src/project/folder.rs
+++ b/rust/src/project/folder.rs
@@ -1,12 +1,14 @@
 use crate::progress::{NoProgressCallback, ProgressCallback};
+use crate::project::file::ProjectFile;
 use crate::project::Project;
-use crate::rc::{CoreArrayProvider, CoreArrayProviderInner, Guard, Ref, RefCountable};
+use crate::rc::{Array, CoreArrayProvider, CoreArrayProviderInner, Guard, Ref, RefCountable};
 use crate::string::{BnString, IntoCStr};
 use binaryninjacore_sys::{
     BNFreeProjectFolder, BNFreeProjectFolderList, BNNewProjectFolderReference, BNProjectFolder,
-    BNProjectFolderExport, BNProjectFolderGetDescription, BNProjectFolderGetId,
-    BNProjectFolderGetName, BNProjectFolderGetParent, BNProjectFolderGetProject,
-    BNProjectFolderSetDescription, BNProjectFolderSetName, BNProjectFolderSetParent,
+    BNProjectFolderExport, BNProjectFolderGetDescription, BNProjectFolderGetFiles,
+    BNProjectFolderGetId, BNProjectFolderGetName, BNProjectFolderGetParent,
+    BNProjectFolderGetProject, BNProjectFolderSetDescription, BNProjectFolderSetName,
+    BNProjectFolderSetParent,
 };
 use std::ffi::c_void;
 use std::fmt::Debug;
@@ -100,6 +102,13 @@ impl ProjectFolder {
             )
         }
     }
+
+    /// Get the files contained in this folder
+    pub fn files(&self) -> Array<ProjectFile> {
+        let mut count = 0;
+        let result = unsafe { BNProjectFolderGetFiles(self.handle.as_ptr(), &mut count) };
+        unsafe { Array::new(result, count, ()) }
+    }
 }
 
 impl PartialEq for ProjectFolder {