File indexing completed on 2024-05-05 16:09:01

 /*
     SPDX-FileCopyrightText: 2018 Alexander Stippich <a.stippich@gmx.net>
 
     SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
 #include "embeddedimagedata.h"
 #include "extractorcollection.h"
 #include "simpleextractionresult.h"
 #include "writedata.h"
 #include "writercollection.h"
 #include "kfilemetadata_debug.h"
 
 #include <QMimeDatabase>
 
 using namespace KFileMetaData;
 
 class Q_DECL_HIDDEN EmbeddedImageData::Private
 {
 public:
     QMimeDatabase mMimeDatabase;
     static const QStringList mMimetypes;
 };
 
 const QStringList EmbeddedImageData::Private::mMimetypes =
 {
     QStringLiteral("audio/flac"),
     QStringLiteral("audio/mp4"),
     QStringLiteral("audio/mpeg"),
     QStringLiteral("audio/mpeg3"),
     QStringLiteral("audio/ogg"),
     QStringLiteral("audio/opus"),
     QStringLiteral("audio/wav"),
     QStringLiteral("audio/x-aiff"),
     QStringLiteral("audio/x-ape"),
     QStringLiteral("audio/x-mpeg"),
     QStringLiteral("audio/x-ms-wma"),
     QStringLiteral("audio/x-musepack"),
     QStringLiteral("audio/x-opus+ogg"),
     QStringLiteral("audio/x-speex+ogg"),
     QStringLiteral("audio/x-vorbis+ogg"),
     QStringLiteral("audio/x-wav"),
     QStringLiteral("audio/x-wavpack"),
 };
 
 EmbeddedImageData::EmbeddedImageData()
     : d(std::unique_ptr<Private>(new Private()))
 {
 }
 
 EmbeddedImageData::~EmbeddedImageData()
 = default;
 
 #if KFILEMETADATA_BUILD_DEPRECATED_SINCE(5, 91)
 
 QStringList EmbeddedImageData::mimeTypes() const
 {
     return d->mMimetypes;
 }
 
 QMap<EmbeddedImageData::ImageType, QByteArray>
 EmbeddedImageData::imageData(const QString &fileUrl,
                              const EmbeddedImageData::ImageTypes types) const
 {
     const auto fileMimeType = d->mMimeDatabase.mimeTypeForFile(fileUrl).name();
     KFileMetaData::ExtractorCollection ec;
     KFileMetaData::SimpleExtractionResult result(fileUrl, fileMimeType, ExtractionResult::ExtractImageData);
 
     auto extractors = ec.fetchExtractors(fileMimeType);
     for (const auto& ex : extractors) {
     ex->extract(&result);
     }
     return result.imageData();
 }
 
 void
 EmbeddedImageData::writeImageData(const QString &fileUrl,
                              QMap<EmbeddedImageData::ImageType, QByteArray> &imageData)
 {
     const auto fileMimeType = d->mMimeDatabase.mimeTypeForFile(fileUrl).name();
     KFileMetaData::WriterCollection wc;
     KFileMetaData::WriteData data(fileUrl, fileMimeType);
     data.addImageData(imageData);
 
     auto writers = wc.fetchWriters(fileMimeType);
     for (const auto& w : writers) {
     w->write(data);
     }
 }
 #endif