feat(pbxproj): PBXProject helper methods

Author: kkharji

src/pbxproj/object/product_type.rs

@@ -1,7 +1,7 @@
 use tap::Pipe;
 
 /// Target Product Type
-#[derive(Default, Debug, Hash, PartialEq, Eq)]
+#[derive(Debug, Hash, PartialEq, Eq)]
 pub enum PBXProductType {
     /// Application
     Application,
@@ -62,10 +62,15 @@ pub enum PBXProductType {
     /// SystemExtension
     SystemExtension,
     /// None Identified
-    #[default]
     None,
 }
 
+impl Default for PBXProductType {
+    fn default() -> Self {
+        Self::None
+    }
+}
+
 impl PBXProductType {
     /// Return file extension for product type
     pub fn file_extension(&self) -> Option<&str> {

src/pbxproj/object/project.rs

@@ -1,3 +1,5 @@
+use tap::Pipe;
+
 use crate::pbxproj::*;
 use std::{cell::RefCell, collections::HashMap, rc::Rc};
 
@@ -32,7 +34,7 @@ pub struct PBXProject {
     /// The objects are a reference to a PBXTarget element.
     target_references: Vec<String>,
     /// Project references.
-    project_references: Vec<HashMap<String, String>>,
+    project_references: Option<Vec<HashMap<String, String>>>,
     /// The object is a reference to a PBXGroup element.
     products_group_reference: Option<String>,
     /// The object is a reference to a PBXGroup element.
@@ -64,7 +66,6 @@ impl PBXProject {
     }
 
     /// Get targets for given reference
-    /// TODO: Wrap target
     #[must_use]
     pub fn targets(&self) -> Vec<Rc<RefCell<PBXTarget>>> {
         let objects = if let Some(objects) = self.objects.upgrade() {
@@ -84,12 +85,111 @@ impl PBXProject {
 
     /// Returns the attributes of a given target.
     #[must_use]
-    fn get_attributes_for_target_reference(&self, target_reference: &str) -> Option<&PBXHashMap> {
+    pub fn get_attributes_for_target_reference(
+        &self,
+        target_reference: &str,
+    ) -> Option<&PBXHashMap> {
         self.target_attribute_references
             .as_ref()?
             .get(target_reference)?
             .as_object()
     }
+
+    /// Returns the attributes of a given target.
+    #[must_use]
+    pub fn get_attributes_for_target_reference_mut(
+        &mut self,
+        target_reference: &str,
+    ) -> Option<&mut PBXHashMap> {
+        self.target_attribute_references
+            .as_mut()?
+            .get_mut(target_reference)?
+            .as_object_mut()
+    }
+
+    /// Git build configuration list
+    pub fn build_configuration_list(&self) -> Option<Rc<RefCell<XCConfigurationList>>> {
+        self.objects
+            .upgrade()?
+            .borrow()
+            .get(&self.build_configuration_list_reference)?
+            .as_xc_configuration_list()?
+            .clone()
+            .pipe(Some)
+    }
+
+    /// Get project projects
+    pub fn projects(&self) -> Option<Vec<HashMap<&String, PBXObject>>> {
+        self.project_references
+            .as_ref()?
+            .iter()
+            .map(|v| {
+                v.iter()
+                    .map(|(k, v)| Some((k, self.objects.upgrade()?.borrow().get(v)?.clone())))
+                    .flatten()
+                    .collect::<HashMap<_, _>>()
+            })
+            .collect::<Vec<_>>()
+            .pipe(Some)
+    }
+
+    /// Get Project main group.
+    pub fn main_group(&self) -> Rc<RefCell<PBXGroup>> {
+        self.objects
+            .upgrade()
+            .expect("objects weak is valid")
+            .borrow()
+            .get(&self.main_group_reference)
+            .expect("PBXProject should contain mainGroup")
+            .as_pbx_group()
+            .expect("given reference point to PBXGroup")
+            .clone()
+    }
+
+    /// Products Group
+    pub fn products_group(&self) -> Option<Rc<RefCell<PBXGroup>>> {
+        let products_group = self.products_group_reference.as_ref()?;
+        self.objects
+            .upgrade()?
+            .borrow()
+            .get(products_group)?
+            .as_pbx_group()?
+            .clone()
+            .pipe(Some)
+    }
+
+    /// Get attributes for a given target reference
+    pub fn get_target_attributes(&mut self, target_reference: &str) -> Option<&PBXHashMap> {
+        let target_attributes = self.target_attribute_references.as_mut()?;
+        target_attributes.get(target_reference)?.as_object()
+    }
+
+    /// Sets the attributes for the given target.
+    pub fn set_target_attributes(&mut self, attributes: PBXHashMap, target_reference: &str) {
+        let target_attributes = self
+            .target_attribute_references
+            .get_or_insert(Default::default());
+        target_attributes.insert(target_reference.into(), attributes.into());
+    }
+
+    /// Remove attributes for a given target reference
+    pub fn remove_target_attributes(&mut self, target_reference: &str) -> Option<PBXHashMap> {
+        if let Some(target_attributes) = self.target_attribute_references.as_mut() {
+            target_attributes
+                .remove(target_reference)?
+                .into_object()
+                .ok()
+        } else {
+            None
+        }
+    }
+
+    /// Removes the all the target attributes
+    pub fn clear_all_target_attributes(&mut self) {
+        if let Some(target_attributes) = self.target_attribute_references.as_mut() {
+            target_attributes.clear();
+        }
+    }
 }
 
 #[test]
@@ -105,11 +205,22 @@ fn test_collections() {
         .find(|s| s.1.is_pbx_project())
         .map(|(_, o)| o.as_pbx_project().unwrap().clone())
         .unwrap();
+
     let project = project.borrow();
     let packages = project.packages();
     let targets = project.targets();
+    let build_configuration_list = project.build_configuration_list();
+    let projects = project.projects();
+    let main_group = project.main_group();
+    let products_group = project.products_group();
+
+    println!("Project: {:#?}", project);
     println!("Packages: {:#?}", packages);
     println!("Targets: {:#?}", targets);
+    println!("build_configuration_list: {:#?}", build_configuration_list);
+    println!("projects: {:#?}", projects);
+    println!("main_group: {:#?}", main_group);
+    println!("products_group: {:#?}", products_group);
 }
 
 impl PBXObjectExt for PBXProject {
@@ -156,21 +267,18 @@ impl PBXObjectExt for PBXProject {
                 .try_remove_value("buildConfigurationList")?
                 .try_into()?,
             target_references: value.try_remove_value("targets")?.try_into()?,
-            project_references: value
-                .remove_vec("projectReferences")
-                .map(|v| {
-                    v.0.into_iter()
-                        .map(|v| v.try_into_object())
-                        .flatten()
-                        .map(|v| {
-                            v.0.into_iter()
-                                .map(|(k, v)| anyhow::Ok((k, v.try_into_string()?)))
-                                .flatten()
-                                .collect()
-                        })
-                        .collect::<Vec<_>>()
-                })
-                .unwrap_or_default(),
+            project_references: value.remove_vec("projectReferences").map(|v| {
+                v.0.into_iter()
+                    .map(|v| v.try_into_object())
+                    .flatten()
+                    .map(|v| {
+                        v.0.into_iter()
+                            .map(|(k, v)| anyhow::Ok((k, v.try_into_string()?)))
+                            .flatten()
+                            .collect()
+                    })
+                    .collect::<Vec<_>>()
+            }),
             main_group_reference: value.try_remove_value("mainGroup")?.try_into()?,
             products_group_reference: value
                 .remove_value("productRefGroup")