move client classes into akonadi2common and add the base class for resource plugins

we can divide up libakonadi2common later once we have a full collection of classes
this makes writing code a bit simpler now as we don't have to figuer out which
libraries to link against or how class dependencies should look. when we have
more infrastructure in place this will mostly become self-evident

1 files changed, 181 insertions, 0 deletions

diff --git a/common/resourceaccess.cpp b/common/resourceaccess.cpp
new file mode 100644
index 0000000..2b58545
--- /dev/null
+++ b/common/resourceaccess.cpp
@@ -0,0 +1,181 @@
+/*
+ * Copyright (C) 2014 Aaron Seigo <aseigo@kde.org>
+ *
+ * This 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) version 3, or any
+ * later version accepted by the membership of KDE e.V. (or its
+ * successor approved by the membership of KDE e.V.), which shall
+ * act as a proxy defined in Section 6 of version 3 of the license.
+ *
+ * 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, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "resourceaccess.h"
+
+#include "common/console.h"
+#include "common/commands.h"
+#include "common/handshake_generated.h"
+#include "common/revisionupdate_generated.h"
+
+#include <QDebug>
+#include <QProcess>
+
+namespace Akonadi2
+{
+
+ResourceAccess::ResourceAccess(const QString &resourceName, QObject *parent)
+    : QObject(parent),
+      m_resourceName(resourceName),
+      m_socket(new QLocalSocket(this)),
+      m_tryOpenTimer(new QTimer(this)),
+      m_startingProcess(false)
+{
+    m_tryOpenTimer->setInterval(50);
+    m_tryOpenTimer->setSingleShot(true);
+    connect(m_tryOpenTimer, &QTimer::timeout,
+           this, &ResourceAccess::open);
+
+    log("Starting access");
+    connect(m_socket, &QLocalSocket::connected,
+            this, &ResourceAccess::connected);
+    connect(m_socket, &QLocalSocket::disconnected,
+            this, &ResourceAccess::disconnected);
+    connect(m_socket, SIGNAL(error(QLocalSocket::LocalSocketError)),
+            this, SLOT(connectionError(QLocalSocket::LocalSocketError)));
+    connect(m_socket, &QIODevice::readyRead,
+            this, &ResourceAccess::readResourceMessage);
+
+}
+
+ResourceAccess::~ResourceAccess()
+{
+
+}
+
+QString ResourceAccess::resourceName() const
+{
+    return m_resourceName;
+}
+
+bool ResourceAccess::isReady() const
+{
+    return m_socket->isValid();
+}
+
+void ResourceAccess::open()
+{
+    if (m_socket->isValid()) {
+        log("Socket valid, so aborting the open");
+        return;
+    }
+
+    m_socket->setServerName(m_resourceName);
+    log(QString("Opening %1").arg(m_socket->serverName()));
+    //FIXME: race between starting the exec and opening the socket?
+    m_socket->open();
+}
+
+void ResourceAccess::close()
+{
+    log(QString("Closing %1").arg(m_socket->fullServerName()));
+    m_socket->close();
+}
+
+void ResourceAccess::connected()
+{
+    m_startingProcess = false;
+    log(QString("Connected: ").arg(m_socket->fullServerName()));
+
+    {
+        auto name = m_fbb.CreateString(QString::number((long long)this).toLatin1());
+        auto command = Akonadi2::CreateHandshake(m_fbb, name);
+        Akonadi2::FinishHandshakeBuffer(m_fbb, command);
+        Commands::write(m_socket, Commands::HandshakeCommand, m_fbb);
+        m_fbb.Clear();
+    }
+
+    emit ready(true);
+}
+
+void ResourceAccess::disconnected()
+{
+    m_socket->close();
+    log(QString("Disconnected from %1").arg(m_socket->fullServerName()));
+    emit ready(false);
+    open();
+}
+
+void ResourceAccess::connectionError(QLocalSocket::LocalSocketError error)
+{
+    log(QString("Connection error: %2").arg(error));
+    if (m_startingProcess) {
+        if (!m_tryOpenTimer->isActive()) {
+            m_tryOpenTimer->start();
+        }
+        return;
+    }
+
+    m_startingProcess = true;
+    log(QString("Attempting to start resource ") + m_resourceName);
+    QStringList args;
+    args << m_resourceName;
+    if (QProcess::startDetached("akonadi2_synchronizer", args)) {
+        m_socket->open();
+    }
+}
+
+void ResourceAccess::readResourceMessage()
+{
+    if (!m_socket->isValid()) {
+        return;
+    }
+
+    m_partialMessageBuffer += m_socket->readAll();
+
+    // should be scheduled rather than processed all at once
+    while (processMessageBuffer()) {}
+}
+
+bool ResourceAccess::processMessageBuffer()
+{
+    static const int headerSize = (sizeof(int) * 2);
+    if (m_partialMessageBuffer.size() < headerSize) {
+        return false;
+    }
+
+    const int commandId = *(int*)m_partialMessageBuffer.constData();
+    const int size = *(int*)(m_partialMessageBuffer.constData() + sizeof(int));
+
+    if (size > m_partialMessageBuffer.size() - headerSize) {
+        return false;
+    }
+
+    switch (commandId) {
+        case Commands::RevisionUpdateCommand: {
+            auto buffer = Akonadi2::GetRevisionUpdate(m_partialMessageBuffer.constData() + headerSize);
+            log(QString("Revision updated to: %1").arg(buffer->revision()));
+            emit revisionChanged(buffer->revision());
+            break;
+        }
+        default:
+            break;
+    }
+
+    m_partialMessageBuffer.remove(0, headerSize + size);
+    return m_partialMessageBuffer.size() >= headerSize;
+}
+
+void ResourceAccess::log(const QString &message)
+{
+    Console::main()->log(m_resourceName + ": " + message);
+}
+
+}