Course bars: Download videos to device and Select download quality bars

Core/Core/Data/Model/UserSettings.swift

@@ -7,13 +7,19 @@
 
 import Foundation
 
-public struct UserSettings: Codable {
+public struct UserSettings: Codable, Hashable {
     public var wifiOnly: Bool
     public var streamingQuality: StreamingQuality
-
-    public init(wifiOnly: Bool, streamingQuality: StreamingQuality) {
+    public var downloadQuality: DownloadQuality
+
+    public init(
+        wifiOnly: Bool,
+        streamingQuality: StreamingQuality,
+        downloadQuality: DownloadQuality
+    ) {
         self.wifiOnly = wifiOnly
         self.streamingQuality = streamingQuality
+        self.downloadQuality = downloadQuality
     }
 }
 
@@ -23,3 +29,10 @@ public enum StreamingQuality: Codable {
     case medium
     case high
 }
+
+public enum DownloadQuality: Codable, CaseIterable {
+    case auto
+    case low_360
+    case medium_540
+    case high_720
+}

Core/Core/Data/Persistence/CoreDataModel.xcdatamodeld/CoreDataModel.xcdatamodel/contents

@@ -1,8 +1,10 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
-<model type="com.apple.IDECoreDataModeler.DataModel" documentVersion="1.0" lastSavedToolsVersion="22222" systemVersion="22G91" minimumToolsVersion="Automatic" sourceLanguage="Swift" userDefinedModelVersionIdentifier="">
+<model type="com.apple.IDECoreDataModeler.DataModel" documentVersion="1.0" lastSavedToolsVersion="22522" systemVersion="22G91" minimumToolsVersion="Automatic" sourceLanguage="Swift" userDefinedModelVersionIdentifier="">
     <entity name="CDDownloadData" representedClassName="CDDownloadData" syncable="YES" codeGenerationType="class">
         <attribute name="courseId" optional="YES" attributeType="String"/>
+        <attribute name="displayName" optional="YES" attributeType="String"/>
         <attribute name="fileName" optional="YES" attributeType="String"/>
+        <attribute name="fileSize" optional="YES" attributeType="Integer 32" defaultValueString="0.0" usesScalarValueType="YES"/>
         <attribute name="id" optional="YES" attributeType="String"/>
         <attribute name="progress" optional="YES" attributeType="Double" defaultValueString="0.0" usesScalarValueType="YES"/>
         <attribute name="resumeData" optional="YES" attributeType="Binary"/>

Core/Core/Data/Persistence/CorePersistenceProtocol.swift

@@ -10,14 +10,15 @@ import Combine
 
 public protocol CorePersistenceProtocol {
     func publisher() -> AnyPublisher<Int, Never>
-    func getAllDownloadData() -> [DownloadData]
-    func addToDownloadQueue(blocks: [CourseBlock])
+    func addToDownloadQueue(blocks: [CourseBlock], quality: DownloadQuality)
     func getNextBlockForDownloading() -> DownloadData?
-    func getDownloadsForCourse(_ courseId: String) -> [DownloadData]
-    func downloadData(by blockId: String) -> DownloadData?
     func updateDownloadState(id: String, state: DownloadState, resumeData: Data?)
     func deleteDownloadData(id: String) throws
     func saveDownloadData(data: DownloadData)
+    func downloadData(by blockId: String) -> DownloadData?
+    func downloadData(by blockId: String, completion: @escaping (DownloadData?) -> Void)
+    func getAllDownloadData(completion: @escaping ([DownloadData]) -> Void)
+    func getDownloadsForCourse(_ courseId: String, completion: @escaping ([DownloadData]) -> Void)
 }
 
 public final class CoreBundle {

Core/Core/Domain/Model/CourseBlockModel.swift

@@ -8,6 +8,10 @@
 import Foundation
 
 public struct CourseStructure: Equatable {
+    public static func == (lhs: CourseStructure, rhs: CourseStructure) -> Bool {
+        return lhs.id == rhs.id
+    }
+
     public let id: String
     public let graded: Bool
     public let completion: Double
@@ -42,11 +46,24 @@ public struct CourseStructure: Equatable {
         self.media = media
         self.certificate = certificate
     }
-
-    public static func == (lhs: CourseStructure, rhs: CourseStructure) -> Bool {
-        return lhs.id == rhs.id
+
+    public func blocksTotalSizeInBytes(quality: DownloadQuality) -> Int {
+        childs.flatMap {
+            $0.childs.flatMap { $0.childs.flatMap { $0.childs.compactMap { $0 } } }
+        }
+        .filter { $0.isDownloadable }
+        .compactMap { $0.video(quality: quality)?.fileSize }
+        .reduce(.zero) { $0 + $1 }
     }
-
+
+    public func blocksTotalSizeInMb(quality: DownloadQuality) -> Double {
+        Double(blocksTotalSizeInBytes(quality: quality)) / 1024.0 / 1024.0
+    }
+
+    public func blocksTotalSizeInGb(quality: DownloadQuality) -> Double {
+        Double(blocksTotalSizeInBytes(quality: quality)) / 1024.0 / 1024.0 / 1024.0
+    }
+
 }
 
 public struct CourseChapter: Identifiable {
@@ -102,7 +119,11 @@ public struct CourseSequential: Identifiable {
     }
 }
 
-public struct CourseVertical {
+public struct CourseVertical: Identifiable, Hashable {
+    public func hash(into hasher: inout Hasher) {
+        hasher.combine(id)
+    }
+
     public let blockId: String
     public let id: String
     public let courseId: String
@@ -114,7 +135,7 @@ public struct CourseVertical {
     public var isDownloadable: Bool {
         return childs.first(where: { $0.isDownloadable }) != nil
     }
-    
+
     public init(
         blockId: String,
         id: String,
@@ -144,7 +165,11 @@ public struct SubtitleUrl: Equatable {
     }
 }
 
-public struct CourseBlock: Equatable {
+public struct CourseBlock: Hashable {
+    public func hash(into hasher: inout Hasher) {
+        hasher.combine(id)
+    }
+
     public let blockId: String
     public let id: String
     public let courseId: String
@@ -155,13 +180,43 @@ public struct CourseBlock: Equatable {
     public let displayName: String
     public let studentUrl: String
     public let subtitles: [SubtitleUrl]?
-    public let videoUrl: String?
-    public let youTubeUrl: String?
-
+    public let fallback: CourseBlockVideo?
+    public let desktopMP4: CourseBlockVideo?
+    public let mobileHigh: CourseBlockVideo?
+    public let mobileLow: CourseBlockVideo?
+    public let hls: CourseBlockVideo?
+    public let youTube: CourseBlockVideo?
+
     public var isDownloadable: Bool {
-        return videoUrl != nil
+        [hls, desktopMP4, mobileHigh, mobileLow, fallback]
+            .contains { $0?.isDownloadable == true }
     }
-
+
+    public func video(quality: DownloadQuality) -> CourseBlockVideo? {
+        switch quality {
+        case .auto:
+            [mobileLow, mobileHigh, desktopMP4, fallback, hls]
+                .first(where: { $0?.isDownloadable == true })?
+                .flatMap { $0 }
+        case .high_720:
+            [desktopMP4, mobileHigh, mobileLow, fallback, hls]
+                .first(where: { $0?.isDownloadable == true })?
+                .flatMap { $0 }
+        case .medium_540:
+            [mobileHigh, mobileLow, desktopMP4, fallback, hls]
+                .first(where: { $0?.isDownloadable == true })?
+                .flatMap { $0 }
+        case .low_360:
+            [mobileLow, mobileHigh, desktopMP4, fallback, hls]
+                .first(where: { $0?.isDownloadable == true })?
+                .flatMap { $0 }
+        }
+    }
+
+    public var youTubeUrl: String? {
+        youTube?.url
+    }
+
     public init(
         blockId: String,
         id: String,
@@ -173,8 +228,12 @@ public struct CourseBlock: Equatable {
         displayName: String,
         studentUrl: String,
         subtitles: [SubtitleUrl]? = nil,
-        videoUrl: String? = nil,
-        youTubeUrl: String? = nil
+        fallback: CourseBlockVideo?,
+        youTube: CourseBlockVideo?,
+        desktopMP4: CourseBlockVideo?,
+        mobileHigh: CourseBlockVideo?,
+        mobileLow: CourseBlockVideo?,
+        hls: CourseBlockVideo?
     ) {
         self.blockId = blockId
         self.id = id
@@ -186,7 +245,31 @@ public struct CourseBlock: Equatable {
         self.displayName = displayName
         self.studentUrl = studentUrl
         self.subtitles = subtitles
-        self.videoUrl = videoUrl
-        self.youTubeUrl = youTubeUrl
+        self.fallback = fallback
+        self.youTube = youTube
+        self.desktopMP4 = desktopMP4
+        self.mobileHigh = mobileHigh
+        self.mobileLow = mobileLow
+        self.hls = hls
+    }
+}
+
+public struct CourseBlockVideo: Equatable {
+    public let url: String?
+    public let fileSize: Int?
+    public let streamPriority: Int?
+
+    public init(url: String?, fileSize: Int?, streamPriority: Int?) {
+        self.url = url
+        self.fileSize = fileSize
+        self.streamPriority = streamPriority
+    }
+
+    public var isVideoURL: Bool {
+        [".mp4", ".m3u8"].contains(where: { url?.contains($0) == true })
+    }
+
+    public var isDownloadable: Bool {
+        [".mp4"].contains(where: { url?.contains($0) == true })
     }
 }