57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from typing import Annotated
|
|
|
|
import asyncpg
|
|
from dishka.integrations.fastapi import DishkaRoute, FromDishka
|
|
from fastapi import APIRouter, HTTPException, Query
|
|
from fastapi.responses import FileResponse
|
|
|
|
from utils.files import content_disposition, story_file_name
|
|
from utils.read import peers
|
|
from utils.read.models import DEFAULT_LIMIT, Page, StoryView
|
|
from utils.storage import ContentAddressedStorage
|
|
|
|
router = APIRouter(prefix="/api", tags=["stories"], route_class=DishkaRoute)
|
|
|
|
AccountId = Annotated[int, Query()]
|
|
|
|
_STORY_MIME = {"photo": "image/jpeg", "video": "video/mp4"}
|
|
|
|
|
|
@router.get("/stories")
|
|
async def list_stories(
|
|
pool: FromDishka[asyncpg.Pool],
|
|
account_id: AccountId,
|
|
peer_id: Annotated[int | None, Query()] = None,
|
|
limit: Annotated[int, Query()] = DEFAULT_LIMIT,
|
|
offset: Annotated[int, Query()] = 0,
|
|
) -> list[StoryView]:
|
|
return await peers.get_stories(
|
|
pool, account_id, Page(limit=limit, offset=offset), peer_id=peer_id
|
|
)
|
|
|
|
|
|
@router.get("/stories/{peer_id}/{story_id}/media")
|
|
async def serve_story_media(
|
|
pool: FromDishka[asyncpg.Pool],
|
|
storage: FromDishka[ContentAddressedStorage],
|
|
peer_id: int,
|
|
story_id: int,
|
|
account_id: AccountId,
|
|
download: Annotated[bool, Query()] = False,
|
|
) -> FileResponse:
|
|
story = await peers.get_story(pool, account_id, peer_id, story_id)
|
|
if story is None:
|
|
raise HTTPException(status_code=404, detail="story not found")
|
|
if not story.downloaded or story.storage_key is None:
|
|
raise HTTPException(status_code=409, detail="story media not downloaded")
|
|
headers = {}
|
|
if download:
|
|
headers["Content-Disposition"] = content_disposition(
|
|
story_file_name(peer_id, story_id, story.media_kind), attachment=True
|
|
)
|
|
return FileResponse(
|
|
storage.url(story.storage_key),
|
|
media_type=_STORY_MIME.get(story.media_kind or "", "application/octet-stream"),
|
|
headers=headers,
|
|
)
|