VideoModel.swift 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. /*****************************************************************************
  2. * VideoModel.swift
  3. *
  4. * Copyright © 2018 VLC authors and VideoLAN
  5. * Copyright © 2018 Videolabs
  6. *
  7. * Authors: Soomin Lee <bubu@mikan.io>
  8. *
  9. * Refer to the COPYING file of the official project for license.
  10. *****************************************************************************/
  11. class VideoModel: MediaModel {
  12. typealias MLType = VLCMLMedia
  13. var sortModel = SortModel([.alpha, .duration, .insertionDate, .releaseDate, .fileSize])
  14. var updateView: (() -> Void)?
  15. var files = [VLCMLMedia]()
  16. var cellType: BaseCollectionViewCell.Type { return MovieCollectionViewCell.self }
  17. var medialibrary: MediaLibraryService
  18. var indicatorName: String = NSLocalizedString("MOVIES", comment: "")
  19. required init(medialibrary: MediaLibraryService) {
  20. self.medialibrary = medialibrary
  21. medialibrary.addObserver(self)
  22. files = medialibrary.media(ofType: .video)
  23. }
  24. }
  25. // MARK: - Sort
  26. extension VideoModel {
  27. func sort(by criteria: VLCMLSortingCriteria, desc: Bool) {
  28. files = medialibrary.media(ofType: .video,
  29. sortingCriteria: criteria,
  30. desc: desc)
  31. sortModel.currentSort = criteria
  32. sortModel.desc = desc
  33. updateView?()
  34. }
  35. }
  36. // MARK: - MediaLibraryObserver
  37. extension VideoModel: MediaLibraryObserver {
  38. func medialibrary(_ medialibrary: MediaLibraryService, didAddVideos videos: [VLCMLMedia]) {
  39. videos.forEach({ append($0) })
  40. updateView?()
  41. }
  42. func medialibrary(_ medialibrary: MediaLibraryService, didModifyVideos videos: [VLCMLMedia]) {
  43. if !videos.isEmpty {
  44. files = swapModels(with: videos)
  45. updateView?()
  46. }
  47. }
  48. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteMediaWithIds ids: [NSNumber]) {
  49. files = files.filter() {
  50. for id in ids where $0.identifier() == id.int64Value {
  51. return false
  52. }
  53. return true
  54. }
  55. updateView?()
  56. }
  57. // MARK: - Thumbnail
  58. func medialibrary(_ medialibrary: MediaLibraryService,
  59. thumbnailReady media: VLCMLMedia,
  60. type: VLCMLThumbnailSizeType, success: Bool) {
  61. guard success else {
  62. return
  63. }
  64. files = swapModels(with: [media])
  65. updateView?()
  66. }
  67. }