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

#include "volume-manager.h"
#include <mutex>
#include <QMutexLocker>
#include <QDebug>
static std::once_flag flag;
static VolumeManager *global_intance = nullptr;

Volume::Volume()
{
}

Volume::Volume(const QString& device)
{
    m_device = device;
}

QString Volume::device()
{
    return m_device;
}

QStringList Volume::duplicatesMountPoints() const
{
    QStringList tmp;
    for (auto iter = m_mountPoints.constBegin(); iter != m_mountPoints.constEnd(); ++iter) {
        if(iter.value().isEmpty()) {
            tmp.append(iter.key());
        }
    }
    if(tmp.size() == 1) {
        tmp.clear();
    }
    return tmp;
}

QMap<QString, QString> Volume::subVolumes() const
{
    QMap<QString, QString> tmp;
    for (auto iter = m_mountPoints.constBegin(); iter != m_mountPoints.constEnd(); ++iter) {
        if(!iter.value().isEmpty()) {
            tmp.insert(iter.key(), iter.value());
        }
    }
    return tmp;
}

QStringList Volume::mountPoints() const
{
    return m_mountPoints.keys();
}

void Volume::setDevice(const QString& device)
{
    m_device = device;
}

void Volume::addMountPoint(const QString &mountPoint, const QString &subVolume)
{
    m_mountPoints.insert(mountPoint, subVolume);
}

VolumeManager *VolumeManager::self()
{
    std::call_once(flag, [ & ] {
        global_intance = new VolumeManager();
    });
    return global_intance;
}

VolumeManager::VolumeManager(QObject *parent)
    : QObject{parent}
{
    this->refresh();
    m_volumeMonitor = g_volume_monitor_get();
    if (!m_volumeMonitor) {
        qDebug() << "Fail to init volume monitor";
        return;
    }

    m_mountAddHandle = g_signal_connect(m_volumeMonitor, "mount-added", G_CALLBACK(mountAddCallback), this);
    m_mountRemoveHandle = g_signal_connect(m_volumeMonitor, "mount-removed", G_CALLBACK(mountRemoveCallback), this);
}

VolumeManager::~VolumeManager()
{
    if(m_volumeMonitor){
        g_signal_handler_disconnect(m_volumeMonitor, m_mountAddHandle);
        g_signal_handler_disconnect(m_volumeMonitor, m_mountRemoveHandle);
        g_object_unref(m_volumeMonitor);
        m_volumeMonitor = nullptr;
    }
}

QVector<QStringList> VolumeManager::getDuplicates()
{
    QMutexLocker locker(&m_mutex);
    QVector<QStringList> tmp;
    for (auto iter = m_volumes.constBegin(); iter != m_volumes.constEnd(); ++iter) {
        QStringList duplicates = iter.value().duplicatesMountPoints();
        if(!duplicates.isEmpty()) {
            tmp.append(duplicates);
        }
    }
    return tmp;

}

QVector<Volume> VolumeManager::volumesHaveSubVolumes()
{
    QMutexLocker locker(&m_mutex);
    QVector<Volume> tmp;
    for (auto iter = m_volumes.constBegin(); iter != m_volumes.constEnd(); ++iter) {
        if(!iter.value().subVolumes().isEmpty()) {
            tmp.append(iter.value());
        }
    }
    return tmp;
}

QVector<Volume> VolumeManager::volumes()
{
    QMutexLocker locker(&m_mutex);
    return m_volumes.values().toVector();
}

void VolumeManager::refresh()
{
    QMutexLocker locker(&m_mutex);
    m_volumes.clear();
    //遍历当前系统所有挂载设备
    for (const QStorageInfo &storage: QStorageInfo::mountedVolumes()) {
        if (storage.isValid() and storage.isReady()) {
            Volume volume(storage.device());
            if(m_volumes.contains(storage.device())) {
                volume = m_volumes.value(storage.device());
            }
            volume.addMountPoint(storage.rootPath(), storage.subvolume());
            m_volumes.insert(storage.device(), volume);
        }
    }
    Q_EMIT VolumeDataUpdated();
}

void VolumeManager::mountAddCallback(GVolumeMonitor *monitor, GMount *gmount, VolumeManager *pThis)
{
    Q_UNUSED(monitor)
    Q_UNUSED(gmount)
    //TODO 识别U盘等移动设备
    pThis->refresh();
}

void VolumeManager::mountRemoveCallback(GVolumeMonitor *monitor, GMount *gmount, VolumeManager *pThis)
{
    Q_UNUSED(monitor)
    Q_UNUSED(gmount)
    pThis->refresh();
}