From eb9e79c095b248d418c4244740a429215f35fbb0 Mon Sep 17 00:00:00 2001
From: panda <panda_dtdyy@outlook.com>
Date: Mon, 3 Apr 2023 16:48:11 +0800
Subject: [PATCH] ui: fix problem_list edit mode switch (#549)

---
 packages/ui-default/pages/problem_main.page.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/ui-default/pages/problem_main.page.js b/packages/ui-default/pages/problem_main.page.js
index 8a0fb4cc..7eb4838f 100644
--- a/packages/ui-default/pages/problem_main.page.js
+++ b/packages/ui-default/pages/problem_main.page.js
@@ -240,16 +240,16 @@ function processElement(ele) {
 }
 
 const page = new NamedPage(['problem_main'], () => {
-  const doc = document.body;
-  doc.className += ' display-mode';
+  const $body = $('body');
+  $body.addClass('display-mode');
   $('.section.display-mode').removeClass('display-mode');
   buildCategoryFilter();
   parseCategorySelection();
   $(document).on('click', '[name="leave-edit-mode"]', () => {
-    doc.className = doc.className.replace(' edit-mode', ' display-mode');
+    $body.removeClass('edit-mode').addClass('display-mode');
   });
   $(document).on('click', '[name="enter-edit-mode"]', () => {
-    doc.className = doc.className.replace(' display-mode', ' edit-mode');
+    $body.removeClass('display-mode').addClass('edit-mode');
   });
   ['delete', 'hide', 'unhide', 'copy'].forEach((op) => {
     $(document).on('click', `[name="${op}_selected_problems"]`, () => handleOperation(op));