misskey2telegram_web/main.py

import io
import traceback
import uuid
import PIL.Image
from httpx import AsyncClient
from fastapi import FastAPI, Query
from fastapi.templating import Jinja2Templates
from starlette.requests import Request
from starlette.responses import RedirectResponse, StreamingResponse, HTMLResponse

from utils import gen_url, get_auth

app = FastAPI()
with open("default.jpg", "rb") as f:
    default_jpg = io.BytesIO(f.read())
templates = Jinja2Templates(directory="public/templates")


@app.get('/gen')
async def gen(
    *,
    username: str,
    host: str,
    back_host: str,
):
    session = str(uuid.uuid4())
    return RedirectResponse(gen_url(username, host, back_host, session))


@app.get('/1.jpg')
async def jpg(
    *,
    url: str,
) -> StreamingResponse:
    if "/proxy/avatar" not in url:
        return StreamingResponse(default_jpg, media_type="image/jpg")
    url = url.replace("/proxy/avatar.webp", "/proxy/avatar.png")
    # jpg png webp gif to jpg
    try:
        async with AsyncClient() as client:
            print(f"get jpg {url}")
            r = await client.get(url)
            remote = PIL.Image.open(io.BytesIO(r.content))
            remote = remote.convert("RGB")
            file_like = io.BytesIO()
            remote.save(file_like, "jpeg")
            file_like.seek(0)
            return StreamingResponse(file_like, media_type="image/jpg")
    except Exception:
        traceback.print_exc()
        return StreamingResponse(default_jpg, media_type="image/jpg")


@app.get("/{username}/{host}")
async def back_to_telegram(
    *,
    username: str,
    host: str,
    session: str,
):
    data = await get_auth(host, session)
    if data.get("ok", False):
        return RedirectResponse(f"https://t.me/{username}?start={data['token']}")
    else:
        return data


@app.get("/config", response_class=HTMLResponse)
async def debug_config_page(
    request: Request,
    bot_data: str = Query(..., title="bot_data"),
):
    user = {"command": "config", "bot_data": bot_data}
    return templates.TemplateResponse(
        "config.html", {"request": request, "user": user}
    )