Andrew DeMaria
8 years ago
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