inital test + lint

Author: Mikefly123

src/flight-software/main.py

@@ -1,4 +1,5 @@
 import gc
+import json
 import os
 import time
 
@@ -25,6 +26,9 @@
 
 boot_time: float = time.time()
 
+cube_ids = ["Listener1", "Listener2", "Listener3"]
+my_cubesat_id = "Listener1"
+
 rtc = MicrocontrollerManager()
 
 (boot_count := Counter(index=Register.boot_count)).increment()
@@ -134,19 +138,85 @@ def nominal_power_loop():
             bytes_remaining=gc.mem_free(),
         )
 
-        uhf_packet_manager.send(config.radio.license.encode("utf-8"))
-
-        beacon.send()
-
-        cdh.listen_for_commands(10)
-
-        sleep_helper.safe_sleep(config.sleep_duration)
+        message: dict[str, object] = dict()
+        for cube_id in cube_ids:
+            if cube_id == my_cubesat_id:
+                continue
+
+            # --- 1. Send a ping to a single cubesat ---
+            logger.info(f"Pinging {cube_id}...")
+            message["current_time"] = time.monotonic()
+            message["cube_id"] = cube_id
+            message["command"] = "ping"
+            encoded_message = json.dumps(message, separators=(",", ":")).encode("utf-8")
+
+            if not uhf_packet_manager.send(encoded_message):
+                logger.warning(f"Failed to send ping to {cube_id}")
+                sleep_helper.safe_sleep(1)  # Wait a moment before trying the next one
+                continue
+
+            # --- 2. Listen for an immediate pong response ---
+            logger.info(f"Listening for pong from {cube_id} for 10 seconds.")
+            received_message = uhf_packet_manager.listen(10)
+
+            if received_message:
+                try:
+                    decoded_message = json.loads(received_message.decode("utf-8"))
+                    sender_id = decoded_message.get("cube_id")
+                    command = decoded_message.get("command")
+                    if command == "pong" and sender_id == cube_id:
+                        logger.info(f"Success! Received pong for me from {sender_id}.")
+                    else:
+                        logger.warning(
+                            f"Received unexpected message: {decoded_message}"
+                        )
+                except (json.JSONDecodeError, UnicodeError) as e:
+                    logger.error("Could not process received message", e)
+            else:
+                logger.warning(f"No response from {cube_id} within the time limit.")
+
+            # --- 3. Wait before contacting the next satellite ---
+            logger.debug("Waiting 5 seconds before contacting next satellite.")
+            sleep_helper.safe_sleep(5)
+
+    def listener_nominal_power_loop():
+        # logger.debug(
+        #     "FC Board Stats",
+        #     bytes_remaining=gc.mem_free(),
+        # )
+
+        # logger.info("Listening for messages for 60 seconds")
+        received_message: bytes | None = uhf_packet_manager.listen(5)
+
+        if received_message:
+            try:
+                decoded_message = json.loads(received_message.decode("utf-8"))
+                logger.info(f"Received message: {decoded_message}")
+
+                cubesat_id = decoded_message.get("cube_id")
+                if cubesat_id == my_cubesat_id:
+                    command = decoded_message.get("command")
+                    if command == "ping":
+                        logger.info(f"Received ping from {cubesat_id}")
+                        response_message = {
+                            "current_time": time.monotonic(),
+                            "cube_id": my_cubesat_id,
+                            "command": "pong",
+                        }
+                        encoded_response = json.dumps(
+                            response_message, separators=(",", ":")
+                        ).encode("utf-8")
+                        sleep_helper.safe_sleep(1)
+                        uhf_packet_manager.send(encoded_response)
+
+            except (json.JSONDecodeError, UnicodeError) as e:
+                logger.error("Failed to decode message", e)
 
     try:
         logger.info("Entering main loop")
         while True:
             # TODO(nateinaction): Modify behavior based on power state
-            nominal_power_loop()
+            listener_nominal_power_loop()
 
     except Exception as e:
         logger.critical("Critical in Main Loop", e)