// SPDX-FileCopyrightText: 2018 - 2023 UnionTech Software Technology Co., Ltd.
//
// SPDX-License-Identifier: GPL-3.0-or-later

#include "dfwatcher.h"

#include <QDir>
#include <QDBusConnection>
#include <QDBusError>
#include <QtDebug>

const QString dfSuffix = ".desktop";
const QString configSuffix = ".json";

 DFWatcher::DFWatcher(QObject *parent)
  : QObject (parent)
  , watcher(new QFileSystemWatcher(this))
 {
     QDBusConnection con = QDBusConnection::sessionBus();
     if (!con.registerService("org.deepin.dde.DFWatcher1"))
     {
         qInfo() << "register service DFWatcher error:" << con.lastError().message();
         return;
     }

     if (!con.registerObject("/org/deepin/dde/DFWatcher1", this, QDBusConnection::ExportAllSlots | QDBusConnection::ExportAllSignals))
     {
         qInfo() << "register object DFWatcher error:" << con.lastError().message();
         return;
     }

     connect(watcher, &QFileSystemWatcher::fileChanged, this, &DFWatcher::fileUpdated, Qt::QueuedConnection);
     connect(watcher, &QFileSystemWatcher::directoryChanged, this, &DFWatcher::directoryUpdated, Qt::QueuedConnection);
 }

 DFWatcher::~DFWatcher()
 {
     QDBusConnection::sessionBus().unregisterObject("/org/deepin/dde/DFWatcher1");
 }

 void DFWatcher::addDir(const QString &path)
 {
    // 记录当前目录内容
    qInfo() << "addDir :" << path;
    const QDir dir(path);
    dirContentMap[path] = dir.entryList(QDir::NoDotAndDotDot | QDir::AllDirs | QDir::Files, QDir::DirsFirst);

    watcher->addPath(path);
 }

 void DFWatcher::addPaths(const QStringList &paths)
 {
    watcher->addPaths(paths);
 }

 QStringList DFWatcher::files()
 {
     return watcher->files();
 }

 void DFWatcher::removePath(const QString &filepath)
 {
    watcher->removePath(filepath);
 }

 void DFWatcher::fileUpdated(const QString &filePath)
 {
     qInfo() << "event: modify filepath=" << filePath;
     if (filePath.endsWith(dfSuffix) || filePath.endsWith(configSuffix))
        Q_EMIT Event(filePath, int(event::Mod));
 }

 void DFWatcher::directoryUpdated(const QString &dirPath)
 {
    QStringList recoderedContent = dirContentMap[dirPath];

    const QDir dir(dirPath);
    QStringList newEntries = dir.entryList(QDir::NoDotAndDotDot  | QDir::AllDirs | QDir::Files, QDir::DirsFirst);

    QSet<QString> newDirSet = QSet<QString>::fromList(newEntries);
    QSet<QString> currentDirSet = QSet<QString>::fromList(recoderedContent);

    // 添加的文件
    QSet<QString> newFiles = newDirSet - currentDirSet;
    QStringList newFile = newFiles.toList();

    // 移除的文件
    QSet<QString> deletedFiles = currentDirSet - newDirSet;
    QStringList deleteFile = deletedFiles.toList();

    // 更新目录记录
    dirContentMap[dirPath] = newEntries;

    // 新增文件
    if (newFile.size() > 0) {
        for (auto &file : newFile) {
            //qInfo() << "event: add filepath=" << file;
            if (file.endsWith(dfSuffix)) {
                Q_EMIT Event(dirPath + file, event::Add);
            }
        }
    }

    // 移除文件
    if (deleteFile.size() > 0) {
        for (auto &file : deleteFile) {
            //qInfo() << "event: del filepath=" << file;
            if (file.endsWith(dfSuffix)) {
                Q_EMIT Event(dirPath + file, event::Del);
            }
        }
    }
 }