VLCMediaLibraryManager.swift 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207
  1. /*****************************************************************************
  2. * VLCMediaLibraryManager.swift
  3. * VLC for iOS
  4. *****************************************************************************
  5. * Copyright © 2018 VideoLAN. All rights reserved.
  6. * Copyright © 2018 Videolabs
  7. *
  8. * Authors: Soomin Lee <bubu # mikan.io>
  9. *
  10. * Refer to the COPYING file of the official project for license.
  11. *****************************************************************************/
  12. extension Notification.Name {
  13. static let VLCNewFileAddedNotification = Notification.Name("NewFileAddedNotification")
  14. }
  15. // For objc
  16. extension NSNotification {
  17. @objc static let VLCNewFileAddedNotification = Notification.Name.VLCNewFileAddedNotification
  18. }
  19. @objc protocol MediaLibraryObserver: class {
  20. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  21. didUpdateVideo video: [VLCMLMedia])
  22. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  23. didAddVideo video: [VLCMLMedia])
  24. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  25. didAddShowEpisode showEpisode: [VLCMLMedia])
  26. // Audio
  27. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  28. didAddAudio audio: [VLCMLMedia])
  29. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  30. didAddAlbumTrack audio: [VLCMLMedia])
  31. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  32. didAddArtist artist: [VLCMLArtist])
  33. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  34. didAddAlbum album: [VLCMLAlbum])
  35. @objc optional func medialibrary(_ medialibrary: VLCMediaLibraryManager,
  36. didAddGenre genre: [VLCMLGenre])
  37. }
  38. class VLCMediaLibraryManager: NSObject {
  39. private static let databaseName: String = "medialibrary.db"
  40. private var databasePath: String!
  41. private var thumbnailPath: String!
  42. // Using ObjectIdentifier to avoid duplication and facilitate
  43. // identification of observing object
  44. private var observers = [ObjectIdentifier: Observer]()
  45. private lazy var medialib: VLCMediaLibrary = {
  46. let medialibrary = VLCMediaLibrary()
  47. medialibrary.delegate = self
  48. return medialibrary
  49. }()
  50. override init() {
  51. super.init()
  52. setupMediaLibrary()
  53. NotificationCenter.default.addObserver(self, selector: #selector(reload),
  54. name: .VLCNewFileAddedNotification, object: nil)
  55. }
  56. // MARK: Private
  57. private func setupMediaLibrary() {
  58. guard let documentPath = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true).first,
  59. let dbPath = NSSearchPathForDirectoriesInDomains(.libraryDirectory, .userDomainMask, true).first else {
  60. preconditionFailure("VLCMediaLibraryManager: Unable to init medialibrary.")
  61. }
  62. databasePath = dbPath + "/" + VLCMediaLibraryManager.databaseName
  63. thumbnailPath = documentPath
  64. let medialibraryStatus = medialib.setupMediaLibrary(databasePath: databasePath,
  65. thumbnailPath: thumbnailPath)
  66. switch medialibraryStatus {
  67. case .success:
  68. guard medialib.start() else {
  69. assertionFailure("VLCMediaLibraryManager: Medialibrary failed to start.")
  70. return
  71. }
  72. medialib.reload()
  73. medialib.discover(onEntryPoint: "file://" + documentPath)
  74. case .alreadyInitialized:
  75. assertionFailure("VLCMediaLibraryManager: Medialibrary already initialized.")
  76. case .failed:
  77. preconditionFailure("VLCMediaLibraryManager: Failed to setup medialibrary.")
  78. case .dbReset:
  79. // should still start and discover but warn the user that the db has been wipped
  80. assertionFailure("VLCMediaLibraryManager: The database was resetted, please re-configure.")
  81. }
  82. }
  83. // MARK: Internal
  84. @objc private func reload() {
  85. medialib.reload()
  86. }
  87. /// Returns number of *ALL* files(audio and video) present in the medialibrary database
  88. func numberOfFiles() -> Int {
  89. var media = medialib.audioFiles(with: .filename, desc: false)
  90. media += medialib.videoFiles(with: .filename, desc: false)
  91. return media.count
  92. }
  93. /// Returns *ALL* file found for a specified VLCMLMediaType
  94. ///
  95. /// - Parameter type: Type of the media
  96. /// - Returns: Array of VLCMLMedia
  97. func media(ofType type: VLCMLMediaType, sortingCriteria sort: VLCMLSortingCriteria = .filename, desc: Bool = false) -> [VLCMLMedia] {
  98. return type == .video ? medialib.videoFiles(with: sort, desc: desc) : medialib.audioFiles(with: sort, desc: desc)
  99. }
  100. func genre(sortingCriteria sort: VLCMLSortingCriteria = .default, desc: Bool = false) -> [VLCMLGenre] {
  101. return medialib.genres(with: sort, desc: desc)
  102. }
  103. }
  104. // MARK: - Observer
  105. private extension VLCMediaLibraryManager {
  106. struct Observer {
  107. weak var observer: MediaLibraryObserver?
  108. }
  109. }
  110. extension VLCMediaLibraryManager {
  111. func addObserver(_ observer: MediaLibraryObserver) {
  112. let identifier = ObjectIdentifier(observer)
  113. observers[identifier] = Observer(observer: observer)
  114. }
  115. func removeObserver(_ observer: MediaLibraryObserver) {
  116. let identifier = ObjectIdentifier(observer)
  117. observers.removeValue(forKey: identifier)
  118. }
  119. }
  120. // MARK: MediaDataSource - Audio methods
  121. extension VLCMediaLibraryManager {
  122. func getArtists() -> [VLCMLArtist] {
  123. return medialib.artists(with: .artist, desc: false, all: true)
  124. }
  125. func getAlbums() -> [VLCMLAlbum] {
  126. return medialib.albums(with: .album, desc: false)
  127. }
  128. }
  129. // MARK: MediaDataSource - Video methods
  130. extension VLCMediaLibraryManager {
  131. }
  132. // MARK: VLCMediaLibraryDelegate
  133. extension VLCMediaLibraryManager: VLCMediaLibraryDelegate {
  134. func medialibrary(_ medialibrary: VLCMediaLibrary, didAddMedia media: [VLCMLMedia]) {
  135. let video = media.filter {( $0.type() == .video )}
  136. let audio = media.filter {( $0.type() == .audio )}
  137. let showEpisode = media.filter {( $0.subtype() == .showEpisode )}
  138. let albumTrack = media.filter {( $0.subtype() == .albumTrack )}
  139. for observer in observers {
  140. observer.value.observer?.medialibrary?(self, didAddVideo: video)
  141. observer.value.observer?.medialibrary?(self, didAddAudio: audio)
  142. observer.value.observer?.medialibrary?(self, didAddShowEpisode: showEpisode)
  143. observer.value.observer?.medialibrary?(self, didAddAlbumTrack: albumTrack)
  144. }
  145. }
  146. func medialibrary(_ medialibrary: VLCMediaLibrary, didAdd artists: [VLCMLArtist]) {
  147. for observer in observers {
  148. observer.value.observer?.medialibrary?(self, didAddArtist: artists)
  149. }
  150. }
  151. func medialibrary(_ medialibrary: VLCMediaLibrary, didAdd albums: [VLCMLAlbum]) {
  152. for observer in observers {
  153. observer.value.observer?.medialibrary?(self, didAddAlbum: albums)
  154. }
  155. }
  156. func medialibrary(_ medialibrary: VLCMediaLibrary, didStartDiscovery entryPoint: String) {
  157. }
  158. func medialibrary(_ medialibrary: VLCMediaLibrary, didCompleteDiscovery entryPoint: String) {
  159. }
  160. func medialibrary(_ medialibrary: VLCMediaLibrary, didProgressDiscovery entryPoint: String) {
  161. }
  162. func medialibrary(_ medialibrary: VLCMediaLibrary, didUpdateParsingStatsWithPercent percent: UInt32) {
  163. }
  164. }