/* BEGIN_COMMON_COPYRIGHT_HEADER
 * (c)LGPL2+
 *
 * Razor - a lightweight, Qt based, desktop toolset
 * http://razor-qt.org
 *
 * Copyright: 2012 Razor team
 * Authors:
 *   Alexander Sokoloff <sokoloff.a@gmail.com>
 *   Petr Vanek <petr@scribus.info>
 *
 * This program or library is free software; you can redistribute it
 * and/or modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.

 * You should have received a copy of the GNU Lesser General
 * Public License along with this library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301 USA
 *
 * END_COMMON_COPYRIGHT_HEADER */


#ifndef RAZORMOUNT_RZMOUNTPROVIDERS_H
#define RAZORMOUNT_RZMOUNTPROVIDERS_H

#include <QObject>
#include "razormount.h"
#include <QtDBus/QDBusObjectPath>
class QDBusInterface;
class QDBusMessage;
class QDBusError;

class RzMountProvider : public QObject
{
          Q_OBJECT
public:
          explicit RzMountProvider(QObject *parent = 0);
          virtual ~RzMountProvider() {}

          const RazorMountDeviceList devices() const { return mDevices; }

          bool isValid() const { return mIsValid; }
          virtual void update()=0;

signals:
          void deviceAdded(RazorMountDevice *device);
          void deviceRemoved(RazorMountDevice *device);
          void deviceChanged(RazorMountDevice *device);

protected:
          RazorMountDeviceList mDevices;
          bool mIsValid;
};



class UDiskMountDevice: public RazorMountDevice
{
          Q_OBJECT
public:
          UDiskMountDevice(const QDBusObjectPath &path);
          bool update();
          QString udiskPath() const { return mUdiskPath; }

          virtual bool mount();
          virtual bool unmount();
          virtual bool eject();

private:
          QDBusInterface *mDbus;
          QString mUdiskPath;

          MediaType calcMediaType();
          QString calcLabel();
          bool calcIsExternal();
          QString calcIconName();

private slots:
          void dbusError(const QDBusError &err, const QDBusMessage &);
};



class UDiskProvider: public RzMountProvider
{
          Q_OBJECT
public:
          explicit UDiskProvider(QObject *parent = 0);

          virtual void update();

private slots:
          void dbusDeviceAdded(const QDBusObjectPath &path);
          void dbusDeviceRemoved(const QDBusObjectPath &path);
          void dbusDeviceChanged(const QDBusObjectPath &path);

private:
          QHash<QString, UDiskMountDevice*> mDevicesByPath;
          UDiskMountDevice *getDevice(const QDBusObjectPath &path) const;

          void addDevice(UDiskMountDevice *device);
          void delDevice(UDiskMountDevice *device);
};

#endif // RAZORMOUNT_RZMOUNTPROVIDERS_H
