From 95e0b658569ee0f5c1e953602d50c43a66fa3180 Mon Sep 17 00:00:00 2001
From: wuchangjun233 <34960661+wuchangjun233@users.noreply.github.com>
Date: Wed, 5 Apr 2023 18:27:01 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20`event`=E6=96=B0=E5=A2=9E`file=5Fty?=
 =?UTF-8?q?pe`=20(#8)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gsuid_core/handler.py | 4 ++++
 gsuid_core/models.py  | 1 +
 2 files changed, 5 insertions(+)

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):