drivemanager.cpp

#include "drivemanager.h"
#include <QDebug>

DriveManager::DriveManager(QObject *parent)
    : QObject(parent), m_timer(new QTimer(this)) {
  // Initial check
  m_currentDrives = getFilteredDrives();

  // Poll for changes every 2 seconds
  connect(m_timer, &QTimer::timeout, this, &DriveManager::checkForChanges);
  m_timer->start(2000);
}

QList<QStorageInfo> DriveManager::getDrives() const { return m_currentDrives; }

QList<QStorageInfo> DriveManager::getFilteredDrives() const {
  QList<QStorageInfo> allDrives = QStorageInfo::mountedVolumes();
  QList<QStorageInfo> filteredDrives;

  for (const QStorageInfo &drive : allDrives) {
    // Filter out loop devices and system filesystems
    if (drive.device().startsWith("/dev/loop") ||
        drive.fileSystemType() == "squashfs" ||
        drive.fileSystemType() == "tmpfs" ||
        drive.fileSystemType() == "overlay" ||
        drive.fileSystemType() == "proc" || drive.fileSystemType() == "sysfs" ||
        drive.fileSystemType() == "devtmpfs" ||
        drive.fileSystemType() == "devpts") {
      continue;
    }
    filteredDrives.append(drive);
  }
  return filteredDrives;
}

void DriveManager::checkForChanges() {
  QList<QStorageInfo> newDrives = getFilteredDrives();

  // Simple check if the number of drives changed or if paths are different
  bool changed = false;
  if (newDrives.size() != m_currentDrives.size()) {
    changed = true;
  } else {
    for (int i = 0; i < newDrives.size(); ++i) {
      if (newDrives[i].rootPath() != m_currentDrives[i].rootPath() ||
          newDrives[i].name() != m_currentDrives[i].name()) {
        changed = true;
        break;
      }
    }
  }

  if (changed) {
    m_currentDrives = newDrives;
    emit drivesUpdated(m_currentDrives);
  }
}