Clementine-audio-player-Mac-Linux-Windows/src/devices/udisks2lister.h

#pragma once

#include <memory>

#include <QReadWriteLock>
#include <QMutex>
#include <QStringList>
#include <QDBusArgument>

#include "devicelister.h"
#include "dbus/metatypes.h"

class OrgFreedesktopDBusObjectManagerInterface;
class OrgFreedesktopUDisks2JobInterface;

class Udisks2Lister : public DeviceLister {
  Q_OBJECT

public:
  Udisks2Lister();
  ~Udisks2Lister();

  QStringList DeviceUniqueIDs();
  QVariantList DeviceIcons(const QString &id);
  QString DeviceManufacturer(const QString &id);
  QString DeviceModel(const QString &id);
  quint64 DeviceCapacity(const QString &id);
  quint64 DeviceFreeSpace(const QString &id);
  QVariantMap DeviceHardwareInfo(const QString &id);

  QString MakeFriendlyName(const QString &id);
  QList<QUrl> MakeDeviceUrls(const QString &id);

  void UnmountDevice(const QString &id);

public slots:
  void UpdateDeviceFreeSpace(const QString &id);

protected:
  void Init();

private slots:
  void DBusInterfaceAdded(const QDBusObjectPath &path, const InterfacesAndProperties &ifaces);
  void DBusInterfaceRemoved(const QDBusObjectPath &path, const QStringList &ifaces);
  void JobCompleted(bool success, const QString &message);

private:
  bool isPendingJob(const QDBusObjectPath &jobPath);
  void RemoveDevice(const QDBusObjectPath &devicePath);
  QList<QDBusObjectPath> GetMountedPartitionsFromDBusArgument(const QDBusArgument &input);

  struct Udisks2Job {
    bool isMount = true;
    QList<QDBusObjectPath> mounted_partitions;
    std::shared_ptr<OrgFreedesktopUDisks2JobInterface> dbus_interface;
  };

  QMutex jobs_lock_;
  QMap<QDBusObjectPath, Udisks2Job> mounting_jobs_;

private:
  struct PartitionData {
    QString unique_id() const;

    QString dbus_path;
    QString friendly_name;

    // Device
    QString serial;
    QString vendor;
    QString model;
    quint64 capacity = 0;
    QString dbus_drive_path;

    // Paritition
    QString label;
    QString uuid;
    quint64 free_space = 0;
    QStringList mount_paths;
  };

  PartitionData ReadPartitionData(const QDBusObjectPath &path);
  void HandleFinishedMountJob(const Udisks2Lister::PartitionData &partitionData);
  void HandleFinishedUnmountJob(const Udisks2Lister::PartitionData &partitionData, const QDBusObjectPath &mountedObject);

  QReadWriteLock device_data_lock_;
  QMap<QString, PartitionData> device_data_;

private:
  std::unique_ptr<OrgFreedesktopDBusObjectManagerInterface> udisks2_interface_;

  static const QString udisks2service_;
};