epmet-cloud-rizhao0920/epmet-user/epmet-user-server/src/main/java/com/epmet/controller/IcResiUserController.java

/**
 * Copyright 2018 人人开源 https://www.renren.io
 * <p>
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * <p>
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * <p>
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.epmet.controller;

import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.TemplateExportParams;
import cn.afterturn.easypoi.excel.entity.params.ExcelExportEntity;
import cn.afterturn.easypoi.excel.export.ExcelExportService;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.epmet.commons.tools.annotation.LoginUser;
import com.epmet.commons.tools.exception.EpmetErrorCode;
import com.epmet.commons.tools.exception.RenException;
import com.epmet.commons.tools.page.PageData;
import com.epmet.commons.tools.security.dto.TokenDto;
import com.epmet.commons.tools.utils.ExcelUtils;
import com.epmet.commons.tools.utils.Result;
import com.epmet.commons.tools.validator.AssertUtils;
import com.epmet.commons.tools.validator.ValidatorUtils;
import com.epmet.commons.tools.validator.group.AddGroup;
import com.epmet.commons.tools.validator.group.DefaultGroup;
import com.epmet.commons.tools.validator.group.UpdateGroup;
import com.epmet.dto.IcResiUserDTO;
import com.epmet.dto.form.CustomerFormQueryDTO;
import com.epmet.dto.form.IcResiDetailFormDTO;
import com.epmet.dto.form.IcResiUserFormDTO;
import com.epmet.dto.form.IcResiUserPageFormDTO;
import com.epmet.dto.result.CustomerFormResultDTO;
import com.epmet.dto.result.FormGroupDTO;
import com.epmet.dto.result.FormItem;
import com.epmet.dto.result.HomeUserResultDTO;
import com.epmet.excel.IcResiUserExcel;
import com.epmet.feign.OperCustomizeOpenFeignClient;
import com.epmet.service.IcResiUserService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.*;


/**
 * 用户基础信息
 *
 * @author generator generator@elink-cn.com
 * @since v1.0.0 2021-10-26
 */
@Slf4j
@RestController
@RequestMapping("icresiuser")
public class IcResiUserController {
    private static final  String BASE_TABLE_NAME = "ic_resi_user";

    @Autowired
    private OperCustomizeOpenFeignClient operCustomizeOpenFeignClient;
    @Autowired
    private IcResiUserService icResiUserService;

    @GetMapping("page")
    public Result<PageData<IcResiUserDTO>> page(@RequestParam Map<String, Object> params){
        PageData<IcResiUserDTO> page = icResiUserService.page(params);
        return new Result<PageData<IcResiUserDTO>>().ok(page);
    }

    @GetMapping("{id}")
    public Result<IcResiUserDTO> get(@PathVariable("id") String id){
        IcResiUserDTO data = icResiUserService.get(id);
        return new Result<IcResiUserDTO>().ok(data);
    }

    @PostMapping
    public Result save(@RequestBody IcResiUserDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, AddGroup.class, DefaultGroup.class);
        icResiUserService.save(dto);
        return new Result();
    }

    @PutMapping
    public Result update(@RequestBody IcResiUserDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, UpdateGroup.class, DefaultGroup.class);
        icResiUserService.update(dto);
        return new Result();
    }

    @DeleteMapping
    public Result delete(@RequestBody String[] ids){
        //效验数据
        AssertUtils.isArrayEmpty(ids, "id");
        icResiUserService.delete(ids);
        return new Result();
    }

    @GetMapping("export")
    public void export(@RequestParam Map<String, Object> params, HttpServletResponse response) throws Exception {
        List<IcResiUserDTO> list = icResiUserService.list(params);
        ExcelUtils.exportExcelToTarget(response, null, list, IcResiUserExcel.class);
    }

    /**
     * @Author sun
     * @Description 党建互联平台--保存居民信息
     **/
    @PostMapping("add")
    public Result add(@LoginUser TokenDto tokenDto, @RequestBody List<IcResiUserFormDTO> formDTO) {
        icResiUserService.add(tokenDto, formDTO);
        return new Result();
    }

    /**
     * @Author sun
     * @Description 党建互联平台--修改居民信息
     **/
    @PostMapping("edit")
    public Result edit(@LoginUser TokenDto tokenDto, @RequestBody List<IcResiUserFormDTO> formDTO) {
        icResiUserService.edit(tokenDto, formDTO);
        return new Result();
    }

    @GetMapping("download/template")
    public void downloadTemplate(@RequestParam String customerId) throws Exception {
        CustomerFormQueryDTO queryDTO = new CustomerFormQueryDTO();
        queryDTO.setFormCode("resi_base_info");
        queryDTO.setCustomerId(customerId);
        Result<CustomerFormResultDTO> resultForm = operCustomizeOpenFeignClient.getCustomerForm(queryDTO);
        if (resultForm == null || !resultForm.success() ||resultForm.getData() == null){
            throw  new RenException(EpmetErrorCode.SERVER_ERROR.getCode());
        }
        System.out.println(JSON.toJSONString(resultForm.getData()));

        XSSFWorkbook workbook = new XSSFWorkbook();
        Map<String, List<ExcelExportEntity>> sheetHeaderMap = buildHeaderByItem(resultForm);
        //Workbook workbook = null;
        for (Map.Entry<String, List<ExcelExportEntity>> entry : sheetHeaderMap.entrySet()) {
            String sheetName = entry.getKey();
            List<ExcelExportEntity> headers = entry.getValue();
            System.out.println("headers:"+sheetName+JSON.toJSONString(headers));
            ExportParams exportParams = new ExportParams(null,sheetName);
            //exportParams.setAutoSize(true);
            List<Map<String,String>> dataSet = new ArrayList<>();
            HashMap<String, String> map = new HashMap<>();
            map.put("1","2");
            dataSet.add(map);
            new ExcelExportService().createSheetForMap(workbook, exportParams, headers, dataSet);
        }

        FileOutputStream fos = new FileOutputStream("//Users/liujianjun/Downloads/基础信息表/Dow.tt.xls");
        workbook.write(fos);
        fos.close();
    }

    @NotNull
    private Map<String,List<ExcelExportEntity>> buildHeaderByItem(Result<CustomerFormResultDTO> resultForm) {
        //form中的itemlist 为一级表头 但是要排除每个item中含有childGroup的

        List<FormItem> itemList = resultForm.getData().getItemList();
        List<FormGroupDTO> groupList = resultForm.getData().getGroupList();

        Map<String,List<ExcelExportEntity>> everySheetHeaderMap = new LinkedHashMap<>();

        List<ExcelExportEntity> firstSheetHeaderList = new ArrayList<>();
        //Map<String, String> groupNameMap = groupList.stream().collect(Collectors.toMap(FormGroupDTO::getGroupId,FormGroupDTO::getLabel));
        itemList.forEach(item->{
            if (StringUtils.isBlank(item.getColumnName())){
                return;
            }

            if (item.getChildGroup() == null){
                ExcelExportEntity header = new ExcelExportEntity(item.getLabel(),item.getColumnName().concat(String.valueOf(item.getColumnNum())),30);
                header.setNeedMerge(true);
                firstSheetHeaderList.add(header);
                return;
            }
            everySheetHeaderMap.putIfAbsent(resultForm.getData().getFormName(),firstSheetHeaderList);

            //这些是动态的 formGroup
            if (item.getChildGroup() != null){
                //baseTableName单独的一个sheet
                System.out.println("childGroup:"+item.getLabel());
                if (BASE_TABLE_NAME.equals(item.getTableName())){
                   // header = new ExcelExportEntity(item.getChildGroup().getLabel(),item.getChildGroup().getTableName());
                    //header.setNeedMerge(true);
                    List<ExcelExportEntity> otherSheetHeaderList = new ArrayList<>();
                    //这里是设置除基础信息之外的sheet的表头
                    item.getChildGroup().getItemList().forEach(item2->{
                        if (!BASE_TABLE_NAME.equals(item2.getTableName())){
                            everySheetHeaderMap.putIfAbsent(item.getLabel(),otherSheetHeaderList);
                        }
                        ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum())));

                        otherSheetHeaderList.add(secondHeader);
                        if (!"radio".equals(item2.getItemType()) && CollectionUtils.isNotEmpty(item2.getOptions())){
                            secondHeader.setNeedMerge(true);
                            List<ExcelExportEntity> thirdHeaderList = new ArrayList<>();
                            item2.getOptions().forEach(child->{
                                ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel());
                                thirdHeaderList.add(thirdHeader);
                            });
                            secondHeader.setList(thirdHeaderList);
                        }

                    });
                    //header.setList(secondHeaderList);
                    //otherSheetHeaderList.add(header);
                }
            }
        });
        groupList.forEach(item->{
           /* if (!"兴趣爱好".equals(item.getLabel()) && !"宗教信仰".equals(item.getLabel())){
                return;
            }*/
            if (!BASE_TABLE_NAME.equals(item.getTableName())){
               return;
           }
            ExcelExportEntity header = new ExcelExportEntity(item.getLabel(),item.getTableName());
            header.setNeedMerge(true);
            List<ExcelExportEntity> secondHeaderList = new ArrayList<>();
            item.getItemList().forEach(item2->{
                ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum())));
                secondHeader.setNeedMerge(true);
                secondHeaderList.add(secondHeader);
                if (com.baomidou.mybatisplus.core.toolkit.CollectionUtils.isNotEmpty(item2.getOptions())){
                    List<ExcelExportEntity> thirdHeaderList = new ArrayList<>();
                    item2.getOptions().forEach(child->{
                        ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel(),child.getValue()+ new Random(1).nextInt(2));
                        thirdHeader.setNeedMerge(true);
                        thirdHeaderList.add(thirdHeader);
                    });
                    secondHeader.setList(thirdHeaderList);
                }

            });
            header.setList(secondHeaderList);
            firstSheetHeaderList.add(header);
            System.out.println(JSON.toJSONString(firstSheetHeaderList));
        });
        return everySheetHeaderMap;
    }

    private void buildHeader(Map<String, List<ExcelExportEntity>> everySheetHeaderMap, FormItem item, ExcelExportEntity header) {
        List<ExcelExportEntity> firstSheetHeaderList = new ArrayList<>();
        List<ExcelExportEntity> secondHeaderList = new ArrayList<>();
        item.getChildGroup().getItemList().forEach(item2->{
            if (!BASE_TABLE_NAME.equals(item2.getTableName())){
                everySheetHeaderMap.putIfAbsent(item.getLabel(),firstSheetHeaderList);
            }
            ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum())));
            secondHeader.setNeedMerge(true);
            secondHeaderList.add(secondHeader);
            if (CollectionUtils.isNotEmpty(item2.getOptions())){
                List<ExcelExportEntity> thirdHeaderList = new ArrayList<>();
                item2.getOptions().forEach(child->{
                    ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel());
                    thirdHeaderList.add(thirdHeader);
                });
                secondHeader.setList(thirdHeaderList);
            }

        });
        header.setList(secondHeaderList);
        firstSheetHeaderList.add(header);
        everySheetHeaderMap.putIfAbsent(item.getLabel(),firstSheetHeaderList);
    }

    /**
     * @Description 根据房间号查人
     * @Param formDTO
     * @Return {@link Result<List<HomeUserResultDTO>>}
     * @Author zhaoqifeng
     * @Date 2021/11/1 11:04
     */
    @PostMapping("getpeoplebyroom")
    public Result<List<HomeUserResultDTO>> getPeopleByRoom(@RequestBody IcResiUserDTO formDTO) {
        return new Result<List<HomeUserResultDTO>>().ok(icResiUserService.getPeopleByRoom(formDTO.getHomeId()));
    }

    @PostMapping("listresi")
    public Result<PageData<Map<String,Object>>> queryListResi1(@LoginUser TokenDto tokenDto, @RequestBody IcResiUserPageFormDTO pageFormDTO){
        //pageFormDTO.setCustomerId("45687aa479955f9d06204d415238f7cc");
        pageFormDTO.setCustomerId(tokenDto.getCustomerId());
        ValidatorUtils.validateEntity(pageFormDTO,IcResiUserPageFormDTO.AddUserInternalGroup.class);
        return new Result<PageData<Map<String,Object>>>().ok(icResiUserService.pageResiMap(pageFormDTO));
    }
    /**
     * 编辑页面，显示居民信息详情
     *
     * @param pageFormDTO
     * @return  com.epmet.commons.tools.utils.Result
     * @author yinzuomei
     * @date 2021/10/28 10:29 上午
     */
    @PostMapping("detail")
    public Result queryIcResiDetail(@LoginUser TokenDto tokenDto,@RequestBody IcResiDetailFormDTO pageFormDTO){
        //pageFormDTO.setCustomerId("45687aa479955f9d06204d415238f7cc");
        pageFormDTO.setCustomerId(tokenDto.getCustomerId());
        ValidatorUtils.validateEntity(pageFormDTO,IcResiDetailFormDTO.AddUserInternalGroup.class);
        return new Result().ok(icResiUserService.queryIcResiDetail(pageFormDTO));
    }

    @RequestMapping(value = "/exportExcel")
    public void exportExcel(/*@LoginUser TokenDto tokenDto,*/ @RequestBody IcResiUserPageFormDTO pageFormDTO, HttpServletResponse response) throws IOException {
        pageFormDTO.setCustomerId("45687aa479955f9d06204d415238f7cc");
        pageFormDTO.setPageFlag(false);
        PageData<Map<String, Object>> mapPageData = icResiUserService.pageResiMap(pageFormDTO);
        List<Map<String, Object>> list = mapPageData.getList();
        Map<String, Object> mapData = new HashMap<>();
        log.info("list:{}", JSON.toJSONString(list));

        mapData.put("list", list);
        String templatePath = "excel/ic_resi_info_cid.xls";


        Workbook workbook = ExcelExportUtil.exportExcel(new TemplateExportParams(templatePath), mapData);
        //header
        response.setHeader("content-Type", "application/vnd.ms-excel");
        response.addHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode("继续追踪导出详情-" + ".xlsx", "UTF-8"));
        workbook.write(response.getOutputStream());

       /*
        //方式1 通过mv导出
       ModelAndView mv = new ModelAndView(new EasypoiTemplateExcelView());
        mv.addObject(TemplateExcelConstants.FILE_NAME, "继续追踪导出详情-".concat(DateUtils.formatDate()));
        mv.addObject(TemplateExcelConstants.PARAMS, new TemplateExportParams(templatePath,dataType));
        mv.addObject(TemplateExcelConstants.MAP_DATA, mapData);
        return mv;
        */

    }

}