From ebda011bc7dbb0b87915d8a5a8b7d12294e83ba4 Mon Sep 17 00:00:00 2001
From: Blizzard Finnegan <blizzardfinnegan@gmail.com>
Date: Mon, 19 Jun 2023 14:28:03 -0400
Subject: [PATCH] Add parsing for DebugInit

---
 src/device.rs | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/src/device.rs b/src/device.rs
index 8c117d3..0cf30b8 100755
--- a/src/device.rs
+++ b/src/device.rs
@@ -351,17 +351,21 @@ impl Device{
         self.usb_tty.write_to_device(Command::Login);
         while self.usb_tty.read_from_device(None) != Response::ShellPrompt {}
         self.usb_tty.write_to_device(Command::GetSerial);
-        match self.usb_tty.read_from_device(None){
-            Response::Serial(Some(contains_serial)) =>{
-                for line in contains_serial.split("\n").collect::<Vec<&str>>(){
-                    if !line.contains(':') { continue; }
-                    let (section,value) = line.split_once(':').unwrap();
-                    if section.contains(SERIAL_HEADER){
-                        self.serial = value.trim().replace("\"","");
+        loop{
+            match self.usb_tty.read_from_device(None){
+                Response::Serial(Some(contains_serial)) =>{
+                    for line in contains_serial.split("\n").collect::<Vec<&str>>(){
+                        if !line.contains(':') { continue; }
+                        let (section,value) = line.split_once(':').unwrap();
+                        if section.contains(SERIAL_HEADER){
+                            self.serial = value.trim().replace("\"","");
+                        }
                     }
-                }
-            },
-            _ => todo!(),
+                    break;
+                },
+                Response::DebugInit => { continue; }
+                _ => todo!(),
+            }
         }
         self.reboot();
         //self.serial = serial.to_string();