diff --git a/README.md b/README.md index 8e2ed76..101ed96 100755 --- a/README.md +++ b/README.md @@ -19,6 +19,13 @@ You can hard code the server URL, federation port and access token into the [har Your access token can be found in Element > Settings > Help & About, your user account must first be upgraded to a server admin. +This script also requires you to install the following PIP packages: +``` +pip3 install python-whois +pip3 install requests +pip3 install pyAesCrypt +``` + *** ## Upgrade user to 'server admin' diff --git a/moderation_tool.py b/moderation_tool.py index 041c531..290d58f 100755 --- a/moderation_tool.py +++ b/moderation_tool.py @@ -57,6 +57,7 @@ while pass_token == False: print("40) Delete and block a specific media.\t\t\t\t70) Generate user report.") print("41) Purge remote media repository up to a certain date.\t\t71) Decrypt user report .zip file.") print("42) Prepare database for copying events of multiple rooms.") + print("43) Lookup homeserver admin contact email.") print("\n#### rdlist ####") print("50) Block all rooms with specific rdlist tags.") print("51) Block all rooms with recommended rdlist tags.") @@ -139,6 +140,8 @@ while pass_token == False: server_commands.purge_remote_media_repo() elif menu_input == "42": server_commands.prepare_database_copy_of_multiple_rooms() + elif menu_input == "43": + server_commands.lookup_homeserver_admin_email('') elif menu_input == "50": rdlist_commands.block_all_rooms_with_rdlist_tags(False,'','','','','') elif menu_input == "51": diff --git a/server_commands.py b/server_commands.py index 05960dd..d212bcc 100644 --- a/server_commands.py +++ b/server_commands.py @@ -1,8 +1,10 @@ +import os import subprocess import csv import time -import os +import json +import whois import requests import datetime import hardcoded_variables @@ -136,3 +138,52 @@ def prepare_database_copy_of_multiple_rooms(): print(chown_command_process.stdout) print("\nThe sql query files have been generated, as postgres user in container run:\n# docker exec -it matrix-postgres /bin/bash\nbash-5.0$ export PGPASSWORD=your-db-password\nbash-5.0$ for f in /var/lib/postgresql/data/ramdisk/*/dump_room_data.sql; do psql --host=127.0.0.1 --port=5432 --username=synapse -w -f $f; done\n\nAfter copying the data to a cloud location law enforcement can access, clean up the ramdisk like so:\n# rm -r /matrix/postgres/data/ramdisk/*\n# umount /matrix/postgres/data/ramdisk") + +def lookup_homeserver_admin_email(preset_homeserver): + if preset_homeserver == '': + homeserver = input("\nEnter the base URL to collect the admin contact details (Example: matrix.org): ") + elif preset_homeserver != '': + homeserver = preset_homeserver + + # Check target homserver for MSC1929 support email + url = f"https://{homeserver}/.well-known/matrix/support" + try: + response = requests.get(url) + except requests.exceptions.RequestException as e: + print(f"Error: Unable to connect to server {homeserver}. Trying WHOIS data...") + response = None + + # If the request was successful, the status code will be 200 + if response and response.status_code == 200: + # Parse the response as JSON + data = json.loads(response.text) + + # Extract the emails from the admins field and remove duplicates + admin_emails = list({admin['email_address'] for admin in data['admins']}) + + print("Admin contact emails for " + homeserver + " are: " + str(admin_emails)) + + # Create a dictionary with homeserver as key and emails as value + email_dict = {homeserver: admin_emails} + + # Convert the dictionary to a JSON string and print it + email_json = json.dumps(email_dict, indent=4) + print("Admin contact emails for " + homeserver + " in JSON format: " + email_json) + + return email_dict, False + else: + print(f"Error: Unable to collect admin email from server {homeserver}") + print("Attempting to collect admin email from WHOIS data...") + + # Get WHOIS data + try: + w = whois.whois(homeserver) + if w.emails: + print("Admin contact email(s) for " + homeserver + " are: " + str(w.emails)) + return {homeserver: list(w.emails)}, True + else: + print(f"Error: Unable to collect admin email from WHOIS data for {homeserver}") + return None, False + except: + print(f"Error: Unable to collect WHOIS data for {homeserver}") + return None, False