OutputInvoiceController.java

package pwc.taxtech.atms.controller;

import com.github.pagehelper.PageInfo;
import com.github.pagehelper.util.StringUtil;
import com.google.common.collect.Lists;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import pwc.taxtech.atms.common.util.BeanUtil;
import pwc.taxtech.atms.constant.enums.EnumModule;
import pwc.taxtech.atms.constant.enums.KeyValueConfigResultType;
import pwc.taxtech.atms.dpo.PagingResultDto;
import pwc.taxtech.atms.dto.vatdto.ImportOutputInvoiceDto;
import pwc.taxtech.atms.dto.vatdto.QueryEvidenceDto;
import pwc.taxtech.atms.service.impl.IdentityServiceImpl;
import pwc.taxtech.atms.thirdparty.ExcelUtil;
import pwc.taxtech.atms.vat.dpo.OutputVATInvoiceInfoDto;
import pwc.taxtech.atms.vat.dpo.QueryOutputDto;
import pwc.taxtech.atms.vat.entity.ImportFile;
import pwc.taxtech.atms.vat.entity.OutputInvoice;
import pwc.taxtech.atms.vat.service.impl.FileUploadAdapter;
import pwc.taxtech.atms.vat.service.impl.OutputInvoiceServiceImpl;
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;
import static pwc.taxtech.atms.constant.Constant.FILE_NAME;
import static pwc.taxtech.atms.constant.Constant.TEMP_FILE_NAME;

@RestController
@RequestMapping(value = "api/v1/outputInvoiceImport")
public class OutputInvoiceController {

    @Autowired
    private BeanUtil beanUtil;
    @Autowired
    OutputInvoiceServiceImpl outputInvoiceService;
    @Autowired
    private IdentityServiceImpl identityService;
    @Autowired
    private FileUploadAdapter fileUploadAdapter;

    @RequestMapping(value = "queryOutputInvoiceList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public PageInfo<OutputInvoice> queryOutputInvoiceList(@RequestBody QueryOutputDto queryDto,
                                                          @RequestHeader("from") String projectId) {
            return outputInvoiceService.queryOutputInvoiceList(queryDto,projectId);
    }

    @RequestMapping(value = "queryOutputInvoiceAllList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public List<OutputInvoice> queryOutputInvoiceAllList(@RequestBody QueryOutputDto queryDto,
                                                          @RequestHeader("from") String projectId) {
        List<OutputInvoice> list = outputInvoiceService.getOutputInvoice(queryDto, projectId);
        double hjje = 0;
        double hjse = 0;
        for(OutputInvoice outputInvoice : list){
            hjje += FormulaHelper.roundValue(new BigDecimal(outputInvoice.getHJJE()), KeyValueConfigResultType.Accounting, true, null).doubleValue();
            hjse += FormulaHelper.roundValue(new BigDecimal(outputInvoice.getHJSE()), KeyValueConfigResultType.Accounting, true, null).doubleValue();
        }
        list = Lists.newArrayList();
        OutputInvoice o = new OutputInvoice();
        o.setHJJE(hjje + "");
        o.setHJSE(hjse + "");
        list.add(o);
        return list;
    }

    @RequestMapping(value = "getExportOutputInvoiceList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public void downloadInvoiceQueryData(@RequestBody QueryOutputDto paras, @RequestHeader("from") String projectId, HttpServletResponse response) {
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.addHeader("Access-Control-Expose-Headers", "Content-Type,Content-Disposition,x-file-name");
        String fileName = paras.getPeriodStart() + "-" + paras.getPeriodEnd();
        response.setCharacterEncoding("UTF-8");
        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, projectId, os);

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

    @RequestMapping(value = "getEvidenceList", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity getEvidenceList(@RequestBody QueryEvidenceDto queryDto) {
        return ResponseEntity.ok().body(outputInvoiceService.getEvidenceList(queryDto));
    }

    @RequestMapping(value = "getOutputInvoiceList/{period}", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity getEvidenceList(@PathVariable Integer period) {
        return ResponseEntity.ok().body(outputInvoiceService.getOutputInvoiceList(period));
    }

    private int getDownloadFilePath(QueryOutputDto paras, String projectId, OutputStream outputStream) {
        List<OutputVATInvoiceInfoDto> list = outputInvoiceService.getExportOutputInvoiceList(paras, projectId).getData();
        if (list.size() == 0) {
            return 0;
        }
        pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto outputVATInvoiceInfoDto = new pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto();

        outputVATInvoiceInfoDto.setTaxAmount(list.stream().map(OutputVATInvoiceInfoDto::getTaxAmount).reduce(BigDecimal::add).get());
        outputVATInvoiceInfoDto.setAmount(list.stream().map(OutputVATInvoiceInfoDto::getAmount).reduce(BigDecimal::add).get());
        pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto black = new pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto();
//        list.add(black);
//        list.add(outputVATInvoiceInfoDto);
        List<pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto> list2 = new ArrayList<>();
        list.stream().forEach(x -> {
                pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto d =
                new pwc.taxtech.atms.dto.vatdto.OutputVATInvoiceInfoDto();
                d = beanUtil.copyProperties(x,d);
                d.setInvoiceTypeName(d.getInvoiceTypeName());
                d.setTaxRateS(x.getTaxRate().doubleValue() * 100  + "%");
                list2.add(d);
        });
        list2.add(black);
        list2.add(outputVATInvoiceInfoDto);

        Map<String, String> header = new LinkedHashMap<>();
        header.put("InvoiceTypeName", "发票类型");
        header.put("ClassCode", "发票代码");
        header.put("InvoiceNumber", "发票号码");
        header.put("BuyerName", "购方企业名称");
        header.put("BuyerTaxNumber", "购方税号");
//        header.put("BankAccount", "银行账号");
//        header.put("PhoneNum", "地址电话");
        header.put("InvoiceDate", "开票日期");
//        header.put("CodeVersion", "商品编码版本号");
//        header.put("DocumentNum", "单据号");
        header.put("ProductName", "商品名称");
//        header.put("ProductStandar", "规格");
//        header.put("Unit", "单位");
        header.put("Quantity", "数量");
        header.put("UnitPrice", "单价");
        header.put("Amount", "金额");
        header.put("TaxRateS", "税率");
        header.put("TaxAmount", "税额");
//        header.put("TaxClassCode", "税收分类编码");

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

    @RequestMapping(value = "importOutputInvoice", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity ImportInputInvoiceData(@RequestBody ImportOutputInvoiceDto importDto) {
        return ResponseEntity.ok(outputInvoiceService.ImportOutputInvoiceData(importDto, identityService.getIdentityUser().getId()));
    }

    @RequestMapping(value = "UploadEvidenceFile", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity UploadEvidenceFile(String period, MultipartHttpServletRequest request) {
        String userId = identityService.getIdentityUser().getId();
        String fullPath = fileUploadAdapter.uploadEvidence(request, EnumModule.Evidence, userId);
        if (StringUtil.isNotEmpty(fullPath)) {
            String fileName = request.getParameter(FILE_NAME);
            String tempFileName = request.getParameter(TEMP_FILE_NAME);

            ImportFile fileDto = new ImportFile();
            fileDto.setFileId(tempFileName.substring(1, tempFileName.length() - 5));
            fileDto.setFilePath(fullPath);
            fileDto.setFileName(fileName);
            fileDto.setPeriodId(Integer.parseInt(period));
            fileDto.setFileType(GetExtension(fileName));
            fileDto.setFileImportType(EnumModule.Evidence.getCode());
            fileDto.setCreatorId(userId);
            fileDto.setCreateTime(new Date());
            return ResponseEntity.ok(outputInvoiceService.addImportFile(fileDto));
        }
        return ResponseEntity.ok(false);
    }

    @RequestMapping(value = "getOutputInvoiceList/{period}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity GetOutputInvoiceList(Integer period) {
        return ResponseEntity.ok(outputInvoiceService.getOutputInvoiceList(period));
    }

    @RequestMapping(value = "queryOutputInvoiceItemList/{invoiceId}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public ResponseEntity QueryOutputInvoiceItemList(@PathVariable String invoiceId,@RequestParam(required = false) String tag) {
        return ResponseEntity.ok(outputInvoiceService.QueryOutputInvoiceItemList(invoiceId,tag));
    }

    private String GetExtension(String file) {
        return file.substring(file.lastIndexOf(".") + 1, file.length());
    }

}