/***************************************************************************** * 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 sortModel = SortModel([.alpha, .duration, .insertionDate, .releaseDate, .fileSize]) var updateView: (() -> Void)? var files = [VLCMLMedia]() var cellType: BaseCollectionViewCell.Type { return MovieCollectionViewCell.self } var medialibrary: MediaLibraryService var indicatorName: String = NSLocalizedString("MOVIES", comment: "") required init(medialibrary: MediaLibraryService) { self.medialibrary = medialibrary medialibrary.addObserver(self) files = medialibrary.media(ofType: .video) } } // MARK: - Sort extension VideoModel { func sort(by criteria: VLCMLSortingCriteria, desc: Bool) { files = medialibrary.media(ofType: .video, sortingCriteria: criteria, desc: desc) sortModel.currentSort = criteria sortModel.desc = desc updateView?() } } // MARK: - MediaLibraryObserver extension VideoModel: MediaLibraryObserver { func medialibrary(_ medialibrary: MediaLibraryService, didAddVideos videos: [VLCMLMedia]) { videos.forEach({ append($0) }) updateView?() } func medialibrary(_ medialibrary: MediaLibraryService, didModifyVideos videos: [VLCMLMedia]) { if !videos.isEmpty { files = swapModels(with: videos) updateView?() } } func medialibrary(_ medialibrary: MediaLibraryService, didDeleteMediaWithIds ids: [NSNumber]) { files = files.filter() { for id in ids where $0.identifier() == id.int64Value { return false } return true } updateView?() } // MARK: - Thumbnail func medialibrary(_ medialibrary: MediaLibraryService, thumbnailReady media: VLCMLMedia, type: VLCMLThumbnailSizeType, success: Bool) { guard success else { return } files = swapModels(with: [media]) updateView?() } }