Add rust

execute.py

@@ -48,6 +48,8 @@ def check_correctness(check_program, timeout, task_id, completion_id, language):
         p = multiprocessing.Process(target=unsafe_execute_java, args=(check_program, result, timeout))
     elif language == "javascript":
         p = multiprocessing.Process(target=unsafe_execute_js, args=(check_program, result, timeout))
+    elif language == "rust":
+        p = multiprocessing.Process(target=unsafe_execute_rust, args=(check_program, result, timeout))
     else:
         raise ValueError(f"Language {language} not supported. Feel free to add it :)")
 
@@ -220,6 +222,27 @@ def unsafe_execute_js(check_program, result, timeout):
         except subprocess.TimeoutExpired as e:
             result.append("timed out")            
 
+def unsafe_execute_rust(check_program, result, timeout):
+
+    with create_tempdir():
+        open(f"test.rs", 'w').write(check_program)
+
+        log_path = "test.jsonl"
+        cargo_check: str = "cargo check --bin test --message-format json >> " + log_path
+        returned_val_compilation = os.system(cargo_check)
+        if returned_val_compilation == 0:
+
+            cargo_test: str = "cargo test --bin test --message-format json >> " + log_path
+            returned_val_execution = os.system(cargo_test)
+
+            if returned_val_execution == 0:
+                result.append("passed")
+            else:
+                result.append("failed: execution error")
+        else:
+            result.append("failed: compilation error")
+
+
 
 @contextlib.contextmanager
 def time_limit(seconds):