ukui-search/libsearch/index/file-reader.cpp

/*
 * Copyright (C) 2020, KylinSoft Co., Ltd.
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * Authors: zhangpengfei <zhangpengfei@kylinos.cn>
 *
 */
#include "file-reader.h"
#include <ocr-utils.h>
#include <mime-utils.h>
#include <mutex>
#include "file-extraction-result.h"
#include "common.h"
#include "file-indexer-config.h"

using namespace UkuiSearch;
FileReader *g_instance = nullptr;
std::once_flag g_instanceFlag;
FileReader *FileReader::getInstance()
{
    std::call_once(g_instanceFlag, [] () {
        g_instance = new FileReader;
    });
    return g_instance;
}

FileReader::FileReader()
= default;
void FileReader::getTextContent(const QString &path, QString &textContent, const QString &suffix)
{
    if(FileIndexerConfig::getInstance()->ocrContentIndexTarget()[suffix]) {
        textContent = UkuiFileMetadata::OcrUtils::getTextInPicture(path);
        return;
    }
    QString mimeType = UkuiFileMetadata::MimeUtils::strictMimeType(path, {}).name();
    QList<UkuiFileMetadata::Extractor*> extractors = m_extractorManager.fetchExtractors(mimeType);
    FileExtractionResult result(path, mimeType, UkuiFileMetadata::ExtractionResult::Flag::ExtractPlainText);
    for(auto extractor : extractors) {
        extractor->extract(&result);
        if(!result.text().isEmpty()) {
            textContent = result.text();
            break;
        }
    }
}

QImage FileReader::getThumbnail(const QString &path, const QSize &targetSize, qreal dpr)
{
    QString mimeType = UkuiFileMetadata::MimeUtils::strictMimeType(path, {}).name();
    QList<UkuiFileMetadata::Extractor*> extractors = m_extractorManager.fetchExtractors(mimeType);
    FileExtractionResult result(path, mimeType, UkuiFileMetadata::ExtractionResult::Flag::ExtractThumbnail);
    result.setThumbnailRequest(UkuiFileMetadata::ThumbnailRequest(targetSize, dpr));
    for(auto extractor : extractors) {
        extractor->extract(&result);
        if(result.thumbnail().isValid()) {
            return result.thumbnail().image();
        }
    }
    return {};
}