/* This file is part of Clementine.
   Copyright 2010, David Sansome <me@davidsansome.com>

   Clementine 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
   (at your option) any later version.

   Clementine 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 Clementine.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef GIOLISTER_H
#define GIOLISTER_H

#include "core/scopedgobject.h"
#include "devicelister.h"

// Work around compile issue with glib >= 2.25
#ifdef signals
#undef signals
#endif

#include <gio/gio.h>

#include <QMutex>
#include <QStringList>

class GioLister : public DeviceLister {
  Q_OBJECT

 public:
  GioLister();
  ~GioLister();

  int priority() const { return 50; }

  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);
  bool DeviceNeedsMount(const QString& id);

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

  int MountDevice(const QString& id);
  void UnmountDevice(const QString& id);

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

 protected:
  void Init();

 private:
  struct DeviceInfo {
    DeviceInfo()
        : drive_removable(false),
          filesystem_size(0),
          filesystem_free(0),
          is_mountpoint(false),
          invalid_enclosing_mount(false) {}

    QString unique_id() const;
    bool is_suitable() const;

    // Convert a gio property string to a QString and free the original.
    static QString ConvertAndFree(char* str);
    // Like ConvertAndFree, but perform url decoding.
    static QString DecodeAndFree(char* str);

    void ReadDriveInfo(GDrive* drive);
    void ReadVolumeInfo(GVolume* volume);
    void ReadMountInfo(GMount* mount);

    // Only available if it's a physical drive
    ScopedGObject<GVolume> volume_;
    QString volume_name;
    QString volume_unix_device;
    QString volume_root_uri;
    QString volume_uuid;

    // Only available if it's a physical drive
    ScopedGObject<GDrive> drive_;
    QString drive_name;
    bool drive_removable;

    // Only available if it's mounted
    ScopedGObject<GMount> mount_;
    QString mount_path;
    QString mount_uri;
    QString mount_name;
    QVariantList mount_icon_names;
    QString mount_uuid;
    quint64 filesystem_size;
    quint64 filesystem_free;
    QString filesystem_type;

    bool is_mountpoint;  // unix::is-mountpoint

    bool invalid_enclosing_mount;
  };

  void VolumeAdded(GVolume* volume);
  void VolumeRemoved(GVolume* volume);

  void MountAdded(GMount* mount);
  void MountChanged(GMount* mount);
  void MountRemoved(GMount* mount);

  static void VolumeAddedCallback(GVolumeMonitor*, GVolume*, gpointer);
  static void VolumeRemovedCallback(GVolumeMonitor*, GVolume*, gpointer);

  static void MountAddedCallback(GVolumeMonitor*, GMount*, gpointer);
  static void MountChangedCallback(GVolumeMonitor*, GMount*, gpointer);
  static void MountRemovedCallback(GVolumeMonitor*, GMount*, gpointer);

  static void VolumeMountFinished(GObject* object, GAsyncResult* result,
                                  gpointer);
  static void VolumeEjectFinished(GObject* object, GAsyncResult* result,
                                  gpointer);
  static void MountEjectFinished(GObject* object, GAsyncResult* result,
                                 gpointer);
  static void MountUnmountFinished(GObject* object, GAsyncResult* result,
                                   gpointer);

  // You MUST hold the mutex while calling this function
  QString FindUniqueIdByMount(GMount* mount) const;
  QString FindUniqueIdByVolume(GVolume* volume) const;

  template <typename T>
  T LockAndGetDeviceInfo(const QString& id, T DeviceInfo::*field);

 private slots:
  void DoMountDevice(const QString& id, int request_id);

 private:
  ScopedGObject<GVolumeMonitor> monitor_;
  QList<gulong> signals_;

  QMutex mutex_;
  QMap<QString, DeviceInfo> devices_;

  QStringList scheme_blacklist_;
};

template <typename T>
T GioLister::LockAndGetDeviceInfo(const QString& id, T DeviceInfo::*field) {
  QMutexLocker l(&mutex_);
  if (!devices_.contains(id)) return T();

  return devices_[id].*field;
}

#endif  // GIOLISTER_H