From 0473db2a28b3dee06d781bf9d692a2bd257b4eef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurentfainsin@protonmail.com>
Date: Tue, 2 Nov 2021 19:23:47 +0100
Subject: [PATCH] feat(emulator): created `parse_message` function

---
 src/emulator.py | 65 +++++++++++++++++++++++++------------------------
 1 file changed, 33 insertions(+), 32 deletions(-)

diff --git a/src/emulator.py b/src/emulator.py
index f5d4c0e..5fd199b 100644
--- a/src/emulator.py
+++ b/src/emulator.py
@@ -39,48 +39,49 @@ mgba.log.silence()
 logging.basicConfig(level=logging.DEBUG)
 
 
+async def parse_message(message: dict[str, str]):
+    if "action" in message:
+        data = message["action"]
+        if data in KEYMAP:
+            key = KEYMAP[data]
+            core.set_keys(key)
+            logging.debug(f"pressing: {key}")
+        elif data == "null":
+            pass
+        else:
+            logging.error(f"unsupported action: {data}")
+
+    if "admin" in message:
+        data = message["admin"]
+        if data == "save":  # voodoo magic incomming
+            state = core.save_raw_state()
+            with open("states/test.state", "wb") as state_file:
+                for byte in state:
+                    state_file.write(byte.to_bytes(4, byteorder="big", signed=False))
+        elif data == "load":  # black magic incomming
+            state = ffi.new("unsigned char[397312]")
+            with open("states/test.state", "rb") as state_file:
+                for i in range(len(state)):
+                    state[i] = int.from_bytes(state_file.read(4), byteorder="big", signed=False)
+            core.load_raw_state(state)
+        else:
+            logging.error(f"unsupported admin: {data}")
+
+
 async def main():
     with pyvirtualcam.Camera(width=WIDTH, height=HEIGHT, fps=FPS) as cam:
         logging.debug(f"Using virtual camera: {cam.device}")
         async with websockets.connect(URI) as websocket:
             await websocket.send('{"auth":"password"}')
             logging.debug(f"connected to: {websocket}")
+
             while True:
                 if not (core.frame_counter % POLLING_RATE):
-
                     core.clear_keys(*KEYMAP.values())
                     await websocket.send('{"emu":"get"}')
-                    message = await websocket.recv()
-                    data = json.loads(message)
-
-                    if "action" in data:
-                        action = data["action"]
-                        if action in KEYMAP:
-                            key = KEYMAP[action]
-                            core.set_keys(key)
-                            logging.debug(f"pressing: {key}")
-                        elif action == "null":
-                            pass
-                        else:
-                            logging.error(f"unsupported action: {data}")
-
-                    if "admin" in data:
-                        admin = data["admin"]
-                        if admin == "save":
-                            state = core.save_raw_state()
-                            with open("states/test.state", "wb") as state_file:
-                                for byte in state:
-                                    state_file.write(byte.to_bytes(4, byteorder="big", signed=False))
-                        elif admin == "load":
-                            state = ffi.new("unsigned char[397312]")
-                            with open("states/test.state", "rb") as state_file:
-                                for i in range(len(state)):
-                                    ptdr = state_file.read(4)
-                                    state[i] = int.from_bytes(ptdr, byteorder="big", signed=False)
-                            core.load_raw_state(state)
-
-                        else:
-                            logging.error(f"unsupported admin: {data}")
+                    json_message = await websocket.recv()
+                    message = json.loads(json_message)
+                    parse_message(message)
 
                 core.run_frame()