mirror of
https://github.com/strawberrymusicplayer/strawberry
synced 2024-12-15 10:14:43 +01:00
549 lines
18 KiB
C++
549 lines
18 KiB
C++
/* This file is part of Strawberry.
|
|
Copyright 2021, Jonas Kvinge <jonas@jkvinge.net>
|
|
|
|
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 "tagreadertagparser.h"
|
|
|
|
#include <string>
|
|
#include <memory>
|
|
#include <algorithm>
|
|
#include <vector>
|
|
#include <sys/stat.h>
|
|
|
|
#include <tagparser/mediafileinfo.h>
|
|
#include <tagparser/diagnostics.h>
|
|
#include <tagparser/progressfeedback.h>
|
|
#include <tagparser/tag.h>
|
|
#include <tagparser/abstracttrack.h>
|
|
|
|
#include <QtGlobal>
|
|
#include <QFile>
|
|
#include <QFileInfo>
|
|
#include <QByteArray>
|
|
#include <QString>
|
|
#include <QUrl>
|
|
#include <QDateTime>
|
|
#include <QtDebug>
|
|
|
|
#include "core/logging.h"
|
|
#include "core/messagehandler.h"
|
|
#include "utilities/timeconstants.h"
|
|
|
|
TagReaderTagParser::TagReaderTagParser() = default;
|
|
|
|
TagReaderTagParser::~TagReaderTagParser() = default;
|
|
|
|
bool TagReaderTagParser::IsMediaFile(const QString &filename) const {
|
|
|
|
qLog(Debug) << "Checking for valid file" << filename;
|
|
|
|
QFileInfo fileinfo(filename);
|
|
if (!fileinfo.exists() || fileinfo.suffix().compare("bak", Qt::CaseInsensitive) == 0) return false;
|
|
|
|
try {
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
taginfo.open(true);
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTracks(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
const auto tracks = taginfo.tracks();
|
|
for (TagParser::AbstractTrack *track : tracks) {
|
|
if (track->mediaType() == TagParser::MediaType::Audio) {
|
|
taginfo.close();
|
|
return true;
|
|
}
|
|
}
|
|
taginfo.close();
|
|
}
|
|
catch(...) {}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
bool TagReaderTagParser::ReadFile(const QString &filename, spb::tagreader::SongMetadata *song) const {
|
|
|
|
qLog(Debug) << "Reading tags from" << filename;
|
|
|
|
const QFileInfo fileinfo(filename);
|
|
|
|
if (!fileinfo.exists() || fileinfo.suffix().compare("bak", Qt::CaseInsensitive) == 0) return false;
|
|
|
|
const QByteArray url(QUrl::fromLocalFile(filename).toEncoded());
|
|
const QByteArray basefilename = fileinfo.fileName().toUtf8();
|
|
|
|
song->set_basefilename(basefilename.constData(), basefilename.size());
|
|
song->set_url(url.constData(), url.size());
|
|
song->set_filesize(fileinfo.size());
|
|
|
|
song->set_mtime(fileinfo.lastModified().isValid() ? std::max(fileinfo.lastModified().toSecsSinceEpoch(), 0LL) : 0LL);
|
|
song->set_ctime(fileinfo.birthTime().isValid() ? std::max(fileinfo.birthTime().toSecsSinceEpoch(), 0LL) : fileinfo.lastModified().isValid() ? std::max(fileinfo.lastModified().toSecsSinceEpoch(), 0LL) : 0LL);
|
|
|
|
if (song->ctime() <= 0) {
|
|
song->set_ctime(song->mtime());
|
|
}
|
|
|
|
song->set_lastseen(QDateTime::currentDateTime().toSecsSinceEpoch());
|
|
|
|
try {
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
|
|
#ifdef Q_OS_WIN32
|
|
taginfo.setPath(filename.toStdWString().toStdString());
|
|
#else
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
#endif
|
|
|
|
taginfo.open(true);
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTracks(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTags(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
std::vector<TagParser::AbstractTrack*> tracks = taginfo.tracks();
|
|
for (TagParser::AbstractTrack *track : tracks) {
|
|
switch (track->format().general) {
|
|
case TagParser::GeneralMediaFormat::Flac:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_FLAC);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::WavPack:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_WAVPACK);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::MonkeysAudio:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_APE);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::WindowsMediaAudio:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_ASF);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Vorbis:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_OGGVORBIS);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Opus:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_OGGOPUS);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Speex:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_OGGSPEEX);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Mpeg1Audio:
|
|
switch (track->format().sub) {
|
|
case TagParser::SubFormats::Mpeg1Layer3:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_MPEG);
|
|
break;
|
|
case TagParser::SubFormats::None:
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Mpc:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_MPC);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Pcm:
|
|
song->set_filetype(spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_PCM);
|
|
break;
|
|
case TagParser::GeneralMediaFormat::Unknown:
|
|
default:
|
|
break;
|
|
}
|
|
song->set_length_nanosec(track->duration().totalMilliseconds() * kNsecPerMsec);
|
|
song->set_samplerate(track->samplingFrequency());
|
|
song->set_bitdepth(track->bitsPerSample());
|
|
song->set_bitrate(std::max(track->bitrate(), track->maxBitrate()));
|
|
}
|
|
|
|
if (song->filetype() == spb::tagreader::SongMetadata_FileType::SongMetadata_FileType_UNKNOWN) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
for (TagParser::Tag *tag : taginfo.tags()) {
|
|
song->set_albumartist(tag->value(TagParser::KnownField::AlbumArtist).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_artist(tag->value(TagParser::KnownField::Artist).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_album(tag->value(TagParser::KnownField::Album).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_title(tag->value(TagParser::KnownField::Title).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_genre(tag->value(TagParser::KnownField::Genre).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_composer(tag->value(TagParser::KnownField::Composer).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_performer(tag->value(TagParser::KnownField::Performers).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_grouping(tag->value(TagParser::KnownField::Grouping).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_comment(tag->value(TagParser::KnownField::Comment).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_lyrics(tag->value(TagParser::KnownField::Lyrics).toString(TagParser::TagTextEncoding::Utf8));
|
|
song->set_year(tag->value(TagParser::KnownField::RecordDate).toInteger());
|
|
song->set_originalyear(tag->value(TagParser::KnownField::ReleaseDate).toInteger());
|
|
song->set_track(tag->value(TagParser::KnownField::TrackPosition).toInteger());
|
|
song->set_disc(tag->value(TagParser::KnownField::DiskPosition).toInteger());
|
|
if (!tag->value(TagParser::KnownField::Cover).empty() && tag->value(TagParser::KnownField::Cover).dataSize() > 0) {
|
|
song->set_art_embedded(true);
|
|
}
|
|
const float rating = ConvertPOPMRating(tag->value(TagParser::KnownField::Rating));
|
|
if (song->rating() <= 0 && rating > 0.0 && rating <= 1.0) {
|
|
song->set_rating(rating);
|
|
}
|
|
}
|
|
|
|
// Set integer fields to -1 if they're not valid
|
|
if (song->track() <= 0) { song->set_track(-1); }
|
|
if (song->disc() <= 0) { song->set_disc(-1); }
|
|
if (song->year() <= 0) { song->set_year(-1); }
|
|
if (song->originalyear() <= 0) { song->set_originalyear(-1); }
|
|
if (song->samplerate() <= 0) { song->set_samplerate(-1); }
|
|
if (song->bitdepth() <= 0) { song->set_bitdepth(-1); }
|
|
if (song->bitrate() <= 0) { song->set_bitrate(-1); }
|
|
if (song->lastplayed() <= 0) { song->set_lastplayed(-1); }
|
|
|
|
song->set_valid(true);
|
|
|
|
taginfo.close();
|
|
|
|
return true;
|
|
|
|
}
|
|
catch(...) {
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
bool TagReaderTagParser::SaveFile(const spb::tagreader::SaveFileRequest &request) const {
|
|
|
|
if (request.filename().empty()) return false;
|
|
|
|
const QString filename = QString::fromUtf8(request.filename().data(), request.filename().size());
|
|
const spb::tagreader::SongMetadata song = request.metadata();
|
|
const bool save_tags = request.has_save_tags() && request.save_tags();
|
|
const bool save_playcount = request.has_save_playcount() && request.save_playcount();
|
|
const bool save_rating = request.has_save_rating() && request.save_rating();
|
|
const bool save_cover = request.has_save_cover() && request.save_cover();
|
|
|
|
QStringList save_tags_options;
|
|
if (save_tags) {
|
|
save_tags_options << "tags";
|
|
}
|
|
if (save_playcount) {
|
|
save_tags_options << "playcount";
|
|
}
|
|
if (save_rating) {
|
|
save_tags_options << "rating";
|
|
}
|
|
if (save_cover) {
|
|
save_tags_options << "embedded cover";
|
|
}
|
|
|
|
qLog(Debug) << "Saving" << save_tags_options.join(", ") << "to" << filename;
|
|
|
|
const QByteArray cover_data = LoadCoverDataFromRequest(request);
|
|
|
|
try {
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
#ifdef Q_OS_WIN32
|
|
taginfo.setPath(filename.toStdWString().toStdString());
|
|
#else
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
#endif
|
|
taginfo.open(false);
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTracks(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTags(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
if (taginfo.tags().size() <= 0) {
|
|
taginfo.createAppropriateTags();
|
|
}
|
|
|
|
for (TagParser::Tag *tag : taginfo.tags()) {
|
|
if (save_tags) {
|
|
tag->setValue(TagParser::KnownField::AlbumArtist, TagParser::TagValue(song.albumartist(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Artist, TagParser::TagValue(song.artist(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Album, TagParser::TagValue(song.album(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Title, TagParser::TagValue(song.title(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Genre, TagParser::TagValue(song.genre(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Composer, TagParser::TagValue(song.composer(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Performers, TagParser::TagValue(song.performer(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Grouping, TagParser::TagValue(song.grouping(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Comment, TagParser::TagValue(song.comment(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::Lyrics, TagParser::TagValue(song.lyrics(), TagParser::TagTextEncoding::Utf8, tag->proposedTextEncoding()));
|
|
tag->setValue(TagParser::KnownField::TrackPosition, TagParser::TagValue(song.track()));
|
|
tag->setValue(TagParser::KnownField::DiskPosition, TagParser::TagValue(song.disc()));
|
|
tag->setValue(TagParser::KnownField::RecordDate, TagParser::TagValue(song.year()));
|
|
tag->setValue(TagParser::KnownField::ReleaseDate, TagParser::TagValue(song.originalyear()));
|
|
}
|
|
if (save_playcount) {
|
|
SaveSongPlaycountToFile(tag, song);
|
|
}
|
|
if (save_rating) {
|
|
SaveSongRatingToFile(tag, song);
|
|
}
|
|
if (save_cover) {
|
|
SaveEmbeddedArt(tag, cover_data);
|
|
}
|
|
}
|
|
|
|
taginfo.applyChanges(diag, progress);
|
|
taginfo.close();
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch(...) {}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
QByteArray TagReaderTagParser::LoadEmbeddedArt(const QString &filename) const {
|
|
|
|
if (filename.isEmpty()) return QByteArray();
|
|
|
|
qLog(Debug) << "Loading art from" << filename;
|
|
|
|
try {
|
|
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
|
|
#ifdef Q_OS_WIN32
|
|
taginfo.setPath(filename.toStdWString().toStdString());
|
|
#else
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
#endif
|
|
|
|
taginfo.open();
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return QByteArray();
|
|
}
|
|
|
|
taginfo.parseTags(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return QByteArray();
|
|
}
|
|
|
|
for (TagParser::Tag *tag : taginfo.tags()) {
|
|
if (!tag->value(TagParser::KnownField::Cover).empty() && tag->value(TagParser::KnownField::Cover).dataSize() > 0) {
|
|
QByteArray data(tag->value(TagParser::KnownField::Cover).dataPointer(), tag->value(TagParser::KnownField::Cover).dataSize());
|
|
taginfo.close();
|
|
return data;
|
|
}
|
|
}
|
|
|
|
taginfo.close();
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
}
|
|
catch(...) {}
|
|
|
|
return QByteArray();
|
|
|
|
}
|
|
|
|
void TagReaderTagParser::SaveEmbeddedArt(TagParser::Tag *tag, const QByteArray &data) const {
|
|
|
|
tag->setValue(TagParser::KnownField::Cover, TagParser::TagValue(data.toStdString()));
|
|
|
|
}
|
|
|
|
bool TagReaderTagParser::SaveEmbeddedArt(const spb::tagreader::SaveEmbeddedArtRequest &request) const {
|
|
|
|
if (request.filename().empty()) return false;
|
|
|
|
const QString filename = QString::fromUtf8(request.filename().data(), request.filename().size());
|
|
|
|
qLog(Debug) << "Saving art to" << filename;
|
|
|
|
const QByteArray cover_data = LoadCoverDataFromRequest(request);
|
|
|
|
try {
|
|
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
|
|
#ifdef Q_OS_WIN32
|
|
taginfo.setPath(filename.toStdWString().toStdString());
|
|
#else
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
#endif
|
|
|
|
taginfo.open();
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTags(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
if (taginfo.tags().size() <= 0) {
|
|
taginfo.createAppropriateTags();
|
|
}
|
|
|
|
for (TagParser::Tag *tag : taginfo.tags()) {
|
|
SaveEmbeddedArt(tag, cover_data);
|
|
}
|
|
|
|
taginfo.applyChanges(diag, progress);
|
|
taginfo.close();
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
catch(...) {}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
void TagReaderTagParser::SaveSongPlaycountToFile(TagParser::Tag*, const spb::tagreader::SongMetadata&) const {}
|
|
|
|
bool TagReaderTagParser::SaveSongPlaycountToFile(const QString&, const spb::tagreader::SongMetadata&) const { return false; }
|
|
|
|
void TagReaderTagParser::SaveSongRatingToFile(TagParser::Tag *tag, const spb::tagreader::SongMetadata &song) const {
|
|
|
|
tag->setValue(TagParser::KnownField::Rating, TagParser::TagValue(ConvertToPOPMRating(song.rating())));
|
|
|
|
}
|
|
|
|
bool TagReaderTagParser::SaveSongRatingToFile(const QString &filename, const spb::tagreader::SongMetadata &song) const {
|
|
|
|
if (filename.isEmpty()) return false;
|
|
|
|
qLog(Debug) << "Saving song rating to" << filename;
|
|
|
|
try {
|
|
TagParser::MediaFileInfo taginfo;
|
|
TagParser::Diagnostics diag;
|
|
TagParser::AbortableProgressFeedback progress;
|
|
#ifdef Q_OS_WIN32
|
|
taginfo.setPath(filename.toStdWString().toStdString());
|
|
#else
|
|
taginfo.setPath(QFile::encodeName(filename).toStdString());
|
|
#endif
|
|
taginfo.open(false);
|
|
|
|
taginfo.parseContainerFormat(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTracks(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
taginfo.parseTags(diag, progress);
|
|
if (progress.isAborted()) {
|
|
taginfo.close();
|
|
return false;
|
|
}
|
|
|
|
if (taginfo.tags().size() <= 0) {
|
|
taginfo.createAppropriateTags();
|
|
}
|
|
|
|
for (TagParser::Tag *tag : taginfo.tags()) {
|
|
SaveSongRatingToFile(tag, song);
|
|
}
|
|
|
|
taginfo.applyChanges(diag, progress);
|
|
taginfo.close();
|
|
|
|
for (const TagParser::DiagMessage &msg : diag) {
|
|
qLog(Debug) << QString::fromStdString(msg.message());
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch(...) {}
|
|
|
|
return false;
|
|
|
|
}
|