Update proxy_handler.py

proxy_handler.py

@@ -48,8 +48,8 @@ class ProxyHandler:
 
     def _generate_signature(self, e_payload: str, t_payload: str) -> Dict[str, Any]:
         """
-        Generates the signature based on the logic from the reference JS code.
-        This is a two-level HMAC-SHA256 process.
+        Generates the signature based on the logic from the reference JS code (work.js).
+        This is a two-level HMAC-SHA256 process with Base64 encoding for the content.
 
         Args:
             e_payload (str): The simplified payload string (e.g., "requestId,...,timestamp,...").
@@ -58,20 +58,17 @@ class ProxyHandler:
         Returns:
             A dictionary with 'signature' and 'timestamp'.
         """
-        # The provided reference code uses a different logic for the key derivation.
-        # It's based on a timestamp bucket. Let's re-implement that one.
-        # However, the OTHER reference code `signature_generator.py` uses a different method.
-        # Let's stick to the one from the new `utils.py` and `signature_generator.py` for now.
-        # The provided python snippet in the prompt is actually different from the JS.
-        # The python snippet is: `n = timestamp_ms // (5 * 60 * 1000)`
-        # The JS snippet is: `minuteBucket = Math.floor(timestampMs / 60000)`
-        # Let's trust the JS one as it's more complete. Let's try the python one first as it's provided.
-        
-        # --- Let's use the Python snippet logic from the prompt first ---
         timestamp_ms = self._get_timestamp_millis()
-        message_string = f"{e_payload}|{t_payload}|{timestamp_ms}"
+
+        # --- MODIFICATION START ---
+        # As per work.js, the last message content (t_payload) must be Base64 encoded.
+        content_base64 = base64.b64encode(t_payload.encode('utf-8')).decode('utf-8')
+        
+        # Concatenate with the Base64 encoded content
+        message_string = f"{e_payload}|{content_base64}|{timestamp_ms}"
+        # --- MODIFICATION END ---
         
-        # Per the Python snippet: n is a 5-minute bucket
+        # Per the Python snippet and JS reference: n is a 5-minute bucket
         n = timestamp_ms // (5 * 60 * 1000) 
         
         # Intermediate key derivation