SideStore/AltStoreCore/Model/Source.swift

//
//  Source.swift
//  AltStore
//
//  Created by Riley Testut on 7/30/19.
//  Copyright © 2019 Riley Testut. All rights reserved.
//

import CoreData
import UIKit

public extension Source
{
    static var altStoreIdentifier: String {
        let appVersion = Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String
        
        if appVersion != nil {
            if appVersion!.contains("beta") {
                return Bundle.Info.appbundleIdentifier + ".Beta"
            }
            if appVersion!.contains("nightly") {
                return Bundle.Info.appbundleIdentifier + ".Nightly"
            }
        }
        
        return Bundle.Info.appbundleIdentifier
    }
    
    static let altStoreSourceBaseURL = "https://apps.sidestore.io/"
    
    static var altStoreSourceURL: URL {
        let appVersion = Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String
        
        if appVersion != nil {
            if appVersion!.contains("beta") {
                return URL(string: altStoreSourceBaseURL + "beta")!
            }
            if appVersion!.contains("nightly") {
                return URL(string: altStoreSourceBaseURL + "nightly")!
            }
        }
        
        return URL(string: altStoreSourceBaseURL)!
    }
}

public struct AppPermissionFeed: Codable {
    let type: String // ALTAppPermissionType
    let usageDescription: String
       
    enum CodingKeys: String, CodingKey
    {
        case type
        case usageDescription
    }
}

public struct AppVersionFeed: Codable {
    /* Properties */
    let version: String
    let date: Date
    let localizedDescription: String?
    
    let downloadURL: URL
    let size: Int64
    
    enum CodingKeys: String, CodingKey
    {
        case version
        case date
        case localizedDescription
        case downloadURL
        case size
    }
}

public struct PlatformURLFeed: Codable {
    /* Properties */
    let platform: Platform
    let downloadURL: URL
    
    
    private enum CodingKeys: String, CodingKey
    {
        case platform
        case downloadURL
    }
}


public struct StoreAppFeed: Codable {
    let name: String
    let bundleIdentifier: String
    let subtitle: String?
    
    let developerName: String
    let localizedDescription: String
    let size: Int64
    
    let iconURL: URL
    let screenshotURLs: [URL]
    
    let version: String
    let versionDate: Date
    let versionDescription: String?
    let downloadURL: URL
    let platformURLs: [PlatformURLFeed]?
    
    let tintColor: String? // UIColor?
    let isBeta: Bool
    
    //    let source: Source?
    let appPermission: [AppPermissionFeed]
    let versions: [AppVersionFeed]
    
    enum CodingKeys: String, CodingKey
    {
        case bundleIdentifier
        case developerName
        case downloadURL
        case iconURL
        case isBeta = "beta"
        case localizedDescription
        case name
        case appPermission = "permissions"
        case platformURLs
        case screenshotURLs
        case size
        case subtitle
        case tintColor
        case version
        case versionDate
        case versionDescription
        case versions
    }
}

public struct NewsItemFeed: Codable {
    let identifier: String
    let date: Date
    
    let title: String
    let caption: String
    let tintColor: String //UIColor
    let notify: Bool
    
    let imageURL: URL?
    let externalURL: URL?
    
    let appID: String?
    
    private enum CodingKeys: String, CodingKey
    {
        case identifier
        case date
        case title
        case caption
        case tintColor
        case imageURL
        case externalURL = "url"
        case appID
        case notify
    }
}


public struct SourceJSON: Codable {
    let name: String
    let identifier: String
    let sourceURL: URL
    let userInfo: [String:String]? //[ALTSourceUserInfoKey:String]?
    let apps: [StoreAppFeed]
    let news: [NewsItemFeed]
    
    enum CodingKeys: String, CodingKey
    {
        case name
        case identifier
        case sourceURL
        case userInfo
        case apps
        case news
    }
    
}

@objc(Source)
public class Source: NSManagedObject, Fetchable, Decodable
{
    /* Properties */
    @NSManaged public var name: String
    @NSManaged public var identifier: String
    @NSManaged public var sourceURL: URL
    
    @NSManaged public var error: NSError?
    
    /* Non-Core Data Properties */
    public var userInfo: [ALTSourceUserInfoKey: String]?
    
    /* Relationships */
    @objc(apps) @NSManaged public private(set) var _apps: NSOrderedSet
    @objc(newsItems) @NSManaged public private(set) var _newsItems: NSOrderedSet
    
    @nonobjc public var apps: [StoreApp] {
        get {
            return self._apps.array as! [StoreApp]
        }
        set {
            self._apps = NSOrderedSet(array: newValue)
        }
    }
    
    @nonobjc public 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)
    }
    
    public 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: context)
        
        do
        {
            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)
            }
            self._apps = NSMutableOrderedSet(array: apps)
            
            let newsItems = try container.decodeIfPresent([NewsItem].self, forKey: .news) ?? []
            for (index, item) in newsItems.enumerated()
            {
                item.sourceIdentifier = self.identifier
                item.sortIndex = Int32(index)
            }
                                
            for newsItem in newsItems
            {
                guard let appID = newsItem.appID else { continue }
                
                if let storeApp = appsByID[appID]
                {
                    newsItem.storeApp = storeApp
                }
                else
                {
                    newsItem.storeApp = nil
                }
            }
            self._newsItems = NSMutableOrderedSet(array: newsItems)
        }
        catch
        {
            if let context = self.managedObjectContext
            {
                context.delete(self)
            }
            
            throw error
        }
    }
}

public 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 = "SideStore Offical"
        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
    }
}