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.params.ExcelExportEntity;
import cn.afterturn.easypoi.excel.export.ExcelExportService;
import com.alibaba.fastjson.JSON;
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.IcResiUserFormDTO;
import com.epmet.dto.result.CustomerFormResultDTO;
import com.epmet.dto.result.FormGroupDTO;
import com.epmet.dto.result.FormItem;
import com.epmet.excel.IcResiUserExcel;
import com.epmet.feign.OperCustomizeOpenFeignClient;
import com.epmet.service.IcResiUserService;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Workbook;
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.util.*;


/**
 * 用户基础信息
 *
 * @author generator generator@elink-cn.com
 * @since v1.0.0 2021-10-26
 */
@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()));


        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();
            ExportParams exportParams = new ExportParams(null, sheetName);

            if (workbook == null) {
                List<Map<String, String>> dataList =new ArrayList<>();
                Map<String, String> dataMap = new HashMap<>();
                dataMap.put("GENDER0", "1");
                dataMap.put("ID_CARD0", "371888991");
                dataList.add(dataMap);
                workbook = ExcelExportUtil.exportExcel(exportParams, headers, dataList);
            }
            new ExcelExportService().createSheetForMap(workbook, exportParams, headers,new ArrayList<>());
        }


        //exportParams.setDictHandler(new ExcelDiceAddressListHandlerImpl());
        System.out.println("header:"+JSON.toJSONString(sheetHeaderMap));
        //Workbook workbook = ExcelExportUtil.exportExcel( new ExportParams(), sheetHeaderMap.get(), dataList);
        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) {


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

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

        itemList.forEach(item->{
            if (StringUtils.isBlank(item.getColumnName())){
                return;
            }
            ExcelExportEntity header = new ExcelExportEntity(item.getLabel(),item.getColumnName().concat(String.valueOf(item.getColumnNum())));
            header.setNeedMerge(true);

            //这些是动态的 formGroup
            if (item.getChildGroup() != null){
                //baseTableName单独的一个sheet
                System.out.println(item.getLabel());
                if (BASE_TABLE_NAME.equals(item.getTableName())){
                    header = new ExcelExportEntity(item.getChildGroup().getLabel(),item.getChildGroup().getTableName());
                    header.setNeedMerge(true);
                    buildHeader(everySheetHeaderMap, item, header);
                }
            }
        });
        List<ExcelExportEntity> firstSheetHeaderList = new ArrayList<>();
        groupList.forEach(item->{
           /* if (!"兴趣爱好".equals(item.getLabel()) && !"宗教信仰".equals(item.getLabel())){
                return;
            }*/
           if (!BASE_TABLE_NAME.equals(item.getTableName())){
               System.out.println(item.getLabel()+"--"+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())),30);
                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(),child.getValue()+ new Random(1).nextInt(2),30);
                        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);
    }

}