Improve history, resume videos, mark watched videos (fixes #42)

Model/HistoryModel.swift

@@ -0,0 +1,73 @@
+import CoreData
+import CoreMedia
+import Foundation
+
+extension PlayerModel {
+    func historyVideo(_ id: String) -> Video? {
+        historyVideos.first { $0.videoID == id }
+    }
+
+    func loadHistoryVideoDetails(_ id: Video.ID) {
+        guard historyVideo(id).isNil else {
+            return
+        }
+
+        accounts.api.video(id).load().onSuccess { [weak self] response in
+            guard let video: Video = response.typedContent() else {
+                return
+            }
+
+            self?.historyVideos.append(video)
+        }
+    }
+
+    func updateWatch(finished: Bool = false) {
+        guard let id = currentVideo?.videoID else {
+            return
+        }
+
+        let time = player.currentTime().seconds
+
+        let watch: Watch!
+        let watchFetchRequest = Watch.fetchRequest()
+        watchFetchRequest.predicate = NSPredicate(format: "videoID = %@", id as String)
+
+        let results = try? context.fetch(watchFetchRequest)
+
+        if results?.isEmpty ?? true {
+            if time < 1 {
+                return
+            }
+            watch = Watch(context: context)
+            watch.videoID = id
+        } else {
+            watch = results?.first
+        }
+
+        if let seconds = playerItemDuration?.seconds {
+            watch.videoDuration = seconds
+        }
+
+        if finished {
+            watch.stoppedAt = watch.videoDuration
+        } else if time.isFinite, time > 0 {
+            watch.stoppedAt = time
+        }
+
+        watch.watchedAt = Date()
+
+        try? context.save()
+    }
+
+    func removeWatch(_ watch: Watch) {
+        context.delete(watch)
+        try? context.save()
+    }
+
+    func removeAllWatches() {
+        let watchesFetchRequest = NSFetchRequest<NSFetchRequestResult>(entityName: "Watch")
+        let deleteRequest = NSBatchDeleteRequest(fetchRequest: watchesFetchRequest)
+        _ = try? context.execute(deleteRequest)
+        _ = try? context.save()
+    }
+}

Model/PersistenceController.swift

@@ -0,0 +1,47 @@
+import CoreData
+
+struct PersistenceController {
+    static let shared = PersistenceController()
+
+    static var preview: PersistenceController = {
+        let result = PersistenceController(inMemory: true)
+        let viewContext = result.container.viewContext
+
+        do {
+            try viewContext.save()
+        } catch {
+            // Replace this implementation with code to handle the error appropriately.
+            // fatalError() causes the application to generate a crash log and terminate. You should not use this function in a shipping application, although it may be useful during development.
+            let nsError = error as NSError
+            fatalError("Unresolved error \(nsError), \(nsError.userInfo)")
+        }
+        return result
+    }()
+
+    let container: NSPersistentContainer
+
+    init(inMemory: Bool = false) {
+        container = NSPersistentContainer(name: "Yattee")
+        if inMemory {
+            container.persistentStoreDescriptions.first!.url = URL(fileURLWithPath: "/dev/null")
+        }
+        container.viewContext.automaticallyMergesChangesFromParent = true
+        container.viewContext.mergePolicy = NSMergeByPropertyObjectTrumpMergePolicy
+        container.loadPersistentStores { _, error in
+            if let error = error as NSError? {
+                // Replace this implementation with code to handle the error appropriately.
+                // fatalError() causes the application to generate a crash log and terminate. You should not use this function in a shipping application, although it may be useful during development.
+
+                /*
+                 Typical reasons for an error here include:
+                 * The parent directory does not exist, cannot be created, or disallows writing.
+                 * The persistent store is not accessible, due to permissions or data protection when the device is locked.
+                 * The device is out of space.
+                 * The store could not be migrated to the current model version.
+                 Check the error message to determine what the actual problem was.
+                 */
+                fatalError("Unresolved error \(error), \(error.userInfo)")
+            }
+        }
+    }
+}

Model/Player/PlayerModel.swift

@@ -1,4 +1,5 @@
 import AVKit
+import CoreData
 import Defaults
 import Foundation
 import Logging
@@ -27,8 +28,8 @@ final class PlayerModel: ObservableObject {
     @Published var streamSelection: Stream? { didSet { rebuildTVMenu() } }
 
     @Published var queue = [PlayerQueueItem]() { didSet { Defaults[.queue] = queue } }
-    @Published var currentItem: PlayerQueueItem! { didSet { Defaults[.lastPlayed] = currentItem } }
-    @Published var history = [PlayerQueueItem]() { didSet { Defaults[.history] = history } }
+    @Published var currentItem: PlayerQueueItem!
+    @Published var historyVideos = [Video]()
 
     @Published var preservedTime: CMTime?
 
@@ -47,6 +48,8 @@ final class PlayerModel: ObservableObject {
     var composition = AVMutableComposition()
     var loadedCompositionAssets = [AVMediaType]()
 
+    var context: NSManagedObjectContext = PersistenceController.shared.container.viewContext
+
     private var currentArtwork: MPMediaItemArtwork?
     private var frequentTimeObserver: Any?
     private var infrequentTimeObserver: Any?
@@ -131,7 +134,7 @@ final class PlayerModel: ObservableObject {
     }
 
     var live: Bool {
-        currentItem?.video?.live ?? false
+        currentVideo?.live ?? false
     }
 
     var playerItemDuration: CMTime? {
@@ -162,9 +165,17 @@ final class PlayerModel: ObservableObject {
         player.pause()
     }
 
-    func upgradeToStream(_ stream: Stream) {
-        if !self.stream.isNil, self.stream != stream {
-            playStream(stream, of: currentVideo!, preservingTime: true, upgrading: true)
+    func play(_ video: Video, at time: TimeInterval? = nil, inNavigationView: Bool = false) {
+        playNow(video, at: time)
+
+        guard !playingInPictureInPicture else {
+            return
+        }
+
+        if inNavigationView {
+            playerNavigationLinkActive = true
+        } else {
+            show()
         }
     }
 
@@ -205,6 +216,12 @@ final class PlayerModel: ObservableObject {
         }
     }
 
+    func upgradeToStream(_ stream: Stream) {
+        if !self.stream.isNil, self.stream != stream {
+            playStream(stream, of: currentVideo!, preservingTime: true, upgrading: true)
+        }
+    }
+
     private func handleAvailableStreamsChange() {
         rebuildTVMenu()
 
@@ -469,18 +486,20 @@ final class PlayerModel: ObservableObject {
     }
 
     @objc func itemDidPlayToEndTime() {
-        currentItem.playbackTime = playerItemDuration
+        prepareCurrentItemForHistory(finished: true)
 
         if queue.isEmpty {
             #if !os(macOS)
                 try? AVAudioSession.sharedInstance().setActive(false)
             #endif
-            addCurrentItemToHistory()
             resetQueue()
             #if os(tvOS)
-                controller?.dismiss(animated: true)
+                controller?.playerView.dismiss(animated: false) { [weak self] in
+                    self?.controller?.dismiss(animated: true)
+                }
+            #else
+                hide()
             #endif
-            hide()
         } else {
             advanceToNextItem()
         }
@@ -551,7 +570,7 @@ final class PlayerModel: ObservableObject {
             }
 
             self.timeObserverThrottle.execute {
-                self.updateCurrentItemIntervals()
+                self.updateWatch()
             }
         }
     }
@@ -581,26 +600,15 @@ final class PlayerModel: ObservableObject {
             #endif
 
             self.timeObserverThrottle.execute {
-                self.updateCurrentItemIntervals()
+                self.updateWatch()
             }
         }
     }
 
-    private func updateCurrentItemIntervals() {
-        currentItem?.playbackTime = player.currentTime()
-        currentItem?.videoDuration = player.currentItem?.asset.duration.seconds
-    }
-
     fileprivate func updateNowPlayingInfo() {
-        var duration: Int?
-        if !currentItem.video.live {
-            let itemDuration = currentItem.videoDuration ?? 0
-            duration = itemDuration.isFinite ? Int(itemDuration) : nil
-        }
         var nowPlayingInfo: [String: AnyObject] = [
             MPMediaItemPropertyTitle: currentItem.video.title as AnyObject,
             MPMediaItemPropertyArtist: currentItem.video.author as AnyObject,
-            MPMediaItemPropertyPlaybackDuration: duration as AnyObject,
             MPNowPlayingInfoPropertyIsLiveStream: currentItem.video.live as AnyObject,
             MPNowPlayingInfoPropertyElapsedPlaybackTime: player.currentTime().seconds as AnyObject,
             MPNowPlayingInfoPropertyPlaybackQueueCount: queue.count as AnyObject,
@@ -611,6 +619,15 @@ final class PlayerModel: ObservableObject {
             nowPlayingInfo[MPMediaItemPropertyArtwork] = currentArtwork as AnyObject
         }
 
+        if !currentItem.video.live {
+            let itemDuration = currentItem.videoDuration ?? 0
+            let duration = itemDuration.isFinite ? Int(itemDuration) : nil
+
+            if !duration.isNil {
+                nowPlayingInfo[MPMediaItemPropertyPlaybackDuration] = duration as AnyObject
+            }
+        }
+
         MPNowPlayingInfoCenter.default().nowPlayingInfo = nowPlayingInfo
     }
 
@@ -647,7 +664,7 @@ final class PlayerModel: ObservableObject {
             if let channel: Channel = response.typedContent() {
                 self?.channelWithDetails = channel
                 withAnimation {
-                    self?.currentItem.video.channel = channel
+                    self?.currentItem?.video.channel = channel
                 }
             }
         }
@@ -671,7 +688,7 @@ final class PlayerModel: ObservableObject {
     }
 
     func closeCurrentItem() {
-        addCurrentItemToHistory()
+        prepareCurrentItemForHistory()
         currentItem = nil
         player.replaceCurrentItem(with: nil)
     }

Model/Player/PlayerQueue.swift

@@ -29,11 +29,11 @@ extension PlayerModel {
     }
 
     func playNow(_ video: Video, at time: TimeInterval? = nil) {
-        if !playingInPictureInPicture || closePiPOnNavigation {
+        if playingInPictureInPicture, closePiPOnNavigation {
             closePiP()
         }
 
-        addCurrentItemToHistory()
+        prepareCurrentItemForHistory()
 
         enqueueVideo(video, prepending: true) { _, item in
             self.advanceToItem(item, at: time)
@@ -83,7 +83,7 @@ extension PlayerModel {
     }
 
     func advanceToNextItem() {
-        addCurrentItemToHistory()
+        prepareCurrentItemForHistory()
 
         if let nextItem = queue.first {
             advanceToItem(nextItem)
@@ -91,7 +91,7 @@ extension PlayerModel {
     }
 
     func advanceToItem(_ newItem: PlayerQueueItem, at time: TimeInterval? = nil) {
-        addCurrentItemToHistory()
+        prepareCurrentItemForHistory()
 
         remove(newItem)
 
@@ -148,20 +148,15 @@ extension PlayerModel {
         return item
     }
 
-    func addCurrentItemToHistory() {
-        if let item = currentItem, Defaults[.saveHistory] {
-            addItemToHistory(item)
+    func prepareCurrentItemForHistory(finished: Bool = false) {
+        if !currentItem.isNil, Defaults[.saveHistory] {
+            if let video = currentVideo, !historyVideos.contains(where: { $0 == video }) {
+                historyVideos.append(video)
+            }
+            updateWatch(finished: finished)
         }
     }
 
-    func addItemToHistory(_ item: PlayerQueueItem) {
-        if let index = history.firstIndex(where: { $0.video?.videoID == item.video?.videoID }) {
-            history.remove(at: index)
-        }
-
-        history.insert(currentItem, at: 0)
-    }
-
     func playHistory(_ item: PlayerQueueItem) {
         var time = item.playbackTime
 
@@ -172,67 +167,9 @@ extension PlayerModel {
         let newItem = enqueueVideo(item.video, atTime: time, prepending: true)
 
         advanceToItem(newItem!)
-
-        if let historyItemIndex = history.firstIndex(of: item) {
-            history.remove(at: historyItemIndex)
-        }
-    }
-
-    @discardableResult func removeHistory(_ item: PlayerQueueItem) -> PlayerQueueItem? {
-        if let index = history.firstIndex(where: { $0 == item }) {
-            return history.remove(at: index)
-        }
-
-        return nil
     }
 
     func removeQueueItems() {
         queue.removeAll()
     }
-
-    func removeHistoryItems() {
-        history.removeAll()
-    }
-
-    func loadHistoryDetails() {
-        guard !accounts.current.isNil else {
-            return
-        }
-
-        queue = Defaults[.queue]
-        queue.forEach { item in
-            accounts.api.loadDetails(item) { newItem in
-                if let index = self.queue.firstIndex(where: { $0.id == item.id }) {
-                    self.queue[index] = newItem
-                }
-            }
-        }
-
-        var savedHistory = Defaults[.history]
-
-        if let lastPlayed = Defaults[.lastPlayed] {
-            if let index = savedHistory.firstIndex(where: { $0.videoID == lastPlayed.videoID }) {
-                var updatedLastPlayed = savedHistory[index]
-
-                updatedLastPlayed.playbackTime = lastPlayed.playbackTime
-                updatedLastPlayed.videoDuration = lastPlayed.videoDuration
-
-                savedHistory.remove(at: index)
-                savedHistory.insert(updatedLastPlayed, at: 0)
-            } else {
-                savedHistory.insert(lastPlayed, at: 0)
-            }
-
-            Defaults[.lastPlayed] = nil
-        }
-
-        history = savedHistory
-        history.forEach { item in
-            accounts.api.loadDetails(item) { newItem in
-                if let index = self.history.firstIndex(where: { $0.id == item.id }) {
-                    self.history[index] = newItem
-                }
-            }
-        }
-    }
 }

Model/Player/PlayerQueueItem.swift

@@ -11,6 +11,15 @@ struct PlayerQueueItem: Hashable, Identifiable, Defaults.Serializable {
     var playbackTime: CMTime?
     var videoDuration: TimeInterval?
 
+    static func from(_ watch: Watch, video: Video? = nil) -> Self {
+        .init(
+            video,
+            videoID: watch.videoID,
+            playbackTime: CMTime.secondsInDefaultTimescale(watch.stoppedAt),
+            videoDuration: watch.videoDuration
+        )
+    }
+
     init(_ video: Video? = nil, videoID: Video.ID? = nil, playbackTime: CMTime? = nil, videoDuration: TimeInterval? = nil) {
         self.video = video
         self.videoID = videoID ?? video!.videoID

Model/Video.swift

@@ -1,6 +1,7 @@
 import Alamofire
 import AVKit
 import Foundation
+import SwiftUI
 import SwiftyJSON
 
 struct Video: Identifiable, Equatable, Hashable {
@@ -105,10 +106,24 @@ struct Video: Identifiable, Equatable, Hashable {
     }
 
     static func == (lhs: Video, rhs: Video) -> Bool {
-        lhs.id == rhs.id
+        let videoIDIsEqual = lhs.videoID == rhs.videoID
+
+        if !lhs.indexID.isNil, !rhs.indexID.isNil {
+            return videoIDIsEqual && lhs.indexID == rhs.indexID
+        }
+
+        return videoIDIsEqual
     }
 
     func hash(into hasher: inout Hasher) {
         hasher.combine(id)
     }
+
+    var watchFetchRequest: FetchRequest<Watch> {
+        FetchRequest<Watch>(
+            entity: Watch.entity(),
+            sortDescriptors: [],
+            predicate: NSPredicate(format: "videoID = %@", videoID)
+        )
+    }
 }

Model/Watch.swift

@@ -0,0 +1,47 @@
+import CoreData
+import Defaults
+import Foundation
+
+@objc(Watch)
+final class Watch: NSManagedObject, Identifiable {
+    @Default(.watchedThreshold) private var watchedThreshold
+}
+
+extension Watch {
+    @nonobjc class func fetchRequest() -> NSFetchRequest<Watch> {
+        NSFetchRequest<Watch>(entityName: "Watch")
+    }
+
+    @NSManaged var videoID: String
+    @NSManaged var videoDuration: Double
+
+    @NSManaged var watchedAt: Date?
+    @NSManaged var stoppedAt: Double
+
+    var progress: Double {
+        guard videoDuration.isFinite, !videoDuration.isZero else {
+            return 0
+        }
+
+        let progress = (stoppedAt / videoDuration) * 100
+        return min(max(progress, 0), 100)
+    }
+
+    var finished: Bool {
+        progress >= Double(watchedThreshold)
+    }
+
+    var watchedAtString: String? {
+        guard let watchedAt = watchedAt else {
+            return nil
+        }
+
+        if watchedAt.timeIntervalSinceNow < 5 {
+            return "just now"
+        }
+
+        let formatter = RelativeDateTimeFormatter()
+        formatter.unitsStyle = .full
+        return formatter.localizedString(for: watchedAt, relativeTo: Date())
+    }
+}

Model/Yattee.xcdatamodeld/Yattee.xcdatamodel/contents

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
+<model type="com.apple.IDECoreDataModeler.DataModel" documentVersion="1.0" lastSavedToolsVersion="19574" systemVersion="21D5025f" minimumToolsVersion="Automatic" sourceLanguage="Swift" userDefinedModelVersionIdentifier="1">
+    <entity name="Watch" representedClassName="Watch" syncable="YES">
+        <attribute name="stoppedAt" optional="YES" attributeType="Double" defaultValueString="0.0" usesScalarValueType="YES"/>
+        <attribute name="videoDuration" optional="YES" attributeType="Double" defaultValueString="0.0" usesScalarValueType="YES"/>
+        <attribute name="videoID" attributeType="String"/>
+        <attribute name="watchedAt" optional="YES" attributeType="Date" usesScalarValueType="NO"/>
+        <uniquenessConstraints>
+            <uniquenessConstraint>
+                <constraint value="videoID"/>
+            </uniquenessConstraint>
+        </uniquenessConstraints>
+    </entity>
+    <elements>
+        <element name="Watch" positionX="-63" positionY="-18" width="128" height="89"/>
+    </elements>
+</model>