1 files changed, 77 insertions, 0 deletions
diff --git a/src/browser/BrowserClients.cpp b/src/browser/BrowserClients.cpp
new file mode 100755
index 000000000..9c47fdd46
--- /dev/null
+++ b/src/browser/BrowserClients.cpp
@@ -0,0 +1,77 @@
+/*
+*  Copyright (C) 2017 Sami Vänttinen <sami.vanttinen@protonmail.com>
+*  Copyright (C) 2017 KeePassXC Team <team@keepassxc.org>
+*
+*  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 3 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, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <QJsonValue>
+#include <QJsonParseError>
+#include "BrowserClients.h"
+
+BrowserClients::BrowserClients(BrowserService& browserService) :
+    m_mutex(QMutex::Recursive),
+    m_browserService(browserService)
+{
+    m_clients.reserve(1000);
+}
+
+QJsonObject BrowserClients::readResponse(const QByteArray& arr)
+{
+    QJsonObject json;
+    const QJsonObject message = byteArrayToJson(arr);
+    const QString clientID = getClientID(message);
+
+    if (!clientID.isEmpty()) {
+        const ClientPtr client = getClient(clientID);
+        if (client->browserAction) {
+            json = client->browserAction->readResponse(message);
+        }
+    }
+
+    return json;
+}
+
+QJsonObject BrowserClients::byteArrayToJson(const QByteArray& arr) const
+{
+    QJsonObject json;
+    QJsonParseError err;
+    QJsonDocument doc(QJsonDocument::fromJson(arr, &err));
+    if (doc.isObject()) {
+        json = doc.object();
+    }
+
+    return json;
+}
+
+QString BrowserClients::getClientID(const QJsonObject& json) const
+{
+    return json["clientID"].toString();
+}
+
+BrowserClients::ClientPtr BrowserClients::getClient(const QString& clientID)
+{
+    QMutexLocker locker(&m_mutex);
+    for (const auto &i : m_clients) {
+        if (i->clientID.compare(clientID, Qt::CaseSensitive) == 0) {
+            return i;
+        }
+    }
+
+    // clientID not found, create a new client
+    QSharedPointer<BrowserAction> ba = QSharedPointer<BrowserAction>::create(m_browserService);
+    ClientPtr client = ClientPtr::create(clientID, ba);
+    m_clients.push_back(client);
+    return m_clients.back();
+}