Initial commit.

src/playlistparsers/m3uparser.cpp

@@ -0,0 +1,141 @@
+/*
+ * Strawberry Music Player
+ * This file was part of Clementine.
+ * Copyright 2010, David Sansome <me@davidsansome.com>
+ *
+ * Strawberry 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.
+ *
+ * Strawberry 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 Strawberry.  If not, see <http://www.gnu.org/licenses/>.
+ * 
+ */
+
+#include "config.h"
+
+#include "m3uparser.h"
+#include "core/logging.h"
+#include "core/timeconstants.h"
+
+#include "playlist/playlist.h"
+
+#include <QBuffer>
+#include <QtDebug>
+
+M3UParser::M3UParser(CollectionBackendInterface *collection, QObject *parent)
+    : ParserBase(collection, parent) {}
+
+SongList M3UParser::Load(QIODevice *device, const QString &playlist_path, const QDir &dir) const {
+
+  SongList ret;
+
+  M3UType type = STANDARD;
+  Metadata current_metadata;
+
+  QString data = QString::fromUtf8(device->readAll());
+  data.replace('\r', '\n');
+  data.replace("\n\n", "\n");
+  QByteArray bytes = data.toUtf8();
+  QBuffer buffer(&bytes);
+  buffer.open(QIODevice::ReadOnly);
+
+  QString line = QString::fromUtf8(buffer.readLine()).trimmed();
+  if (line.startsWith("#EXTM3U")) {
+    // This is in extended M3U format.
+    type = EXTENDED;
+    line = QString::fromUtf8(buffer.readLine()).trimmed();
+  }
+
+  forever {
+    if (line.startsWith('#')) {
+      // Extended info or comment.
+      if (type == EXTENDED && line.startsWith("#EXT")) {
+        if (!ParseMetadata(line, &current_metadata)) {
+          qLog(Warning) << "Failed to parse metadata: " << line;
+        }
+      }
+    }
+    else if (!line.isEmpty()) {
+      Song song = LoadSong(line, 0, dir);
+      if (!current_metadata.title.isEmpty()) {
+        song.set_title(current_metadata.title);
+      }
+      if (!current_metadata.artist.isEmpty()) {
+        song.set_artist(current_metadata.artist);
+      }
+      if (current_metadata.length > 0) {
+        song.set_length_nanosec(current_metadata.length);
+      }
+      ret << song;
+
+      current_metadata = Metadata();
+    }
+    if (buffer.atEnd()) {
+      break;
+    }
+    line = QString::fromUtf8(buffer.readLine()).trimmed();
+  }
+
+  return ret;
+
+}
+
+bool M3UParser::ParseMetadata(const QString &line, M3UParser::Metadata *metadata) const {
+
+  // Extended info, eg.
+  // #EXTINF:123,Sample Artist - Sample title
+  QString info = line.section(':', 1);
+  QString l = info.section(',', 0, 0);
+  bool ok = false;
+  int length = l.toInt(&ok);
+  if (!ok) {
+    return false;
+  }
+  metadata->length = length * kNsecPerSec;
+
+  QString track_info = info.section(',', 1);
+  QStringList list = track_info.split(" - ");
+  if (list.size() <= 1) {
+    metadata->title = track_info;
+    return true;
+  }
+  metadata->artist = list[0].trimmed();
+  metadata->title = list[1].trimmed();
+  return true;
+}
+
+void M3UParser::Save(const SongList &songs, QIODevice *device, const QDir &dir, Playlist::Path path_type) const {
+
+  device->write("#EXTM3U\n");
+
+  QSettings s;
+  s.beginGroup(Playlist::kSettingsGroup);
+  bool writeMetadata = s.value(Playlist::kWriteMetadata, true).toBool();
+  s.endGroup();
+
+  for (const Song &song : songs) {
+    if (song.url().isEmpty()) {
+      continue;
+    }
+    if (writeMetadata) {
+      QString meta = QString("#EXTINF:%1,%2 - %3\n")
+                         .arg(song.length_nanosec() / kNsecPerSec)
+                         .arg(song.artist())
+                         .arg(song.title());
+      device->write(meta.toUtf8());
+    }
+    device->write(URLOrFilename(song.url(), dir, path_type).toUtf8());
+    device->write("\n");
+  }
+}
+
+bool M3UParser::TryMagic(const QByteArray &data) const {
+  return data.contains("#EXTM3U") || data.contains("#EXTINF");
+}