fix single file evaluate

Author: kunyuanxu-star

src/main.rs

@@ -11,7 +11,7 @@ struct Exercise {
     path: String,
     #[serde(rename = "type")]
     exercise_type: String,  
-    score: i32,  // Added: Each exercise score
+    score: i32, 
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -25,15 +25,15 @@ struct ExerciseConfig {
 struct ExerciseResult {
     name: String,
     result: bool,
-    score: i32,  // Store score for each exercise
+    score: i32, 
 }
 
 #[derive(Serialize, Deserialize, Debug)]
 struct Statistics {
     total_exercises: usize,
     total_successes: usize,
     total_failures: usize,
-    total_score: i32,  // Total score for the assessment
+    total_score: i32,  
     total_time: u64,
 }
 
@@ -53,7 +53,7 @@ fn main() {
     let mode = &args[1];
     let start_time = Instant::now();
 
-    // Load the exercise config
+    
     let config = match load_exercise_config("exercise_config.json") {
         Ok(cfg) => cfg,
         Err(e) => {
@@ -68,54 +68,52 @@ fn main() {
             total_exercises: 0,
             total_successes: 0,
             total_failures: 0,
-            total_score: 0, // Initialize total score to 0
+            total_score: 0,
             total_time: 0,
         },
     };
 
-    // Evaluate exercises from config
+    
     evaluate_exercises_from_config(mode, config, &mut report);
 
-    // Calculate total time
+    
     report.statistics.total_time = start_time.elapsed().as_secs();
     report.statistics.total_exercises = report.statistics.total_successes + report.statistics.total_failures;
 
-    // Output summary
+    
     println!("\nSummary:");
     println!("Total exercises: {}", report.statistics.total_exercises);
     println!("Total successes: {}", report.statistics.total_successes);
     println!("Total failures: {}", report.statistics.total_failures);
-    println!("Total score: {}", report.statistics.total_score);  // Output the total score
+    println!("Total score: {}", report.statistics.total_score);
 
-    // Save the report to a JSON file
+    
     if let Err(e) = save_report_to_json("report.json", &report) {
         eprintln!("Error saving report: {}", e);
     }
 }
 
-// Load exercise configuration from JSON
+
 fn load_exercise_config(file_path: &str) -> Result<ExerciseConfig, io::Error> {
     let file = File::open(file_path)?;
     let config: ExerciseConfig = serde_json::from_reader(file)?;
     Ok(config)
 }
 
-// Evaluate all exercises from the configuration
+
 fn evaluate_exercises_from_config(mode: &str, config: ExerciseConfig, report: &mut Report) {
     let all_exercises = [config.easy, config.normal, config.hard].concat();
-    
+
     for exercise in all_exercises {
         println!("\nEvaluating {}: {}", exercise.exercise_type, exercise.name);
         let result = evaluate_exercise(&exercise);
 
-        // Calculate score based on result
         let score = if result { exercise.score } else { 0 };
 
-        // Add result to the report
         report.exercises.push(ExerciseResult {
             name: exercise.name.clone(),
             result,
-            score, 
+            score,
         });
 
         if result {
@@ -124,7 +122,6 @@ fn evaluate_exercises_from_config(mode: &str, config: ExerciseConfig, report: &m
             report.statistics.total_failures += 1;
         }
 
-        // Add score to total score
         report.statistics.total_score += score;
 
         if mode == "watch" && !ask_to_continue() {
@@ -133,7 +130,7 @@ fn evaluate_exercises_from_config(mode: &str, config: ExerciseConfig, report: &m
     }
 }
 
-// Evaluate a single exercise
+
 fn evaluate_exercise(exercise: &Exercise) -> bool {
     let exercise_path = PathBuf::from(&format!("./exercises/{}", exercise.path));
     match exercise.exercise_type.as_str() {
@@ -146,30 +143,61 @@ fn evaluate_exercise(exercise: &Exercise) -> bool {
     }
 }
 
-// Evaluate a single file Rust exercise
+// 评测单文件 Rust 习题（使用 rustc --test 并执行测试）
 fn evaluate_single_file(file_path: &PathBuf) -> bool {
-    let output = Command::new("rustc")
+    // 获取文件名（不带扩展名）
+    let test_binary = file_path.with_extension(""); 
+
+    // 编译测试文件
+    let compile_output = Command::new("rustc")
+        .arg("--test")  // 使用 rustc --test 进行编译
         .arg(file_path)
+        .arg("-o")
+        .arg(&test_binary)  // 指定输出文件
         .output();
 
-    match output {
-        Ok(out) => {
-            if out.status.success() {
-                println!("\x1b[32m{}: PASSED\x1b[0m", file_path.display());
-                true
+    if let Ok(output) = compile_output {
+        if output.status.success() {
+            // 编译成功，运行测试二进制文件
+            let test_output = Command::new(&test_binary)
+                .output();
+
+            let test_passed = match test_output {
+                Ok(test_run) => {
+                    if test_run.status.success() {
+                        println!("\x1b[32m{}: TEST PASSED\x1b[0m", file_path.display());
+                        true
+                    } else {
+                        println!("\x1b[31m{}: TEST FAILED\x1b[0m", file_path.display());
+                        false
+                    }
+                }
+                Err(_) => {
+                    eprintln!("Error running test executable for {}", file_path.display());
+                    false
+                }
+            };
+
+            // 删除测试二进制文件
+            if let Err(e) = fs::remove_file(&test_binary) {
+                eprintln!("Failed to remove test binary {}: {}", test_binary.display(), e);
             } else {
-                println!("\x1b[31m{}: FAILED\x1b[0m", file_path.display());
-                false
+                println!("Successfully removed test binary: {}", test_binary.display());
             }
+
+            return test_passed;
+        } else {
+            // 编译失败
+            eprintln!("\x1b[31m{}: COMPILATION FAILED\x1b[0m", file_path.display());
+            return false;
         }
-        Err(_) => {
-            eprintln!("Error executing rustc for {}", file_path.display());
-            false
-        }
+    } else {
+        eprintln!("Error executing rustc --test for {}", file_path.display());
+        return false;
     }
 }
 
-// Evaluate a cargo project
+// 评测 Cargo 项目
 fn evaluate_cargo_project(proj_path: &PathBuf) -> bool {
     let build_success = run_cargo_command(proj_path, "build");
     let test_success = run_cargo_command(proj_path, "test");
@@ -183,13 +211,12 @@ fn evaluate_cargo_project(proj_path: &PathBuf) -> bool {
         println!("\x1b[31m{}: FAILED\x1b[0m", proj_path.display());
     }
 
-    // Clean up the target directory after evaluation
     clean_target_directory(proj_path);
 
     passed
 }
 
-// Run a cargo command (build, test, clippy)
+// 运行 Cargo 命令
 fn run_cargo_command(proj_path: &PathBuf, command: &str) -> bool {
     let output = Command::new("cargo")
         .arg(command)
@@ -202,7 +229,7 @@ fn run_cargo_command(proj_path: &PathBuf, command: &str) -> bool {
     }
 }
 
-// Clean up the target directory after evaluating a cargo project
+// 清理 target 目录
 fn clean_target_directory(proj_path: &PathBuf) {
     let target_dir = proj_path.join("target");
 
@@ -215,15 +242,15 @@ fn clean_target_directory(proj_path: &PathBuf) {
     }
 }
 
-// Ask the user whether to continue after each evaluation
+// 用户确认是否继续
 fn ask_to_continue() -> bool {
     let mut input = String::new();
     println!("\nPress any key to continue, or 'q' to quit.");
     io::stdin().read_line(&mut input).unwrap();
     input.trim().to_lowercase() != "q"
 }
 
-// Save the report to a JSON file
+// 保存评测报告
 fn save_report_to_json(file_name: &str, report: &Report) -> io::Result<()> {
     let file = File::create(file_name)?;
     serde_json::to_writer_pretty(file, report)?;