From 7151f71417755c3a63ca2a550fb0f2de6600e263 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Tue, 8 Feb 2022 20:04:49 +0800
Subject: [PATCH] ui: problem_files: allow image preview

---
 packages/ui-default/package.json               |  2 +-
 .../ui-default/pages/problem_files.page.js     | 18 +++++++++++++++---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/packages/ui-default/package.json b/packages/ui-default/package.json
index 60f4baec..39ab29ff 100644
--- a/packages/ui-default/package.json
+++ b/packages/ui-default/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@hydrooj/ui-default",
-  "version": "4.31.5",
+  "version": "4.31.6",
   "author": "undefined <i@undefined.moe>",
   "license": "AGPL-3.0",
   "main": "hydro.js",
diff --git a/packages/ui-default/pages/problem_files.page.js b/packages/ui-default/pages/problem_files.page.js
index b12f2a86..4692c389 100644
--- a/packages/ui-default/pages/problem_files.page.js
+++ b/packages/ui-default/pages/problem_files.page.js
@@ -1,7 +1,9 @@
 import _ from 'lodash';
 import { NamedPage } from 'vj/misc/Page';
 import Notification from 'vj/components/notification';
-import { ConfirmDialog, ActionDialog, Dialog } from 'vj/components/dialog/index';
+import {
+  ConfirmDialog, ActionDialog, InfoDialog, Dialog,
+} from 'vj/components/dialog/index';
 import download from 'vj/components/zipDownloader';
 import createHint from 'vj/components/hint';
 import request from 'vj/utils/request';
@@ -217,8 +219,18 @@ const page = new NamedPage('problem_files', () => {
     if (ev) {
       const link = $(ev.currentTarget).find('a').attr('href');
       if (!link) return;
-      if (filesize > 8 * 1024 * 1024) {
-        Notification.error(i18n('file too large'));
+      const ext = filename.split('.').pop();
+      if (['png', 'jpeg', 'jpg', 'gif', 'webp', 'bmp'].includes(ext)) {
+        await new InfoDialog({
+          $body: tpl`<div class="typo"><img src="${link}"></img></div>`,
+        }).open();
+        return;
+      }
+      if (['zip', 'rar', '7z'].includes(ext) || filesize > 8 * 1024 * 1024) {
+        const action = await new ActionDialog({
+          $body: tpl.typoMsg(i18n('Cannot preview this file. Download now?')),
+        }).open();
+        if (action === 'ok') window.open(link);
         return;
       }
       Notification.info(i18n('Loading file...'));