yattee/Model/ContentItem.swift

import Foundation

struct ContentItem: Identifiable {
    enum ContentType: String {
        case video, playlist, channel, placeholder

        private var sortOrder: Int {
            switch self {
            case .channel:
                return 1
            case .playlist:
                return 2
            default:
                return 3
            }
        }

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

    static var placeholders: [Self] {
        (0 ..< 9).map { i in .init(id: String(i)) }
    }

    var video: Video!
    var playlist: ChannelPlaylist!
    var channel: Channel!

    var id: String = UUID().uuidString

    static func array(of videos: [Video]) -> [Self] {
        videos.map { Self(video: $0) }
    }

    static func array(of playlists: [ChannelPlaylist]) -> [Self] {
        playlists.map { Self(playlist: $0) }
    }

    static func array(of channels: [Channel]) -> [Self] {
        channels.map { Self(channel: $0) }
    }

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

    var contentType: ContentType {
        video.isNil ? (channel.isNil ? (playlist.isNil ? .placeholder : .playlist) : .channel) : .video
    }

    var cacheKey: String {
        switch contentType {
        case .video:
            return video.cacheKey
        case .playlist:
            return playlist.cacheKey
        case .channel:
            return channel.cacheKey
        case .placeholder:
            return id
        }
    }
}