From 04d9de1bd0305efd49249bb6dad56211b58cae4b Mon Sep 17 00:00:00 2001
From: Dhiraj Mishra <mishra.dhiraj95@gmail.com>
Date: Wed, 13 Nov 2024 22:12:30 +0400
Subject: [PATCH] Update background.js

---
 SmuggleShield/background.js | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/SmuggleShield/background.js b/SmuggleShield/background.js
index abf34e5..176c775 100644
--- a/SmuggleShield/background.js
+++ b/SmuggleShield/background.js
@@ -13,38 +13,40 @@ class WeakLRUCache {
   constructor(maxSize) {
     this.maxSize = maxSize;
     this.cache = new Map();
-    this.keyMap = new WeakMap();
+    this.keyMap = new Map();
   }
 
   get(key) {
-    const keyObj = this.keyMap.get(key);
-    if (!keyObj) return undefined;
-    const value = this.cache.get(keyObj);
+    const keyString = this.getKeyString(key);
+    const value = this.cache.get(keyString);
     if (value) {
-      this.cache.delete(keyObj);
-      this.cache.set(keyObj, value);
+      this.cache.delete(keyString);
+      this.cache.set(keyString, value);
     }
     return value;
   }
 
   set(key, value) {
-    let keyObj = this.keyMap.get(key);
-    if (!keyObj) {
-      keyObj = { key };
-      this.keyMap.set(key, keyObj);
-    }
-    if (this.cache.has(keyObj)) {
-      this.cache.delete(keyObj);
+    const keyString = this.getKeyString(key);
+    if (this.cache.has(keyString)) {
+      this.cache.delete(keyString);
     } else if (this.cache.size >= this.maxSize) {
       const oldestKey = this.cache.keys().next().value;
       this.cache.delete(oldestKey);
     }
-    this.cache.set(keyObj, value);
+    this.cache.set(keyString, value);
+    this.keyMap.set(keyString, key);
   }
 
   clear() {
     this.cache.clear();
-    this.keyMap = new WeakMap();
+    this.keyMap.clear();
+  }
+
+  getKeyString(key) {
+    return typeof key === 'object' ? 
+      JSON.stringify(key) : 
+      String(key);
   }
 }