CellConfigTranslater.java

package pwc.taxtech.atms.service.impl;

import org.apache.commons.lang3.StringUtils;
import pwc.taxtech.atms.constant.Constant;
import pwc.taxtech.atms.constant.enums.CellDataSourceType;
import pwc.taxtech.atms.dto.CellTemplateConfigDto;
import pwc.taxtech.atms.dto.vatdto.CellCalcInfoDto;
import pwc.taxtech.atms.entity.CellTemplate;
import pwc.taxtech.atms.entity.CellTemplateConfig;
import pwc.taxtech.atms.vat.entity.PeriodCellTemplateConfig;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;

public final class CellConfigTranslater {

    public static CellTemplateConfigDto getConfigDto(CellTemplate template, List<CellTemplateConfig> configList) {
        if (template == null) {
            return null;
        }

        return getConfigDto(template.getId(), template.getReportTemplateId(), template.getRowIndex(), template.getRowName(), template.getColumnIndex(),
                template.getColumnName(), template.getDataType(), template.getIsReadOnly(), template.getComment(), configList);
    }

    public static CellTemplateConfigDto getConfigDto(CellCalcInfoDto x) {
        return CellConfigTranslater.getPeriodConfigDto(Long.parseLong(x.getCellTemplateId()), x.getReportTemplateId(), x.getRowIndex(),
                x.getRowName(), x.getColumnIndex(), x.getColumnName(),
                x.getDataType(), x.getIsReadOnly(),
                x.getDescription(), x.getConfigList());
    }

    private static CellTemplateConfigDto getConfigDto(Long configId, Long templateId, int rowIndex, String rowName, int columnIndex,
                                                      String columnName, Integer dataType, Boolean isReadOnly, String description,
                                                      List<CellTemplateConfig> configList) {
        if (configList == null) {
            return null;
        }

        CellTemplateConfigDto cellTemplateConfigDto = new CellTemplateConfigDto();
        cellTemplateConfigDto.setCellTemplateId(configId.toString());
        cellTemplateConfigDto.setTemplateId(templateId.toString());
        cellTemplateConfigDto.setRowIndex(rowIndex);
        cellTemplateConfigDto.setRowName(rowName);
        cellTemplateConfigDto.setColumnIndex(columnIndex);
        cellTemplateConfigDto.setColumnName(columnName);
        cellTemplateConfigDto.setDataType(dataType);
        cellTemplateConfigDto.setIsReadOnly(isReadOnly);
        cellTemplateConfigDto.setFormulaDescription(description);

        Optional<CellTemplateConfig> formulaItem = configList.stream().filter(a -> a.getDataSourceType().equals(CellDataSourceType.Formula.getCode())).findFirst();
        if (formulaItem.isPresent()) {
            cellTemplateConfigDto.setHasFormula(true);
            cellTemplateConfigDto.setFormula(formulaItem.get().getFormula());
            if (StringUtils.isEmpty(cellTemplateConfigDto.getFormulaDescription())
                    && !StringUtils.isEmpty(formulaItem.get().getFormulaDescription())) {
                cellTemplateConfigDto.setFormulaDescription(formulaItem.get().getFormulaDescription());
            }
            cellTemplateConfigDto.setFormulaDataSource(formulaItem.get().getFormulaDataSource());
            cellTemplateConfigDto.setCellTemplateId(formulaItem.get().getCellTemplateId().toString());
        }

        Optional<CellTemplateConfig> voucherItem = configList.stream().filter(x -> x.getDataSourceType().equals(CellDataSourceType.Voucher.getCode())).findFirst();
        if (voucherItem.isPresent()) {
            cellTemplateConfigDto.setHasVoucher(true);
            cellTemplateConfigDto.setVoucherKeyword(voucherItem.get().getVoucherKeyword() == null ? "" : voucherItem.get().getVoucherKeyword());
            if (!StringUtils.isEmpty(voucherItem.get().getAccountCodes())) {
                cellTemplateConfigDto.setAccountCodes(getList(voucherItem.get().getAccountCodes()));
            }
        }

        Optional<CellTemplateConfig> invoiceItem = configList
                .stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.OutputInvoice.getCode())
                        || x.getDataSourceType().equals(CellDataSourceType.InputInvoice.getCode())
                        || x.getDataSourceType().equals(CellDataSourceType.CustomInvoice.getCode()))
                .findFirst();
        if (invoiceItem.isPresent()) {
            cellTemplateConfigDto.setHasInvoice(true);
            cellTemplateConfigDto.setInvoiceType(invoiceItem.get().getInvoiceType());
            cellTemplateConfigDto.setInvoiceAmountType(invoiceItem.get().getInvoiceAmountType());
            if (!StringUtils.isEmpty(invoiceItem.get().getTaxRate())) {
                cellTemplateConfigDto.setTaxRate(getList(invoiceItem.get().getTaxRate()));
            }

            if (!StringUtils.isEmpty(invoiceItem.get().getInvoiceCategory())) {
                List<String> invoiceCategoryStrs;
                invoiceCategoryStrs = getList(invoiceItem.get().getInvoiceCategory());
                List<Integer> ints = new ArrayList<>();
                assert invoiceCategoryStrs != null;
                for (String categoryStr : invoiceCategoryStrs) {
                    int categoryVal;
                    categoryVal = Integer.parseInt(categoryStr);
                    ints.add(categoryVal);
                }
                cellTemplateConfigDto.setInvoiceCategory(ints);
            }
        }

        Optional<CellTemplateConfig> keyInItem = configList.stream().filter(x -> x.getDataSourceType().equals(CellDataSourceType.KeyIn.getCode())).findFirst();
        if (keyInItem.isPresent()) {
            cellTemplateConfigDto.setHasKeyIn(true);
        }

        Optional<CellTemplateConfig> modelItem = configList.stream().filter(x -> x.getDataSourceType().equals(CellDataSourceType.RelatedModel.getCode())).findFirst();
        if (modelItem.isPresent()) {
            cellTemplateConfigDto.setHasModel(true);
            cellTemplateConfigDto.setModelIds(getList(modelItem.get().getModelIds()));
        }

        Optional<CellTemplateConfig> validationItem = configList.stream().filter(x -> x.getDataSourceType().equals(CellDataSourceType.Validation.getCode())).findFirst();
        if (validationItem.isPresent()) {
            cellTemplateConfigDto.setHasValidation(true);
            cellTemplateConfigDto.setValidation(validationItem.get().getValidation());
            cellTemplateConfigDto.setValidationDescription(validationItem.get().getValidationDescription());
            cellTemplateConfigDto.setCellTemplateId(validationItem.get().getCellTemplateId().toString());
        }
        return cellTemplateConfigDto;
    }


    private static CellTemplateConfigDto getPeriodConfigDto(Long cellTemplateId, Long templateId, int rowIndex, String rowName, int columnIndex,
                                                            String columnName, Integer dataType, Boolean isReadOnly,
                                                            String description, List<PeriodCellTemplateConfig> configList) {
        //already finished by today 20180711
        if (configList == null) {
            return null;
        }

        CellTemplateConfigDto cellTemplateConfigDto = new CellTemplateConfigDto();
        cellTemplateConfigDto.setCellTemplateId(cellTemplateId.toString());
        cellTemplateConfigDto.setTemplateId(templateId.toString());
        cellTemplateConfigDto.setRowIndex(rowIndex);
        cellTemplateConfigDto.setRowName(rowName);
        cellTemplateConfigDto.setColumnIndex(columnIndex);
        cellTemplateConfigDto.setColumnName(columnName);
        cellTemplateConfigDto.setDataType(dataType);
        cellTemplateConfigDto.setIsReadOnly(isReadOnly);
        cellTemplateConfigDto.setFormulaDescription(description);

        Optional<PeriodCellTemplateConfig> formulaItem = configList.stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.Formula.getCode()))
                .findFirst();

        if (formulaItem.isPresent()) {
            cellTemplateConfigDto.setHasFormula(true);
            cellTemplateConfigDto.setFormula(formulaItem.get().getFormula());
            cellTemplateConfigDto.setParsedFormula(formulaItem.get().getParsedFormula());
            if (StringUtils.isBlank(cellTemplateConfigDto.getFormulaDescription())
                    && StringUtils.isNotBlank(formulaItem.get().getFormulaDescription())) {
                cellTemplateConfigDto.setFormulaDescription(formulaItem.get().getFormulaDescription());
            }
            cellTemplateConfigDto.setFormulaDataSource(formulaItem.get().getFormulaDataSource());
            cellTemplateConfigDto.setCellTemplateId(formulaItem.get().getCellTemplateId().toString());
        }

        Optional<PeriodCellTemplateConfig> voucherItem = configList.stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.Voucher.getCode())).findFirst();
        if (voucherItem.isPresent()) {
            cellTemplateConfigDto.setHasVoucher(true);
            cellTemplateConfigDto.setVoucherKeyword(voucherItem.get().getVoucherKeyword() == null
                    ? StringUtils.EMPTY : voucherItem.get().getVoucherKeyword());
            if (StringUtils.isNotBlank(voucherItem.get().getAccountCodes())) {
                cellTemplateConfigDto.setAccountCodes(getList(voucherItem.get().getAccountCodes()));
            }
        }

        Optional<PeriodCellTemplateConfig> invoiceItem = configList.stream()
                .filter(x -> (x.getDataSourceType().equals(CellDataSourceType.OutputInvoice.getCode()))
                        || (x.getDataSourceType().equals(CellDataSourceType.InputInvoice.getCode())
                        || (x.getDataSourceType().equals(CellDataSourceType.CustomInvoice.getCode())))).findFirst();
        if (invoiceItem.isPresent()) {
            cellTemplateConfigDto.setHasInvoice(true);
            cellTemplateConfigDto.setInvoiceType(invoiceItem.get().getInvoiceType());
            cellTemplateConfigDto.setInvoiceAmountType(invoiceItem.get().getInvoiceAmountType());
            if (StringUtils.isNotBlank(invoiceItem.get().getTaxRate())) {
                cellTemplateConfigDto.setTaxRate(getList(invoiceItem.get().getTaxRate()));
            }

            if (StringUtils.isNotBlank(invoiceItem.get().getInvoiceCategory())) {
                List<String> invoiceCategoryStrs = getList(invoiceItem.get().getInvoiceCategory());
                List<Integer> invoiceCategoryList = new ArrayList<>();
                invoiceCategoryStrs.forEach(s -> {
                    Integer categoryVal;
                    try {
                        categoryVal = Integer.parseInt(s);
                        invoiceCategoryList.add(categoryVal);
                    } catch (Exception e) {
                        //转换错误就不做添加，直接吃掉这里的异常，不进行阻塞
                    }
                });
                cellTemplateConfigDto.setInvoiceCategory(invoiceCategoryList);
            }
        }

        Optional<PeriodCellTemplateConfig> keyInItem = configList.stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.KeyIn.getCode()))
                .findFirst();

        if (keyInItem.isPresent()) {
            cellTemplateConfigDto.setHasKeyIn(true);
        }

        Optional<PeriodCellTemplateConfig> modelItem = configList.stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.RelatedModel.getCode()))
                .findFirst();

        if (modelItem.isPresent()) {
            cellTemplateConfigDto.setHasModel(true);
            cellTemplateConfigDto.setModelIds(getList(modelItem.get().getModelIds()));
        }

        Optional<PeriodCellTemplateConfig> validationItem = configList.stream()
                .filter(x -> x.getDataSourceType().equals(CellDataSourceType.Validation.getCode()))
                .findFirst();

        if (validationItem.isPresent()) {
            cellTemplateConfigDto.setHasValidation(true);
            cellTemplateConfigDto.setValidation(validationItem.get().getValidation());
            cellTemplateConfigDto.setParsedFormula(validationItem.get().getParsedFormula());
            cellTemplateConfigDto.setValidationDescription(validationItem.get().getValidationDescription());
            cellTemplateConfigDto.setCellTemplateId(validationItem.get().getCellTemplateId().toString());
        }

        return cellTemplateConfigDto;
    }


    private static List<String> getList(String joinString) {
        if (StringUtils.isEmpty(joinString)) {
            return null;
        }

        return new ArrayList<>(Arrays.asList(joinString.split(Constant.Comma)));
    }
}