vlc-ios/SharedSources/MediaLibraryModel/MediaModel.swift

/*****************************************************************************
 * MediaModel.swift
 *
 * Copyright © 2018 VLC authors and VideoLAN
 * Copyright © 2018 Videolabs
 *
 * Authors: Soomin Lee <bubu@mikan.io>
 *
 * Refer to the COPYING file of the official project for license.
 *****************************************************************************/

protocol MediaModel: MLBaseModel where MLType == VLCMLMedia { }

extension MediaModel {
    func append(_ item: VLCMLMedia) {
        if !files.contains { $0 == item } {
            files.append(item)
        }
    }

    func delete(_ items: [VLCMLObject]) {
        do {
            for case let media as VLCMLMedia in items {
                if let mainFile = media.mainFile() {
                    try FileManager.default.removeItem(atPath: mainFile.mrl.path)
                }
            }
            medialibrary.reload()
        }
        catch let error as NSError {
            assertionFailure("MediaModel: Delete failed: \(error.localizedDescription)")
        }
    }

    func createPlaylist(_ name: String, _ fileIndexes: Set<IndexPath>? = nil) {
        guard let playlist = medialibrary.createPlaylist(with: name) else {
            assertionFailure("MediaModel: createPlaylist: Failed to create a playlist.")
            return
        }

        guard let fileIndexes = fileIndexes else {
            return
        }

        for index in fileIndexes  where index.row < files.count {
            playlist.appendMedia(withIdentifier: files[index.row].identifier())
        }
    }
}

// MARK: - Helpers

extension VLCMLMedia {
    static func == (lhs: VLCMLMedia, rhs: VLCMLMedia) -> Bool {
        return lhs.identifier() == rhs.identifier()
    }
}

extension VLCMLMedia {
    @objc func mediaDuration() -> String {
        return String(format: "%@", VLCTime(int: Int32(duration())))
    }

    @objc func formatSize() -> String {
        return ByteCountFormatter.string(fromByteCount: Int64(mainFile()?.size() ?? 0),
                                         countStyle: .file)
    }

    func mediaProgress() -> Float {
        guard let string = metadata(of: .progress).str as NSString? else {
            return 0.0
        }
        return string.floatValue
    }

    func isNew() -> Bool {
        let integer = metadata(of: .seen).integer()
        return integer == 0
    }
}