From 0a6a7e08d510f41d995cff8d281c9574422e1f9d Mon Sep 17 00:00:00 2001
From: MrOrz <johnsonliang7@gmail.com>
Date: Wed, 28 Feb 2024 23:08:08 +0800
Subject: [PATCH] fix(Article): attachmentUrl now supports video's preview &
 thumbnail variants

---
 src/graphql/models/Article.js | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/graphql/models/Article.js b/src/graphql/models/Article.js
index 8164a7f7..e7289d14 100644
--- a/src/graphql/models/Article.js
+++ b/src/graphql/models/Article.js
@@ -32,7 +32,9 @@ import ArticleReference from 'graphql/models/ArticleReference';
 import User, { userFieldResolver } from 'graphql/models/User';
 import mediaManager, {
   IMAGE_PREVIEW,
+  VIDEO_PREVIEW,
   IMAGE_THUMBNAIL,
+  VIDEO_THUMBNAIL,
 } from 'util/mediaManager';
 import ArticleReplyStatusEnum from './ArticleReplyStatusEnum';
 import ArticleReply from './ArticleReply';
@@ -528,17 +530,18 @@ const Article = new GraphQLObjectType({
         if (!attachmentHash) return null;
 
         let variant = 'original';
-        switch (variantArg) {
-          case 'PREVIEW':
-            if (articleType === 'IMAGE') {
-              variant = IMAGE_PREVIEW;
-            }
+        switch (`${articleType}_${variantArg}`) {
+          case 'IMAGE_PREVIEW':
+            variant = IMAGE_PREVIEW;
             break;
-
-          case 'THUMBNAIL':
-            if (articleType === 'IMAGE') {
-              variant = IMAGE_THUMBNAIL;
-            }
+          case 'VIDEO_PREVIEW':
+            variant = VIDEO_PREVIEW;
+            break;
+          case 'IMAGE_THUMBNAIL':
+            variant = IMAGE_THUMBNAIL;
+            break;
+          case 'VIDEO_THUMBNAIL':
+            variant = VIDEO_THUMBNAIL;
             break;
         }