package jnpf.base.model.ocr.model;

import io.swagger.v3.oas.annotations.media.Schema;
import jnpf.base.model.ocr.OcrResModel;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Data
@Schema(description = "ocr解析发票模型（项目详情）")
@NoArgsConstructor
public class InvoiceDetail {
    @Schema(description = "发票明细-货物或应税劳务/服务名称")
    private String itemName;
    @Schema(description = "发票明细-规格型号")
    private String itemSpec;
    @Schema(description = "发票明细-单位")
    private String itemUnit;
    @Schema(description = "发票明细-数量")
    private String itemQuantity;
    @Schema(description = "发票明细-单价")
    private String itemUnitPrice;
    @Schema(description = "发票明细-金额")
    private String itemAmount;
    @Schema(description = "发票明细-税率")
    private String itemTaxRate;
    @Schema(description = "发票明细-税额")
    private String itemTaxAmount;

    public static List<InvoiceDetail> getList(Map<String, List<OcrResModel>> map) {
        List<OcrResModel> nameList = map.get("nameList") != null ? map.get("nameList") : new ArrayList<>();
        List<OcrResModel> typeList = map.get("typeList") != null ? map.get("typeList") : new ArrayList<>();
        List<OcrResModel> unitList = map.get("unitList") != null ? map.get("unitList") : new ArrayList<>();
        List<OcrResModel> numList = map.get("numList") != null ? map.get("numList") : new ArrayList<>();
        List<OcrResModel> priceList = map.get("priceList") != null ? map.get("priceList") : new ArrayList<>();
        List<OcrResModel> amountList = map.get("amountList") != null ? map.get("amountList") : new ArrayList<>();
        List<OcrResModel> rateList = map.get("rateList") != null ? map.get("rateList") : new ArrayList<>();
        List<OcrResModel> taxAmountList = map.get("taxAmountList") != null ? map.get("taxAmountList") : new ArrayList<>();
        List<InvoiceDetail> list = new ArrayList<>();
        if (priceList.size() == 1) {
            InvoiceDetail invoiceDetail = new InvoiceDetail();
            invoiceDetail.setItemName(String.join("", nameList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemSpec(String.join("", typeList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemUnit(String.join("", unitList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemQuantity(String.join("", numList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemUnitPrice(String.join("", priceList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemAmount(String.join("", amountList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemTaxRate(String.join("", rateList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            invoiceDetail.setItemTaxAmount(String.join("", taxAmountList.stream().map(OcrResModel::getText).collect(Collectors.toList())));
            list.add(invoiceDetail);
        } else if (priceList.size() > 1) {
            for (int i = 0; i < priceList.size(); i++) {
                OcrResModel ocrResModel = priceList.get(i);
                Integer y1 = ocrResModel.getY1() - 8;
                Integer y2;
                InvoiceDetail invoiceDetail = new InvoiceDetail();
                if (i + 1 < priceList.size()) {
                    OcrResModel nextModel = priceList.get(i + 1);
                    y2 = nextModel.getY1() - 8;
                } else {
                    y2 = 0;
                }
                invoiceDetail.setItemName(String.join("", nameList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemSpec(String.join("", typeList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemUnit(String.join("", unitList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemQuantity(String.join("", numList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemUnitPrice(String.join("", priceList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemAmount(String.join("", amountList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemTaxRate(String.join("", rateList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                invoiceDetail.setItemTaxAmount(String.join("", taxAmountList.stream()
                        .filter(t -> y2 == 0 ? t.getY1() > y1 : (t.getY1() > y1 && t.getY1() < y2)).map(OcrResModel::getText).collect(Collectors.toList())));
                list.add(invoiceDetail);
            }
        }
        return list;
    }
}