import json
import user_commands
import room_commands
import server_commands
import ipinfo_commands
import rdlist_commands
import report_commands
import hardcoded_variables

# check if homeserver url is hard coded, if not set it

if hardcoded_variables.homeserver_url == "matrix.example.org":
	homeserver_url = input("What is the URL of your server? Eg: matrix.example.org ")

# check if base url is hard coded, if not set it

if hardcoded_variables.base_url == "example.org":
	base_url = input("What is the URL of your server? Eg: example.org ")

# check if access token is hard coded, if not set it

length_access_token = len(hardcoded_variables.access_token)

if length_access_token == 0:
	access_token = input("Please enter access token for server admin account: ")

# loop menu for various moderation actions

pass_token = False
while pass_token == False:
	print("\n##########################")
	print("# MATRIX MODERATION TOOL #")
	print("##########################")
	print("\nA tool for making common Synapse moderation tasks easier. Created by @PC-Admin.")
	print("\n----------------------------------------------")
	print("\n#### User Account Commands ####\t\t\t#### Room Commands ####")
	print("1) Deactivate a user account.\t\t\t20) List details of a room.")
	print("2) Deactivate multiple user accounts.\t\t21) List the members of a room.")
	print("3) Create a user account.\t\t\t22) Export the state events of a target room.")
	print("4) Create multiple user accounts.\t\t23) List rooms in public directory.")
	print("5) Reset a users password.\t\t\t24) Remove a room from the public directory.")
	print("6) Whois user account.\t\t\t\t25) Remove multiple rooms from the public directory.")
	print("7) Whois multiple user accounts.\t\t26) Redact a room event.")
	print("8) Query user account.\t\t\t\t27) List/Download all media in a room.")
	print("9) Query multiple user accounts.\t\t28) Download media from multiple rooms.")
	print("10) List room memberships of user.\t\t29) Quarantine all media in a room.")
	print("11) Promote a user to server admin.\t\t30) Shutdown a room.")
	print("12) List all user accounts.\t\t\t31) Shutdown multiple rooms.")
	print("13) Quarantine all media a users uploaded.\t32) Delete a room.")
	print("14) Collect account data.\t\t\t33) Delete multiple rooms.")
	print("15) List account pushers.\t\t\t34) Purge the event history of a room to a specific timestamp.")
	print("16) Get rate limit of a user account.\t\t35) Purge the event history of multiple rooms to a specific timestamp.")
	print("17) Set rate limit of a user account.\t\t36) Get blocked status for room.")
	print("18) Delete rate limit of a user account.\t37) Block a room.")
	print("19) Check if user account exists.\t\t38) Unblock a room.")
	print("\n#### Server Commands ####\t\t\t\t\t#### Report Generation ####")
	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.\t72) Lookup homeserver admin contact email.")
	print("\t\t\t\t\t\t\t\t73) Send a test email.")
	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.")
	print("\nPlease enter a number from the above menu, or enter 'q' or 'e' to exit.\n")
	menu_input = input()
	if menu_input == "1":
		user_commands.deactivate_account('')
	elif menu_input == "2":
		user_commands.deactivate_multiple_accounts()
	elif menu_input == "3":
		user_commands.create_account('','')
	elif menu_input == "4":
		user_commands.create_multiple_accounts()
	elif menu_input == "5":
		user_commands.reset_password('','')
	elif menu_input == "6":
		whois_account_dict = user_commands.whois_account('')
		print(json.dumps(whois_account_dict, indent=4, sort_keys=True))
	elif menu_input == "7":
		user_commands.whois_multiple_accounts()
	elif menu_input == "8":
		query_account_dict = user_commands.query_account()
		print(json.dumps(query_account_dict, indent=4, sort_keys=True))
	elif menu_input == "9":
		user_commands.query_multiple_accounts()
	elif menu_input == "10":
		joined_rooms_dict = user_commands.list_joined_rooms('')
		print(json.dumps(joined_rooms_dict, indent=4, sort_keys=True))
	elif menu_input == "11":
		set_user_server_admin_dict = user_commands.set_user_server_admin('')
		print(json.dumps(set_user_server_admin_dict, indent=4, sort_keys=True))
	elif menu_input == "12":
		user_commands.list_accounts()
	elif menu_input == "13":
		user_commands.quarantine_users_media()
	elif menu_input == "14":
		account_data_dict = user_commands.collect_account_data('')
		print(json.dumps(account_data_dict, indent=4, sort_keys=True))
	elif menu_input == "15":
		user_commands.list_account_pushers('')
	elif menu_input == "16":
		user_commands.get_rate_limit()
	elif menu_input == "17":
		user_commands.set_rate_limit()
	elif menu_input == "18":
		user_commands.delete_rate_limit()
	elif menu_input == "19":
		user_account_exists = user_commands.check_user_account_exists('')
		if user_account_exists == True:
			print("\nUser account exists.\n")
		elif user_account_exists == False:
			print("\nUser account does not exist.\n")
	elif menu_input == "20":
		room_details_dict = room_commands.get_room_details('')
		print(json.dumps(room_details_dict, indent=4, sort_keys=True))
	elif menu_input == "21":
		room_members_dict = room_commands.get_room_members('',False)
		print(json.dumps(room_members_dict, indent=4, sort_keys=True))
	elif menu_input == "22":
		room_commands.export_room_state('','',True)
	elif menu_input == "23":
		room_commands.public_directory_rooms()
	elif menu_input == "24":
		room_commands.remove_room_from_directory('')
	elif menu_input == "25":
		room_commands.remove_multiple_rooms_from_directory()
	elif menu_input == "26":
		room_commands.redact_room_event()
	elif menu_input == "27":
		room_commands.list_and_download_media_in_room('','','','./')
	elif menu_input == "28":
		room_commands.download_media_from_multiple_rooms()
	elif menu_input == "29":
		room_commands.quarantine_media_in_room()
	elif menu_input == "30":
		room_commands.shutdown_room('','','','','','')
	elif menu_input == "31":
		room_commands.shutdown_multiple_rooms()
	elif menu_input == "32":
		room_commands.delete_room('')
	elif menu_input == "33":
		room_commands.delete_multiple_rooms()
	elif menu_input == "34":
		room_commands.purge_room_to_timestamp('','')
	elif menu_input == "35":
		room_commands.purge_multiple_rooms_to_timestamp()
	elif menu_input == "36":
		blocked_status = room_commands.get_block_status('')
		if blocked_status == True:
			print("\nRoom is blocked.\n")
		elif blocked_status == False:
			print("\nRoom is not blocked.\n")
		print(json.dumps(blocked_status, indent=4, sort_keys=True))
	elif menu_input == "37":
		room_commands.set_block_status('', True)
	elif menu_input == "38":
		room_commands.set_block_status('', False)
	elif menu_input == "40":
		server_commands.delete_block_media()
	elif menu_input == "41":
		server_commands.purge_remote_media_repo()
	elif menu_input == "42":
		server_commands.prepare_database_copy_of_multiple_rooms()
	elif menu_input == "50":
		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":
		ipinfo_commands.analyse_multiple_account_ips()
	elif menu_input == "70":
		report_commands.generate_user_report('','')
	elif menu_input == "71":
		report_commands.decrypt_zip_file()
	elif menu_input == "72":
		report_commands.lookup_homeserver_admin_email('')
	elif menu_input == "73":
		report_commands.test_send_email()
	elif menu_input == "74":
		report_commands.test_send_incident_reports()
	elif menu_input == "q" or menu_input == "Q" or menu_input == "e" or menu_input == "E":
		print("\nExiting...\n")
		pass_token = True
	else:
		print("\nIncorrect input detected, please select a number from 1 to 74!\n")