nikita
4194f3da48
- Updated mab to v0.5.0 - Added `!short` command to add named tokens - Fixed `last_access_at` field not being updated
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""API Routes"""
|
|
|
|
import html
|
|
import traceback
|
|
|
|
from fastapi import APIRouter, Request, Depends
|
|
from fastapi import HTTPException, status
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from mab import MatrixBot
|
|
from database import Database
|
|
from datatypes import *
|
|
|
|
router = APIRouter(prefix="/api")
|
|
|
|
#
|
|
# MODELS
|
|
#
|
|
class ModelGetNotify(BaseModel):
|
|
channel: str = Field(
|
|
...,
|
|
min_length=11,
|
|
max_length=11,
|
|
description="Notification channel (use `!info` bot command)"
|
|
)
|
|
text: str = Field(
|
|
...,
|
|
min_length=1,
|
|
max_length=1024,
|
|
description="Text to use as notification body"
|
|
)
|
|
service: str | None = Field(
|
|
None,
|
|
min_length=1,
|
|
max_length=1024,
|
|
description="Service name to use (ignored if token has name)"
|
|
)
|
|
|
|
#
|
|
# ENDPOINTS
|
|
#
|
|
@router.get("/notify")
|
|
async def _get_notify(request: Request, params: ModelGetNotify = Depends()):
|
|
# improve readability
|
|
database: Database = request.app.state.database
|
|
matrix: MatrixBot = request.app.state.matrix
|
|
token: ObjectToken = request.state.token
|
|
# check if room exists
|
|
room = await database.room_get(params.channel)
|
|
if room is None:
|
|
raise HTTPException(status.HTTP_200_OK, "No such channel")
|
|
# prepare service name
|
|
service = token.name or params.service or "Unnamed"
|
|
# prepare text of the notification
|
|
text = f"<strong>Уведомление от службы <code>{html.escape(service)}</code></strong>"
|
|
text += "<br>" * 2
|
|
text += html.escape(params.text)
|
|
# try to send the notification
|
|
try:
|
|
notification_id = await matrix.send_text(
|
|
room.matrix_id,
|
|
text,
|
|
is_html=True
|
|
)
|
|
except:
|
|
traceback.print_exc()
|
|
raise HTTPException(status.HTTP_200_OK, "Failed to send matrix message")
|
|
# success
|
|
return {"notification_id": notification_id} |