package com.ots.project.exam.controller;
import com.ots.common.utils.security.ShiroUtils;
import com.ots.framework.web.controller.BaseController;
import com.ots.framework.web.page.TableDataInfo;
import com.ots.project.exam.domain.SysUserExtend;
import com.ots.project.system.user.domain.User;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import java.util.Objects;

@Controller
@RequestMapping("/mail/index")
public class MainIndexController extends BaseController {
    
    @PostMapping("/listCompleteTestRecord")
    @ResponseBody
    public TableDataInfo listCompleteTestRecord(SysUserExtend sysUserExtend) {
        startPage();
        User sysUser = ShiroUtils.getSysUser();
        if ("01".equals(sysUser.getUserType())) {
            sysUserExtend.setParentUserId(sysUser.getUserId());
        } else {
            sysUserExtend.setUserType("02");
        }
        if (Objects.nonNull(sysUserExtend.getParentUserTempId())) {
            sysUserExtend.setParentUserId(sysUserExtend.getParentUserTempId());
        }

        return getDataTable(null);
    }
    
    @PostMapping("/listAnsweringTestRecord")
    @ResponseBody
    public TableDataInfo listAnsweringTestRecord(SysUserExtend sysUserExtend) {
        startPage();
        User sysUser = ShiroUtils.getSysUser();
        if ("01".equals(sysUser.getUserType())) {
            sysUserExtend.setParentUserId(sysUser.getUserId());
        } else {
            sysUserExtend.setUserType("02");
        }
        if (Objects.nonNull(sysUserExtend.getParentUserTempId())) {
            sysUserExtend.setParentUserId(sysUserExtend.getParentUserTempId());
        }

        return getDataTable(null);
    }
    
    @PostMapping("/listRecentlyCreatedTestPackage")
    @ResponseBody
    public TableDataInfo listRecentlyCreatedTestPackage(SysUserExtend sysUserExtend) {
        startPage();
        User sysUser = ShiroUtils.getSysUser();
        if ("01".equals(sysUser.getUserType())) {
            sysUserExtend.setParentUserId(sysUser.getUserId());
        } else {
            sysUserExtend.setUserType("02");
        }
        if (Objects.nonNull(sysUserExtend.getParentUserTempId())) {
            sysUserExtend.setParentUserId(sysUserExtend.getParentUserTempId());
        }

        return getDataTable(null);
    }
}