dfdsf

Provider/matrix.py

@@ -0,0 +1,94 @@
+from __future__ import annotations
+
+import mimetypes
+from pathlib import Path
+from typing import Any
+
+import requests
+
+from Provider._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}"