yattee/Model/Video.swift

import Alamofire
import AVKit
import Foundation
import SwiftUI
import SwiftyJSON

struct Video: Identifiable, Equatable, Hashable {
    enum VideoID {
        static func isValid(_ id: Video.ID) -> Bool {
            isYouTube(id) || isPeerTube(id)
        }

        static func isYouTube(_ id: Video.ID) -> Bool {
            id.count == 11
        }

        static func isPeerTube(_ id: Video.ID) -> Bool {
            id.count == 36
        }
    }

    var instanceID: Instance.ID?
    var app: VideosApp
    var instanceURL: URL?

    var id: String
    var videoID: String
    var videoURL: URL?
    var title: String
    var thumbnails: [Thumbnail]
    var author: String
    var length: TimeInterval
    var published: String
    var views: Int
    var description: String?
    var genre: String?

    // index used when in the Playlist
    var indexID: String?

    var live: Bool
    var upcoming: Bool

    var streams = [Stream]()

    var publishedAt: Date?
    var likes: Int?
    var dislikes: Int?
    var keywords = [String]()

    var channel: Channel

    var related = [Video]()
    var chapters = [Chapter]()

    var captions = [Captions]()

    init(
        instanceID: Instance.ID? = nil,
        app: VideosApp,
        instanceURL: URL? = nil,
        id: String? = nil,
        videoID: String,
        videoURL: URL? = nil,
        title: String = "",
        author: String = "",
        length: TimeInterval = .zero,
        published: String = "",
        views: Int = 0,
        description: String? = nil,
        genre: String? = nil,
        channel: Channel = .init(id: "", name: ""),
        thumbnails: [Thumbnail] = [],
        indexID: String? = nil,
        live: Bool = false,
        upcoming: Bool = false,
        publishedAt: Date? = nil,
        likes: Int? = nil,
        dislikes: Int? = nil,
        keywords: [String] = [],
        streams: [Stream] = [],
        related: [Video] = [],
        chapters: [Chapter] = [],
        captions: [Captions] = []
    ) {
        self.instanceID = instanceID
        self.app = app
        self.instanceURL = instanceURL
        self.id = id ?? UUID().uuidString
        self.videoID = videoID
        self.videoURL = videoURL
        self.title = title
        self.author = author
        self.length = length
        self.published = published
        self.views = views
        self.description = description
        self.genre = genre
        self.channel = channel
        self.thumbnails = thumbnails
        self.indexID = indexID
        self.live = live
        self.upcoming = upcoming
        self.publishedAt = publishedAt
        self.likes = likes
        self.dislikes = dislikes
        self.keywords = keywords
        self.streams = streams
        self.related = related
        self.chapters = chapters
        self.captions = captions
    }

    static func local(_ url: URL) -> Video {
        Video(
            app: .local,
            videoID: url.absoluteString,
            streams: [.init(localURL: url)]
        )
    }

    var cacheKey: String {
        switch app {
        case .local:
            return videoID
        case .invidious:
            return "youtube-\(videoID)"
        case .piped:
            return "youtube-\(videoID)"
        case .peerTube:
            return "peertube-\(instanceURL?.absoluteString ?? "unknown-instance")-\(videoID)"
        }
    }

    var json: JSON {
        let dateFormatter = ISO8601DateFormatter()
        let publishedAt = self.publishedAt == nil ? "" : dateFormatter.string(from: self.publishedAt!)
        return [
            "instanceID": instanceID ?? "",
            "app": app.rawValue,
            "instanceURL": instanceURL?.absoluteString ?? "",
            "id": id,
            "videoID": videoID,
            "videoURL": videoURL?.absoluteString ?? "",
            "title": title,
            "author": author,
            "length": length,
            "published": published,
            "views": views,
            "description": description ?? "",
            "genre": genre ?? "",
            "channel": channel.json.object,
            "thumbnails": thumbnails.compactMap { $0.json.object },
            "indexID": indexID ?? "",
            "live": live,
            "upcoming": upcoming,
            "publishedAt": publishedAt
        ]
    }

    static func from(_ json: JSON) -> Self {
        let dateFormatter = ISO8601DateFormatter()

        return Video(
            instanceID: json["instanceID"].stringValue,
            app: .init(rawValue: json["app"].stringValue) ?? AccountsModel.shared.current.app ?? .local,
            instanceURL: URL(string: json["instanceURL"].stringValue) ?? AccountsModel.shared.current.instance.apiURL,
            id: json["id"].stringValue,
            videoID: json["videoID"].stringValue,
            videoURL: json["videoURL"].url,
            title: json["title"].stringValue,
            author: json["author"].stringValue,
            length: json["length"].doubleValue,
            published: json["published"].stringValue,
            views: json["views"].intValue,
            description: json["description"].string,
            genre: json["genre"].string,
            channel: Channel.from(json["channel"]),
            thumbnails: json["thumbnails"].arrayValue.compactMap { Thumbnail.from($0) },
            indexID: json["indexID"].stringValue,
            live: json["live"].boolValue,
            upcoming: json["upcoming"].boolValue,
            publishedAt: dateFormatter.date(from: json["publishedAt"].stringValue)
        )
    }

    var instance: Instance! {
        if let instance = InstancesModel.shared.find(instanceID) {
            return instance
        }

        if let url = instanceURL?.absoluteString {
            return Instance(app: app, id: instanceID, apiURLString: url, proxiesVideos: false)
        }

        return nil
    }

    var isLocal: Bool {
        !VideoID.isValid(videoID) && videoID != Self.fixtureID
    }

    var displayTitle: String {
        if isLocal {
            return localStreamFileName ?? localStream?.description ?? title
        }

        return title
    }

    var displayAuthor: String {
        if isLocal, localStreamIsRemoteURL {
            return remoteUrlHost ?? "Unknown"
        }

        return author
    }

    var publishedDate: String? {
        (published.isEmpty || published == "0 seconds ago") ? publishedAt?.timeIntervalSince1970.formattedAsRelativeTime() : published
    }

    var viewsCount: String? {
        views != 0 ? views.formattedAsAbbreviation() : nil
    }

    var likesCount: String? {
        guard let likes else {
            return nil
        }

        return likes.formattedAsAbbreviation()
    }

    var dislikesCount: String? {
        guard let dislikes else { return nil }

        return dislikes.formattedAsAbbreviation()
    }

    func thumbnailURL(quality: Thumbnail.Quality) -> URL? {
        thumbnails.first { $0.quality == quality }?.url
    }

    static func == (lhs: Video, rhs: Video) -> Bool {
        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)
        )
    }

    var localStream: Stream? {
        guard isLocal else { return nil }
        return streams.first
    }

    var localStreamImageSystemName: String {
        guard localStream != nil else { return "" }

        if localStreamIsDirectory {
            return "folder"
        }
        if localStreamIsFile {
            return "doc"
        }

        return "globe"
    }

    var localStreamIsFile: Bool {
        guard let localStream else { return false }
        return localStream.localURL.isFileURL
    }

    var localStreamIsRemoteURL: Bool {
        guard let localStream else { return false }
        return !localStream.localURL.isFileURL
    }

    var localStreamIsDirectory: Bool {
        guard let localStream else { return false }
        #if os(iOS)
            return DocumentsModel.shared.isDirectory(localStream.localURL)
        #else
            return false
        #endif
    }

    var remoteUrlHost: String? {
        localStreamURLComponents?.host
    }

    var localStreamFileName: String? {
        guard let path = localStream?.localURL?.lastPathComponent else { return nil }

        if let localStreamFileExtension {
            return String(path.dropLast(localStreamFileExtension.count + 1))
        }
        return String(path)
    }

    var localStreamFileExtension: String? {
        guard let path = localStreamURLComponents?.path else { return nil }
        return path.contains(".") ? path.components(separatedBy: ".").last?.uppercased() : nil
    }

    private var localStreamURLComponents: URLComponents? {
        guard let localStream else { return nil }
        return URLComponents(url: localStream.localURL, resolvingAgainstBaseURL: false)
    }
}