95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
|
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}"
|