fix: only easy difficulty getting hidden, not medium hard

Author: zubyj

manifest.json

@@ -28,8 +28,8 @@
             "js": [
                 "dist/content-script/get-gpt-access-token.js",
                 "dist/content-script/get-user-code.js",
-                "dist/content-script/update-solutions.js",
-                "dist/content-script/update-description.js"
+                "dist/content-script/update-solutions-tab.js",
+                "dist/content-script/update-description-tab.js"
             ],
             "matches": [
                 "https://leetcode.com/problems/*"

src/content-script/update-description.ts renamed to src/content-script/update-description-tab.ts

@@ -25,8 +25,14 @@ function showExamples() {
 function showDifficulty() {
     chrome.storage.local.get(['showDifficulty'], (result) => {
         let showDifficulty = result.showDifficulty;
-        let difficultyContainer = document.querySelectorAll('div.bg-olive')[0];
-        difficultyContainer.style.display = showDifficulty ? 'block' : 'none';
+
+        let colors = ['bg-olive', 'bg-yellow', 'bg-red'];
+        for (let color in colors) {
+            let difficultyContainer = document.querySelectorAll('div.' + colors[color])[0];
+            if (difficultyContainer) {
+                difficultyContainer.style.display = showDifficulty ? 'block' : 'none';
+            }
+        }
     });
 }