From 270a5299484c1a305dddd7b2ea6f61c969e05c77 Mon Sep 17 00:00:00 2001
From: boojack <stevenlgtm@gmail.com>
Date: Sat, 5 Aug 2023 20:17:33 +0800
Subject: [PATCH] chore: update resource type checks (#2081)

---
 web/src/components/MemoResourceListView.tsx | 12 ++++----
 web/src/components/ResourceCover.tsx        | 28 +------------------
 web/src/components/ResourceItemDropdown.tsx |  4 +--
 web/src/utils/resource.ts                   | 31 +++++++++++++++++++++
 4 files changed, 39 insertions(+), 36 deletions(-)

diff --git a/web/src/components/MemoResourceListView.tsx b/web/src/components/MemoResourceListView.tsx
index 4509e951..20ed4511 100644
--- a/web/src/components/MemoResourceListView.tsx
+++ b/web/src/components/MemoResourceListView.tsx
@@ -1,6 +1,6 @@
 import { ImageList, ImageListItem, useMediaQuery } from "@mui/material";
 import { absolutifyLink } from "@/helpers/utils";
-import { getResourceUrl } from "@/utils/resource";
+import { getResourceType, getResourceUrl } from "@/utils/resource";
 import SquareDiv from "./kit/SquareDiv";
 import MemoResource from "./MemoResource";
 import showPreviewImageDialog from "./PreviewImageDialog";
@@ -24,17 +24,15 @@ const MemoResourceListView: React.FC<Props> = (props: Props) => {
     ...props,
   };
   const matches = useMediaQuery("(min-width:640px)");
-  const imageResourceList = resourceList.filter((resource) => resource.type.startsWith("image"));
+  const imageResourceList = resourceList.filter((resource) => getResourceType(resource).startsWith("image"));
   const videoResourceList = resourceList.filter((resource) => resource.type.startsWith("video"));
   const otherResourceList = resourceList.filter(
     (resource) => !imageResourceList.includes(resource) && !videoResourceList.includes(resource)
   );
 
-  const imgUrls = imageResourceList
-    .filter((resource) => resource.type.startsWith("image"))
-    .map((resource) => {
-      return getResourceUrl(resource);
-    });
+  const imgUrls = imageResourceList.map((resource) => {
+    return getResourceUrl(resource);
+  });
 
   const handleImageClick = (imgUrl: string) => {
     const index = imgUrls.findIndex((url) => url === imgUrl);
diff --git a/web/src/components/ResourceCover.tsx b/web/src/components/ResourceCover.tsx
index 76565c63..d7ff04da 100644
--- a/web/src/components/ResourceCover.tsx
+++ b/web/src/components/ResourceCover.tsx
@@ -1,5 +1,5 @@
 import React from "react";
-import { getResourceUrl } from "@/utils/resource";
+import { getResourceType, getResourceUrl } from "@/utils/resource";
 import Icon from "./Icon";
 import SquareDiv from "./kit/SquareDiv";
 import showPreviewImageDialog from "./PreviewImageDialog";
@@ -9,32 +9,6 @@ interface ResourceCoverProps {
   resource: Resource;
 }
 
-const getResourceType = (resource: Resource) => {
-  if (resource.type.startsWith("image")) {
-    return "image/*";
-  } else if (resource.type.startsWith("video")) {
-    return "video/*";
-  } else if (resource.type.startsWith("audio")) {
-    return "audio/*";
-  } else if (resource.type.startsWith("text")) {
-    return "text/*";
-  } else if (resource.type.startsWith("application/epub+zip")) {
-    return "application/epub+zip";
-  } else if (resource.type.startsWith("application/pdf")) {
-    return "application/pdf";
-  } else if (resource.type.includes("word")) {
-    return "application/msword";
-  } else if (resource.type.includes("excel")) {
-    return "application/msexcel";
-  } else if (resource.type.startsWith("application/zip")) {
-    return "application/zip";
-  } else if (resource.type.startsWith("application/x-java-archive")) {
-    return "application/x-java-archive";
-  } else {
-    return "application/octet-stream";
-  }
-};
-
 const ResourceCover = ({ resource }: ResourceCoverProps) => {
   const resourceType = getResourceType(resource);
   const resourceUrl = getResourceUrl(resource);
diff --git a/web/src/components/ResourceItemDropdown.tsx b/web/src/components/ResourceItemDropdown.tsx
index 2615e193..17fcc361 100644
--- a/web/src/components/ResourceItemDropdown.tsx
+++ b/web/src/components/ResourceItemDropdown.tsx
@@ -3,7 +3,7 @@ import React from "react";
 import toast from "react-hot-toast";
 import { useResourceStore } from "@/store/module";
 import { useTranslate } from "@/utils/i18n";
-import { getResourceUrl } from "@/utils/resource";
+import { getResourceType, getResourceUrl } from "@/utils/resource";
 import showChangeResourceFilenameDialog from "./ChangeResourceFilenameDialog";
 import { showCommonDialog } from "./Dialog/CommonDialog";
 import Icon from "./Icon";
@@ -20,7 +20,7 @@ const ResourceItemDropdown = ({ resource }: Props) => {
 
   const handlePreviewBtnClick = (resource: Resource) => {
     const resourceUrl = getResourceUrl(resource);
-    if (resource.type.startsWith("image")) {
+    if (getResourceType(resource).startsWith("image")) {
       showPreviewImageDialog([getResourceUrl(resource)], 0);
     } else {
       window.open(resourceUrl);
diff --git a/web/src/utils/resource.ts b/web/src/utils/resource.ts
index 26ffa865..1216a125 100644
--- a/web/src/utils/resource.ts
+++ b/web/src/utils/resource.ts
@@ -5,3 +5,34 @@ export const getResourceUrl = (resource: Resource, withOrigin = true) => {
 
   return `${withOrigin ? window.location.origin : ""}/o/r/${resource.id}`;
 };
+
+export const getResourceType = (resource: Resource) => {
+  if (resource.type.startsWith("image") && isImage(resource.type)) {
+    return "image/*";
+  } else if (resource.type.startsWith("video")) {
+    return "video/*";
+  } else if (resource.type.startsWith("audio")) {
+    return "audio/*";
+  } else if (resource.type.startsWith("text")) {
+    return "text/*";
+  } else if (resource.type.startsWith("application/epub+zip")) {
+    return "application/epub+zip";
+  } else if (resource.type.startsWith("application/pdf")) {
+    return "application/pdf";
+  } else if (resource.type.includes("word")) {
+    return "application/msword";
+  } else if (resource.type.includes("excel")) {
+    return "application/msexcel";
+  } else if (resource.type.startsWith("application/zip")) {
+    return "application/zip";
+  } else if (resource.type.startsWith("application/x-java-archive")) {
+    return "application/x-java-archive";
+  } else {
+    return "application/octet-stream";
+  }
+};
+
+// isImage returns true if the given mime type is an image.
+export const isImage = (t: string) => {
+  return t === "image/jpeg" || t === "image/png" || t === "image/gif" || t === "image/svg+xml" || t === "image/webp";
+};