source-code

source-code/ALTs/AltStore/Model/Source.swift

@@ -0,0 +1,162 @@
+//
+//  Source.swift
+//  AltStore
+//
+//  Created by Riley Testut on 7/30/19.
+//  Copyright © 2019 Riley Testut. All rights reserved.
+//
+
+import CoreData
+
+extension Source
+{
+    #if ALPHA
+    static let altStoreIdentifier = "com.rileytestut.AltStore.Alpha"
+    #else
+    static let altStoreIdentifier = "com.rileytestut.AltStore"
+    #endif
+    
+    #if STAGING
+    
+    #if ALPHA
+    static let altStoreSourceURL = URL(string: "https://f000.backblazeb2.com/file/altstore-staging/sources/alpha/apps-alpha-staging.json")!
+    #else
+    static let altStoreSourceURL = URL(string: "https://f000.backblazeb2.com/file/altstore-staging/apps-staging.json")!
+    #endif
+    
+    #else
+    
+    #if ALPHA
+    static let altStoreSourceURL = URL(string: "https://alpha.altstore.io/")!
+    #else
+    static let altStoreSourceURL = URL(string: "https://apps.altstore.io/")!
+    #endif
+    
+    #endif
+}
+
+@objc(Source)
+class Source: NSManagedObject, Fetchable, Decodable
+{
+    /* Properties */
+    @NSManaged var name: String
+    @NSManaged var identifier: String
+    @NSManaged var sourceURL: URL
+    
+    /* Non-Core Data Properties */
+    var userInfo: [ALTSourceUserInfoKey: String]?
+    
+    /* Relationships */
+    @objc(apps) @NSManaged private(set) var _apps: NSOrderedSet
+    @objc(newsItems) @NSManaged private(set) var _newsItems: NSOrderedSet
+    
+    @nonobjc var apps: [StoreApp] {
+        get {
+            return self._apps.array as! [StoreApp]
+        }
+        set {
+            self._apps = NSOrderedSet(array: newValue)
+        }
+    }
+    
+    @nonobjc var newsItems: [NewsItem] {
+        get {
+            return self._newsItems.array as! [NewsItem]
+        }
+        set {
+            self._newsItems = NSOrderedSet(array: newValue)
+        }
+    }
+    
+    private enum CodingKeys: String, CodingKey
+    {
+        case name
+        case identifier
+        case sourceURL
+        case userInfo
+        case apps
+        case news
+    }
+    
+    private override init(entity: NSEntityDescription, insertInto context: NSManagedObjectContext?)
+    {
+        super.init(entity: entity, insertInto: context)
+    }
+    
+    required init(from decoder: Decoder) throws
+    {
+        guard let context = decoder.managedObjectContext else { preconditionFailure("Decoder must have non-nil NSManagedObjectContext.") }
+        guard let sourceURL = decoder.sourceURL else { preconditionFailure("Decoder must have non-nil sourceURL.") }
+        
+        super.init(entity: Source.entity(), insertInto: nil)
+        
+        self.sourceURL = sourceURL
+        
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        self.name = try container.decode(String.self, forKey: .name)
+        self.identifier = try container.decode(String.self, forKey: .identifier)
+        
+        let userInfo = try container.decodeIfPresent([String: String].self, forKey: .userInfo)
+        self.userInfo = userInfo?.reduce(into: [:]) { $0[ALTSourceUserInfoKey($1.key)] = $1.value }
+        
+        let apps = try container.decodeIfPresent([StoreApp].self, forKey: .apps) ?? []
+        let appsByID = Dictionary(apps.map { ($0.bundleIdentifier, $0) }, uniquingKeysWith: { (a, b) in return a })
+        
+        for (index, app) in apps.enumerated()
+        {
+            app.sourceIdentifier = self.identifier
+            app.sortIndex = Int32(index)
+        }
+        
+        let newsItems = try container.decodeIfPresent([NewsItem].self, forKey: .news) ?? []
+        for (index, item) in newsItems.enumerated()
+        {
+            item.sourceIdentifier = self.identifier
+            item.sortIndex = Int32(index)
+        }
+        
+        context.insert(self)
+                
+        for newsItem in newsItems
+        {            
+            guard let appID = newsItem.appID else { continue }
+            
+            if let storeApp = appsByID[appID]
+            {
+                newsItem.storeApp = storeApp
+            }
+            else
+            {
+                newsItem.storeApp = nil
+            }
+        }
+        
+        // Must assign after we're inserted into context.
+        self._apps = NSMutableOrderedSet(array: apps)
+        self._newsItems = NSMutableOrderedSet(array: newsItems)
+    }
+}
+
+extension Source
+{
+    @nonobjc class func fetchRequest() -> NSFetchRequest<Source>
+    {
+        return NSFetchRequest<Source>(entityName: "Source")
+    }
+    
+    class func makeAltStoreSource(in context: NSManagedObjectContext) -> Source
+    {
+        let source = Source(context: context)
+        source.name = "AltStore"
+        source.identifier = Source.altStoreIdentifier
+        source.sourceURL = Source.altStoreSourceURL
+        
+        return source
+    }
+    
+    class func fetchAltStoreSource(in context: NSManagedObjectContext) -> Source?
+    {
+        let source = Source.first(satisfying: NSPredicate(format: "%K == %@", #keyPath(Source.identifier), Source.altStoreIdentifier), in: context)
+        return source
+    }
+}