joshlei-cookies 1.0.0 → 2.5.0

package/index.js

@@ -94,7 +94,7 @@ const initCrypto = async () => {
         }
       };
       
-      console.warn('⚠️  Using fallback crypto implementation. For production use, ensure Web Crypto API or Node.js crypto is available.');
+      console.warn('Using fallback crypto implementation. For production use, ensure Web Crypto API or Node.js crypto is available.');
     }
   }
   
@@ -303,7 +303,29 @@ export const SetJSCookie = async (name, data, options = {}) => {
   }
   
   try {
-    const encryptedData = await CryptoManager.encrypt(JSON.stringify(data), hashedKey);
+    let dataToStore = data;
+
+    if (isBrowser && typeof data === 'object' && data !== null && typeof data.token === 'string') {
+      const tokenParts = data.token.split('.');
+      if (tokenParts.length === 3) {
+        try {
+          const encryptedToken = await CryptoManager.encrypt(data.token, hashedKey);
+          sessionStorage.setItem(`jstudio_token_${name}`, encryptedToken);
+          const { token, ...rest } = data;
+          dataToStore = { ...rest, __tokenSaved: true };
+        } catch (storageErr) {
+          try {
+            sessionStorage.setItem(`jstudio_token_${name}`, data.token);
+            const { token, ...rest } = data;
+            dataToStore = { ...rest, __tokenSaved: true };
+          } catch (e) {
+            dataToStore = data;
+          }
+        }
+      }
+    }
+
+    const encryptedData = await CryptoManager.encrypt(JSON.stringify(dataToStore), hashedKey);
     CookieManager.set(name, encryptedData, options);
   } catch (error) {
     throw new Error(`Failed to set cookie: ${error.message}`);
@@ -323,7 +345,25 @@ export const GetJSCookie = async (name) => {
     if (!encryptedData) return null;
     
     const decryptedData = await CryptoManager.decrypt(encryptedData, hashedKey);
-    return JSON.parse(decryptedData);
+    const parsed = JSON.parse(decryptedData);
+
+    if (isBrowser && parsed && parsed.__tokenSaved) {
+      try {
+        const stored = sessionStorage.getItem(`jstudio_token_${name}`);
+        if (stored) {
+          try {
+            const decryptedToken = await CryptoManager.decrypt(stored, hashedKey);
+            parsed.token = decryptedToken;
+          } catch (decryptErr) {
+            parsed.token = stored;
+          }
+        }
+        delete parsed.__tokenSaved;
+      } catch (storageErr) {
+      }
+    }
+
+    return parsed;
   } catch (error) {
     try {
       CookieManager.remove(name);
@@ -339,6 +379,10 @@ export const RemoveJSCookie = (name, options = {}) => {
   }
   
   try {
+    try {
+      if (isBrowser) sessionStorage.removeItem(`jstudio_token_${name}`);
+    } catch (e) {}
+
     CookieManager.remove(name, options);
   } catch (error) {
     throw new Error(`Failed to remove cookie: ${error.message}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "joshlei-cookies",
-  "version": "1.0.0",
+  "version": "2.5.0",
   "description": "A secure cookie management library with built-in AES encryption for browser environments",
   "main": "index.js",
   "type": "module",
@@ -34,5 +34,8 @@
       "import": "./index.js",
       "types": "./index.d.ts"
     }
+  },
+  "dependencies": {
+    "joshlei-cookies": "^1.0.0"
   }
 }