fix: initialize script on the case of disabling color toggle

razonyang · Jan 27, 2023 · 4e82cc6 · 4e82cc6
1 parent dbe5b0a
commit 4e82cc6
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/assets/js/mode/index.ts b/assets/js/mode/index.ts
@@ -73,11 +73,13 @@ class ModeToggle implements Component {
       }
     });
 
-    const icon = document
-      .querySelector('.mode-item[data-color-mode="' + mode + '"] .mode-icon')
-      .cloneNode(true) as HTMLElement;
+    const icon = document.querySelector('.mode-item[data-color-mode="' + mode + '"] .mode-icon')?.cloneNode(true) as HTMLElement;
+    if (!icon) {
+      return
+    }
+
     icon.setAttribute('id', 'modeIcon');
-    document.querySelector('#modeIcon').replaceWith(icon);
+    document.querySelector('#modeIcon')?.replaceWith(icon);
   }
 
   setMode(value: string) {