wait for submission

Author: Heliozoa

tmc-langs-cli/src/main.rs

@@ -749,6 +749,14 @@ fn run() -> Result<()> {
                 .context("Failed to submit")?;
 
             print_result_as_json(&new_submission)?;
+        } else if let Some(matches) = matches.subcommand_matches("wait-for-submission") {
+            let submission_url = matches.value_of("submission-url").unwrap();
+            let submission_finished = core
+                .wait_for_submission(submission_url)
+                .context("Failed while waiting for submissions")?;
+            let submission_finished = serde_json::to_string(&submission_finished)
+                .context("Failed to serialize submission results")?;
+            println!("{}", submission_finished);
         } else if let Some(_matches) = matches.subcommand_matches("get-exercise-updates") {
             let course_id = matches.value_of("course-id").unwrap();
             let course_id = into_usize(course_id)?;

tmc-langs-core/src/tmc_core.rs

@@ -16,6 +16,8 @@ use std::collections::HashMap;
 use std::io::Write;
 use std::path::Path;
 use std::path::PathBuf;
+use std::thread;
+use std::time::Duration;
 use tempfile::NamedTempFile;
 use tmc_langs_util::task_executor;
 
@@ -387,6 +389,18 @@ impl TmcCore {
         result
     }
 
+    pub fn wait_for_submission(&self, submission_url: &str) -> Result<SubmissionFinished> {
+        loop {
+            match self.check_submission(submission_url)? {
+                SubmissionProcessingStatus::Finished(f) => return Ok(*f),
+                SubmissionProcessingStatus::Processing(_p) => {
+                    // todo: report status
+                    thread::sleep(Duration::from_secs(1));
+                }
+            }
+        }
+    }
+
     /// Fetches the course's exercises from the server,
     /// and finds new or updated exercises.
     /// If an exercise's id is not found in the checksum map, it is considered new.