AlbumModel.swift 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. /*****************************************************************************
  2. * AlbumModel.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 AlbumModel: MLBaseModel {
  12. typealias MLType = VLCMLAlbum
  13. var sortModel = SortModel([.alpha, .duration, .releaseDate, .trackNumber])
  14. var updateView: (() -> Void)?
  15. var files = [VLCMLAlbum]()
  16. var cellType: BaseCollectionViewCell.Type { return MediaCollectionViewCell.self }
  17. var medialibrary: MediaLibraryService
  18. var indicatorName: String = NSLocalizedString("ALBUMS", comment: "")
  19. required init(medialibrary: MediaLibraryService) {
  20. self.medialibrary = medialibrary
  21. medialibrary.addObserver(self)
  22. files = medialibrary.albums()
  23. }
  24. func append(_ item: VLCMLAlbum) {
  25. files.append(item)
  26. }
  27. func delete(_ items: [VLCMLObject]) {
  28. preconditionFailure("AlbumModel: Albums can not be deleted, they disappear when their last title got deleted")
  29. }
  30. }
  31. // MARK: - Sort
  32. extension AlbumModel {
  33. func sort(by criteria: VLCMLSortingCriteria) {
  34. files = medialibrary.albums(sortingCriteria: criteria)
  35. sortModel.currentSort = criteria
  36. updateView?()
  37. }
  38. }
  39. // MARK: - Edit
  40. extension AlbumModel: EditableMLModel {
  41. func editCellType() -> BaseCollectionViewCell.Type {
  42. return MediaEditCell.self
  43. }
  44. }
  45. // MARK: - MediaLibraryObserver
  46. extension AlbumModel: MediaLibraryObserver {
  47. func medialibrary(_ medialibrary: MediaLibraryService, didAddAlbums albums: [VLCMLAlbum]) {
  48. albums.forEach({ append($0) })
  49. updateView?()
  50. }
  51. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteAlbumsWithIds albumsIds: [NSNumber]) {
  52. files.removeAll {
  53. albumsIds.contains(NSNumber(value: $0.identifier()))
  54. }
  55. updateView?()
  56. }
  57. }
  58. extension VLCMLAlbum: MediaCollectionModel {
  59. func sortModel() -> SortModel? {
  60. return nil
  61. }
  62. func files() -> [VLCMLMedia]? {
  63. return tracks
  64. }
  65. func title() -> String? {
  66. return title
  67. }
  68. }