ArtistModel.swift 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. /*****************************************************************************
  2. * ArtistModel.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 ArtistModel: AudioCollectionModel {
  12. typealias MLType = VLCMLArtist
  13. var sortModel = SortModel([.alpha])
  14. var updateView: (() -> Void)?
  15. var files = [VLCMLArtist]()
  16. var cellType: BaseCollectionViewCell.Type { return MediaCollectionViewCell.self }
  17. var medialibrary: MediaLibraryService
  18. var indicatorName: String = NSLocalizedString("ARTISTS", comment: "")
  19. required init(medialibrary: MediaLibraryService) {
  20. self.medialibrary = medialibrary
  21. medialibrary.addObserver(self)
  22. files = medialibrary.artists()
  23. }
  24. func append(_ item: VLCMLArtist) {
  25. files.append(item)
  26. }
  27. }
  28. // MARK: - Sort
  29. extension ArtistModel {
  30. func sort(by criteria: VLCMLSortingCriteria, desc: Bool) {
  31. files = medialibrary.artists(sortingCriteria: criteria, desc: desc)
  32. sortModel.currentSort = criteria
  33. sortModel.desc = desc
  34. updateView?()
  35. }
  36. }
  37. // MARK: - Edit
  38. extension ArtistModel: EditableMLModel {
  39. func editCellType() -> BaseCollectionViewCell.Type {
  40. return MediaEditCell.self
  41. }
  42. }
  43. // MARK: - Search
  44. extension VLCMLArtist: SearchableMLModel {
  45. func contains(_ searchString: String) -> Bool {
  46. return name.lowercased().contains(searchString)
  47. }
  48. }
  49. // MARK: - MediaLibraryObserver
  50. extension ArtistModel: MediaLibraryObserver {
  51. func medialibrary(_ medialibrary: MediaLibraryService, didAddArtists artists: [VLCMLArtist]) {
  52. artists.forEach({ append($0) })
  53. updateView?()
  54. }
  55. func medialibrary(_ medialibrary: MediaLibraryService,
  56. didModifyArtistsWithIds artistsIds: [NSNumber]) {
  57. var artists = [VLCMLArtist]()
  58. artistsIds.forEach() {
  59. guard let safeArtist = medialibrary.medialib.artist(withIdentifier: $0.int64Value)
  60. else {
  61. return
  62. }
  63. artists.append(safeArtist)
  64. }
  65. files = swapModels(with: artists)
  66. updateView?()
  67. }
  68. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteArtistsWithIds artistsIds: [NSNumber]) {
  69. files.removeAll {
  70. artistsIds.contains(NSNumber(value: $0.identifier()))
  71. }
  72. updateView?()
  73. }
  74. }
  75. extension VLCMLArtist: MediaCollectionModel {
  76. func sortModel() -> SortModel? {
  77. return SortModel([.alpha])
  78. }
  79. func files() -> [VLCMLMedia]? {
  80. return tracks()
  81. }
  82. func title() -> String {
  83. return name
  84. }
  85. func sortFilesInCollection(with criteria: VLCMLSortingCriteria,
  86. desc: Bool) -> [VLCMLMedia]? {
  87. return nil
  88. }
  89. }
  90. extension VLCMLArtist {
  91. func numberOfTracksString() -> String {
  92. let tracksString = tracks()?.count == 1 ? NSLocalizedString("TRACK", comment: "") : NSLocalizedString("TRACKS", comment: "")
  93. return String(format: tracksString, tracks()?.count ?? 0)
  94. }
  95. @objc func thumbnail() -> UIImage? {
  96. var image = UIImage(contentsOfFile: artworkMRL()?.path ?? "")
  97. if image == nil {
  98. for track in files() ?? [] where track.isThumbnailGenerated() {
  99. image = UIImage(contentsOfFile: track.thumbnail()?.path ?? "")
  100. break
  101. }
  102. }
  103. if image == nil {
  104. let isDarktheme = PresentationTheme.current == PresentationTheme.darkTheme
  105. image = isDarktheme ? UIImage(named: "artist-placeholder-dark") : UIImage(named: "artist-placeholder-white")
  106. }
  107. return image
  108. }
  109. func artistName() -> String {
  110. if identifier() == UnknownArtistID {
  111. return NSLocalizedString("UNKNOWN_ARTIST", comment: "")
  112. } else if identifier() == VariousArtistID {
  113. return NSLocalizedString("VARIOUS_ARTIST", comment: "")
  114. } else {
  115. return name
  116. }
  117. }
  118. func accessibilityText() -> String? {
  119. return artistName() + " " + numberOfTracksString()
  120. }
  121. }