Add some missing references. Add referrers to allow a remove an objec…

…t from them
phimage · Nov 11, 2022 · 608a29b · 608a29b
1 parent 19a06ab
commit 608a29b
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 5 deletions.
diff --git a/Sources/PBXBuildFile.swift b/Sources/PBXBuildFile.swift
@@ -17,11 +17,11 @@ public class PBXBuildFile: PBXProjectItem {
     }
 
     #if LAZY
-    public lazy var fileRef: PBXReference? = self.object(PBXKeys.fileRef)
+    public lazy var fileRef: PBXFileReference? = self.object(PBXKeys.fileRef)
     public lazy var productRef: XCSwiftPackageProductDependency? = self.object(PBXKeys.productRef)
     public lazy var settings: [String: Any]? = self.dictionary(PBXKeys.settings)
     #else
-    public var fileRef: PBXReference? { self.object(PBXKeys.fileRef) }
+    public var fileRef: PBXFileReference? { self.object(PBXKeys.fileRef) }
     public var productRef: XCSwiftPackageProductDependency? { self.object(PBXKeys.productRef) }
     public var settings: [String: Any]? { self.dictionary(PBXKeys.settings) }
     #endif

diff --git a/Sources/PBXObject.swift b/Sources/PBXObject.swift
@@ -18,6 +18,8 @@ public /* abstract */ class PBXObject {
     public var fields: PBXObject.Fields
     let objects: PBXObjectFactory
 
+    public var referrers: [PBXObject] = []
+
     public enum PBXKeys: PBXKey {
         case isa
     }
@@ -148,12 +150,19 @@ extension PBXObject {
         guard let objectKeys = fields[key] as? [String] else {
             return []
         }
-
-        return objectKeys.compactMap(objects.object)
+        let objects: [T] = objectKeys.compactMap(objects.object)
+        for object in objects {
+            object.addReferrers(self)
+        }
+        return objects
     }
 
     func object<T: PBXObject, R: RawRepresentable>(_ key: R) -> T? where R.RawValue == String {
-        return object(key.rawValue)
+        if let object: T = object(key.rawValue) {
+            object.addReferrers(self)
+            return object
+        }
+        return nil
     }
 
     func objects<T: PBXObject, R: RawRepresentable>(_ key: R) -> [T] where R.RawValue == String {
@@ -227,6 +236,37 @@ extension PBXObject {
     public func attach() {
         objects.add(self)
     }
+
+    public func addReferrers(_ object: PBXObject) {
+        self.referrers.append(object)
+    }
+
+    public func removeReferrers(_ object: PBXObject) {
+        self.referrers.removeAll(where: { $0.ref == object.ref})
+    }
+
+    public func removeFromReferrers() {
+        for referrer in self.referrers {
+            for (key, value) in referrer.fields {
+                if let value = value as? String {
+                    if value == self.ref {
+                        referrer.fields[key] = nil
+                    }
+                } else if var values = value as? [String] {
+                    for (index, value) in values.enumerated() where value == self.ref {
+                        values.remove(at: index)
+                        referrer.fields[key] = values
+                        break
+                    }
+                }
+            }
+        }
+    }
+
+    public func remove() {
+        unattach()
+        removeFromReferrers()
+    }
 }
 
 extension PBXObject: CustomStringConvertible {

diff --git a/Sources/PBXTarget.swift b/Sources/PBXTarget.swift
@@ -13,6 +13,8 @@ public /* abstract */ class PBXTarget: PBXProjectItem, PBXBuildConfigurationList
     public enum PBXKeys: PBXKey {
         case name
         case productName
+        case productReference
+        case productType
         case buildPhases
         case buildRules
         case buildConfigurationList
@@ -23,6 +25,8 @@ public /* abstract */ class PBXTarget: PBXProjectItem, PBXBuildConfigurationList
     #if LAZY
     public lazy var name: String = self.string(PBXKeys.name)
     public lazy var productName: String? = self.string(PBXKeys.productName)
+    public lazy var productReference: PBXFileReference? = self.object(PBXKeys.productReference)
+    public lazy var productType: String? = self.string(PBXKeys.productName)
     public lazy var buildPhases: [PBXBuildPhase] = self.objects(PBXKeys.buildPhases)
     public lazy var buildRules: [PBXBuildRule] = self.objects(PBXKeys.buildRules)
     public lazy var buildConfigurationList: XCConfigurationList? = self.object(PBXKeys.buildConfigurationList)
@@ -31,6 +35,8 @@ public /* abstract */ class PBXTarget: PBXProjectItem, PBXBuildConfigurationList
     #else
     public var name: String { self.string(PBXKeys.name) }
     public var productName: String? { self.string(PBXKeys.productName) }
+    public var productReference: PBXFileReference? { self.object(PBXKeys.productReference) }
+    public var productType: String? { self.string(PBXKeys.productName) }
     public var buildPhases: [PBXBuildPhase] { self.objects(PBXKeys.buildPhases) }
     public var buildRules: [PBXBuildRule] { self.objects(PBXKeys.buildRules) }
     public var buildConfigurationList: XCConfigurationList? { self.object(PBXKeys.buildConfigurationList) }

diff --git a/Sources/XCBuildConfiguration.swift b/Sources/XCBuildConfiguration.swift
@@ -12,12 +12,15 @@ public class XCBuildConfiguration: PBXBuildStyle {
 
     public enum PBXKeys: PBXKey {
         case name
+        case baseConfigurationReference
     }
 
     #if LAZY
     public lazy var name: String? = self.string(PBXKeys.name)
+    public var baseConfigurationReference: PBXFileReference? = self.object(PBXKeys.baseConfigurationReference)
     #else
     public var name: String? { self.string(PBXKeys.name) }
+    public var baseConfigurationReference: PBXFileReference? { self.object(PBXKeys.baseConfigurationReference) }
     #endif
 
     public override var comment: String? {