InputInvoiceImportController.java

package pwc.taxtech.atms.controller;

import com.github.pagehelper.PageInfo;
import com.google.common.collect.Lists;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import pwc.taxtech.atms.constant.enums.KeyValueConfigResultType;
import pwc.taxtech.atms.dto.vatdto.*;
import pwc.taxtech.atms.service.impl.IdentityServiceImpl;
import pwc.taxtech.atms.thirdparty.ExcelSheet;
import pwc.taxtech.atms.thirdparty.ExcelUtil;
import pwc.taxtech.atms.vat.dpo.InputInvoicePreviewQueryParam;
import pwc.taxtech.atms.vat.entity.InputInvoice;
import pwc.taxtech.atms.vat.entity.InputInvoiceDetail;
import pwc.taxtech.atms.vat.service.impl.InputInvoiceDataImportServiceImpl;
import pwc.taxtech.atms.vat.service.impl.report.functions.FormulaHelper;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.util.*;

import static javax.servlet.http.HttpServletResponse.SC_NO_CONTENT;
import static javax.servlet.http.HttpServletResponse.SC_OK;

@RestController
@RequestMapping(value = "api/v1/inputInvoiceImport")
public class InputInvoiceImportController {
    @Autowired
    InputInvoiceDataImportServiceImpl inputInvoiceDataImportService;
    @Autowired
    private IdentityServiceImpl identityService;
    private Logger logger = LoggerFactory.getLogger(InputInvoiceImportController.class);

    @RequestMapping(value = "inputInvoicePreviewAllList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public @ResponseBody
    List<InputInvoice> getInputInvoiceTree(@RequestBody InputInvoicePreviewQueryParam paras,
                                           @RequestHeader("from") String projectId) {
        double hjje = 0;
        double hjse = 0;
        for(InputInvoice inputInvoice : inputInvoiceDataImportService.getInputInvoice(paras, projectId)){
            hjje += FormulaHelper.roundValue(new BigDecimal(inputInvoice.getHJJE()), KeyValueConfigResultType.Accounting, true, null).doubleValue();
            hjse += FormulaHelper.roundValue(new BigDecimal(inputInvoice.getHJSE()), KeyValueConfigResultType.Accounting, true, null).doubleValue();
        }
        InputInvoice i = new InputInvoice();
        i.setHJJE(hjje + "");
        i.setHJSE(hjse + "");
        List<InputInvoice> list = Lists.newArrayList();
        list.add(i);
        return list;
    }

    @RequestMapping(value = "inputInvoicePreviewList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public @ResponseBody
    PageInfo<InputInvoice> getInputInvoiceTreeViewData(@RequestBody InputInvoicePreviewQueryParam paras,
                                                       @RequestHeader("from") String projectId) {
        return inputInvoiceDataImportService.getInputInvoiceTreeViewData(paras, projectId);
    }

    @RequestMapping(value = "exportQueryData/get", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public void downloadInvoiceQueryData(@RequestBody InputInvoicePreviewQueryParam paras, HttpServletResponse response,
                                         @RequestHeader("from") String projectId) {
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.addHeader("Access-Control-Expose-Headers", "Content-Type,Content-Disposition,x-file-name");
        response.setCharacterEncoding("UTF-8");
        String fileName = paras.getPeriodStart() + "-" + paras.getPeriodEnd();
        OutputStream os = null;
        try {
            response.addHeader("Content-Disposition", "attachment;filename="
                    + UUID.randomUUID() + ".xls");
            response.addHeader("x-file-name", fileName);
            os = response.getOutputStream();
            int count = getDownloadFilePath(paras, os, projectId);

            if (count == 0) {
                response.setStatus(SC_NO_CONTENT);
            } else {
                response.setStatus(SC_OK);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    @RequestMapping(value = "getInputInvoiceList/{period}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity getInputInvoiceList(@PathVariable Integer period) {
        return ResponseEntity.ok().body(inputInvoiceDataImportService.getInputInvoiceList(period));
    }

    @RequestMapping(value = "getInputInvoiceItemList/{fpid}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity getInputInvoiceItemList(@PathVariable String fpid) {
        return ResponseEntity.ok().body(inputInvoiceDataImportService.getInputInvoiceItemList(fpid));
    }

    private String getFplx(String fplx){
        if(StringUtils.isBlank(fplx)) return "";
        switch (fplx){
            case "004":
                return "增值税专票";
            case "007":
                return "普票";
            case "026":
                return "电子发票";
            default:
                return fplx;
        }
    }
    private int getDownloadFilePath(InputInvoicePreviewQueryParam paras, OutputStream outputStream, String projectId) {
        List<InputInvoice> list = inputInvoiceDataImportService.getInputInvoiceTree(paras, projectId);
        if (list.size() == 0) {
            return 0;
        }
        List<InputInvoiceExportDto> inputInvoiceExportDtos = Lists.newArrayList();
        for(InputInvoice inputInvoice : list){
            InputInvoiceExportDto inputInvoiceExportDto = new InputInvoiceExportDto();
            inputInvoiceExportDto.setAmount(inputInvoice.getHJJE() != null ? new BigDecimal(inputInvoice.getHJJE().replace(",","")): BigDecimal.ZERO);
            inputInvoiceExportDto.setCertificationDate(inputInvoice.getRZSJ());
            inputInvoiceExportDto.setInvoiceCode(inputInvoice.getFPDM());
            inputInvoiceExportDto.setInvoiceDate(inputInvoice.getKPRQ());
            inputInvoiceExportDto.setInvoiceNumber(inputInvoice.getFPHM());
            inputInvoiceExportDto.setInvoiceTypeName(getFplx(inputInvoice.getFPLX()));
            inputInvoiceExportDto.setPeriodId(Integer.parseInt(inputInvoice.getRZSQ().substring(4, 6)));
            inputInvoiceExportDto.setSellerTaxNumber(inputInvoice.getXFSH());
            inputInvoiceExportDto.setTaxAmount(inputInvoice.getHJSE() != null ? new BigDecimal(inputInvoice.getHJSE().replace(",","")) : BigDecimal.ZERO);
            inputInvoiceExportDtos.add(inputInvoiceExportDto);
        }
        Map<String, String> header = new LinkedHashMap<>();
        header.put("PeriodId", "期间");
        header.put("InvoiceDate", "开票日期");
        header.put("InvoiceCode", "发票代码");
        header.put("InvoiceNumber", "发票号码");
        header.put("SellerTaxNumber", "供货方税号");
        header.put("InvoiceTypeName", "发票类型");
        header.put("Amount", "金额");
        header.put("TaxAmount", "税额");
        header.put("CertificationDate", "认证日期");

        InputInvoiceExportDto inputInvoiceExportDto = new InputInvoiceExportDto();
        inputInvoiceExportDto.setTaxAmount(inputInvoiceExportDtos.stream().map(InputInvoiceExportDto::getTaxAmount).reduce(BigDecimal::add).get());
        inputInvoiceExportDto.setAmount(inputInvoiceExportDtos.stream().map(InputInvoiceExportDto::getAmount).reduce(BigDecimal::add).get());
        InputInvoiceExportDto black = new InputInvoiceExportDto();
        inputInvoiceExportDtos.add(black);
        inputInvoiceExportDtos.add(inputInvoiceExportDto);
        ExcelSheet excelSheetA = new ExcelSheet<>();
        excelSheetA.setHeaders(header);
        excelSheetA.setDataset(inputInvoiceExportDtos);
        excelSheetA.setSheetName("sheetA");


        Map<String, String> header2 = new LinkedHashMap<>();
        header2.put("PeriodId", "期间");
        header2.put("InvoiceCode", "发票代码");
        header2.put("InvoiceNumber", "发票号码");
        header2.put("Amount", "金额");
        header2.put("TaxAmount", "税额");
        header2.put("TaxRate", "税率");
        header2.put("ProductionName", "货物名称");

        ExcelSheet excelSheetB = new ExcelSheet<>();
        excelSheetB.setHeaders(header2);
        excelSheetB.setSheetName("sheetB");

        List<InputInvoiceDetailExportDto> inputInvoiceItemList = new ArrayList<>();
        for (InputInvoice item : list) {
            List<InputInvoiceDetail> inputInvoiceDetailList = inputInvoiceDataImportService.getInputInvoiceItemList(item.getID());
            for(InputInvoiceDetail inputInvoiceDetail : inputInvoiceDetailList){
                InputInvoiceDetailExportDto inputInvoiceDetailExportDto = new InputInvoiceDetailExportDto();
                inputInvoiceDetailExportDto.setAmount(inputInvoiceDetail.getJE() != null ? new BigDecimal(inputInvoiceDetail.getJE().replace(",","")) : BigDecimal.ZERO);
                inputInvoiceDetailExportDto.setInvoiceCode(inputInvoiceDetail.getFPDM());
                inputInvoiceDetailExportDto.setInvoiceNumber(inputInvoiceDetail.getFPHM());
                inputInvoiceDetailExportDto.setPeriodId(Integer.parseInt(item.getRZSQ().substring(4, 6)));
                inputInvoiceDetailExportDto.setProductionName(inputInvoiceDetail.getSPMC());
                inputInvoiceDetailExportDto.setTaxAmount(inputInvoiceDetail.getSE() != null ? new BigDecimal(inputInvoiceDetail.getSE().replace(",","")) : BigDecimal.ZERO);
                inputInvoiceDetailExportDto.setTaxRate(inputInvoiceDetail.getSLV());
                inputInvoiceItemList.add(inputInvoiceDetailExportDto);
            }
        }

        InputInvoiceDetailExportDto inputInvoiceDetailExportDto = new InputInvoiceDetailExportDto();

        inputInvoiceDetailExportDto.setTaxAmount(inputInvoiceItemList.stream().map(InputInvoiceDetailExportDto::getTaxAmount).reduce(BigDecimal::add).get());
        inputInvoiceDetailExportDto.setAmount(inputInvoiceItemList.stream().map(InputInvoiceDetailExportDto::getAmount).reduce(BigDecimal::add).get());
        InputInvoiceDetailExportDto detailBlack = new InputInvoiceDetailExportDto();
        inputInvoiceItemList.add(detailBlack);
        inputInvoiceItemList.add(inputInvoiceDetailExportDto);


        excelSheetB.setDataset(inputInvoiceItemList);
        List<ExcelSheet<InputVATInvoiceBaseDto>> sheets = new ArrayList<>();
        sheets.add(excelSheetA);
        sheets.add(excelSheetB);

        ExcelUtil.exportExcel(sheets, outputStream);
        return list.size();
    }

    @RequestMapping(value = "importInputInvoice", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity ImportInputInvoiceData(@RequestBody ImportInputInvoiceDto importParam) {
        return ResponseEntity.ok(inputInvoiceDataImportService.ImportInputInvoiceData(importParam.InputInvoiceList,
                importParam.ImportType, identityService.getIdentityUser().getId()));
    }

    @RequestMapping(value = "importInputInvoiceItem", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity ImportInputInvoiceItemData(@RequestBody ImportInputInvoiceItemDto importParam) {
        return ResponseEntity.ok(inputInvoiceDataImportService.ImportInputInvoiceItemData(importParam.InputInvoiceItemList,
                importParam.ImportType, identityService.getIdentityUser().getId()));
    }

    @RequestMapping(value = "inputInvoiceItemPreviewList/{InputInvoiceId}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity GetInputInvoiceItemTreeViewData(@PathVariable String InputInvoiceId) {
        return ResponseEntity.ok(inputInvoiceDataImportService.getInputInvoiceItemList(InputInvoiceId));
    }
}