api/src/main/java/com/xjrsoft/module/schedule/util/CourseTableExportQueryUtil.java

package com.xjrsoft.module.schedule.util;

import com.xjrsoft.module.courseTable.entity.ClassTime;
import com.xjrsoft.module.schedule.vo.CourseDetailVo;
import com.xjrsoft.module.schedule.vo.CourseTableVo;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author dzx
 * @date 2025/4/9
 */
public class CourseTableExportQueryUtil {

    /**
     * 合并9列
     * @param rowNumber 写入的行号
     */
    public void createThirdTitle(Workbook workbook, Sheet sheet, int rowNumber, String leftTitle, String rightTitle){
        Font font = workbook.createFont();
        font.setFontName("宋体");
        //font.setColor(IndexedColors.RED.getIndex()); // 设置字体颜色为红色
        font.setFontHeightInPoints((short) 14);

        //设置样式
        CellStyle cellStyle = workbook.createCellStyle();
        cellStyle.setFont(font); // 将字体应用到样式
        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        cellStyle.setAlignment(HorizontalAlignment.LEFT);
        cellStyle.setBorderTop(BorderStyle.THIN);
        cellStyle.setBorderBottom(BorderStyle.THIN);
        cellStyle.setBorderLeft(BorderStyle.THIN);
        cellStyle.setBorderRight(BorderStyle.THIN);

        //创建一行
        Row row = sheet.createRow(rowNumber);
        Cell cell = row.createCell(0);
        cell.setCellValue(leftTitle);
        cell.setCellStyle(cellStyle);
        for (int i = 1; i < 4; i ++){
            cell = row.createCell(i);
            cell.setCellStyle(cellStyle);
        }
        sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber, 0, 3));

        cellStyle = workbook.createCellStyle();
        cellStyle.setFont(font); // 将字体应用到样式
        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        cellStyle.setAlignment(HorizontalAlignment.RIGHT);
        cellStyle.setBorderTop(BorderStyle.THIN);
        cellStyle.setBorderBottom(BorderStyle.THIN);
        cellStyle.setBorderLeft(BorderStyle.THIN);
        cellStyle.setBorderRight(BorderStyle.THIN);

        cell = row.createCell(4);
        cell.setCellValue(rightTitle);
        cell.setCellStyle(cellStyle);
        for (int i = 5; i < 9; i ++){
            cell = row.createCell(i);
            cell.setCellStyle(cellStyle);
        }
        sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber, 4, 8));
    }


    public void createContent(Workbook workbook, Sheet sheet, int rowNumber, CourseTableVo data, String courseType){
        int firstRow = rowNumber;
        Map<Integer, ClassTime> timeNumberMap = data.getClassTimeList().stream().collect(Collectors.toMap(ClassTime::getNumber, x -> x));
        Map<String, List<CourseDetailVo>> timeNumberDataMap = data.getCourseList().stream().filter(x -> x.getTimeNumber() != null).collect(Collectors.groupingBy(CourseDetailVo::getTimeNumber));

        List<Integer> weekList = new ArrayList<>();
        weekList.add(1);
        weekList.add(2);
        weekList.add(3);
        weekList.add(4);
        weekList.add(5);
        weekList.add(6);
        weekList.add(7);
        Collections.sort(weekList);

        List<Integer> timeNumberList = data.getClassTimeList().stream().map(ClassTime::getNumber).collect(Collectors.toList());
        Collections.sort(timeNumberList);

        Map<Integer, Integer> timePeriodMap = data.getClassTimeList().stream().collect(Collectors.toMap(ClassTime::getNumber, ClassTime::getTimePeriod));

        ArrayList<ArrayList<String>> dataList = new ArrayList<>();

        for (Integer timeNumber : timeNumberList) {
            ArrayList<String> rowData = new ArrayList<>();
            Integer timePeriod = timePeriodMap.get(timeNumber);
            String timePeriodStr = "";
            if (timePeriod == 1) {
                timePeriodStr = "上午";
            } else if (timePeriod == 2) {
                timePeriodStr = "下午";
            } else if (timePeriod == 3) {
                timePeriodStr = "晚上";
            }
            rowData.add(timePeriodStr);
            ClassTime classTime = timeNumberMap.get(timeNumber);
            String timeNumberStr = classTime.getShortName() + "\r\n" + classTime.getSummerStartTime() + "-" + classTime.getSummerEndTime();
            rowData.add(timeNumberStr);
            List<CourseDetailVo> list1 = timeNumberDataMap.get(timeNumber.toString());
            if (list1 == null) {
                for (Integer week : weekList) {
                    rowData.add("");
                }
                dataList.add(rowData);
                continue;
            }
            Map<Integer, List<CourseDetailVo>> weeksMap = list1.stream().collect(Collectors.groupingBy(CourseDetailVo::getWeeks));
            for (Integer week : weekList) {
                String content = "";
                List<CourseDetailVo> list = weeksMap.get(week);
                if (list != null && !list.isEmpty()) {
                    if (list.size() == 1) {
                        CourseDetailVo courseDetailVo = list.get(0);
                        content += courseDetailVo.getCourseName() + "\r\n";
                        if ("teacher".equals(courseType)) {
                            if(courseDetailVo.getClassroomName() != null){
                                content += courseDetailVo.getClassroomName() + "\r\n";
                            }
                            content += courseDetailVo.getClassName();
                        } else if ("class".equals(courseType)) {
                            if(courseDetailVo.getClassroomName() != null){
                                content += courseDetailVo.getClassroomName() + "\r\n";
                            }

                            content += courseDetailVo.getTeacherName();
                        } else if("classroom".equals(courseType)){
                            content += courseDetailVo.getClassName() + "\r\n";
                            content += courseDetailVo.getTeacherName();
                        }
                    } else {
                        CourseDetailVo courseDetailVo = list.get(0);
                        content += courseDetailVo.getCourseName() + "\r\n";
                        if ("teacher".equals(courseType)) {
                            if(courseDetailVo.getClassroomName() != null){
                                content += courseDetailVo.getClassroomName() + "\r\n";
                            }
                            content += courseDetailVo.getClassName();
                        } else if ("class".equals(courseType)) {
                            if(courseDetailVo.getClassroomName() != null){
                                content += courseDetailVo.getClassroomName() + "\r\n";
                            }
                            Set<String> classNames = list.stream().map(CourseDetailVo::getTeacherName).collect(Collectors.toSet());
                            content += classNames.toString().replace("]", "").replace("[", "").replace(" ", "");
                        } else if("classroom".equals(courseType)){
                            content += courseDetailVo.getClassName() + "\r\n";
                            content += courseDetailVo.getTeacherName();
                        }
                    }
                }
                rowData.add(content);
            }
            dataList.add(rowData);
        }

        Font font = workbook.createFont();
        font.setFontName("宋体");
        font.setFontHeightInPoints((short) 9);

        // 创建一个单元格样式对象
        CellStyle cellStyle = workbook.createCellStyle();
        cellStyle.setFont(font); // 将字体应用到样式
        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        cellStyle.setAlignment(HorizontalAlignment.CENTER);
        cellStyle.setWrapText(true);
        cellStyle.setBorderTop(BorderStyle.THIN);
        cellStyle.setBorderBottom(BorderStyle.THIN);
        cellStyle.setBorderLeft(BorderStyle.THIN);
        cellStyle.setBorderRight(BorderStyle.THIN);

        for (ArrayList<String> strings : dataList) {
            Row row = sheet.createRow(rowNumber);
            int cellNumber = 0;
            for (String string : strings) {
                Cell row1cell1 = row.createCell(cellNumber);
                row1cell1.setCellValue(string);
                row1cell1.setCellStyle(cellStyle);
                cellNumber++;
            }
            rowNumber++;
            row.setHeightInPoints(48);
        }
        //合并
        Map<Integer, List<Integer>> collect = data.getClassTimeList().stream()
                .collect(Collectors.groupingBy(ClassTime::getTimePeriod, Collectors.mapping(ClassTime::getNumber, Collectors.toList())));
        for (Integer i : collect.keySet()) {
            int lastRow = firstRow + collect.get(i).size() - 1;
            if (lastRow == firstRow) {
                continue;
            }
            sheet.addMergedRegion(new CellRangeAddress(firstRow, lastRow, 0, 0));

            firstRow = firstRow + collect.get(i).size();
        }
    }

    public void createAllCourseTableTitle(Workbook workbook, Sheet sheet, List<ClassTime> classTimeList){
        // 创建一个字体对象
        Font font = workbook.createFont();
        font.setBold(true);// 设置为粗体
        font.setFontName("宋体");
        //font.setColor(IndexedColors.RED.getIndex()); // 设置字体颜色为红色
        font.setFontHeightInPoints((short) 12);

        // 创建一个单元格样式对象
        CellStyle cellStyle = workbook.createCellStyle();
        cellStyle.setFont(font); // 将字体应用到样式
        cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        cellStyle.setAlignment(HorizontalAlignment.CENTER);
        cellStyle.setBorderTop(BorderStyle.THIN);
        cellStyle.setBorderBottom(BorderStyle.THIN);
        cellStyle.setBorderLeft(BorderStyle.THIN);
        cellStyle.setBorderRight(BorderStyle.THIN);
        cellStyle.setWrapText(true);

        int rowNumber = 0;
        Row row = sheet.createRow(rowNumber);
        row.setHeightInPoints(48);

        List<String> weekList = new ArrayList<>();
        weekList.add("周一");
        weekList.add("周二");
        weekList.add("周三");
        weekList.add("周四");
        weekList.add("周五");
        weekList.add("周六");
        weekList.add("周日");

        int cellNumber = 0;
        Cell row1cell1 = row.createCell(cellNumber);
        row1cell1.setCellValue("班级");
        row1cell1.setCellStyle(cellStyle);
        sheet.setColumnWidth(0, 24 * 256);
        sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, cellNumber, cellNumber));
        cellNumber ++;

        int size = classTimeList.size();

        Row row2 = sheet.createRow(rowNumber + 1);
        int cellNumber2 = 1;

        for (String week : weekList) {
            row1cell1 = row.createCell(cellNumber);
            row1cell1.setCellValue(week);
            row1cell1.setCellStyle(cellStyle);
            int firstCol = cellNumber;
            int lastCol = firstCol + size - 1;
            sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber, firstCol, lastCol));
            cellNumber = cellNumber + size;

            for (ClassTime classTime : classTimeList) {
                row1cell1 = row2.createCell(cellNumber2);
                row1cell1.setCellValue(classTime.getShortName() + "\r\n" + classTime.getSummerStartTime() + "-" + classTime.getSummerEndTime());
                row1cell1.setCellStyle(cellStyle);
                sheet.setColumnWidth(cellNumber2, 24 * 256);
                cellNumber2 ++;
            }
        }
    }
}