feat(pbxproj): get file by name or path

Author: kkharji

src/pbxproj/object/collection.rs

@@ -86,6 +86,24 @@ impl PBXObjectCollection {
         self.get_fs_references(|fs_reference| fs_reference.is_group())
     }
 
+    /// Get PBXGroup with by name or path
+    pub fn get_group_by_name_or_path<'a, S: AsRef<str>>(
+        &'a self,
+        name_or_path: S,
+    ) -> Option<(String, Rc<RefCell<PBXFSReference>>)> {
+        let name = name_or_path.as_ref();
+        self.groups().into_iter().find(|(_, o)| {
+            let group = o.borrow();
+            if let Some(n) = group.name() {
+                return n == name;
+            } else if let Some(p) = group.path() {
+                return p == name;
+            } else {
+                false
+            }
+        })
+    }
+
     /// Get all PBXProject
     pub fn projects<'a>(&'a self) -> Vec<(String, Rc<RefCell<PBXProject>>)> {
         self.iter()

src/pbxproj/object/fs/mod.rs

@@ -124,6 +124,27 @@ impl PBXFSReference {
     pub fn parent(&self) -> Option<Rc<RefCell<Self>>> {
         self.parent.upgrade()
     }
+
+    /// Get File from the group
+    ///
+    /// NOTE: This will return None if self is file
+    pub fn get_file<S: AsRef<str>>(&self, name: S) -> Option<Rc<RefCell<PBXFSReference>>> {
+        let name = name.as_ref();
+        self.children().into_iter().find(|o| {
+            if !o.borrow().is_file() {
+                return false;
+            }
+            let file = o.borrow();
+
+            if let Some(n) = file.name() {
+                n == name
+            } else if let Some(p) = file.path() {
+                p == name
+            } else {
+                false
+            }
+        })
+    }
 }
 
 impl Eq for PBXFSReference {}
@@ -178,4 +199,17 @@ mod tests {
             main_group.children_references()
         )
     }
+    #[test]
+    fn get_file() {
+        use crate::pbxproj::test_demo_file;
+        let project = test_demo_file!(demo1);
+        let source_group = project
+            .objects()
+            .get_group_by_name_or_path("Source")
+            .unwrap()
+            .1;
+        let source_group = source_group.borrow();
+        let file = source_group.get_file("Log.swift");
+        assert!(file.is_some())
+    }
 }