package com.ruoyi.web.controller.learn;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletResponse;

import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.dz.domain.DzSchool;
import com.ruoyi.dz.service.IDzClassesService;
import com.ruoyi.dz.service.IDzSchoolService;
import com.ruoyi.enums.UserTypeEnum;
import io.swagger.annotations.Api;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.learn.domain.LearnStudent;
import com.ruoyi.learn.dto.StudyRecordStatisticsDTO;
import com.ruoyi.learn.service.ILearnStudentService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;

/**
 * 学生Controller
 *
 * @author ruoyi
 * @date 2025-09-18
 */
@RestController
@RequestMapping("/learn/student")
@Api("后台-学习 - 学生")
public class LearnStudentController extends BaseController
{
    @Autowired
    private ILearnStudentService learnStudentService;

    @Autowired
    private IDzSchoolService schoolService;

    @Autowired
    private IDzClassesService classesService;
    /**
     * 查询学生列表
     */
    @PreAuthorize("@ss.hasPermi('learn:student:list')")
    @GetMapping("/list")
    public TableDataInfo list(LearnStudent learnStudent)
    {
        startPage();
        List<LearnStudent> list = learnStudentService.selectLearnStudentList(learnStudent);
        return getDataTable(list);
    }

    /**
     * 导出学生列表
     */
    @PreAuthorize("@ss.hasPermi('learn:student:export')")
    @Log(title = "学生", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, LearnStudent learnStudent)
    {
        List<LearnStudent> list = learnStudentService.selectLearnStudentList(learnStudent);
        ExcelUtil<LearnStudent> util = new ExcelUtil<LearnStudent>(LearnStudent.class);
        util.exportExcel(response, list, "学生数据");
    }

    /**
     * 获取学生详细信息
     */
    @PreAuthorize("@ss.hasPermi('learn:student:query')")
    @GetMapping(value = "/{studentId}")
    public AjaxResult getInfo(@PathVariable("studentId") Long studentId)
    {
        return success(learnStudentService.selectLearnStudentByStudentId(studentId));
    }

    /**
     * 新增学生
     */
    @PreAuthorize("@ss.hasPermi('learn:student:add')")
    @Log(title = "学生", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody LearnStudent learnStudent)
    {
        return toAjax(learnStudentService.insertLearnStudent(learnStudent));
    }

    /**
     * 修改学生
     */
    @PreAuthorize("@ss.hasPermi('learn:student:edit')")
    @Log(title = "学生", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody LearnStudent learnStudent)
    {
        return toAjax(learnStudentService.updateLearnStudent(learnStudent));
    }

    /**
     * 删除学生
     */
    @PreAuthorize("@ss.hasPermi('learn:student:remove')")
    @Log(title = "学生", businessType = BusinessType.DELETE)
    @DeleteMapping("/{studentIds}")
    public AjaxResult remove(@PathVariable Long[] studentIds)
    {
        return toAjax(learnStudentService.deleteLearnStudentByStudentIds(studentIds));
    }

    /**
     * 统计学习记录
     */
    @PreAuthorize("@ss.hasPermi('learn:student:list')")
    @GetMapping("/statisticStudyRecord")
    public TableDataInfo statisticStudyRecord(StudyRecordStatisticsDTO dto)
    {
        startPage();
        SysUser sysUser = SecurityUtils.getLoginUser().getUser();
        if (UserTypeEnum.isAgent(sysUser.getUserType())) {
            dto.setAgentId(sysUser.getUserTypeId());
        } else if(UserTypeEnum.isSchool(sysUser.getUserType())) {
            dto.setCampusId(sysUser.getUserTypeId());
        } else if(UserTypeEnum.isTeacher(sysUser.getUserType())) {
            dto.setTeacherId(sysUser.getUserTypeId());
        } else if(!UserTypeEnum.isSys(sysUser.getUserType())) {
            dto.setDeptId(sysUser.getDeptId());
        }
        List<StudyRecordStatisticsDTO> list = learnStudentService.statisticStudyRecord(dto);
        // 处理数据，设置学校名称和班级名称
        list = processData(list);
        return getDataTable(list);
    }

    public List<StudyRecordStatisticsDTO> processData(List<StudyRecordStatisticsDTO> list) {
        //将里面的campusId与schoolId汇总为一个list后，通过schoolService查询所有学校的信息List<DzSchool>
        List<Long> schoolIds = list.stream().map(StudyRecordStatisticsDTO::getSchoolId).filter(schoolId -> schoolId != null).collect(Collectors.toList());
        List<Long> campusIds = list.stream().map(StudyRecordStatisticsDTO::getCampusId).filter(campusId -> campusId != null).collect(Collectors.toList());
        // 合并schoolId和campusId，并去重
        List<Long> allIds = new ArrayList<>(schoolIds);
        allIds.addAll(campusIds);
        List<Long> distinctSchoolIds = allIds.stream().distinct().collect(Collectors.toList());
        Map<Long, String> schoolMap = new HashMap<>();
        if (CollectionUtils.isNotEmpty(distinctSchoolIds)) {
            List<DzSchool> schoolList = schoolService.selectDzSchoolListByIds(distinctSchoolIds);
            if (CollectionUtils.isNotEmpty(schoolList)) {
                schoolMap = schoolList.stream()
                        .collect(Collectors.toMap(dzSchool -> dzSchool.getId(), dzSchool -> dzSchool.getName()));
            }
        }

        //将里面的classId与campusClassId汇总为一个list后，通过classesService查询所有班级的信息List<DzClasses>
        List<Long> classIds = list.stream().map(StudyRecordStatisticsDTO::getClassId).filter(classId -> classId != null).collect(Collectors.toList());
        List<Long> campusClassIds = list.stream().map(StudyRecordStatisticsDTO::getCampusClassId).filter(campusClassId -> campusClassId != null).collect(Collectors.toList());
        // 合并classId和campusClassId，并去重
        List<Long> allClassIds = new java.util.ArrayList<>(classIds);
        allClassIds.addAll(campusClassIds);
        List<Long> distinctClassIds = allClassIds.stream().distinct().collect(Collectors.toList());
        Map<Long, String> classMap = new HashMap<>();
        if (CollectionUtils.isNotEmpty(distinctClassIds)) {
            List<com.ruoyi.dz.domain.DzClasses> classList = classesService.selectDzClassesListByIds(distinctClassIds);
            if (CollectionUtils.isNotEmpty(classList)) {
                classMap = classList.stream()
                        .collect(Collectors.toMap(dzClass -> dzClass.getClassId(), dzClass -> dzClass.getName()));
            }
        }
        for (StudyRecordStatisticsDTO dto : list) {
            // 设置注册学校名称（schoolId对应的学校名称）
            if (dto.getSchoolId() != null) {
                dto.setSchoolName(schoolMap.get(dto.getSchoolId()));
            }
            // 设置学校班级名称（classId对应的班级名称）
            if (dto.getClassId() != null) {
                dto.setSchoolClassName(classMap.get(dto.getClassId()));
            }
            // 设置培训校区名称（campusId对应的学校名称）
            if (dto.getCampusId() != null) {
                dto.setCampusName(schoolMap.get(dto.getCampusId()));
            }
            // 设置校区班级名称（campusClassId对应的班级名称）
            if (dto.getCampusClassId() != null) {
                dto.setCampusClassName(classMap.get(dto.getCampusClassId()));
            }
        }
        return list;
    }
}
