from __future__ import annotations

import mimetypes
from pathlib import Path
from typing import Any

import requests

from ProviderCore.base import FileProvider


class Matrix(FileProvider):
	"""File provider for Matrix (Element) chat rooms."""

	def validate(self) -> bool:
		if not self.config:
			return False
		matrix_conf = self.config.get("storage", {}).get("matrix", {})
		return bool(
			matrix_conf.get("homeserver")
			and matrix_conf.get("room_id")
			and (matrix_conf.get("access_token") or matrix_conf.get("password"))
		)

	def upload(self, file_path: str, **kwargs: Any) -> str:
		path = Path(file_path)
		if not path.exists():
			raise FileNotFoundError(f"File not found: {file_path}")

		matrix_conf = self.config.get("storage", {}).get("matrix", {})
		homeserver = matrix_conf.get("homeserver")
		access_token = matrix_conf.get("access_token")
		room_id = matrix_conf.get("room_id")

		if not homeserver:
			raise Exception("Matrix homeserver missing")
		if not access_token:
			raise Exception("Matrix access_token missing")
		if not room_id:
			raise Exception("Matrix room_id missing")

		if not homeserver.startswith("http"):
			homeserver = f"https://{homeserver}"

		# Upload media
		upload_url = f"{homeserver}/_matrix/media/v3/upload"
		headers = {
			"Authorization": f"Bearer {access_token}",
			"Content-Type": "application/octet-stream",
		}

		mime_type, _ = mimetypes.guess_type(path)
		if mime_type:
			headers["Content-Type"] = mime_type

		filename = path.name

		with open(path, "rb") as handle:
			resp = requests.post(upload_url, headers=headers, data=handle, params={"filename": filename})

		if resp.status_code != 200:
			raise Exception(f"Matrix upload failed: {resp.text}")

		content_uri = resp.json().get("content_uri")
		if not content_uri:
			raise Exception("No content_uri returned")

		# Send message
		send_url = f"{homeserver}/_matrix/client/v3/rooms/{room_id}/send/m.room.message"

		# Determine message type
		msgtype = "m.file"
		ext = path.suffix.lower()

		audio_exts = {".mp3", ".flac", ".wav", ".m4a", ".aac", ".ogg", ".opus", ".wma", ".mka", ".alac"}
		video_exts = {".mp4", ".mkv", ".webm", ".mov", ".avi", ".flv", ".mpg", ".mpeg", ".ts", ".m4v", ".wmv"}
		image_exts = {".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp", ".tiff"}

		if ext in audio_exts:
			msgtype = "m.audio"
		elif ext in video_exts:
			msgtype = "m.video"
		elif ext in image_exts:
			msgtype = "m.image"

		info = {"mimetype": mime_type, "size": path.stat().st_size}
		payload = {"msgtype": msgtype, "body": filename, "url": content_uri, "info": info}

		resp = requests.post(send_url, headers=headers, json=payload)
		if resp.status_code != 200:
			raise Exception(f"Matrix send message failed: {resp.text}")

		event_id = resp.json().get("event_id")
		return f"https://matrix.to/#/{room_id}/{event_id}"