/***************************************************************************** * VideoModel.swift * * Copyright © 2018 VLC authors and VideoLAN * Copyright © 2018 Videolabs * * Authors: Soomin Lee * * Refer to the COPYING file of the official project for license. *****************************************************************************/ class VideoModel: MediaModel { typealias MLType = VLCMLMedia var updateView: (() -> Void)? var files = [VLCMLMedia]() var cellType: BaseCollectionViewCell.Type { return MovieCollectionViewCell.self } var medialibrary: VLCMediaLibraryManager var indicatorName: String = NSLocalizedString("MOVIES", comment: "") required init(medialibrary: VLCMediaLibraryManager) { self.medialibrary = medialibrary medialibrary.addObserver(self) files = medialibrary.media(ofType: .video) medialibrary.requestThumbnail(for: files) } } // MARK: - Edit extension VideoModel: EditableMLModel { func editCellType() -> BaseCollectionViewCell.Type { return MediaEditCell.self } } // MARK: - Sort extension VideoModel { func sort(by criteria: VLCMLSortingCriteria) { files = medialibrary.media(ofType: .video, sortingCriteria: criteria, desc: false) updateView?() } } // MARK: - MediaLibraryObserver extension VideoModel: MediaLibraryObserver { func medialibrary(_ medialibrary: VLCMediaLibraryManager, didAddVideos videos: [VLCMLMedia]) { videos.forEach({ append($0) }) updateView?() } func medialibrary(_ medialibrary: VLCMediaLibraryManager, didDeleteMediaWithIds ids: [NSNumber]) { files = files.filter() { for id in ids where $0.identifier() == id.int64Value { return false } return true } updateView?() } } // MARK: MediaLibraryObserver - Thumbnail extension VideoModel { func medialibrary(_ medialibrary: VLCMediaLibraryManager, thumbnailReady media: VLCMLMedia) { for (index, file) in files.enumerated() { if file == media { files[index] = media break } } updateView?() } }