фиксы сокета еще (стало только хуже)

Author: ink-developer

src/pymax/mixins/auth.py

@@ -29,7 +29,7 @@
 class AuthMixin(ClientProtocol):
     def _check_phone(self) -> bool:
         return bool(re.match(PHONE_REGEX, self.phone))
-    
+
     def _split_phone(self, phone: str) -> str:
         placeholder = ""
         result = ""
@@ -61,7 +61,7 @@ def _split_phone(self, phone: str) -> str:
         if phone.startswith("+998"):
             result = "+998 "
             placeholder = "00 000 00 00"
-        phone = phone[len(result)-1:]
+        phone = phone[len(result) - 1 :]
         for char in placeholder:
             if char == "0" and digit_index < len(phone):
                 result += phone[digit_index]

src/pymax/mixins/socket.py

@@ -82,8 +82,8 @@ def _pack_packet(
         payload: dict[str, Any],
     ) -> bytes:
         ver_b = ver.to_bytes(1, "big")
-        cmd_b = cmd.to_bytes(2, "big")
-        seq_b = (seq % 256).to_bytes(1, "big")
+        cmd_b = cmd.to_bytes(1, "big")
+        seq_b = seq.to_bytes(2, "big")
         opcode_b = opcode.to_bytes(2, "big")
         payload_bytes: bytes | None = msgpack.packb(payload)
         if payload_bytes is None:
@@ -255,6 +255,11 @@ async def connect(self, user_agent: UserAgentPayload | None = None) -> dict[str,
         self._outgoing = asyncio.Queue()
         self._pending = {}
         self._recv_task = self._create_safe_task(self._recv_loop(), name="recv_loop socket task")
+        self._recv_task.add_done_callback(
+            lambda t: self.logger.debug(
+                "recv_task done: cancelled=%s, exc=%r", t.cancelled(), t.exception()
+            )
+        )
         self._outgoing_task = self._create_safe_task(
             self._outgoing_loop(), name="outgoing_loop socket task"
         )
@@ -364,11 +369,13 @@ async def _recv_loop(self) -> None:
                 self.logger.debug("Received header: %s", header)
 
                 if not header:
+                    self.logger.error("No header received, exiting recv loop")
                     break
 
                 datas = await self._recv_data(loop, header, sock)
 
                 if not datas:
+                    self.logger.warning("No data received, continuing recv loop")
                     continue
 
                 consecutive_errors = 0
@@ -386,15 +393,15 @@ async def _recv_loop(self) -> None:
 
             except asyncio.CancelledError:
                 self.logger.debug("Recv loop cancelled")
-                raise
+                break
             except (
                 ssl.SSLError,
                 ssl.SSLEOFError,
                 ConnectionResetError,
                 BrokenPipeError,
             ) as ssl_err:
                 consecutive_errors += 1
-                self.logger.error(
+                self.logger.exception(
                     "SSL/Connection error in recv_loop (error %d/%d): %s",
                     consecutive_errors,
                     max_consecutive_errors,