feat: filter by problem category

background.ts

@@ -90,6 +90,7 @@ const generateRandomLeetCodeProblem = async () => {
   try {
     const problemSet = (await storage.get("problemSets")) ?? "all"
     const difficulty = (await storage.get("difficulty")) ?? "all"
+    const category = (await storage.get("category")) ?? "all"
     let leetCodeProblems = []
     if (problemSet === "all") {
       await storage.set("loading", true)
@@ -117,11 +118,18 @@ const generateRandomLeetCodeProblem = async () => {
       }
       const res = await fetch(chrome.runtime.getURL(problemSetURLs[problemSet]))
       leetCodeProblems = await res.json()
+
       if (difficulty !== "all") {
         leetCodeProblems = leetCodeProblems.filter((problem) => {
           return problem.difficulty.toLowerCase() === difficulty.toLowerCase()
         })
       }
+      if (category !== "all") {
+        leetCodeProblems = leetCodeProblems.filter((problem) => {
+          console.log(problem.category.toLowerCase(), category.toLowerCase())
+          return problem.category.toLowerCase() === category.toLowerCase()
+        })
+      }
 
       let randomIndex = Math.floor(Math.random() * leetCodeProblems.length)
       // If the problem is premium, then skip it and go to the next problem until you find a non-premium problem

components/SettingDrawer.tsx

@@ -8,6 +8,7 @@ import SettingLabel from "./SettingLabel"
 const SettingDrawer = ({ close, setClose }) => {
   const [problemSets, setProblemSets] = useStorage<string>("problemSets")
   const [difficulty, setDifficulty] = useStorage<string>("difficulty")
+  const [category, setCategory] = useStorage<string>("category")
   const [leetcodeProblemSolved] = useStorage<boolean>("leetCodeProblemSolved")
   const settingList = [
     {
@@ -43,7 +44,40 @@ const SettingDrawer = ({ close, setClose }) => {
           !leetcodeProblemSolved ? updateStorage() : null
         }
       }
+    },
+    {
+      name: "Problem Category",
+      description: "Choose the category you'd like to solve",
+      dropdownProps: {
+        options: {
+          all: "All categories",
+          "arrays & hashing": "Arrays & Hashing",
+          "two pointers": "Two Pointers",
+          "sliding window": "Sliding Window",
+          stack: "Stack",
+          "binary search": "Binary Search",
+          "linked list": "Linked List",
+          trees: "Trees",
+          tries: "Tries",
+          "heap / priority queue": "Heap / Priority Queue",
+          backtracking: "Backtracking",
+          graphs: "Graphs",
+          "advanced graphs": "Advanced Graphs",
+          "1-d dynamic programming": "1-D Dynamic Programming",
+          "2-d dynamic programming": "2-D Dynamic Programming",
+          greedy: "Greedy",
+          intervals: "Intervals",
+          "math & geometry": "Math & Geometry",
+          "bit manipulation": "Bit Manipulation",
+          javascript: "JavaScript"
+        },
+        defaultValue: category,
+        handleChange: async (e) => {
+          setCategory(e.target.value)
+          !leetcodeProblemSolved ? updateStorage() : null
+        }
     }
+  }
     /* TODO: Add this feature later
     {
       name: "Number of problems to solve",