From f6c761c90b923f8cbfd39e9855a45a20a60d8075 Mon Sep 17 00:00:00 2001
From: Sorunome <mail@sorunome.de>
Date: Fri, 18 Oct 2019 13:10:54 +0200
Subject: [PATCH] room information

---
 include/matrixclient.h  | 13 +++++++
 source/matrixclient.cpp | 85 ++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 96 insertions(+), 2 deletions(-)

diff --git a/include/matrixclient.h b/include/matrixclient.h
index 080ef8c..986027c 100644
--- a/include/matrixclient.h
+++ b/include/matrixclient.h
@@ -2,6 +2,7 @@
 #define _matrixclient_h_
 
 #include <string>
+#include <vector>
 #include <3ds.h>
 #include <jansson.h>
 
@@ -13,6 +14,12 @@ public:
 	virtual std::string getSyncToken() = 0;
 };
 
+struct RoomInfo {
+	std::string name;
+	std::string topic;
+	std::string avatar;
+};
+
 class Client {
 private:
 public:
@@ -37,11 +44,17 @@ public:
 	bool login(std::string username, std::string password);
 	std::string getUserId();
 	std::string resolveRoom(std::string alias);
+	std::vector<std::string> getJoinedRooms();
+	RoomInfo getRoomInfo(std::string roomId);
+	std::string getRoomName(std::string roomId);
+	std::string getRoomTopic(std::string roomId);
+	std::string getRoomAvatar(std::string roomId);
 	std::string sendEmote(std::string roomId, std::string text);
 	std::string sendNotice(std::string roomId, std::string text);
 	std::string sendText(std::string roomId, std::string text);
 	std::string sendMessage(std::string roomId, json_t* content);
 	std::string sendEvent(std::string roomId, std::string eventType, json_t* content);
+	json_t* getStateEvent(std::string roomId, std::string type, std::string stateKey);
 	std::string sendStateEvent(std::string roomId, std::string type, std::string stateKey, json_t* content);
 	std::string redactEvent(std::string roomId, std::string eventId, std::string reason = "");
 	void startSyncLoop();
diff --git a/source/matrixclient.cpp b/source/matrixclient.cpp
index 2437c8a..56ead03 100644
--- a/source/matrixclient.cpp
+++ b/source/matrixclient.cpp
@@ -8,6 +8,8 @@
 #include <string.h>
 #include "util.h"
 #include "memorystore.h"
+#include <string>
+#include <vector>
 
 #include <sys/socket.h>
 
@@ -115,6 +117,79 @@ std::string Client::resolveRoom(std::string alias) {
 	return roomIdStr;
 }
 
+std::vector<std::string> Client::getJoinedRooms() {
+	std::vector<std::string> rooms;
+	json_t* ret = doRequest("GET", "/_matrix/client/r0/joined_rooms");
+	json_t* roomsArr = json_object_get(ret, "joined_rooms");
+	if (!roomsArr) {
+		json_decref(ret);
+		return rooms;
+	}
+	size_t index;
+	json_t* value;
+	json_array_foreach(roomsArr, index, value) {
+		rooms.push_back(json_string_value(value));
+	}
+	json_decref(ret);
+	return rooms;
+}
+
+RoomInfo Client::getRoomInfo(std::string roomId) {
+	// if we resolve the roomId here it only resolves once
+	roomId = resolveRoom(roomId);
+	RoomInfo info = {
+		name: getRoomName(roomId),
+		topic: getRoomTopic(roomId),
+		avatar: getRoomAvatar(roomId),
+	};
+	return info;
+}
+
+std::string Client::getRoomName(std::string roomId) {
+	json_t* ret = getStateEvent(roomId, "m.room.name", "");
+	if (!ret) {
+		return "";
+	}
+	json_t* name = json_object_get(ret, "name");
+	if (!name) {
+		json_decref(ret);
+		return "";
+	}
+	std::string nameStr = json_string_value(name);
+	json_decref(ret);
+	return nameStr;
+}
+
+std::string Client::getRoomTopic(std::string roomId) {
+	json_t* ret = getStateEvent(roomId, "m.room.topic", "");
+	if (!ret) {
+		return "";
+	}
+	json_t* topic = json_object_get(ret, "topic");
+	if (!topic) {
+		json_decref(ret);
+		return "";
+	}
+	std::string topicStr = json_string_value(topic);
+	json_decref(ret);
+	return topicStr;
+}
+
+std::string Client::getRoomAvatar(std::string roomId) {
+	json_t* ret = getStateEvent(roomId, "m.room.avatar", "");
+	if (!ret) {
+		return "";
+	}
+	json_t* url = json_object_get(ret, "url");
+	if (!url) {
+		json_decref(ret);
+		return "";
+	}
+	std::string urlStr = json_string_value(url);
+	json_decref(ret);
+	return urlStr;
+}
+
 std::string Client::sendEmote(std::string roomId, std::string text) {
 	json_t* request = json_object();
 	json_object_set_new(request, "msgtype", json_string("m.emote"));
@@ -164,6 +239,12 @@ std::string Client::sendEvent(std::string roomId, std::string eventType, json_t*
 	return eventIdStr;
 }
 
+json_t* Client::getStateEvent(std::string roomId, std::string type, std::string stateKey) {
+	roomId = resolveRoom(roomId);
+	std::string path = "/_matrix/client/r0/rooms/" + urlencode(roomId) + "/state/" + urlencode(type) + "/" + urlencode(stateKey);
+	return doRequest("GET", path);
+}
+
 std::string Client::sendStateEvent(std::string roomId, std::string type, std::string stateKey, json_t* content) {
 	roomId = resolveRoom(roomId);
 	std::string path = "/_matrix/client/r0/rooms/" + urlencode(roomId) + "/state/" + urlencode(type) + "/" + urlencode(stateKey);
@@ -386,7 +467,7 @@ json_t* Client::doRequestCurl(const char* method, std::string url, json_t* body)
 		return NULL;
 	}
 
-//	D printf_top("%s\n", readBuffer.c_str());
+	D printf_top("%s\n", readBuffer.c_str());
 	json_error_t error;
 	json_t* content = json_loads(readBuffer.c_str(), 0, &error);
 	if (!content) {
@@ -498,7 +579,7 @@ json_t* Client::doRequestHttpc(const char* method, std::string url, json_t* body
 
 	httpcCloseContext(&context);
 
-	//D printf_top("%s\n", buf);
+	D printf_top("%s\n", buf);
 
 	json_error_t error;
 	json_t* content = json_loads((char*)buf, 0, &error);