trace.ai-cli 1.1.6 → 1.1.8

package/utils/encryption.js

@@ -2,8 +2,7 @@ const ENCRYPTION_KEY = 'AlzaSyCVKlzUxK';
 
 function encodeUnicode(str) {
     const utf8Bytes = new TextEncoder().encode(str);
-    const base64 = Buffer.from(utf8Bytes).toString('base64');
-    return base64;
+    return Buffer.from(utf8Bytes).toString('base64');
 }
 
 function decodeUnicode(str) {
@@ -13,14 +12,28 @@ function decodeUnicode(str) {
 
 function encryptData(data) {
     const jsonStr = JSON.stringify(data);
-    const encoded = encodeUnicode(jsonStr);
-    return encoded.split('').reverse().join('');
+    // Match worker.js approach: encode to base64 then reverse
+    const base64 = Buffer.from(
+        Array.from(new TextEncoder().encode(jsonStr))
+            .map(byte => String.fromCharCode(byte))
+            .join('')
+    ).toString('base64');
+    
+    return base64.split('').reverse().join('');
 }
 
 function decryptData(encoded) {
-    const reversed = encoded.split('').reverse().join('');
-    const decoded = decodeUnicode(reversed);
-    return JSON.parse(decoded);
+    try {
+        // Match worker.js approach: reverse then decode
+        const reversed = encoded.split('').reverse().join('');
+        const binaryStr = Buffer.from(reversed, 'base64').toString('binary');
+        const bytes = Uint8Array.from(binaryStr, char => char.charCodeAt(0));
+        const decoded = new TextDecoder().decode(bytes);
+        return JSON.parse(decoded);
+    } catch (error) {
+        console.error('Decryption failed:', error);
+        throw new Error('Invalid encrypted data');
+    }
 }
 
 module.exports = {