GenreModel.swift 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. /*****************************************************************************
  2. * GenreModel.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 GenreModel: MLBaseModel {
  12. typealias MLType = VLCMLGenre
  13. var sortModel = SortModel([.alpha])
  14. var updateView: (() -> Void)?
  15. var files = [VLCMLGenre]()
  16. var cellType: BaseCollectionViewCell.Type { return MediaCollectionViewCell.self }
  17. var medialibrary: MediaLibraryService
  18. var indicatorName: String = NSLocalizedString("GENRES", comment: "")
  19. required init(medialibrary: MediaLibraryService) {
  20. self.medialibrary = medialibrary
  21. medialibrary.addObserver(self)
  22. files = medialibrary.genres()
  23. }
  24. func append(_ item: VLCMLGenre) {
  25. files.append(item)
  26. }
  27. func delete(_ items: [VLCMLObject]) {
  28. preconditionFailure("GenreModel: Genres can not be deleted, they disappear when their last title got deleted")
  29. }
  30. }
  31. // MARK: - MediaLibraryObserver
  32. extension GenreModel: MediaLibraryObserver {
  33. func medialibrary(_ medialibrary: MediaLibraryService, didAddGenres genres: [VLCMLGenre]) {
  34. genres.forEach({ append($0) })
  35. updateView?()
  36. }
  37. func medialibrary(_ medialibrary: MediaLibraryService, didModifyGenres genres: [VLCMLGenre]) {
  38. files = swapModels(with: genres)
  39. updateView?()
  40. }
  41. func medialibrary(_ medialibrary: MediaLibraryService, didDeleteGenresWithIds genreIds: [NSNumber]) {
  42. files.removeAll {
  43. genreIds.contains(NSNumber(value: $0.identifier()))
  44. }
  45. updateView?()
  46. }
  47. }
  48. // MARK: - Sort
  49. extension GenreModel {
  50. func sort(by criteria: VLCMLSortingCriteria, desc: Bool) {
  51. files = medialibrary.genres(sortingCriteria: criteria, desc: desc)
  52. sortModel.currentSort = criteria
  53. sortModel.desc = desc
  54. updateView?()
  55. }
  56. }
  57. // MARK: - Edit
  58. extension GenreModel: EditableMLModel {
  59. func editCellType() -> BaseCollectionViewCell.Type {
  60. return MediaEditCell.self
  61. }
  62. }
  63. // MARK: - Search
  64. extension VLCMLGenre: SearchableMLModel {
  65. func contains(_ searchString: String) -> Bool {
  66. return name.lowercased().contains(searchString)
  67. }
  68. }
  69. // MARK: - Helpers
  70. extension VLCMLGenre {
  71. @objc func numberOfTracksString() -> String {
  72. let numberOftracks = numberOfTracks()
  73. if numberOftracks != 1 {
  74. return String(format: NSLocalizedString("TRACKS", comment: ""), numberOftracks)
  75. }
  76. return String(format: NSLocalizedString("TRACK", comment: ""), numberOftracks)
  77. }
  78. func thumbnail() -> UIImage? {
  79. var image: UIImage? = nil
  80. for track in tracks() ?? [] where track.isThumbnailGenerated() {
  81. image = UIImage(contentsOfFile: track.thumbnail()?.path ?? "")
  82. break
  83. }
  84. if image == nil {
  85. let isDarktheme = PresentationTheme.current == PresentationTheme.darkTheme
  86. image = isDarktheme ? UIImage(named: "song-placeholder-dark") : UIImage(named: "song-placeholder-white")
  87. }
  88. return image
  89. }
  90. func accessibilityText() -> String? {
  91. return name + " " + numberOfTracksString()
  92. }
  93. }
  94. extension VLCMLGenre: MediaCollectionModel {
  95. func sortModel() -> SortModel? {
  96. return SortModel([.alpha])
  97. }
  98. func files() -> [VLCMLMedia]? {
  99. return tracks()
  100. }
  101. func title() -> String? {
  102. return name
  103. }
  104. }