VideoModel.swift 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  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, thumbnailReady media: VLCMLMedia) {
  59. for (index, file) in files.enumerated() where file == media {
  60. files[index] = media
  61. break
  62. }
  63. updateView?()
  64. }
  65. func medialibraryDidStartRescan() {
  66. files.removeAll()
  67. }
  68. }