AlbumModel.swift 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  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: AudioCollectionModel {
  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. }
  28. // MARK: - Sort
  29. extension AlbumModel {
  30. func sort(by criteria: VLCMLSortingCriteria, desc: Bool) {
  31. files = medialibrary.albums(sortingCriteria: criteria, desc: desc)
  32. sortModel.currentSort = criteria
  33. sortModel.desc = desc
  34. updateView?()
  35. }
  36. }
  37. // MARK: - Search
  38. extension VLCMLAlbum: SearchableMLModel {
  39. func contains(_ searchString: String) -> Bool {
  40. var matches = false
  41. matches = matches || title.lowercased().contains(searchString)
  42. matches = matches || String(releaseYear()).lowercased().contains(searchString)
  43. matches = matches || shortSummary.lowercased().contains(searchString)
  44. matches = matches || albumArtist?.contains(searchString) ?? false
  45. matches = matches || tracks?.filter({ $0.contains(searchString)}).isEmpty == false
  46. return matches
  47. }
  48. }
  49. extension VLCMLAlbumTrack: SearchableMLModel {
  50. func contains(_ searchString: String) -> Bool {
  51. var matches = false
  52. matches = matches || artist?.contains(searchString) ?? false
  53. matches = matches || genre?.contains(searchString) ?? false
  54. matches = matches || album?.contains(searchString) ?? false
  55. return matches
  56. }
  57. }
  58. // MARK: - MediaLibraryObserver
  59. extension AlbumModel: MediaLibraryObserver {
  60. func medialibrary(_ medialibrary: MediaLibraryService, didAddAlbums albums: [VLCMLAlbum]) {
  61. albums.forEach({ append($0) })
  62. updateView?()
  63. }
  64. func medialibrary(_ medialibrary: MediaLibraryService,
  65. didModifyAlbumsWithIds albumsIds: [NSNumber]) {
  66. var albums = [VLCMLAlbum]()
  67. albumsIds.forEach() {
  68. guard let safeAlbum = medialibrary.medialib.album(withIdentifier: $0.int64Value)
  69. else {
  70. return
  71. }
  72. albums.append(safeAlbum)
  73. }
  74. files = swapModels(with: albums)
  75. updateView?()
  76. }
  77. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteAlbumsWithIds albumsIds: [NSNumber]) {
  78. files.removeAll {
  79. albumsIds.contains(NSNumber(value: $0.identifier()))
  80. }
  81. updateView?()
  82. }
  83. }
  84. extension VLCMLAlbum: MediaCollectionModel {
  85. func sortModel() -> SortModel? {
  86. return nil
  87. }
  88. func files(with criteria: VLCMLSortingCriteria = .alpha,
  89. desc: Bool = false) -> [VLCMLMedia]? {
  90. return tracks(with: criteria, desc: desc)
  91. }
  92. func title() -> String {
  93. return title
  94. }
  95. }
  96. extension VLCMLAlbum {
  97. func numberOfTracksString() -> String {
  98. let trackCount = numberOfTracks()
  99. let tracksString = trackCount > 1 ? NSLocalizedString("TRACKS", comment: "") : NSLocalizedString("TRACK", comment: "")
  100. return String(format: tracksString, trackCount)
  101. }
  102. func albumName() -> String {
  103. return isUnknownAlbum() ? NSLocalizedString("UNKNOWN_ALBUM", comment: "") : title
  104. }
  105. func albumArtistName() -> String {
  106. guard let artist = albumArtist else {
  107. return NSLocalizedString("UNKNOWN_ARTIST", comment: "")
  108. }
  109. return artist.artistName()
  110. }
  111. func accessibilityText(editing: Bool) -> String? {
  112. if editing {
  113. return albumName() + " " + albumArtistName() + " " + numberOfTracksString()
  114. }
  115. return albumName() + " " + albumArtistName()
  116. }
  117. }
  118. extension VLCMLAlbumTrack {
  119. func albumArtistName() -> String {
  120. guard let artist = artist else {
  121. return NSLocalizedString("UNKNOWN_ARTIST", comment: "")
  122. }
  123. return artist.artistName()
  124. }
  125. }