lib/kagu/playlists.rb in kagu-2.0.3 vs lib/kagu/playlists.rb in kagu-3.0.0

@@ -2,62 +2,47 @@
 
   class Playlists
 
     include Enumerable
 
-    attr_reader :library
-
-    def initialize(library)
-      raise ArgumentError.new("#{self.class}#library must be a library, #{library.inspect} given") unless library.is_a?(Library)
-      @library = library
-    end
-
     def build(attributes = {})
       Playlist.new(attributes)
     end
 
     def create(attributes = {})
       build(attributes).tap(&:save)
     end
 
     def each(&block)
       return unless block_given?
+      Kagu.logger.debug('Kagu') { 'Loading library playlists' }
       tracks = {}.tap do |tracks|
-        library.tracks.each { |track| tracks[track.id] = track }
+        Tracks.new.each { |track| tracks[track.id] = track }
       end
-      Kagu.logger.debug('Kagu') { "Reading library playlists from #{library.path.inspect}" }
-      File.open(library.path, 'r') do |file|
-        begin
-          line = file.readline.strip
-        end while !line.starts_with?('<key>Playlists</key>')
-        playlist_name = nil
-        playlist_tracks = []
-        skip_next = false
-        while !file.eof? && (line = file.readline.strip)
-          if line == '<key>Master</key><true/>'
-            playlist_name = nil
-            skip_next = true
-            next
-          end
-          if line == '</array>'
-            yield(Playlist.new(tracks: playlist_tracks, xml_name: playlist_name)) if playlist_name.present? && playlist_tracks.any?
-            playlist_name = nil
-            playlist_tracks = []
-            next
-          end
-          match = line.match(/<key>(.+)<\/key><(\w+)>(.*)<\/\2>/)
-          next unless match
-          name = match[1]
-          value = match[3]
-          if name == 'Name'
-            if skip_next
-              skip_next = false
-            else
-              playlist_name = value
-            end
-          elsif name == 'Track ID'
-            playlist_tracks << tracks[value.to_i]
-          end
+      playlist_name = nil
+      playlist_tracks = []
+      SwiftHelper.execute(%Q{
+        import iTunesLibrary
+
+        let library = try! ITLibrary(apiVersion: "1")
+        for playlist in library.allPlaylists.filter({ !$0.isMaster }) {
+          print("BEGIN_PLAYLIST")
+          print(playlist.name)
+          for track in playlist.items.filter({ $0.mediaKind == ITLibMediaItemMediaKind.kindSong }) {
+            print(String(format: "%02X", track.persistentID.intValue))
+          }
+          print("END_PLAYLIST")
+        }
+      }) do |line|
+        if line == 'BEGIN_PLAYLIST'
+          playlist_name = nil
+          playlist_tracks = []
+        elsif line == 'END_PLAYLIST'
+          yield(Playlist.new(name: playlist_name, tracks: playlist_tracks)) if playlist_name.present?
+        elsif playlist_name.nil?
+          playlist_name = line
+        else
+          playlist_tracks << tracks[line]
         end
       end
     end
 
   end