From c2d1b3628e948b52ad636dc0023fb19ae354e4be Mon Sep 17 00:00:00 2001
From: Riley Testut <riley@rileytestut.com>
Date: Tue, 14 Jan 2020 18:39:08 -0800
Subject: [PATCH] Adds InstalledApp.team relationship

---
 .../AltStore.xcdatamodeld/AltStore 2.xcdatamodel/contents | 8 +++++---
 AltStore/Model/InstalledApp.swift                         | 1 +
 AltStore/Model/Team.swift                                 | 1 +
 AltStore/Operations/InstallAppOperation.swift             | 5 +++++
 4 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/AltStore/Model/AltStore.xcdatamodeld/AltStore 2.xcdatamodel/contents b/AltStore/Model/AltStore.xcdatamodeld/AltStore 2.xcdatamodel/contents
index 6e401c9e..1f95d2e6 100644
--- a/AltStore/Model/AltStore.xcdatamodeld/AltStore 2.xcdatamodel/contents	
+++ b/AltStore/Model/AltStore.xcdatamodeld/AltStore 2.xcdatamodel/contents	
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
-<model type="com.apple.IDECoreDataModeler.DataModel" documentVersion="1.0" lastSavedToolsVersion="14492.1" systemVersion="18G95" minimumToolsVersion="Automatic" sourceLanguage="Swift" userDefinedModelVersionIdentifier="">
+<model type="com.apple.IDECoreDataModeler.DataModel" documentVersion="1.0" lastSavedToolsVersion="15702" systemVersion="19C57" minimumToolsVersion="Automatic" sourceLanguage="Swift" userDefinedModelVersionIdentifier="">
     <entity name="Account" representedClassName="Account" syncable="YES">
         <attribute name="appleID" attributeType="String" syncable="YES"/>
         <attribute name="firstName" attributeType="String" syncable="YES"/>
@@ -26,6 +26,7 @@
         <attribute name="resignedBundleIdentifier" attributeType="String" syncable="YES"/>
         <attribute name="version" attributeType="String" syncable="YES"/>
         <relationship name="storeApp" optional="YES" maxCount="1" deletionRule="Nullify" destinationEntity="StoreApp" inverseName="installedApp" inverseEntity="StoreApp" syncable="YES"/>
+        <relationship name="team" optional="YES" maxCount="1" deletionRule="Nullify" destinationEntity="Team" inverseName="installedApps" inverseEntity="Team"/>
         <uniquenessConstraints>
             <uniquenessConstraint>
                 <constraint value="bundleIdentifier"/>
@@ -117,6 +118,7 @@
         <attribute name="name" attributeType="String" syncable="YES"/>
         <attribute name="type" attributeType="Integer 16" defaultValueString="0" usesScalarValueType="YES" syncable="YES"/>
         <relationship name="account" optional="YES" maxCount="1" deletionRule="Nullify" destinationEntity="Account" inverseName="teams" inverseEntity="Account" syncable="YES"/>
+        <relationship name="installedApps" toMany="YES" deletionRule="Nullify" destinationEntity="InstalledApp" inverseName="team" inverseEntity="InstalledApp"/>
         <uniquenessConstraints>
             <uniquenessConstraint>
                 <constraint value="identifier"/>
@@ -126,12 +128,12 @@
     <elements>
         <element name="Account" positionX="-36" positionY="90" width="128" height="135"/>
         <element name="AppPermission" positionX="-45" positionY="90" width="128" height="90"/>
-        <element name="InstalledApp" positionX="-63" positionY="0" width="128" height="150"/>
+        <element name="InstalledApp" positionX="-63" positionY="0" width="128" height="193"/>
         <element name="NewsItem" positionX="-45" positionY="126" width="128" height="225"/>
         <element name="PatreonAccount" positionX="-45" positionY="117" width="128" height="105"/>
         <element name="RefreshAttempt" positionX="-45" positionY="117" width="128" height="105"/>
         <element name="Source" positionX="-45" positionY="99" width="128" height="120"/>
         <element name="StoreApp" positionX="-63" positionY="-18" width="128" height="330"/>
-        <element name="Team" positionX="-45" positionY="81" width="128" height="120"/>
+        <element name="Team" positionX="-45" positionY="81" width="128" height="133"/>
     </elements>
 </model>
\ No newline at end of file
diff --git a/AltStore/Model/InstalledApp.swift b/AltStore/Model/InstalledApp.swift
index 4acea713..96e8caca 100644
--- a/AltStore/Model/InstalledApp.swift
+++ b/AltStore/Model/InstalledApp.swift
@@ -25,6 +25,7 @@ class InstalledApp: NSManagedObject, Fetchable
     
     /* Relationships */
     @NSManaged var storeApp: StoreApp?
+    @NSManaged var team: Team?
     
     var isSideloaded: Bool {
         return self.storeApp == nil
diff --git a/AltStore/Model/Team.swift b/AltStore/Model/Team.swift
index e130fcf0..d8e1ad32 100644
--- a/AltStore/Model/Team.swift
+++ b/AltStore/Model/Team.swift
@@ -37,6 +37,7 @@ class Team: NSManagedObject, Fetchable
     
     /* Relationships */
     @NSManaged private(set) var account: Account!
+    @NSManaged var installedApps: Set<InstalledApp>
     
     var altTeam: ALTTeam?
     
diff --git a/AltStore/Operations/InstallAppOperation.swift b/AltStore/Operations/InstallAppOperation.swift
index d9820578..d653888c 100644
--- a/AltStore/Operations/InstallAppOperation.swift
+++ b/AltStore/Operations/InstallAppOperation.swift
@@ -66,6 +66,11 @@ class InstallAppOperation: ResultOperation<InstalledApp>
                 installedApp.expirationDate = profile.expirationDate
             }
             
+            if let team = Team.first(satisfying: NSPredicate(format: "%K == %@", #keyPath(Team.isActiveTeam), NSNumber(value: true)), in: backgroundContext)
+            {
+                installedApp.team = team
+            }
+            
             // Temporary directory and resigned .ipa no longer needed, so delete them now to ensure AltStore doesn't quit before we get the chance to.
             self.cleanUp()