parent
							
								
									2ffe582ccf
								
							
						
					
					
						commit
						9035394d3d
					
				| @ -1,204 +0,0 @@ | ||||
| /* | ||||
|  * This file is part of Libresonic. | ||||
|  * | ||||
|  *  Libresonic is free software: you can redistribute it and/or modify | ||||
|  *  it under the terms of the GNU General Public License as published by | ||||
|  *  the Free Software Foundation, either version 3 of the License, or | ||||
|  *  (at your option) any later version. | ||||
|  * | ||||
|  *  Libresonic is distributed in the hope that it will be useful, | ||||
|  *  but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *  GNU General Public License for more details. | ||||
|  * | ||||
|  *  You should have received a copy of the GNU General Public License | ||||
|  *  along with Libresonic.  If not, see <http://www.gnu.org/licenses/>.
 | ||||
|  * | ||||
|  *  Copyright 2015 (C) Sindre Mehus | ||||
|  */ | ||||
| 
 | ||||
| package org.libresonic.player.service; | ||||
| 
 | ||||
| import com.google.common.base.Predicate; | ||||
| import com.google.common.collect.Iterables; | ||||
| import com.google.common.collect.Lists; | ||||
| import org.apache.commons.io.IOUtils; | ||||
| import org.libresonic.player.util.StringUtil; | ||||
| 
 | ||||
| import javax.xml.stream.XMLInputFactory; | ||||
| import javax.xml.stream.XMLStreamConstants; | ||||
| import javax.xml.stream.XMLStreamException; | ||||
| import javax.xml.stream.XMLStreamReader; | ||||
| 
 | ||||
| import java.io.File; | ||||
| import java.io.FileInputStream; | ||||
| import java.io.InputStream; | ||||
| import java.net.URL; | ||||
| import java.util.*; | ||||
| 
 | ||||
| /** | ||||
|  * @author Sindre Mehus | ||||
|  * @version $Id$ | ||||
|  */ | ||||
| public class ITunesParser { | ||||
| 
 | ||||
|     private String iTunesXml; | ||||
|     private final XMLInputFactory inputFactory; | ||||
| 
 | ||||
|     public ITunesParser(String iTunesXml) { | ||||
|         this.iTunesXml = iTunesXml; | ||||
|         inputFactory = XMLInputFactory.newFactory(); | ||||
|         inputFactory.setProperty(XMLInputFactory.SUPPORT_DTD, false); | ||||
|     } | ||||
| 
 | ||||
|     public List<ITunesPlaylist> parse() throws Exception { | ||||
|         List<ITunesPlaylist> playlists = parsePlaylists(); | ||||
|         Map<String, File> tracks = parseTracks(playlists); | ||||
|         populatePlaylistTracks(playlists, tracks); | ||||
| 
 | ||||
|         for (ITunesPlaylist p : playlists) { | ||||
|             System.out.println(p); | ||||
|         } | ||||
| 
 | ||||
|         return playlists; | ||||
|     } | ||||
| 
 | ||||
|     private List<ITunesPlaylist> parsePlaylists() throws Exception { | ||||
|         List<ITunesPlaylist> playlists = new ArrayList<ITunesPlaylist>(); | ||||
| 
 | ||||
|         InputStream in = new FileInputStream(iTunesXml); | ||||
|         try { | ||||
|             ITunesPlaylist playlist = null; | ||||
| 
 | ||||
|             XMLStreamReader streamReader = inputFactory.createXMLStreamReader(in); | ||||
|             while (streamReader.hasNext()) { | ||||
|                 int code = streamReader.next(); | ||||
|                 if (code == XMLStreamReader.START_ELEMENT) { | ||||
|                     String key = readKey(streamReader); | ||||
| 
 | ||||
|                     if ("Playlist ID".equals(key)) { | ||||
|                         playlist = new ITunesPlaylist(readNextTag(streamReader)); | ||||
|                         playlists.add(playlist); | ||||
|                     } | ||||
| 
 | ||||
|                     if (playlist != null) { | ||||
|                         if ("Name".equals(key)) { | ||||
|                             playlist.name = readNextTag(streamReader); | ||||
|                         } else if ("Smart Info".equals(key)) { | ||||
|                             playlist.smart = true; | ||||
|                         } else if ("Visible".equals(key)) { | ||||
|                             playlist.visible = false; | ||||
|                         } else if ("Distinguished Kind".equals(key)) { | ||||
|                             playlist.distinguishedKind = readNextTag(streamReader); | ||||
|                         } else if ("Track ID".equals(key)) { | ||||
|                             playlist.trackIds.add(readNextTag(streamReader)); | ||||
|                         } | ||||
|                     } | ||||
|                 } | ||||
|             } | ||||
|         } finally { | ||||
|             IOUtils.closeQuietly(in); | ||||
|         } | ||||
| 
 | ||||
|         return Lists.newArrayList(Iterables.filter(playlists, new Predicate<ITunesPlaylist>() { | ||||
|             @Override | ||||
|             public boolean apply(ITunesPlaylist input) { | ||||
|                 return input.isIncluded(); | ||||
|             } | ||||
|         })); | ||||
|     } | ||||
| 
 | ||||
|     private Map<String, File> parseTracks(List<ITunesPlaylist> playlists) throws Exception { | ||||
|         Map<String, File> result = new HashMap<String, File>(); | ||||
|         SortedSet<String> trackIds = new TreeSet<String>(); | ||||
|         for (ITunesPlaylist playlist : playlists) { | ||||
|             trackIds.addAll(playlist.trackIds); | ||||
|         } | ||||
| 
 | ||||
|         InputStream in = new FileInputStream(iTunesXml); | ||||
| 
 | ||||
|         try { | ||||
|             XMLStreamReader streamReader = inputFactory.createXMLStreamReader(in); | ||||
|             String trackId = null; | ||||
|             while (streamReader.hasNext()) { | ||||
|                 int code = streamReader.next(); | ||||
|                 if (code == XMLStreamReader.START_ELEMENT) { | ||||
|                     String key = readKey(streamReader); | ||||
|                     if ("Track ID".equals(key)) { | ||||
|                         trackId = readNextTag(streamReader); | ||||
|                     } else if (trackId != null && trackIds.contains(trackId) && "Location".equals(key)) { | ||||
|                         String location = readNextTag(streamReader); | ||||
|                         File file = new File(StringUtil.urlDecode(new URL(location).getFile())); | ||||
|                         result.put(trackId, file); | ||||
|                     } | ||||
|                 } | ||||
|             } | ||||
|         } finally { | ||||
|             IOUtils.closeQuietly(in); | ||||
|         } | ||||
|         return result; | ||||
|     } | ||||
| 
 | ||||
|     private void populatePlaylistTracks(List<ITunesPlaylist> playlists, Map<String, File> tracks) { | ||||
|         for (ITunesPlaylist playlist : playlists) { | ||||
|             for (String trackId : playlist.trackIds) { | ||||
|                 File file = tracks.get(trackId); | ||||
|                 if (file != null) { | ||||
|                     playlist.trackFiles.add(file); | ||||
|                 } | ||||
|             } | ||||
|         } | ||||
|     } | ||||
| 
 | ||||
|     private String readNextTag(XMLStreamReader streamReader) throws XMLStreamException { | ||||
|         while (streamReader.next() != XMLStreamConstants.START_ELEMENT) { | ||||
|         } | ||||
|         return streamReader.getElementText(); | ||||
|     } | ||||
| 
 | ||||
|     private String readKey(XMLStreamReader streamReader) { | ||||
|         try { | ||||
|             if (streamReader.getEventType() == XMLStreamConstants.START_ELEMENT && | ||||
|                 "key".equals(streamReader.getName().getLocalPart())) { | ||||
|                 return streamReader.getElementText(); | ||||
|             } | ||||
|         } catch (XMLStreamException e) { | ||||
|             // TODO
 | ||||
|             System.out.println(streamReader.getName().getLocalPart() + " " + e); | ||||
|         } | ||||
|         return null; | ||||
|     } | ||||
| 
 | ||||
|     // TODO
 | ||||
|     public static void main(String[] args) throws Exception { | ||||
|         new ITunesParser("/Users/sindre/Music/iTunes/iTunes Music Library.xml").parse(); | ||||
|     } | ||||
| 
 | ||||
|     private static class ITunesPlaylist { | ||||
| 
 | ||||
|         private final String id; | ||||
|         private String name; | ||||
|         private boolean smart; | ||||
|         private final List<String> trackIds = new ArrayList<String>(); | ||||
|         private final List<File> trackFiles = new ArrayList<File>(); | ||||
|         private boolean visible = true; | ||||
|         private String distinguishedKind; | ||||
| 
 | ||||
|         public ITunesPlaylist(String id) { | ||||
|             this.id = id; | ||||
|         } | ||||
| 
 | ||||
|         public boolean isIncluded() { | ||||
|             return !smart && visible && distinguishedKind == null; | ||||
|         } | ||||
| 
 | ||||
|         @Override | ||||
|         public String toString() { | ||||
|             StringBuilder s = new StringBuilder(id + " - " + name ); | ||||
|             for (File trackFile : trackFiles) { | ||||
|                 s.append("\n  " + trackFile); | ||||
|             } | ||||
|             return s.toString(); | ||||
|         } | ||||
|     } | ||||
| } | ||||
					Loading…
					
					
				
		Reference in new issue