dde-application-manager/src/modules/apps/alrecorder.cpp

/*
 * Copyright (C) 2021 ~ 2022 Deepin Technology Co., Ltd.
 *
 * Author:     weizhixiang <weizhixiang@uniontech.com>
 *
 * Maintainer: weizhixiang <weizhixiang@uniontech.com>
 *
 * 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
 * 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 <http://www.gnu.org/licenses/>.
 */

#include "alrecorder.h"
#include "dfwatcher.h"

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

const QString userAppsCfgDir = QDir::homePath() + "/.config/deepin/dde-daemon/apps/";

AlRecorder::AlRecorder(DFWatcher *_watcher, QObject *parent)
 : QObject (parent)
 , watcher(_watcher)
 , mutex(QMutex(QMutex::NonRecursive))
{
    QDBusConnection con = QDBusConnection::sessionBus();
    if (!con.registerService("org.deepin.daemon.AlRecorder1"))
    {
        qWarning() << "register service AlRecorder1 error:" << con.lastError().message();
        return;
    }

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

    connect(watcher, &DFWatcher::Event, this, &AlRecorder::onDFChanged, Qt::QueuedConnection);
    Q_EMIT ServiceRestarted();
}

AlRecorder::~AlRecorder()
{
    QDBusConnection::sessionBus().unregisterObject("/org/deepin/daemon/AlRecorder1");
}

// 获取未启动应用列表
QMap<QString, QStringList> AlRecorder::GetNew()
{
    QMap<QString, QStringList> ret;
    QMutexLocker locker(&mutex);
    for (auto is = subRecoders.begin(); is != subRecoders.end(); is++) {
        QStringList apps;
        for (auto il = is.value().launchedMap.begin(); il != is.value().launchedMap.end(); il++) {
            if (!il.value()) // 未启动应用
                apps.push_back(il.key());
        }

        if (apps.size() > 0)
            ret[is.key()] = apps;
    }

    return ret;
}

// 标记应用已启动状态
void AlRecorder::MarkLaunched(const QString &filePath)
{
    if (!filePath.endsWith(".desktop"))
        return;

    QFileInfo info;
    QMutexLocker locker(&mutex);
    for (auto sri = subRecoders.begin(); sri != subRecoders.end(); sri++) {
        if (!filePath.contains(sri.key()))
            continue;

        info.setFile(filePath);
        QString name = info.baseName();
        for (auto li = sri.value().launchedMap.begin(); li != sri.value().launchedMap.end(); li++) {
            if (li.key() == name && !li.value()) {
                li.value() = true;
                goto end;
            }
        }
    }

    // 根据filePath匹配到唯一应用
    end:
    if (info.isDir()) {
        saveStatusFile(info.absolutePath() + "/");
        Q_EMIT Launched(filePath);
    }
}

// 记录Launcher服务卸载应用信息, 终端卸载不会调用该函数， desktopFiles为绝对路径
void AlRecorder::UninstallHints(const QStringList &desktopFiles)
{
    QMutexLocker locker(&mutex);
    for (auto desktop : desktopFiles) {
        for (auto sri = subRecoders.begin(); sri != subRecoders.end(); sri++) {
            if (!desktop.contains(sri.key()))
                continue;

            QFileInfo info(desktop);
            sri.value().uninstallMap[info.baseName()] = true;
        }
    }
}

// 监控目录
void AlRecorder::WatchDirs(const QStringList &dataDirs)
{
    for (auto dirPath : dataDirs) {
        if (subRecoders.contains(dirPath))
            continue;

        QDir dir(dirPath);
        QStringList files = dir.entryList(QDir::Files);
        for (auto &file : files)
            file = dirPath + file;

        // 监听目录和文件
        watcher->addDir(dirPath);
        if (files.size() > 0)
            watcher->addPaths(files);

        // 初始化对应目录和应用信息
        initSubRecoder(dirPath);
    }
}

// 初始化应用目录记录
void AlRecorder::initSubRecoder(const QString &dirPath)
{
    subRecorder sub;
    QByteArray encryText = QCryptographicHash::hash(dirPath.toLatin1(), QCryptographicHash::Md5);
    QString statusFile = userAppsCfgDir + "launched-" + encryText.toHex() + ".csv";

    // 读取App状态记录
    QMap<QString, bool> launchedApp;
    QFile file(statusFile);
    if (file.open(QIODevice::ReadWrite | QIODevice::Text)) {
        while (!file.atEnd()){
            QString line(file.readLine());
            QStringList strs = line.split(",");
            if (strs.length() != 2)
                continue;

            if (strs[1].size() > 0 && strs[1][0] == 't')
                launchedApp[strs[0]] = true;
            else
                launchedApp[strs[0]] = false;
        }
        file.close();
    } else {
        // 读取app desktop
        QDir dir(dirPath);
        QStringList files = dir.entryList(QDir::Files);
        QStringList apps;
        for (QString file : files) {
            if (!file.endsWith(".desktop"))
                continue;

            int index = file.lastIndexOf(".");
            file.truncate(index);
            qInfo() << "entry =" << file;
            apps.push_back(file);
        }

        // fallback 都打开过
        for (auto app : apps)
            launchedApp[app] = true;
    }

    sub.statusFile = statusFile;
    sub.launchedMap = launchedApp;
    subRecoders[dirPath] = sub;
}

void AlRecorder::onDFChanged(const QString &filePath, uint32_t op)
{
    QFileInfo info(filePath);
    QString dirPath = info.absolutePath() + "/";
    QString name = info.baseName();
    subRecorder &sub = subRecoders[dirPath];
    QMap<QString, bool> &launchedMap = sub.launchedMap;

    // 过滤文件， 仅保留.desktop类型
    if (!filePath.endsWith(".desktop"))
        return;

    switch (op) {
    case DFWatcher::event::Add:
        qInfo() << "AlRecorder: Add " << filePath;
        if (!launchedMap.contains(name)) {
            bool launched = false;
            if (sub.removedLaunchedMap.find(name) != sub.removedLaunchedMap.end()) {
                launched = sub.removedLaunchedMap[name];
                sub.removedLaunchedMap.remove(name);
            }
            launchedMap[name] = launched;
        }
        sub.uninstallMap.remove(name);
        saveStatusFile(dirPath); // 刷新状态文件
        break;
    case DFWatcher::event::Del:
        qInfo() << "AlRecorder: Del" << filePath;
        if (launchedMap.contains(name)) {
            if (!sub.uninstallMap.contains(name))
                sub.removedLaunchedMap[name] = launchedMap[name];

            launchedMap.remove(name);
        }
        saveStatusFile(dirPath); //刷新状态文件
        break;
    case DFWatcher::event::Mod:
        break;
    }
}

void AlRecorder::saveStatusFile(const QString &dirPath)
{
    subRecorder sub = subRecoders[dirPath];
    QString tmpFile = sub.statusFile + "_tmp";
    QFile fp(tmpFile);
    bool ok = false;
    qInfo() << "saveStatusFile=" << dirPath << "create file=" << tmpFile;
    if (fp.open(QIODevice::ReadWrite | QIODevice::Text)) {
        QTextStream out(&fp);
        out << "# " << dirPath << endl;
        for (auto rl = sub.launchedMap.begin(); rl != sub.launchedMap.end(); rl++) {
            out << rl.key() << "," << ((rl.value() == true) ? "t" : "f") << endl;
        }
        ok = true;
        fp.close();
    }

    // 覆盖原文件
    QFile::remove(sub.statusFile);
    QFile::rename(tmpFile, sub.statusFile);
    Q_EMIT StatusSaved(dirPath, sub.statusFile, ok);
}