1from dataclasses import dataclass
2 
3from aiogram.enums import MessageOriginType
4from aiogram.types import Message
5 
6 
7@dataclass(slots=True, frozen=True)
8class ForwardOriginDTO:
9    origin_type: str
10    chat_id: int | None = None
11    user_id: int | None = None
12    display_name: str | None = None
13 
14 
15def extract_forward_origin(message: Message) -> ForwardOriginDTO | None:
16    origin = message.forward_origin
17    if origin is None:
18        return None
19 
20    if origin.type == MessageOriginType.USER:
21        return ForwardOriginDTO(
22            origin_type=origin.type,
23            user_id=origin.sender_user.id,
24            display_name=origin.sender_user.full_name,
25        )
26    if origin.type == MessageOriginType.HIDDEN_USER:
27        return ForwardOriginDTO(
28            origin_type=origin.type,
29            display_name=origin.sender_user_name,
30        )
31    if origin.type == MessageOriginType.CHAT:
32        return ForwardOriginDTO(
33            origin_type=origin.type,
34            chat_id=origin.sender_chat.id,
35            display_name=origin.sender_chat.title,
36        )
37    if origin.type == MessageOriginType.CHANNEL:
38        return ForwardOriginDTO(
39            origin_type=origin.type,
40            chat_id=origin.chat.id,
41            display_name=origin.chat.title,
42        )
43 
44    return ForwardOriginDTO(origin_type=origin.type)
45