refactor: move cmd_id into net

game_server/net/gateway.py

@@ -1,7 +1,8 @@
-from game_server.config.log import Info
+from utils.logger import Info
 from game_server.net.session import Session
 import asyncio
 
+
 class Gateway:
     def __init__(self, server_ip, game_server_port) -> None:
         self.server_ip = server_ip
@@ -16,10 +17,9 @@ class Gateway:
 
     async def start_server(self):
         session = Session()
-        server = await asyncio.start_server(session.handle_connection, self.server_ip, self.game_server_port)
+        server = await asyncio.start_server(
+            session.handle_connection, self.server_ip, self.game_server_port
+        )
         Info("Gateway listening...")
         async with server:
             await server.serve_forever()
-
-    
-

game_server/net/packet.py

@@ -1,5 +1,5 @@
 import struct
-from game_server.protocol.cmd_id import CmdID
+from game_server.net.cmd_id import CmdID
 
 class Packet:
     def __init__(self, buf: bytes):

game_server/net/session.py

@@ -1,5 +1,5 @@
-from game_server.config.log import Error, Info
-from game_server.protocol.cmd_id import CmdID
+from utils.logger import Error, Info
+from game_server.net.cmd_id import CmdID
 from game_server.net.packet import Packet
 from lib import proto as protos
 import traceback
@@ -11,7 +11,8 @@ from game_server.game.player import Player
 
 
 class Session:
-    player : Player
+    player: Player
+
     def __init__(self) -> None:
         self.writer = None
         self.pending_notifies = []
@@ -33,7 +34,7 @@ class Session:
 
     async def handle_connection(self, reader, writer):
         self.writer = writer
-        addr = writer.get_extra_info('peername')
+        addr = writer.get_extra_info("peername")
         Info(f"Accepted connection from {addr}")
 
         prefix = bytes([0x01, 0x23, 0x45, 0x67])
@@ -82,7 +83,7 @@ class Session:
     def create_packet(self, proto_message: betterproto.Message) -> Packet:
         return Packet.send_packet(proto_message)
 
-    def is_valid_packet(self,data: bytes) -> bool:
+    def is_valid_packet(self, data: bytes) -> bool:
         hex_string = data.hex().upper()
         return hex_string.startswith("01234567") and hex_string.endswith("89ABCDEF")
 
@@ -105,12 +106,13 @@ class Session:
             await self.send(packet)
         self.pending_notifies.clear()
 
-    async def process_packet(self, packet : Packet):
+    async def process_packet(self, packet: Packet):
         if packet.cmd_id not in CmdID._value2member_map_:
             Error(f"CmdId {packet.cmd_id} not recognized!")
             return
         request_name = CmdID(packet.cmd_id).name
-        if request_name == "KeepAliveNotify": return #await self.send(packet.send_packet(protos.KeepAliveNotify()))
+        if request_name == "KeepAliveNotify":
+            return  # await self.send(packet.send_packet(protos.KeepAliveNotify()))
         try:
             try:
                 req: betterproto.Message = getattr(protos, request_name)()
@@ -120,7 +122,9 @@ class Session:
 
             try:
                 Info(f"RECV packet: {request_name} ({packet.cmd_id})")
-                handle_module = importlib.import_module(f"game_server.packet.handlers.{request_name}")
+                handle_module = importlib.import_module(
+                    f"game_server.packet.handlers.{request_name}"
+                )
                 handle_function = handle_module.handle
                 handle_result = await handle_function(self, req)
                 if not handle_result: