added udisks2 plugin

git-svn-id: http://svn.code.sf.net/p/qmmp-dev/code/trunk/qmmp@3200 90c681e8-e032-0410-971d-27865f9a5e38

1 files changed, 264 insertions, 0 deletions

diff --git a/src/plugins/General/udisks2/udisks2plugin.cpp b/src/plugins/General/udisks2/udisks2plugin.cpp
new file mode 100644
index 000000000..3ee249c79
--- /dev/null
+++ b/src/plugins/General/udisks2/udisks2plugin.cpp
@@ -0,0 +1,264 @@
+/***************************************************************************
+ *   Copyright (C) 2013 by Ilya Kotov                                      *
+ *   forkotov02@hotmail.ru                                                 *
+ *                                                                         *
+ *   This program 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 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ *   This program 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 this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
+ ***************************************************************************/
+
+#include <QtDBus>
+#include <QActionGroup>
+#include <QApplication>
+#include <QStyle>
+#include <qmmpui/uihelper.h>
+#include <qmmpui/mediaplayer.h>
+#include <qmmpui/playlistmanager.h>
+#include <qmmpui/playlistitem.h>
+#include <qmmp/qmmp.h>
+#include "udisks2device.h"
+#include "udisks2manager.h"
+#include "udisks2plugin.h"
+
+UDisks2Plugin::UDisks2Plugin(QObject *parent) : QObject(parent)
+{
+    qDBusRegisterMetaType<QVariantMapMap>();
+    qDBusRegisterMetaType<QByteArrayList>();
+
+    m_manager = new UDisks2Manager(this);
+    m_actions = new QActionGroup(this);
+    connect(m_manager,SIGNAL(deviceAdded(QDBusObjectPath)), SLOT(addDevice(QDBusObjectPath)));
+    connect(m_manager,SIGNAL(deviceRemoved(QDBusObjectPath)), SLOT(removeDevice(QDBusObjectPath)));
+    connect(m_actions,SIGNAL(triggered (QAction *)), SLOT(processAction(QAction *)));
+    //load settings
+    QSettings settings(Qmmp::configFile(), QSettings::IniFormat);
+    settings.beginGroup("UDisks2");
+    m_detectCDA = settings.value("cda", true).toBool();
+    m_detectRemovable = settings.value("removable", true).toBool();
+    m_addTracks = false; //do not load tracks on startup
+    m_addFiles = false;
+    //find existing devices
+    QList<QDBusObjectPath> devs = m_manager->findAllDevices();
+    foreach(QDBusObjectPath o, devs)
+        addDevice(o);
+    //load remaining settings
+    m_addTracks = settings.value("add_tracks", false).toBool();
+    m_removeTracks = settings.value("remove_tracks", false).toBool();
+    m_addFiles = settings.value("add_files", false).toBool();
+    m_removeFiles = settings.value("remove_files", false).toBool();
+    settings.endGroup();
+}
+
+UDisks2Plugin::~UDisks2Plugin()
+{
+}
+
+void UDisks2Plugin::removeDevice(QDBusObjectPath o)
+{
+    foreach(UDisks2Device *device, m_devices)
+    {
+        if (device->objectPath() == o)
+        {
+            m_devices.removeAll(device);
+            delete device;
+            qDebug("UDisks2Plugin: device \"%s\" removed", qPrintable(o.path()));
+            updateActions();
+            break;
+        }
+    }
+}
+
+void UDisks2Plugin::addDevice(QDBusObjectPath o)
+{
+    foreach(UDisks2Device *device, m_devices) //is it already exists?
+    {
+        if (device->objectPath() == o)
+            return;
+    }
+    UDisks2Device *device = new UDisks2Device(o, this);
+
+    //audio cd
+    if (device->isAudio())
+    {
+        if (m_detectCDA)
+        {
+            qDebug("UDisks2Plugin: device \"%s\" added (cd audio)", qPrintable(o.path()));
+            m_devices << device;
+            connect(device, SIGNAL(changed()), SLOT(updateActions()));
+            updateActions();
+        }
+        else
+            delete device;
+        return;
+    }
+
+    if(!device->isRemovable())
+    {
+        delete device;
+        return;
+    }
+
+    if (device->property("Size").toLongLong() < 17000000000LL &&
+            (device->property("IdType").toString() == "vfat" ||
+             device->property("IdType").toString() == "iso9660" ||
+             device->property("IdType").toString() == "udf" ||
+             device->property("IdType").toString() == "ext2"))
+    {
+        if (m_detectRemovable)
+        {
+            qDebug("UDisks2Plugin: device \"%s\" added (removable)", qPrintable(o.path()));
+            m_devices << device;
+            updateActions();
+            connect(device, SIGNAL(changed()), SLOT(updateActions()));
+        }
+        else
+            delete device;
+        return;
+    }
+    delete device;
+}
+
+void UDisks2Plugin::updateActions()
+{
+    // add action for cd audio or mounted volume
+    foreach(UDisks2Device *device, m_devices)
+    {
+        QString dev_path;
+        if (device->isAudio()) //cd audio
+        {
+            dev_path = "cdda://" + device->deviceFile();
+            qDebug("dev path = %s", qPrintable(dev_path));
+        }
+        else if (device->isMounted()) //mounted volume
+            dev_path = device->mountPoints().first();
+        else
+            continue;
+
+        if (!findAction(dev_path))
+        {
+            QAction *action = new QAction(this);
+            QString actionText;
+            if (device->isAudio())
+                actionText = QString(tr("Add CD \"%1\"")).arg(device->deviceFile());
+            else
+            {
+                QString name = device->property("IdLabel").toString();
+                if (name.isEmpty())
+                    name = dev_path;
+
+                actionText = QString(tr("Add Volume \"%1\"")).arg(name);
+            }
+
+            if (device->isOptical())
+            {
+                if(device->property("IdType").toString() == "iso9660")
+                    action->setIcon(qApp->style()->standardIcon(QStyle::SP_DriveDVDIcon));
+                else
+                    action->setIcon(qApp->style()->standardIcon(QStyle::SP_DriveCDIcon));
+            }
+            else
+                action->setIcon(qApp->style()->standardIcon(QStyle::SP_DriveHDIcon));
+
+            action->setText(actionText);
+            action->setData(dev_path);
+            m_actions->addAction(action);
+            UiHelper::instance()->addAction(action, UiHelper::TOOLS_MENU);
+            addPath(dev_path);
+        }
+    }
+    // remove action if device is unmounted/removed
+    foreach(QAction *action, m_actions->actions ())
+    {
+        if (!findDevice(action))
+        {
+            m_actions->removeAction(action);
+            UiHelper::instance()->removeAction(action);
+            removePath(action->data().toString());
+            action->deleteLater();
+        }
+    }
+}
+
+void UDisks2Plugin::processAction(QAction *action)
+{
+    qDebug("UDisks2Plugin: action triggered: %s", qPrintable(action->data().toString()));
+    QString path = action->data().toString();
+    MediaPlayer::instance()->playListManager()->selectedPlayList()->add(path);
+}
+
+QAction *UDisks2Plugin::findAction(const QString &dev_path)
+{
+    foreach(QAction *action, m_actions->actions ())
+    {
+        if (action->data().toString() == dev_path)
+            return action;
+    }
+    return 0;
+}
+
+UDisks2Device *UDisks2Plugin::findDevice(QAction *action)
+{
+    foreach(UDisks2Device *device, m_devices)
+    {
+        QString dev_path;
+        if (device->isAudio())
+        {
+            dev_path = "cdda://" + device->deviceFile();
+            if (dev_path == action->data().toString())
+                return device;
+        }
+        if (device->isMounted())
+        {
+            dev_path = device->mountPoints().first();
+            if (dev_path == action->data().toString())
+                return device;
+        }
+    }
+    return 0;
+}
+
+void UDisks2Plugin::addPath(const QString &path)
+{
+    foreach(PlayListItem *item, PlayListManager::instance()->selectedPlayList()->items()) // Is it already exist?
+    {
+        if (item->url().startsWith(path))
+            return;
+    }
+
+    if (path.startsWith("cdda://") && m_addTracks)
+    {
+        PlayListManager::instance()->selectedPlayList()->add(path);
+        return;
+    }
+    else if (!path.startsWith("cdda://") && m_addFiles)
+        PlayListManager::instance()->selectedPlayList()->add(path);
+}
+
+void UDisks2Plugin::removePath(const QString &path)
+{
+    if ((path.startsWith("cdda://") && !m_removeTracks) ||
+            (!path.startsWith("cdda://") && !m_removeFiles)) //process settings
+        return;
+
+    PlayListModel *model = PlayListManager::instance()->selectedPlayList();
+
+    int i = 0;
+    while (model->count() > 0 && i < model->count())
+    {
+        if (model->item(i)->url().startsWith(path))
+            model->removeAt (i);
+        else
+            ++i;
+    }
+}