added state success in PendingState

Signed-off-by: dvishal485 <dvishal485@gmail.com>

Author: dvishal485

src/leetcode_api/execution.rs

@@ -28,7 +28,6 @@ pub struct Success {
     pub total_correct: u8,
     pub total_testcases: u8,
     pub status_memory: String,
-    pub pretty_lang: String,
     pub submission_id: String,
     pub status_msg: String,
     pub state: String,
@@ -63,7 +62,6 @@ pub struct LimitExceeded {
     pub total_correct: Option<u8>,
     pub total_testcases: Option<u8>,
     pub status_memory: String,
-    pub pretty_lang: String,
     pub submission_id: String,
     pub status_msg: String,
     pub state: String,

src/leetcode_api/leetcode.rs

@@ -304,7 +304,7 @@ impl LeetCode<Authorized> {
 
         let interpret_id = data.interpret_id;
         println!("Executing testcases...");
-        let mut last_state = 0;
+        let mut last_state = PendingState::Unknown;
         loop {
             let url = format!("https://leetcode.com/submissions/detail/{interpret_id}/check/");
             // std::thread::sleep(std::time::Duration::from_secs(7));
@@ -317,29 +317,32 @@ impl LeetCode<Authorized> {
   };
             match data {
                 ExecutionResult::PendingResult(data) => {
-                    last_state = match data.state.as_str() {
-                        "PENDING" => {
-                            if last_state == 0 {
-                                println!("Status : Pending");
+                    let curr_state = data.state();
+                    match curr_state {
+                        PendingState::Pending => {
+                            if last_state != PendingState::Pending {
+                                println!("Status : Evalutaion Pending");
                             }
-                            1
                         }
-                        "STARTED" => {
-                            if last_state == 1 {
+                        PendingState::Started => {
+                            if last_state != PendingState::Started {
                                 println!("Status : Execution Started");
                             }
-                            2
                         }
-                        _ => {
-                            if last_state == 2 {
-                                println!(
-                                    "Status : {}\nKindly report this state to developer",
-                                    data.state.as_str()
-                                );
-                            }
-                            3
+                        PendingState::Success => {
+                            println!("Your code was executed successfully but we failed to parse result\nCheck on leetcode manually");
+                            std::process::exit(1);
+                        }
+                        PendingState::Unknown => {
+                            println!(
+                                "Status : {}\nKindly report this state to developer",
+                                data.state.as_str()
+                            );
+                            std::process::exit(1);
                         }
                     };
+                    last_state = curr_state;
+
                     continue;
                 }
                 data => return Ok(data),
@@ -385,7 +388,7 @@ impl LeetCode<Authorized> {
    };
         println!("Evaluating solution...");
         let submission_id = data.submission_id;
-        let mut last_state = 0;
+        let mut last_state = PendingState::Unknown;
 
         loop {
             let url = format!("https://leetcode.com/submissions/detail/{submission_id}/check/");
@@ -398,29 +401,31 @@ impl LeetCode<Authorized> {
   };
             match data {
                 SubmissionResult::PendingResult(data) => {
-                    last_state = match data.state.as_str() {
-                        "PENDING" => {
-                            if last_state == 0 {
+                    let curr_state = data.state();
+                    match curr_state {
+                        PendingState::Pending => {
+                            if last_state != PendingState::Pending {
                                 println!("Status : Evalutaion Pending");
                             }
-                            1
                         }
-                        "STARTED" => {
-                            if last_state == 1 {
+                        PendingState::Started => {
+                            if last_state != PendingState::Started {
                                 println!("Status : Execution Started");
                             }
-                            2
                         }
-                        _ => {
-                            if last_state == 2 {
-                                println!(
-                                    "Status : {}\nKindly report this state to developer",
-                                    data.state.as_str()
-                                );
-                            }
-                            3
+                        PendingState::Success => {
+                            println!("Your code was executed successfully but we failed to parse result\nCheck on leetcode manually");
+                            std::process::exit(1);
+                        }
+                        PendingState::Unknown => {
+                            println!(
+                                "Status : {}\nKindly report this state to developer",
+                                data.state.as_str()
+                            );
+                            std::process::exit(1);
                         }
                     };
+                    last_state = curr_state;
                     continue;
                 }
                 data => return Ok(data),

src/leetcode_api/submission.rs

@@ -47,7 +47,6 @@ pub struct SubmitWrong {
     pub task_finish_time: u64,
     pub total_correct: u8,
     pub total_testcases: u8,
-    pub pretty_lang: String,
     pub submission_id: String,
     pub status_msg: String,
     pub state: String,
@@ -72,7 +71,6 @@ pub struct SubmitRuntimeError {
     pub task_finish_time: u64,
     pub total_correct: u8,
     pub total_testcases: u8,
-    pub pretty_lang: String,
     pub submission_id: String,
     pub status_msg: String,
     pub state: String,
@@ -96,7 +94,6 @@ pub struct SubmitCompileError {
     pub task_finish_time: u64,
     pub total_correct: u8,
     pub total_testcases: u8,
-    pub pretty_lang: String,
     pub submission_id: String,
     pub status_msg: String,
     pub state: String,

src/leetcode_api/utils.rs

@@ -26,10 +26,11 @@ pub enum ExecutionResult {
     Unknown(Unknown),
 }
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq)]
 pub enum PendingState {
     Pending,
     Started,
+    Success,
     Unknown,
 }
 
@@ -54,6 +55,7 @@ impl PendingResult {
         match self.state.as_str() {
             "PENDING" => PendingState::Pending,
             "STARTED" => PendingState::Started,
+            "SUCCESS" => PendingState::Success,
             unknown_state => {
                 println!(
                     "Unknown state : {}\nKindly inform about this to the developer",