ukui-search/ukui-search-service-dir-manager/dirwatcher/config.cpp

/*
 *
 * Copyright (C) 2023, 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: iaom <zhangpengfei@kylinos.cn>
 */
#include "config.h"
#include <QDir>
#include <QUrl>
#include <mutex>
#include <QFile>

static const QString HOME_PATH = QDir::homePath();
static const QString CURRENT_INDEXABLE_DIR_SETTINGS = HOME_PATH + "/.config/org.ukui/ukui-search/ukui-search-current-indexable-dir.conf";
static const QString INDEXABLE_DIR_KEY = "IndexableDir"; //兼容历史版本
static const QString CONFIG_VERSION_KEY = "ConfigVersion";
static const QString CONFIG_VERSION = "1.0";
static const QString GLOBAL_BLACK_LIST_KEY = "GlobalBlackList";
static const QStringList GLOBAL_BLACK_LIST{"/proc", "/sys", "/dev", "/tmp", "/run"};
static const QString GLOBAL_SETTINGS_GROUP = "GlobalSettings";
static const QString SEARCH_DIRS_GROUP = "SearchDirs";

static std::once_flag flag;
static Config *global_instance = nullptr;
Config *Config::self()
{
    std::call_once(flag, [ & ] {
        global_instance = new Config();
    });
    return global_instance;
}

void Config::addDir(const SearchDir &dir)
{
    m_searchDirs.append(dir);
    m_settings->beginGroup(SEARCH_DIRS_GROUP);
    m_settings->setValue(QUrl::fromLocalFile(dir.getPath()).toString(), dir.getBlackList());
    m_settings->endGroup();
    m_settings->sync();
}

QStringList Config::removeDir(const SearchDir &dir)
{
    QString url = QUrl::fromLocalFile(dir.getPath()).toString();
    QStringList blackDirs = m_settings->value(SEARCH_DIRS_GROUP + "/" + url).toStringList();
    removeDir(url);
    m_searchDirs.removeOne(dir);
    return blackDirs;
}

bool Config::isCompatibilityMode()
{
    return m_compatibilityMode;
}

QStringList Config::searchDirs()
{
    QStringList tmp;
    for(const SearchDir& dir : m_searchDirs) {
        if(QFile::exists(dir.getPath())) {
            tmp.append(dir.getPath());
        } else {
            m_searchDirs.removeOne(dir);
            removeDir(QUrl::fromLocalFile(dir.getPath()).toString());
        }
    }
    return tmp;
}

QStringList Config::blackDirs()
{
    QStringList tmp;
    for(const SearchDir& dir : m_searchDirs) {
        if(QFile::exists(dir.getPath())) {
            tmp.append(dir.getBlackList());
        } else {
            m_searchDirs.removeOne(dir);
            removeDir(QUrl::fromLocalFile(dir.getPath()).toString());
        }
    }
    return tmp;
}

QStringList Config::globalBlackList() const
{
    return m_settings->value(GLOBAL_SETTINGS_GROUP + "/" + GLOBAL_BLACK_LIST_KEY).toStringList();
}

QVector<SearchDir> Config::searchDirObjects()
{
    for(const SearchDir& dir : m_searchDirs) {
        if(!QFile::exists(dir.getPath())) {
            m_searchDirs.removeOne(dir);
            removeDir(QUrl::fromLocalFile(dir.getPath()).toString());
        }
    }
    return m_searchDirs;
}

Config::Config()
{
    m_settings = new QSettings(CURRENT_INDEXABLE_DIR_SETTINGS, QSettings::IniFormat);
    if(!(m_settings->value(GLOBAL_SETTINGS_GROUP + "/" + CONFIG_VERSION_KEY).toString() == CONFIG_VERSION)) {
        m_compatibilityMode = true;
        for(const QString& path : m_settings->value(INDEXABLE_DIR_KEY + "/" + INDEXABLE_DIR_KEY).toStringList()) {
            if(QFile::exists(path)) {
                SearchDir dir(path, false);
                m_compatibleCache.append(dir);
            }
        }
        m_settings->remove(INDEXABLE_DIR_KEY);
        m_settings->beginGroup(GLOBAL_SETTINGS_GROUP);
        m_settings->setValue(CONFIG_VERSION_KEY, CONFIG_VERSION);
        m_settings->endGroup();
    } else {
        m_settings->beginGroup(SEARCH_DIRS_GROUP);
        for(const QString& urlStr : m_settings->allKeys()) {
            QString path = QUrl(urlStr).toLocalFile();
            if(QFile::exists(path)) {
                SearchDir dir(path, false);
                dir.setBlackList(m_settings->value(urlStr).toStringList());
                m_searchDirs.append(dir);
            } else {
                removeDir(urlStr);
            }
        }
        m_settings->endGroup();
    }

    m_settings->beginGroup(GLOBAL_SETTINGS_GROUP);
    if(m_settings->value(GLOBAL_BLACK_LIST_KEY).isNull()) {
        m_settings->setValue(GLOBAL_BLACK_LIST_KEY, GLOBAL_BLACK_LIST);
    }
    m_settings->endGroup();

    m_settings->sync();
}

void Config::removeDir(const QString &url)
{
    m_settings->beginGroup(SEARCH_DIRS_GROUP);
    m_settings->remove(url);
    m_settings->endGroup();
    m_settings->sync();
}

void Config::processCompatibleCache()
{
    if(this->isCompatibilityMode()) {
        m_settings->beginGroup(SEARCH_DIRS_GROUP);
        if (m_compatibleCache.isEmpty()) {
            SearchDir defaultDir(QDir::homePath());
            if (defaultDir.error() == SearchDir::ErrorInfo::Successful) {
                m_searchDirs.append(defaultDir);
                m_settings->setValue(QUrl::fromLocalFile(defaultDir.getPath()).toString(), defaultDir.getBlackList());
            }
        } else {
            for (SearchDir dir : m_compatibleCache) {
                dir.generateBlackList();
                if (dir.error() == SearchDir::ErrorInfo::Successful) {
                    m_searchDirs.append(dir);
                    m_settings->setValue(QUrl::fromLocalFile(dir.getPath()).toString(), dir.getBlackList());
                }
            }
        }
        m_settings->endGroup();

        m_settings->sync();
        m_compatibleCache.clear();
        m_compatibilityMode = false;
    }
}