yattee/Model/Stream.swift

import AVFoundation
import Defaults
import Foundation

// swiftlint:disable:next final_class
class Stream: Equatable, Hashable, Identifiable {
    enum Resolution: String, CaseIterable, Comparable, Defaults.Serializable {
        case hd4320p60
        case hd4320p
        case hd2160p60
        case hd2160p50
        case hd2160p48
        case hd2160p
        case hd1440p60
        case hd1440p50
        case hd1440p48
        case hd1440p
        case hd1080p60
        case hd1080p50
        case hd1080p48
        case hd1080p
        case hd720p60
        case hd720p50
        case hd720p48
        case hd720p
        case sd480p
        case sd360p
        case sd240p
        case sd144p
        case unknown

        var name: String {
            "\(height)p\(refreshRate != -1 ? ", \(refreshRate) fps" : "")"
        }

        var height: Int {
            if self == .unknown {
                return -1
            }

            let resolutionPart = rawValue.components(separatedBy: "p").first!
            return Int(resolutionPart.components(separatedBy: CharacterSet.decimalDigits.inverted).joined())!
        }

        var refreshRate: Int {
            if self == .unknown {
                return -1
            }

            let refreshRatePart = rawValue.components(separatedBy: "p")[1]
            return Int(refreshRatePart.components(separatedBy: CharacterSet.decimalDigits.inverted).joined()) ?? -1
        }

        static func from(resolution: String) -> Resolution {
            allCases.first { "\($0)".contains(resolution) } ?? .unknown
        }

        static func < (lhs: Resolution, rhs: Resolution) -> Bool {
            lhs.height < rhs.height
        }
    }

    enum Kind: String, Comparable {
        case stream, adaptive, hls

        private var sortOrder: Int {
            switch self {
            case .hls:
                return 0
            case .stream:
                return 1
            case .adaptive:
                return 2
            }
        }

        static func < (lhs: Kind, rhs: Kind) -> Bool {
            lhs.sortOrder < rhs.sortOrder
        }
    }

    enum Format: String, Comparable {
        case webm
        case avc1
        case av1
        case mp4
        case unknown

        private var sortOrder: Int {
            switch self {
            case .webm:
                return 0
            case .mp4:
                return 1
            case .avc1:
                return 2
            case .av1:
                return 3
            case .unknown:
                return 4
            }
        }

        static func < (lhs: Self, rhs: Self) -> Bool {
            lhs.sortOrder < rhs.sortOrder
        }

        static func from(_ string: String) -> Self {
            let lowercased = string.lowercased()

            if lowercased.contains("webm") {
                return .webm
            } else if lowercased.contains("avc1") {
                return .avc1
            } else if lowercased.contains("av01") {
                return .av1
            } else if lowercased.contains("mpeg_4") || lowercased.contains("mp4") {
                return .mp4
            } else {
                return .unknown
            }
        }
    }

    let id = UUID()

    var instance: Instance!
    var audioAsset: AVURLAsset!
    var videoAsset: AVURLAsset!
    var hlsURL: URL!

    var resolution: Resolution!
    var kind: Kind!
    var format: Format!

    var encoding: String!
    var videoFormat: String!

    init(
        instance: Instance? = nil,
        audioAsset: AVURLAsset? = nil,
        videoAsset: AVURLAsset? = nil,
        hlsURL: URL? = nil,
        resolution: Resolution? = nil,
        kind: Kind = .hls,
        encoding: String? = nil,
        videoFormat: String? = nil
    ) {
        self.instance = instance
        self.audioAsset = audioAsset
        self.videoAsset = videoAsset
        self.hlsURL = hlsURL
        self.resolution = resolution
        self.kind = kind
        self.encoding = encoding
        format = .from(videoFormat ?? "")
    }

    var quality: String {
        if resolution == .hd2160p {
            return "4K (2160p)"
        }

        return kind == .hls ? "adaptive (HLS)" : "\(resolution.name)\(kind == .stream ? " (\(kind.rawValue))" : "")"
    }

    var shortQuality: String {
        if resolution == .hd4320p60 || resolution == .hd4320p {
            return "8K"
        } else if resolution == .hd2160p60 ||
            resolution == .hd2160p50 ||
            resolution == .hd2160p48 ||
            resolution == .hd2160p
        {
            return "4K"
        } else if kind == .hls {
            return "HLS"
        } else {
            return resolution.name
        }
    }

    var description: String {
        let formatString = format == .unknown ? "" : " (\(format.rawValue))"
        return "\(quality)\(formatString) - \(instance?.description ?? "")"
    }

    var assets: [AVURLAsset] {
        [audioAsset, videoAsset]
    }

    var videoAssetContainsAudio: Bool {
        assets.dropFirst().allSatisfy { $0.url == assets.first!.url }
    }

    var singleAssetURL: URL? {
        if kind == .hls {
            return hlsURL
        } else if videoAssetContainsAudio {
            return videoAsset.url
        }

        return nil
    }

    static func == (lhs: Stream, rhs: Stream) -> Bool {
        lhs.id == rhs.id
    }

    func hash(into hasher: inout Hasher) {
        hasher.combine(videoAsset?.url)
        hasher.combine(audioAsset?.url)
        hasher.combine(hlsURL)
    }
}