diff --git a/gsuid_core/handler.py b/gsuid_core/handler.py index 7671d11..9648cbb 100644 --- a/gsuid_core/handler.py +++ b/gsuid_core/handler.py @@ -52,6 +52,10 @@ async def msg_process(msg: MessageReceive) -> Event: data = _msg.data.split('|') event.file_name = data[0] event.file = data[1] + if str(data[0]).startswith(('http', 'https')): + event.file_type = 'url' + else: + event.file_type = 'base64' _content.append(_msg) event.content = _content return event diff --git a/gsuid_core/models.py b/gsuid_core/models.py index 1e6dfe1..51a0ac7 100644 --- a/gsuid_core/models.py +++ b/gsuid_core/models.py @@ -31,6 +31,7 @@ class Event(MessageReceive): reply: Optional[str] = None file_name: Optional[str] = None file: Optional[str] = None + file_type: Optional[Literal['url', 'base64']] = None class MessageSend(Struct):