TrackModel.swift 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. /*****************************************************************************
  2. * TrackModel.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 TrackModel: MediaModel {
  12. typealias MLType = VLCMLMedia
  13. var sortModel = SortModel([.alpha, .album, .duration, .fileSize])
  14. var updateView: (() -> Void)?
  15. var files = [VLCMLMedia]()
  16. var cellType: BaseCollectionViewCell.Type { return MediaCollectionViewCell.self }
  17. var medialibrary: MediaLibraryService
  18. var indicatorName: String = NSLocalizedString("SONGS", comment: "")
  19. required init(medialibrary: MediaLibraryService) {
  20. self.medialibrary = medialibrary
  21. medialibrary.addObserver(self)
  22. files = medialibrary.media(ofType: .audio)
  23. }
  24. }
  25. // MARK: - Sort
  26. extension TrackModel {
  27. func sort(by criteria: VLCMLSortingCriteria, desc: Bool) {
  28. // FIXME: Currently if sorted by name, the files are sorted by filename but displaying title
  29. files = medialibrary.media(ofType: .audio,
  30. sortingCriteria: criteria,
  31. desc: desc)
  32. sortModel.currentSort = criteria
  33. sortModel.desc = desc
  34. updateView?()
  35. }
  36. }
  37. // MARK: - MediaLibraryObserver
  38. extension TrackModel: MediaLibraryObserver {
  39. func medialibrary(_ medialibrary: MediaLibraryService, didAddTracks tracks: [VLCMLMedia]) {
  40. tracks.forEach({ append($0) })
  41. updateView?()
  42. }
  43. func medialibrary(_ medialibrary: MediaLibraryService, didModifyTracks tracks: [VLCMLMedia]) {
  44. if !tracks.isEmpty {
  45. files = swapModels(with: tracks)
  46. updateView?()
  47. }
  48. }
  49. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteMediaWithIds ids: [NSNumber]) {
  50. files = files.filter() {
  51. for id in ids where $0.identifier() == id.int64Value {
  52. return false
  53. }
  54. return true
  55. }
  56. updateView?()
  57. }
  58. }