From ad87420546601a4b1fc1e5593fc98893c21933f8 Mon Sep 17 00:00:00 2001
From: PC-Admin <perthchat@protonmail.com>
Date: Tue, 25 Jul 2023 23:14:53 +0800
Subject: [PATCH] add function to return rdlist tags associated with a room

---
 moderation_tool.py |  4 ++++
 rdlist_commands.py | 23 +++++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/moderation_tool.py b/moderation_tool.py
index c221f4d..0eede7c 100755
--- a/moderation_tool.py
+++ b/moderation_tool.py
@@ -62,6 +62,7 @@ while pass_token == False:
 	print("#### rdlist ####\t\t\t\t\t\t74) Send test incident reports to yourself.")
 	print("50) Block all rooms with specific rdlist tags.")
 	print("51) Block all rooms with recommended rdlist tags.")
+	print("52) Get rdlist tags for a room.")
 	print("\n#### ipinfo.io ####")
 	print("60) Analyse a users country of origin.")
 	print("61) Analyse multiple users country of origin.")
@@ -158,6 +159,9 @@ while pass_token == False:
 		rdlist_commands.block_all_rooms_with_rdlist_tags(False,'','','')
 	elif menu_input == "51":
 		rdlist_commands.block_recommended_rdlist_tags()
+	elif menu_input == "52":
+		rdlist_tags = rdlist_commands.get_rdlist_tags('')
+		print(json.dumps(rdlist_tags, indent=4, sort_keys=True))
 	elif menu_input == "60":
 		ipinfo_commands.analyse_account_ip('')
 	elif menu_input == "61":
diff --git a/rdlist_commands.py b/rdlist_commands.py
index d73ebb9..d2352fc 100644
--- a/rdlist_commands.py
+++ b/rdlist_commands.py
@@ -77,6 +77,29 @@ def sync_rdlist():
 # 	# 	}
 # 	# }
 
+# A function to return the rdlist tags associated with a room
+def get_rdlist_tags(preset_internal_ID):
+	if preset_internal_ID == '':
+		internal_ID = input("\nEnter the internal id of the room you wish to query (Example: !OLkDvaYjpNrvmwnwdj:matrix.org): ")
+	elif preset_internal_ID != '':
+		internal_ID = preset_internal_ID
+
+	# Git clone the rdlist repo to ./rdlist/
+	sync_rdlist()
+
+	# Load the summaries JSON file
+	summaries_path = os.path.join("rdlist", "dist", "summaries.json")
+	with open(summaries_path, 'r') as file:
+		data = json.load(file)
+
+	# Find the room with the given id and return its tags
+	for item in data:
+		if 'room' in item and 'room_id' in item['room'] and item['room']['room_id'] == internal_ID:
+			if 'report_info' in item and 'tags' in item['report_info']:
+				return item['report_info']['tags']
+
+	return None
+
 def block_all_rooms_with_rdlist_tags(rdlist_use_recommended,preset_user_ID,preset_new_room_name,preset_message):
 	# Git clone the rdlist repo to ./rdlist/
 	sync_rdlist()