From ee7f1f8f5e3db37813d21c07967c4eda03f584a2 Mon Sep 17 00:00:00 2001
From: Eric <xplzv@126.com>
Date: Thu, 27 Apr 2023 19:31:58 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E6=94=AF=E6=8C=81=20mp4=20=E5=BD=95?=
 =?UTF-8?q?=E5=83=8F=E6=96=87=E4=BB=B6=E4=B8=8A=E4=BC=A0=E5=92=8C=E6=96=B0?=
 =?UTF-8?q?=E5=A2=9E=20video=20worker=20=E7=B1=BB=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/terminal/api/session/session.py    | 2 ++
 apps/terminal/const.py                  | 1 +
 apps/terminal/models/session/session.py | 4 ++--
 apps/terminal/serializers/session.py    | 2 +-
 4 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/apps/terminal/api/session/session.py b/apps/terminal/api/session/session.py
index 13cd62b7a..0661536f7 100644
--- a/apps/terminal/api/session/session.py
+++ b/apps/terminal/api/session/session.py
@@ -186,6 +186,8 @@ class SessionReplayViewSet(AsyncApiMixin, viewsets.ViewSet):
             tp = 'guacamole'
         if url.endswith('.cast.gz'):
             tp = 'asciicast'
+        if url.endswith('.replay.mp4'):
+            tp = 'mp4'
 
         download_url = reverse('api-terminal:session-replay-download', kwargs={'pk': session.id})
         data = {
diff --git a/apps/terminal/const.py b/apps/terminal/const.py
index fd0421427..cfec290c7 100644
--- a/apps/terminal/const.py
+++ b/apps/terminal/const.py
@@ -48,6 +48,7 @@ class TerminalType(TextChoices):
     magnus = 'magnus', 'Magnus'
     razor = 'razor', 'Razor'
     tinker = 'tinker', 'Tinker'
+    video_worker = 'video_worker', 'Video Worker'
 
     @classmethod
     def types(cls):
diff --git a/apps/terminal/models/session/session.py b/apps/terminal/models/session/session.py
index 2e0f3a5c9..5ddd6e647 100644
--- a/apps/terminal/models/session/session.py
+++ b/apps/terminal/models/session/session.py
@@ -48,8 +48,8 @@ class Session(OrgModelMixin):
 
     upload_to = 'replay'
     ACTIVE_CACHE_KEY_PREFIX = 'SESSION_ACTIVE_{}'
-    SUFFIX_MAP = {1: '.gz', 2: '.replay.gz', 3: '.cast.gz'}
-    DEFAULT_SUFFIXES = ['.replay.gz', '.cast.gz', '.gz']
+    SUFFIX_MAP = {1: '.gz', 2: '.replay.gz', 3: '.cast.gz', 4: '.replay.mp4'}
+    DEFAULT_SUFFIXES = ['.replay.gz', '.cast.gz', '.gz', '.replay.mp4']
 
     # Todo: 将来干掉 local_path, 使用 default storage 实现
     def get_all_possible_local_path(self):
diff --git a/apps/terminal/serializers/session.py b/apps/terminal/serializers/session.py
index 572c1ecc3..67eea06be 100644
--- a/apps/terminal/serializers/session.py
+++ b/apps/terminal/serializers/session.py
@@ -61,7 +61,7 @@ class SessionDisplaySerializer(SessionSerializer):
 
 class ReplaySerializer(serializers.Serializer):
     file = serializers.FileField(allow_empty_file=True)
-    version = serializers.IntegerField(write_only=True, required=False, min_value=2, max_value=3)
+    version = serializers.IntegerField(write_only=True, required=False, min_value=2, max_value=4)
 
 
 class SessionJoinValidateSerializer(serializers.Serializer):