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.TemplateExportParams;
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.ConvertUtils;
import com.epmet.commons.tools.utils.Result;
import com.epmet.commons.tools.validator.ValidatorUtils;
import com.epmet.dto.IcResiUserDTO;
import com.epmet.dto.form.*;
import com.epmet.dto.result.*;
import com.epmet.feign.EpmetUserOpenFeignClient;
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.jetbrains.annotations.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;


/**
 * 用户基础信息
 *
 * @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;
    @Autowired
    private EpmetUserOpenFeignClient epmetUserOpenFeignClient;

    /**
     * @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();
    }

    /**
     * @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());
        pageFormDTO.setStaffId(tokenDto.getUserId());
        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(@RequestHeader String customerId, @RequestBody IcResiUserPageFormDTO pageFormDTO, HttpServletResponse response) throws IOException {
        pageFormDTO.setCustomerId(customerId);
        pageFormDTO.setPageFlag(false);
        CustomerFormResultDTO resiFormItems = getResiFormItems(pageFormDTO.getCustomerId());

        Map<String, Map<String, FormItem>> otherSheetItems = new HashMap<>();

        //主表的
        for (FormItem formItem : resiFormItems.getItemList()) {
            if (StringUtils.isBlank(formItem.getColumnName())) {
                continue;
            }

            Map<String, FormItem> itemMap = otherSheetItems.getOrDefault(formItem.getTableName(), new HashMap<>());
            otherSheetItems.putIfAbsent(formItem.getTableName(), itemMap);
            if (formItem.getItemType().equals("checkbox") || formItem.getItemType().equals("select") || formItem.getItemType().equals("radio")) {
                itemMap.put(formItem.getColumnName().concat(formItem.getColumnNum() == 0 ? "" : formItem.getColumnNum().toString()), formItem);
            }
            if (formItem.getChildGroup() != null) {
                itemMap = otherSheetItems.getOrDefault(formItem.getChildGroup().getTableName(), new HashMap<>());
                otherSheetItems.putIfAbsent(formItem.getChildGroup().getTableName(), itemMap);
                for (FormItem2 item2 : formItem.getChildGroup().getItemList()) {
                    if (StringUtils.isBlank(item2.getColumnName())) {
                        continue;
                    }

                    if (item2.getItemType().equals("checkbox") || item2.getItemType().equals("select") || item2.getItemType().equals("radio")) {
                        itemMap.put(item2.getColumnName().concat(item2.getColumnNum() == 0 ? "" : item2.getColumnNum().toString()), ConvertUtils.sourceToTarget(item2, FormItem.class));
                    }
                }

            }
        }
        //其他sheet

        for (FormGroupDTO groupItem : resiFormItems.getGroupList()) {
            if (groupItem.getItemList() == null) {
                continue;
            }
            Map<String, FormItem> itemMap = otherSheetItems.getOrDefault(groupItem.getTableName(), new HashMap<>());
            otherSheetItems.putIfAbsent(groupItem.getTableName(), itemMap);
            for (FormItem2 formItem2 : groupItem.getItemList()) {
                if (StringUtils.isBlank(formItem2.getColumnName())) {
                    continue;
                }
                if (formItem2.getItemType().equals("checkbox") || formItem2.getItemType().equals("select") || formItem2.getItemType().equals("radio")) {
                    itemMap.put(formItem2.getColumnName().concat(formItem2.getColumnNum() == 0 ? "" : formItem2.getColumnNum().toString()), ConvertUtils.sourceToTarget(formItem2, FormItem.class));
                }
            }
        }

        Map<String, Map<String, Object>> resiMainList = icResiUserService.getDataForExport(otherSheetItems.get(BASE_TABLE_NAME), pageFormDTO.getCustomerId(), pageFormDTO.getFormCode(), BASE_TABLE_NAME, pageFormDTO.getConditions());
        //resiMainList = (List<Map<String, Object>>)JSON.parse("[{\"IS_BDHJ\":\"1\",\"IS_SPECIAL\":\"1\",\"IS_XFRY\":\"0\",\"REMARKS\":\"beizhu\",\"IS_PARTY\":\"1\",\"icResiUserId\":\"yzmtest2\",\"HOME_ID\":\"中海国际社区一里城1号楼1单元101\",\"HOUSE_TYPE\":\"平房\",\"UNIT_NAME\":\"1单元\",\"GRID_ID\":\"市北区-市北区第三网格3\",\"IS_DB\":\"0\",\"GENDER\":\"男\",\"BIRTHDAY\":\"2021-10-04\",\"IS_VETERANS\":\"0\",\"IS_MB\":\"0\",\"IS_UNEMPLOYED\":\"0\",\"DEMAND_NAME\":null,\"IS_KC\":\"0\",\"IS_ENSURE_HOUSE\":\"0\",\"IS_SD\":\"0\",\"NAME\":\"尹作梅\",\"RDSJ\":null,\"IS_VOLUNTEER\":\"1\",\"GRID_ID_VALUE\":\"e74829ffc43d5470eba6b5e060c11e63\",\"IS_SZ\":\"0\",\"IS_CJ\":\"0\",\"HOME_ID_VALUE\":\"200\",\"DEMAND_CATEGORY_IDS\":null,\"VILLAGE_NAME\":\"中海国际社区一里城\",\"IS_DBH\":\"0\",\"IS_SN\":\"0\",\"BUILD_NAME\":\"1号楼\",\"IS_YLFN\":\"0\",\"IS_UNITED_FRONT\":\"0\",\"ID_CARD\":\"371325199310260529\",\"MOBILE\":\"15764229697\",\"IS_OLD_PEOPLE\":\"0\",\"DOOR_NAME\":\"101\"},{\"IS_SPECIAL\":\"1\",\"IS_XFRY\":\"0\",\"REMARKS\":\"beizhu\",\"IS_PARTY\":\"1\",\"icResiUserId\":\"yzmtest\",\"HOME_ID\":\"中海国际社区一里城1号楼1单元101\",\"HOUSE_TYPE\":\"平房\",\"UNIT_NAME\":\"1单元\",\"GRID_ID\":\"市北区-市北区第三网格3\",\"IS_DB\":\"0\",\"GENDER\":\"男\",\"BIRTHDAY\":\"2021-10-04\",\"IS_VETERANS\":\"0\",\"IS_MB\":\"0\",\"IS_UNEMPLOYED\":\"0\",\"DEMAND_NAME\":\"心理咨询\",\"IS_KC\":\"0\",\"IS_ENSURE_HOUSE\":\"0\",\"IS_SD\":\"0\",\"NAME\":\"尹作梅\",\"RDSJ\":\"2021-10-28 00:00:00\",\"IS_VOLUNTEER\":\"1\",\"GRID_ID_VALUE\":\"e74829ffc43d5470eba6b5e060c11e63\",\"IS_SZ\":\"0\",\"IS_CJ\":\"0\",\"HOME_ID_VALUE\":\"200\",\"DEMAND_CATEGORY_IDS\":\"10180002\",\"VILLAGE_NAME\":\"中海国际社区一里城\",\"IS_DBH\":\"0\",\"IS_SN\":\"0\",\"BUILD_NAME\":\"1号楼\",\"IS_YLFN\":\"0\",\"IS_UNITED_FRONT\":\"0\",\"ID_CARD\":\"371325199310260529\",\"MOBILE\":\"15764229697\",\"IS_OLD_PEOPLE\":\"0\",\"DOOR_NAME\":\"101\"}]");
        log.info("resiMainList:{}", JSON.toJSONString(resiMainList));
        String templatePath = "excel/ic_resi_info_cid.xls";
        TemplateExportParams params = new TemplateExportParams(templatePath, true);

        Map<Integer, Map<String, Object>> sheetMap = new HashMap<>();
        Map<String, Object> mapData = new HashMap<>();
        mapData.put("list", resiMainList.values());
        sheetMap.put(0, mapData);
        AtomicInteger n = new AtomicInteger();
        for (FormItem item : resiFormItems.getItemList()) {

            if (item.getChildGroup() != null) {
                if (!item.getChildGroup().getTableName().equals(BASE_TABLE_NAME)) {
                    Map<String, FormItem> itemMap1 = otherSheetItems.get(item.getChildGroup().getTableName());
                    Map<String, Map<String, Object>> resiChildMap = icResiUserService.getDataForExport(itemMap1, pageFormDTO.getCustomerId(), pageFormDTO.getFormCode(), item.getChildGroup().getTableName(), pageFormDTO.getConditions());
                    resiChildMap.forEach((key, value) -> value.putAll(resiMainList.get(key)));

                    Map<String, Object> mapData2 = new HashMap<>();
                    mapData2.put("list", resiChildMap.values());
                    sheetMap.put(n.incrementAndGet(), mapData2);
                }
            }
        }

        Workbook workbook = ExcelExportUtil.exportExcel(sheetMap, params);
        workbook.setActiveSheet(0);

        String fileName = "居民基本信息.xls";
        response.setHeader("content-Type", "application/vnd.ms-excel");
        response.addHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(fileName, "UTF-8"));
        workbook.write(response.getOutputStream());
    }

    /**
     * excel导入居民基本信息
     *
     * @param loginUser
     * @return
     */
    @PostMapping("import/excel")
    public Result importExcelByEasyExcel() {
        Object result = icResiUserService.importIcResiInfoFromExcel();
        return new Result().ok(result);
    }

    @PostMapping("test")
    public Result<List<Map<String, Object>>> test(@RequestBody DynamicQueryFormDTO formDTO) {
        formDTO.setCustomerId("45687aa479955f9d06204d415238f7cc");
        //formDTO.setCustomerId(tokenDto.getCustomerId());
        return new Result<List<Map<String, Object>>>().ok(icResiUserService.dynamicQuery(formDTO.getCustomerId(),
                formDTO.getFormCode(),
                formDTO.getResultTableName(),
                formDTO.getConditions()));
    }

    /**
     * @param formDTO
     * @Description 查询个人数据
     * @author zxc
     * @date 2021/11/3 9:21 上午
     */
    @PostMapping("persondata")
    public Result<PersonDataResultDTO> personData(@RequestBody PersonDataFormDTO formDTO) {
        return new Result<PersonDataResultDTO>().ok(icResiUserService.personData(formDTO));
    }

    /**
     * @param formDTO
     * @param tokenDto
     * @Description 根据名字搜索
     * @author zxc
     * @date 2021/11/3 1:42 下午
     */
    @PostMapping("searchbyname")
    public Result<List<SearchByNameResultDTO>> searchByName(@RequestBody SearchByNameFormDTO formDTO, @LoginUser TokenDto tokenDto) {
        ValidatorUtils.validateEntity(formDTO, SearchByNameFormDTO.SearchByNameForm.class);
        return new Result<List<SearchByNameResultDTO>>().ok(icResiUserService.searchByName(formDTO, tokenDto));
    }

    @NotNull
    private CustomerFormResultDTO getResiFormItems(String customerId) {
        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()));
        return resultForm.getData();
    }

}