File indexing completed on 2024-04-28 08:40:21

 /* AUDEX CDDA EXTRACTOR
  * SPDX-FileCopyrightText: Copyright (C) 2007 Marco Nelles
  * <https://userbase.kde.org/Audex>
  *
  * SPDX-License-Identifier: GPL-3.0-or-later
  */
 
 #include "playlist.h"
 #include <QRegularExpression>
 
 Playlist::Playlist()
 {
 }
 
 Playlist::Playlist(const QByteArray &playlist)
 {
     addPlaylist(playlist);
 }
 
 Playlist::~Playlist()
 {
 }
 
 void Playlist::addPlaylist(const QByteArray &playlist)
 {
     QString format = p_playlist_format(playlist);
 
     if (format == "m3u") {
         p_add_M3U(playlist);
     } else if (format == "pls") {
         p_add_PLS(playlist);
     } else if (format == "xspf") {
         p_add_XSPF(playlist);
     }
 }
 
 void Playlist::clear()
 {
     p_playlist.clear();
 }
 
 void Playlist::appendItem(const PlaylistItem &item)
 {
     p_playlist.append(item);
 }
 
 QByteArray Playlist::toM3U(const QString &playlistPath, const bool utf8) const
 {
     QStringList playlist;
     playlist.append("#EXTM3U");
 
     for (int i = 0; i < p_playlist.count(); ++i) {
         PlaylistItem pi = p_playlist[i];
         if (pi.filename().isEmpty())
             continue;
 
         if (!pi.artist().isEmpty()) {
             playlist.append(QString("#EXTINF:%1,%2 - %3").arg(QString::number(pi.length()), pi.artist(), pi.title()));
         } else {
             playlist.append(QString("#EXTINF:%1,%2").arg(QString::number(pi.length()), pi.title()));
         }
         if (!playlistPath.isEmpty()) {
             QDir dir(playlistPath);
             playlist.append(dir.relativeFilePath(pi.filename()));
         } else {
             playlist.append(pi.filename());
         }
     }
 
     if (utf8)
         return playlist.join("\n").append("\n").toUtf8();
     else
         return playlist.join("\n").append("\n").toLatin1();
 }
 
 QByteArray Playlist::toPLS(const QString &playlistPath, const bool utf8) const
 {
     QStringList playlist;
     playlist.append("[Playlist]");
 
     int j = 0;
     for (int i = 0; i < p_playlist.count(); ++i) {
         PlaylistItem pi = p_playlist[i];
 
         if (pi.filename().isEmpty())
             continue;
         ++j;
 
         if (!playlistPath.isEmpty()) {
             QDir dir(playlistPath);
             playlist.append(QString("File%1=%2").arg(i + 1).arg(dir.relativeFilePath(pi.filename())));
         } else {
             playlist.append(QString("File%1=%2").arg(i + 1).arg(pi.filename()));
         }
 
         if (!pi.artist().isEmpty()) {
             playlist.append(QString("Title%1=%2 - %3").arg(QString::number(i + 1), pi.artist(), pi.title()));
         } else {
             playlist.append(QString("Title%1=%2").arg(i + 1).arg(pi.title()));
         }
 
         playlist.append(QString("Length%1=%2").arg(i + 1).arg(pi.length()));
     }
 
     playlist.append(QString("NumberOfEntries=%1").arg(j));
     playlist.append(QString("Version=2"));
 
     if (utf8)
         return playlist.join("\n").append("\n").toUtf8();
     else
         return playlist.join("\n").append("\n").toLatin1();
 }
 
 QByteArray Playlist::toXSPF() const
 {
     QDomDocument doc;
     QDomElement root = doc.createElement("playlist");
     root.setAttribute("version", "1");
     root.setAttribute("xmlns", "http://xspf.org/ns/0");
 
     QDomElement creator = doc.createElement("creator");
     QDomText text = doc.createTextNode("audex");
     creator.appendChild(text);
     root.appendChild(creator);
 
     QDomElement tracklist = doc.createElement("trackList");
 
     int j = 0;
     for (int i = 0; i < p_playlist.count(); ++i) {
         PlaylistItem pi = p_playlist[i];
 
         if (pi.filename().isEmpty())
             continue;
         ++j;
 
         QDomElement track = doc.createElement("track");
 
         QDomElement ch = doc.createElement("location");
         QDomText text = doc.createTextNode(pi.filename());
 
         ch.appendChild(text);
         track.appendChild(ch);
 
         if (!pi.artist().isEmpty()) {
             ch = doc.createElement("creator");
             text = doc.createTextNode(pi.artist());
             ch.appendChild(text);
             track.appendChild(ch);
         }
 
         ch = doc.createElement("title");
         text = doc.createTextNode(pi.title());
         ch.appendChild(text);
         track.appendChild(ch);
 
         ch = doc.createElement("trackNum");
         text = doc.createTextNode(QString::number(j));
         ch.appendChild(text);
         track.appendChild(ch);
 
         if (pi.length() > 0) {
             ch = doc.createElement("duration");
             text = doc.createTextNode(QString::number(pi.length() * 1000));
             ch.appendChild(text);
             track.appendChild(ch);
         }
 
         tracklist.appendChild(track);
     }
 
     root.appendChild(tracklist);
     doc.appendChild(root);
     QByteArray xml_header("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
     return doc.toByteArray().prepend(xml_header);
 }
 
 const QString Playlist::p_playlist_format(const QByteArray &playlist)
 {
     if (playlist.contains("#EXTM3U") || playlist.contains("#EXTINF"))
         return "m3u";
     if (playlist.toLower().contains("[playlist]"))
         return "pls";
     if (playlist.contains("<playlist") && playlist.contains("<trackList"))
         return "xspf";
 
     return QString();
 }
 
 void Playlist::p_add_M3U(const QByteArray &playlist)
 {
     QTextStream stream(playlist, QIODevice::ReadOnly);
 
     QString line = stream.readLine().trimmed();
 
     bool extended = false;
     if (line.startsWith(QLatin1String("#EXTM3U"))) {
         extended = true;
         line = stream.readLine().trimmed();
     }
 
     PlaylistItem pi;
     while (!line.isNull()) {
         if (line.startsWith('#')) {
             if (extended && line.startsWith(QLatin1String("#EXT"))) {
                 pi = p_parse_m3u_metadata_line(line);
             }
         } else if (!line.isEmpty()) {
             pi.setFilename(line);
             if (!pi.filename().isEmpty()) {
                 p_playlist.append(pi);
                 pi.clear();
             }
         }
 
         line = stream.readLine().trimmed();
     }
 }
 
 void Playlist::p_add_PLS(const QByteArray &playlist)
 {
     QTextStream stream(playlist, QIODevice::ReadOnly);
 
     QString line = stream.readLine().trimmed();
 
     QMap<int, PlaylistItem> items;
     while (!line.isNull()) {
         int equals = line.indexOf('=');
         QString key = line.left(equals).toLower();
         QString value = line.mid(equals + 1);
 
         static QRegularExpression n_re("\\d+$");
         auto match = n_re.match(key);
         int n = match.captured(0).toInt();
 
         if (key.startsWith(QLatin1String("file"))) {
             items[n].setFilename(value);
         } else if (key.startsWith(QLatin1String("title"))) {
             items[n].setTitle(value);
         } else if (key.startsWith(QLatin1String("length"))) {
             bool ok;
             int seconds = value.toInt(&ok);
             if (ok)
                 items[n].setLength(seconds);
         }
 
         line = stream.readLine().trimmed();
     }
 
     QMap<int, PlaylistItem>::const_iterator i = items.constBegin();
     while (i != items.constEnd()) {
         p_playlist.append(i.value());
         ++i;
     }
 }
 
 void Playlist::p_add_XSPF(const QByteArray &playlist)
 {
     QDomDocument doc;
     QString errorMsg;
     int errorCol;
     int errorRow;
 
     if (!doc.setContent(QString(playlist), &errorMsg, &errorRow, &errorCol))
         return;
 
     QDomElement rootElement = doc.firstChildElement("playlist");
     if (rootElement.isNull())
         return;
 
     QDomElement tracklistElement = rootElement.firstChildElement("trackList");
     if (tracklistElement.isNull())
         return;
 
     QMap<int, PlaylistItem> items;
     QMap<int, PlaylistItem> items_no_tracknum;
     QDomElement child = tracklistElement.firstChildElement("track");
     int m = 0;
     while (!child.isNull()) {
         bool ok;
         int n = child.firstChildElement("trackNum").text().toInt(&ok);
         if (!ok) {
             items_no_tracknum[m].setFilename(child.firstChildElement("location").text());
             items_no_tracknum[m].setArtist(child.firstChildElement("creator").text());
             items_no_tracknum[m].setTitle(child.firstChildElement("title").text());
             items_no_tracknum[m].setLength(child.firstChildElement("duration").text().toInt() / 1000);
             ++m;
         } else {
             items[n].setFilename(child.firstChildElement("location").text());
             items[n].setArtist(child.firstChildElement("creator").text());
             items[n].setTitle(child.firstChildElement("title").text());
             items[n].setLength(child.firstChildElement("duration").text().toInt() / 1000);
         }
 
         child = child.nextSiblingElement();
     }
 
     QMap<int, PlaylistItem>::const_iterator i = items.constBegin();
     while (i != items.constEnd()) {
         p_playlist.append(i.value());
         ++i;
     }
     QMap<int, PlaylistItem>::const_iterator j = items_no_tracknum.constBegin();
     while (j != items_no_tracknum.constEnd()) {
         p_playlist.append(j.value());
         ++j;
     }
 }
 
 const PlaylistItem Playlist::p_parse_m3u_metadata_line(const QString &line)
 {
     PlaylistItem pi;
 
     QString info = line.section(':', 1);
     QString l = info.section(',', 0, 0);
     bool ok;
     int length = l.toInt(&ok);
     if (!ok)
         return pi;
     pi.setLength(length);
 
     QString track_info = info.section(',', 1);
     QStringList list = track_info.split('-');
     if (list.length() <= 1) {
         pi.setTitle(track_info);
         return pi;
     }
     pi.setArtist(list[0].trimmed());
     pi.setTitle(list[1].trimmed());
     return pi;
 }