from maubot import Plugin, MessageEvent
from maubot.handlers import command

# Needed for configuration
from typing import Type
from mautrix.util.config import BaseProxyConfig, ConfigUpdateHelper

import aiohttp

# Ensures a running instance gets an updated config from the Maubot interface
class Config(BaseProxyConfig):
	def do_update(self, helper: ConfigUpdateHelper) -> None:
		helper.copy("command_prefix")
		helper.copy("allowed_locations_enabled")
		helper.copy("allowed_locations")
		helper.copy("allowed_senders")
		helper.copy("briar_url")
		helper.copy("briar_token")
		helper.copy("briar_sendto")
		helper.copy("send_reaction")

class ConsumerBriar(Plugin):
	# Get configuration at startup
	async def start(self) -> None:
		self.config.load_and_update()
	
	# Get config
	@classmethod
	def get_config_class(cls) -> Type[BaseProxyConfig]:
		return Config
	
	# Get !command_name setting from config to register it
	def get_command_name(self) -> str:
		return self.config["command_prefix"]
	
	# What gets called when !command_name message is sent
	@command.new(name=get_command_name, help="Report Something")
	@command.argument("message", pass_raw=True)
	async def report(self, evt: MessageEvent, message: str) -> None:
		# Split command (minus !command_name) into tokens
		tokens = message.split()
		st_desig = tokens[0].lower()
		
		# Each command must have a state/territory designation and a message
		if len(tokens) < 2: return
		
		# Check locations whitelist
		if self.config["allowed_locations_enabled"]:
			if not tokens[0].lower() in self.config["allowed_locations"]:
				return
		
		# Check allowed senders
		allowed_senders = self.config["allowed_senders"]
		
		
		if st_desig in self.config["allowed_senders"]:
			if not evt.sender in self.config['allowed_senders'][st_desig]:
				return
		
		# Sending notification to Briar
		sent = None # Indicates after the following loop if something has been sent,
		             # AND that it has been sent to all contacts
		
		# Loop through each user
		for userId in self.config["briar_sendto"].keys():
			# Reset sent status first time, as we want to detect if 
			if sent is None: sent = True
			
			url = self.config["briar_url"] + "/v1/messages/" + str(userId)
			
			body = {"text": ' '.join(tokens[1:])}
			
			headers = {'Authorization': "Bearer " + self.config["briar_token"]}
			
			async with self.http.post(url, json=body, headers=headers) as response:
				if not response.status == 200:
					sent = False
		
		if self.config["send_reaction"]:
			if sent is True:
				await evt.react("👍")
			if sent is False:
				await evt.react("👎")