fixes #10 & added TLE prediction message

Author: dvishal485

src/handlers/execution.rs

@@ -134,7 +134,8 @@ impl std::fmt::Display for Success {
         for i in 0..self.code_answer.len() {
             let is_correct = self.code_answer[i] == self.expected_code_answer[i];
             part2.push(format!(
-                "{}\n{}\n{}\nOutput   : {:?}\nExpected : {:?}\n\n{}",
+                "{1}\n{2}\n{3}\n{0:10}: {4:?}\n{7:10}: {5:?}\n\n{6}",
+                "Output",
                 seperator.yellow(),
                 if is_correct {
                     format!("Testcase {} execution success", i + 1).green()
@@ -148,24 +149,43 @@ impl std::fmt::Display for Success {
                     format!("Std Output :\n{}\n", self.std_output[i])
                 } else {
                     String::new()
-                }
+                },
+                "Expected"
             ));
         }
 
+        let may_tle = {
+            // added 200 because leetcode data is not very reliabale
+            self.expected_elapsed_time + 200 < self.elapsed_time
+        };
+
         let part3 = format!(
-            "{}\nRuntime  : {}\nOutput   : {}\nExpected : {}\n",
+            "{}\n{:10}: {:6} ({}%)\n{}",
             seperator.yellow(),
+            "Runtime",
             self.status_runtime.cyan(),
-            self.elapsed_time,
-            self.expected_status_runtime
+            100 - (self.elapsed_time as u128 * 100)
+                .checked_div(self.expected_elapsed_time as u128 + 100)
+                .unwrap_or(100)
+                .min(100),
+            if may_tle {
+                "High runtime detected! May lead to TLE\n"
+                    .red()
+                    .italic()
+                    .to_string()
+            } else {
+                "".to_string()
+            },
         );
 
         let part4 = format!(
-            "{}\nMemory   : {}\nOutput   : {}\nExpected : {}\n",
-            seperator.yellow(),
+            "{2:10}: {:6} ({}%)\n",
             self.status_memory.cyan(),
-            self.memory,
-            self.expected_memory
+            100 - (self.memory as u128 * 100)
+                .checked_div(self.expected_memory as u128)
+                .unwrap_or(100)
+                .min(100),
+            "Memory",
         );
         write!(f, "{}{}{}{}", part1, part2.join(""), part3, part4)
     }

src/handlers/leetcode.rs

@@ -30,35 +30,35 @@ impl LeetCode<Unauthorized> {
     pub fn authenticate(&self, cookie: &str) -> Result<LeetCode<Authorized>, &str> {
         let mut headers = reqwest::header::HeaderMap::with_capacity(5);
         let Some(csrf_token) = cookie
-   .split(';')
-   .find(|s| s.contains("csrftoken"))
-   else{return  Err("No csrf token found"); };
-        let Some(csrf_token) = csrf_token.split('=').last() else{return Err("No csrf token found"); };
-        let csrf_token = csrf_token.to_string();
-        headers.insert(
-            reqwest::header::COOKIE,
-            reqwest::header::HeaderValue::from_str(&cookie).unwrap(),
-        );
-        headers.insert(
-   reqwest::header::USER_AGENT,
-   reqwest::header::HeaderValue::from_str("Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36").unwrap(),
-  );
-        headers.insert(
-            reqwest::header::REFERER,
-            reqwest::header::HeaderValue::from_str("https://leetcode.com/").unwrap(),
-        );
-        headers.insert(
-            reqwest::header::HeaderName::from_static("x-csrftoken"),
-            reqwest::header::HeaderValue::from_str(csrf_token.as_str()).unwrap(),
-        );
-        let client = reqwest::blocking::Client::builder()
-            .default_headers(headers.clone())
-            .build()
-            .unwrap();
-        Ok(LeetCode {
-            state: std::marker::PhantomData::<Authorized>,
-            client,
-        })
+            .split(';')
+            .find(|s| s.contains("csrftoken"))
+            else{return  Err("No csrf token found"); };
+            let Some(csrf_token) = csrf_token.split('=').last() else{return Err("No csrf token found"); };
+            let csrf_token = csrf_token.to_string();
+            headers.insert(
+                reqwest::header::COOKIE,
+                reqwest::header::HeaderValue::from_str(&cookie).unwrap(),
+                );
+            headers.insert(
+                reqwest::header::USER_AGENT,
+                reqwest::header::HeaderValue::from_str("Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36").unwrap(),
+                );
+            headers.insert(
+                reqwest::header::REFERER,
+                reqwest::header::HeaderValue::from_str("https://leetcode.com/").unwrap(),
+                );
+            headers.insert(
+                reqwest::header::HeaderName::from_static("x-csrftoken"),
+                reqwest::header::HeaderValue::from_str(csrf_token.as_str()).unwrap(),
+                );
+            let client = reqwest::blocking::Client::builder()
+                .default_headers(headers.clone())
+                .build()
+                .unwrap();
+            Ok(LeetCode {
+                state: std::marker::PhantomData::<Authorized>,
+                client,
+            })
     }
 }
 

src/handlers/submission.rs

@@ -141,9 +141,13 @@ impl std::fmt::Display for SubmitCorrect {
         let seperator = "-------------------------------";
         write!(
             f,
-            "\n{seperator}\n{}\n{seperator}\nStatus   : {}\nLanguage : {}\nRuntime  : {}\t( beats {:.2}% )\nMemory   : {}\t( beats {:.2}% )\n",
+            "\n{seperator}\n{}\n{seperator}\n{status:10}: {}\n{lang:10}: {}\n{rt:10}: {:6}\t( beats {:.2}% )\n{mem:10}: {:6}\t( beats {:.2}% )\n",
             "Submission Correct!".green().bold(),
-            self.status_msg, self.lang, self.status_runtime.cyan(),self.runtime_percentile, self.status_memory.cyan(), self.memory_percentile
+            self.status_msg, self.lang, self.status_runtime.cyan(),self.runtime_percentile, self.status_memory.cyan(), self.memory_percentile,
+            status = "Status",
+            lang = "Language",
+            rt = "Runtime",
+            mem = "Memory"
         )
     }
 }

src/handlers/user.rs

@@ -17,9 +17,19 @@ impl UserMetadata {
 
 impl std::fmt::Display for UserMetadata {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
-        write!(f,
-            "User name        :\t{}\nQuestions Solved :\n\t- Easy   :\t{}\n\t- Medium :\t{}\n\t- Hard   :\t{}",
-            self.user_name, self.ac_easy, self.ac_medium, self.ac_hard
+        write!(
+            f,
+            "{:18}:{:>15}\n{qs:18}:\n{empty:8}{easy:10}:{ec:>15}\n{empty:8}{med:10}:{mc:>15}\n{empty:8}{hard:10}:{hc:>15}",
+            "Username",
+            self.user_name,
+            ec = self.ac_easy,
+            mc = self.ac_medium,
+            hc = self.ac_hard,
+            qs = "Questions Solved",
+            easy = "- Easy",
+            med = "- Medium",
+            hard = "- Hard",
+            empty = ""
         )
     }
 }