matrixion/src/MatrixClient.cc

586 lines
14 KiB
C++
Raw Normal View History

2017-04-06 02:06:42 +03:00
/*
* nheko Copyright (C) 2017 Konstantinos Sideris <siderisk@auth.gr>
*
* 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 <QDebug>
#include <QJsonArray>
2017-04-06 02:06:42 +03:00
#include <QJsonDocument>
#include <QJsonObject>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <QPixmap>
2017-04-06 02:06:42 +03:00
#include <QSettings>
#include <QUrl>
#include <QUrlQuery>
#include "Login.h"
#include "MatrixClient.h"
#include "Profile.h"
2017-04-08 02:53:23 +03:00
#include "Register.h"
2017-04-06 02:06:42 +03:00
MatrixClient::MatrixClient(QString server, QObject *parent)
: QNetworkAccessManager(parent)
{
server_ = "https://" + server;
api_url_ = "/_matrix/client/r0";
token_ = "";
QSettings settings;
txn_id_ = settings.value("client/transaction_id", 1).toInt();
connect(this, SIGNAL(finished(QNetworkReply *)), this, SLOT(onResponse(QNetworkReply *)));
}
2017-04-09 18:09:09 +03:00
void MatrixClient::reset() noexcept
{
next_batch_ = "";
server_ = "";
token_ = "";
txn_id_ = 0;
}
2017-04-06 02:06:42 +03:00
void MatrixClient::onVersionsResponse(QNetworkReply *reply)
{
reply->deleteLater();
qDebug() << "Handling the versions response";
auto data = reply->readAll();
auto json = QJsonDocument::fromJson(data);
qDebug() << json;
}
void MatrixClient::onLoginResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status_code = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status_code == 403) {
emit loginError("Wrong username or password");
return;
}
if (status_code == 404) {
emit loginError("Login endpoint was not found on the server");
return;
}
2017-04-08 02:53:23 +03:00
if (status_code >= 400) {
qWarning() << "Login error: " << reply->errorString();
emit loginError("An unknown error occured. Please try again.");
return;
2017-04-06 02:06:42 +03:00
}
auto data = reply->readAll();
auto json = QJsonDocument::fromJson(data);
LoginResponse response;
try {
response.deserialize(json);
emit loginSuccess(response.getUserId(),
response.getHomeServer(),
response.getAccessToken());
} catch (DeserializationException &e) {
qWarning() << "Malformed JSON response" << e.what();
emit loginError("Malformed response. Possibly not a Matrix server");
}
}
void MatrixClient::onLogoutResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status != 200) {
qWarning() << "Logout error: " << reply->errorString();
return;
}
emit loggedOut();
}
2017-04-06 02:06:42 +03:00
void MatrixClient::onRegisterResponse(QNetworkReply *reply)
{
reply->deleteLater();
2017-04-08 02:53:23 +03:00
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
auto data = reply->readAll();
auto json = QJsonDocument::fromJson(data);
if (status == 0 || status >= 400) {
if (json.isObject() && json.object().contains("error"))
emit registerError(json.object().value("error").toString());
else
emit registerError(reply->errorString());
return;
}
RegisterResponse response;
try {
response.deserialize(json);
emit registerSuccess(response.getUserId(),
response.getHomeServer(),
response.getAccessToken());
} catch (DeserializationException &e) {
qWarning() << "Register" << e.what();
emit registerError("Received malformed response.");
}
2017-04-06 02:06:42 +03:00
}
void MatrixClient::onGetOwnProfileResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status >= 400) {
qWarning() << reply->errorString();
return;
}
auto data = reply->readAll();
auto json = QJsonDocument::fromJson(data);
ProfileResponse response;
try {
response.deserialize(json);
emit getOwnProfileResponse(response.getAvatarUrl(), response.getDisplayName());
} catch (DeserializationException &e) {
2017-04-08 02:53:23 +03:00
qWarning() << "Profile:" << e.what();
2017-04-06 02:06:42 +03:00
}
}
void MatrixClient::onInitialSyncResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
qWarning() << reply->errorString();
return;
}
auto data = reply->readAll();
if (data.isEmpty())
return;
auto json = QJsonDocument::fromJson(data);
SyncResponse response;
try {
response.deserialize(json);
} catch (DeserializationException &e) {
qWarning() << "Sync malformed response" << e.what();
return;
2017-04-06 02:06:42 +03:00
}
emit initialSyncCompleted(response);
2017-04-06 02:06:42 +03:00
}
void MatrixClient::onSyncResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
2017-04-11 02:35:09 +03:00
emit syncFailed(reply->errorString());
2017-04-06 02:06:42 +03:00
return;
}
auto data = reply->readAll();
if (data.isEmpty())
return;
auto json = QJsonDocument::fromJson(data);
SyncResponse response;
try {
response.deserialize(json);
emit syncCompleted(response);
} catch (DeserializationException &e) {
qWarning() << "Sync malformed response" << e.what();
}
}
void MatrixClient::onSendTextMessageResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
qWarning() << reply->errorString();
return;
}
auto data = reply->readAll();
if (data.isEmpty())
return;
auto json = QJsonDocument::fromJson(data);
if (!json.isObject()) {
qDebug() << "Send message response is not a JSON object";
return;
}
auto object = json.object();
if (!object.contains("event_id")) {
2017-04-08 02:53:23 +03:00
qDebug() << "SendTextMessage: missing event_id from response";
2017-04-06 02:06:42 +03:00
return;
}
emit messageSent(object.value("event_id").toString(),
reply->property("roomid").toString(),
2017-04-06 02:06:42 +03:00
reply->property("txn_id").toInt());
}
void MatrixClient::onRoomAvatarResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
qWarning() << reply->errorString();
return;
}
auto img = reply->readAll();
if (img.size() == 0)
return;
auto roomid = reply->property("roomid").toString();
QPixmap pixmap;
pixmap.loadFromData(img);
emit roomAvatarRetrieved(roomid, pixmap);
}
void MatrixClient::onGetOwnAvatarResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
qWarning() << reply->errorString();
return;
}
auto img = reply->readAll();
if (img.size() == 0)
return;
QPixmap pixmap;
pixmap.loadFromData(img);
emit ownAvatarRetrieved(pixmap);
}
2017-04-28 14:56:45 +03:00
void MatrixClient::onImageResponse(QNetworkReply *reply)
{
reply->deleteLater();
int status = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
if (status == 0 || status >= 400) {
qWarning() << reply->errorString();
return;
}
auto img = reply->readAll();
if (img.size() == 0)
return;
QPixmap pixmap;
pixmap.loadFromData(img);
auto event_id = reply->property("event_id").toString();
emit imageDownloaded(event_id, pixmap);
}
2017-04-06 02:06:42 +03:00
void MatrixClient::onResponse(QNetworkReply *reply)
{
switch (reply->property("endpoint").toInt()) {
case Endpoint::Versions:
onVersionsResponse(reply);
break;
case Endpoint::Login:
onLoginResponse(reply);
break;
case Endpoint::Logout:
onLogoutResponse(reply);
break;
2017-04-06 02:06:42 +03:00
case Endpoint::Register:
onRegisterResponse(reply);
2017-04-08 02:53:23 +03:00
break;
2017-04-06 02:06:42 +03:00
case Endpoint::GetOwnProfile:
onGetOwnProfileResponse(reply);
2017-04-08 02:53:23 +03:00
break;
2017-04-28 14:56:45 +03:00
case Endpoint::Image:
onImageResponse(reply);
break;
2017-04-06 02:06:42 +03:00
case Endpoint::InitialSync:
onInitialSyncResponse(reply);
2017-04-08 02:53:23 +03:00
break;
2017-04-06 02:06:42 +03:00
case Endpoint::Sync:
onSyncResponse(reply);
2017-04-08 02:53:23 +03:00
break;
2017-04-06 02:06:42 +03:00
case Endpoint::SendTextMessage:
onSendTextMessageResponse(reply);
2017-04-08 02:53:23 +03:00
break;
case Endpoint::RoomAvatar:
onRoomAvatarResponse(reply);
break;
case Endpoint::GetOwnAvatar:
onGetOwnAvatarResponse(reply);
break;
2017-04-06 02:06:42 +03:00
default:
break;
}
}
2017-04-09 18:09:09 +03:00
void MatrixClient::login(const QString &username, const QString &password) noexcept
2017-04-06 02:06:42 +03:00
{
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/login");
QNetworkRequest request(endpoint);
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
LoginRequest body(username, password);
QNetworkReply *reply = post(request, body.serialize());
reply->setProperty("endpoint", Endpoint::Login);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::logout() noexcept
{
QUrlQuery query;
query.addQueryItem("access_token", token_);
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/logout");
endpoint.setQuery(query);
QNetworkRequest request(endpoint);
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
QJsonObject body{};
QNetworkReply *reply = post(request, QJsonDocument(body).toJson(QJsonDocument::Compact));
reply->setProperty("endpoint", Endpoint::Logout);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::registerUser(const QString &user, const QString &pass, const QString &server) noexcept
2017-04-08 02:53:23 +03:00
{
setServer(server);
QUrlQuery query;
query.addQueryItem("kind", "user");
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/register");
endpoint.setQuery(query);
QNetworkRequest request(QString(endpoint.toEncoded()));
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
RegisterRequest body(user, pass);
QNetworkReply *reply = post(request, body.serialize());
reply->setProperty("endpoint", Endpoint::Register);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::sync() noexcept
2017-04-06 02:06:42 +03:00
{
QJsonObject filter{{"room",
QJsonObject{{"ephemeral", QJsonObject{{"limit", 0}}}}},
{"presence", QJsonObject{{"limit", 0}}}};
QUrlQuery query;
query.addQueryItem("set_presence", "online");
query.addQueryItem("filter", QJsonDocument(filter).toJson(QJsonDocument::Compact));
query.addQueryItem("access_token", token_);
if (next_batch_.isEmpty()) {
qDebug() << "Sync requires a valid next_batch token. Initial sync should be performed.";
return;
}
query.addQueryItem("since", next_batch_);
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/sync");
endpoint.setQuery(query);
QNetworkRequest request(QString(endpoint.toEncoded()));
QNetworkReply *reply = get(request);
reply->setProperty("endpoint", Endpoint::Sync);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::sendTextMessage(const QString &roomid, const QString &msg) noexcept
2017-04-06 02:06:42 +03:00
{
QUrlQuery query;
query.addQueryItem("access_token", token_);
QUrl endpoint(server_);
endpoint.setPath(api_url_ + QString("/rooms/%1/send/m.room.message/%2").arg(roomid).arg(txn_id_));
endpoint.setQuery(query);
QJsonObject body{
{"msgtype", "m.text"},
{"body", msg}};
QNetworkRequest request(QString(endpoint.toEncoded()));
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
QNetworkReply *reply = put(request, QJsonDocument(body).toJson(QJsonDocument::Compact));
reply->setProperty("endpoint", Endpoint::SendTextMessage);
reply->setProperty("txn_id", txn_id_);
reply->setProperty("roomid", roomid);
incrementTransactionId();
2017-04-06 02:06:42 +03:00
}
2017-04-09 18:09:09 +03:00
void MatrixClient::initialSync() noexcept
2017-04-06 02:06:42 +03:00
{
QJsonArray excluded_event_types = {
QString("m.room.member"),
};
2017-04-06 02:06:42 +03:00
QJsonObject filter{{"room",
QJsonObject{{"timeline", QJsonObject{{"limit", 70}}},
{"state", QJsonObject{{"not_types", excluded_event_types}}},
2017-04-06 02:06:42 +03:00
{"ephemeral", QJsonObject{{"limit", 0}}}}},
{"presence", QJsonObject{{"limit", 0}}}};
QUrlQuery query;
query.addQueryItem("full_state", "true");
query.addQueryItem("set_presence", "online");
query.addQueryItem("filter", QJsonDocument(filter).toJson(QJsonDocument::Compact));
query.addQueryItem("access_token", token_);
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/sync");
endpoint.setQuery(query);
QNetworkRequest request(QString(endpoint.toEncoded()));
QNetworkReply *reply = get(request);
reply->setProperty("endpoint", Endpoint::InitialSync);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::versions() noexcept
2017-04-06 02:06:42 +03:00
{
QUrl endpoint(server_);
endpoint.setPath("/_matrix/client/versions");
QNetworkRequest request(endpoint);
QNetworkReply *reply = get(request);
reply->setProperty("endpoint", Endpoint::Versions);
}
2017-04-09 18:09:09 +03:00
void MatrixClient::getOwnProfile() noexcept
2017-04-06 02:06:42 +03:00
{
// FIXME: Remove settings from the matrix client. The class should store the user's matrix ID.
QSettings settings;
auto userid = settings.value("auth/user_id", "").toString();
QUrlQuery query;
query.addQueryItem("access_token", token_);
QUrl endpoint(server_);
endpoint.setPath(api_url_ + "/profile/" + userid);
endpoint.setQuery(query);
QNetworkRequest request(QString(endpoint.toEncoded()));
QNetworkReply *reply = get(request);
reply->setProperty("endpoint", Endpoint::GetOwnProfile);
}
void MatrixClient::fetchRoomAvatar(const QString &roomid, const QUrl &avatar_url)
{
QList<QString> url_parts = avatar_url.toString().split("mxc://");
if (url_parts.size() != 2) {
qDebug() << "Invalid format for room avatar " << avatar_url.toString();
return;
}
QString media_params = url_parts[1];
QString media_url = QString("%1/_matrix/media/r0/download/%2").arg(getHomeServer(), media_params);
QNetworkRequest avatar_request(media_url);
QNetworkReply *reply = get(avatar_request);
reply->setProperty("roomid", roomid);
reply->setProperty("endpoint", Endpoint::RoomAvatar);
}
2017-04-28 14:56:45 +03:00
void MatrixClient::downloadImage(const QString &event_id, const QUrl &url)
{
QNetworkRequest image_request(url);
QNetworkReply *reply = get(image_request);
reply->setProperty("event_id", event_id);
reply->setProperty("endpoint", Endpoint::Image);
}
void MatrixClient::fetchOwnAvatar(const QUrl &avatar_url)
{
QList<QString> url_parts = avatar_url.toString().split("mxc://");
if (url_parts.size() != 2) {
qDebug() << "Invalid format for media " << avatar_url.toString();
return;
}
QString media_params = url_parts[1];
QString media_url = QString("%1/_matrix/media/r0/download/%2").arg(getHomeServer(), media_params);
QNetworkRequest avatar_request(media_url);
QNetworkReply *reply = get(avatar_request);
reply->setProperty("endpoint", Endpoint::GetOwnAvatar);
}