package com.njuzr.eaibackend.controller;

import com.njuzr.eaibackend.service.AIEvaluationService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

/**
 * @author BaiBai
 * @date 2025-01-17 10:58
 */
@Slf4j
@RestController
@RequestMapping("/api/evaluation")
public class EvaluationController {

    private final AIEvaluationService aiEvaluationService;;

    public EvaluationController(AIEvaluationService aiEvaluationService) {
        this.aiEvaluationService = aiEvaluationService;
    }

    @PostMapping("/overall")
    public MyResponse rewrite(
            @RequestParam Long assignmentId,
            @RequestParam Long studentId
    ) {
        return MyResponse.success(aiEvaluationService.rewrite(assignmentId, studentId));
    }

    @PostMapping("/sentence")
    public MyResponse rewritePerSentence(
            @RequestParam Long assignmentId,
            @RequestParam Long studentId
    ) {
        return MyResponse.success(aiEvaluationService.rewritePerSentence(assignmentId, studentId));
    }

    @GetMapping("/select/overall")
    public MyResponse selectOverall(
            @RequestParam Long assignmentId,
            @RequestParam Long studentId,
            @RequestParam(value = "version", defaultValue = "-1") Integer version
    ){
        return MyResponse.success(aiEvaluationService.getRewrite(assignmentId, studentId, version));
    }

    @GetMapping("/select/sentence")
    public MyResponse selectSentence(
            @RequestParam Long assignmentId,
            @RequestParam Long studentId,
            @RequestParam(value = "version", defaultValue = "-1") Integer version
    ){
        return MyResponse.success(aiEvaluationService.getRewritePerSentence(assignmentId, studentId, version));
    }

    @GetMapping("/record/list")
    public MyResponse getRewriteVersionList(
            @RequestParam Long assignmentId,
            @RequestParam Long studentId,
            @RequestParam(value = "version", defaultValue = "-1") Integer version
    ){
        return MyResponse.success(aiEvaluationService.getVersionList(assignmentId, studentId));
    }
}