forked from mirrors/stickerpicker
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
# maunium-stickerpicker - A fast and simple Matrix sticker picker widget.
|
|
# Copyright (C) 2020 Tulir Asokan
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
from typing import Optional, TYPE_CHECKING
|
|
import json
|
|
|
|
from aiohttp import ClientSession
|
|
from yarl import URL
|
|
|
|
access_token: Optional[str] = None
|
|
homeserver_url: Optional[str] = None
|
|
|
|
upload_url: Optional[URL] = None
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import TypedDict
|
|
|
|
|
|
class MediaInfo(TypedDict):
|
|
w: int
|
|
h: int
|
|
size: int
|
|
mimetype: str
|
|
thumbnail_url: Optional[str]
|
|
thumbnail_info: Optional['MediaInfo']
|
|
|
|
|
|
class StickerInfo(TypedDict, total=False):
|
|
body: str
|
|
url: str
|
|
info: MediaInfo
|
|
id: str
|
|
msgtype: str
|
|
else:
|
|
MediaInfo = None
|
|
StickerInfo = None
|
|
|
|
|
|
async def load_config(path: str) -> None:
|
|
global access_token, homeserver_url, upload_url
|
|
try:
|
|
with open(path) as config_file:
|
|
config = json.load(config_file)
|
|
homeserver_url = config["homeserver"]
|
|
access_token = config["access_token"]
|
|
except FileNotFoundError:
|
|
print("Matrix config file not found. Please enter your homeserver and access token.")
|
|
homeserver_url = input("Homeserver URL: ")
|
|
access_token = input("Access token: ")
|
|
whoami_url = URL(homeserver_url) / "_matrix" / "client" / "r0" / "account" / "whoami"
|
|
if whoami_url.scheme not in ("https", "http"):
|
|
whoami_url = whoami_url.with_scheme("https")
|
|
user_id = await whoami(whoami_url, access_token)
|
|
with open(path, "w") as config_file:
|
|
json.dump({
|
|
"homeserver": homeserver_url,
|
|
"user_id": user_id,
|
|
"access_token": access_token
|
|
}, config_file)
|
|
print(f"Wrote config to {path}")
|
|
|
|
upload_url = URL(homeserver_url) / "_matrix" / "media" / "r0" / "upload"
|
|
|
|
|
|
async def whoami(url: URL, access_token: str) -> str:
|
|
headers = {"Authorization": f"Bearer {access_token}"}
|
|
async with ClientSession() as sess, sess.get(url, headers=headers) as resp:
|
|
resp.raise_for_status()
|
|
user_id = (await resp.json())["user_id"]
|
|
print(f"Access token validated (user ID: {user_id})")
|
|
return user_id
|
|
|
|
|
|
async def upload(data: bytes, mimetype: str, filename: str) -> str:
|
|
url = upload_url.with_query({"filename": filename})
|
|
headers = {"Content-Type": mimetype, "Authorization": f"Bearer {access_token}"}
|
|
async with ClientSession() as sess, sess.post(url, data=data, headers=headers) as resp:
|
|
return (await resp.json())["content_uri"]
|