package jnpf.base.model.ocr.model;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.fasterxml.jackson.annotation.JsonIgnore;
import io.swagger.v3.oas.annotations.media.Schema;
import jnpf.base.model.ocr.OcrModel;
import jnpf.base.model.ocr.OcrResModel;
import jnpf.util.DateUtil;
import jnpf.util.StringUtil;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;

import java.util.*;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * ocr解析发票模型
 *
 * @author JNPF开发平台组
 * @version v6.1.0
 * @copyright 引迈信息技术有限公司
 * @date 2025/7/29 16:58:47
 */
@EqualsAndHashCode(callSuper = true)
@Data
@Schema(description = "ocr解析发票模型")
@NoArgsConstructor
public class InvoiceLicModel extends OcrModel {
    @Schema(description = "发票名称")
    private String title;
    @Schema(description = "发票代码")
    private String code;
    @Schema(description = "发票号码")
    private String number;
    @Schema(description = "开票日期")
    private Long date;
    @Schema(description = "校验码")
    private String checkCode;
    @Schema(description = "密码区")
    private String password;
    @Schema(description = "购买方名称")
    private String buyerName;
    @Schema(description = "购买方纳税人识别号")
    private String buyerTaxId;
    @Schema(description = "购买方地址/电话")
    private String buyerContact;
    @Schema(description = "购买方开户行及账号")
    private String buyerBankAccount;
    @Schema(description = "合计金额")
    private String totalAmount;
    @Schema(description = "合计税额")
    private String totalTaxAmount;
    @Schema(description = "价税合计（大写）")
    private String totalAmountUpper;
    @Schema(description = "价税合计（小写）")
    private String totalAmountLower;
    @Schema(description = "销售方名称")
    private String sellerName;
    @Schema(description = "销售方纳税人识别号")
    private String sellerTaxId;
    @Schema(description = "销售方地址/电话")
    private String sellerContact;
    @Schema(description = "销售方开户行及账号")
    private String sellerBankAccount;
    @Schema(description = "备注")
    private String remarks;
    @Schema(description = "收款人")
    private String payee;
    @Schema(description = "复核人")
    private String reviewer;
    @Schema(description = "开票人")
    private String issuer;

    @Schema(description = "项目详情")
    private List<InvoiceDetail> invoiceDetail = new ArrayList<>();

    @Schema(description = "最大宽度")
    @JsonIgnore
    private Integer widthMax = 0;
    @Schema(description = "最大高度")
    @JsonIgnore
    private Integer heightMax = 0;

    @Override
    public void extract(String url, JSONObject body) {
        JSONObject systemJson = this.getSystemJson(url, body);
        List<OcrResModel> param = getParam(systemJson);
        if (StringUtil.isEmpty(this.getContent()) || !this.getContent().contains("发票")) {
            return;
        }
        //两种发票--带和不带密码区
        if (this.getStrList().contains("密") || this.getStrList().contains("码") || this.getStrList().contains("区") || this.getContent().contains("校验码")) {
            this.setInfoOne(param);
        } else {
            this.setInfoTwo(param);
        }
        this.setContent(null);
        this.setStrList(null);
    }

    //数据转换
    private List<OcrResModel> getParam(JSONObject obj) {
        //识别
        List<String> list = new ArrayList<>();
        List<OcrResModel> listJosn = new ArrayList<>();
        if (obj != null && obj.containsKey("status") && Objects.equals("000", String.valueOf(obj.get("status")))) {
            JSONArray results = obj.getJSONArray("results");
            for (Object result : results) {
                JSONArray m = (JSONArray) result;
                for (Object n : m) {
                    JSONObject o = (JSONObject) n;
                    list.add(o.getString("text"));

                    if (StringUtil.isEmpty(o.getString("text"))) continue;
                    OcrResModel resModel = new OcrResModel();
                    resModel.setText(o.getString("text").replace(" ", ""));
                    resModel.setConfidence(o.getBigDecimal("confidence"));
                    JSONArray textRegion = o.getJSONArray("text_region");
                    if (textRegion.size() == 4) {
                        JSONArray o1 = (JSONArray) textRegion.get(0);
                        JSONArray o2 = (JSONArray) textRegion.get(2);
                        resModel.setX1(o1.getInteger(0));
                        resModel.setY1(o1.getInteger(1));
                        resModel.setX2(o2.getInteger(0));
                        resModel.setY2(o2.getInteger(1));
                        if (resModel.getX1() > this.widthMax) {
                            this.widthMax = resModel.getX1();
                        }
                        if (resModel.getX2() > this.widthMax) {
                            this.widthMax = resModel.getX2();
                        }
                        if (resModel.getY1() > this.heightMax) {
                            this.heightMax = resModel.getX1();
                        }
                        if (resModel.getY2() > this.heightMax) {
                            this.heightMax = resModel.getX1();
                        }
                        listJosn.add(resModel);
                    }
                }
            }
        }
        this.setStrList(list);
        this.setContent(String.join("", list));
        return listJosn;
    }

    //第一种发票类型
    private void setInfoOne(List<OcrResModel> param) {
        //购买方底部线
        Integer h1 = param.stream().filter(t -> "单位".equals(t.getText()) || "数量".equals(t.getText()) || "单价".equals(t.getText()))
                .collect(Collectors.groupingBy(OcrResModel::getY1, Collectors.counting()))
                .entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(0);
        //货物服务清单以上部分
        List<OcrResModel> collect1 = param.stream().filter(t -> t.getY1() < h1).collect(Collectors.toList());
        //头部信息
        topInfo(collect1);
        //购买方信息
        buyerInfo(collect1);
        //货物服务清单以下部分
        List<OcrResModel> collect2 = param.stream().filter(t -> t.getY1() >= h1 - 10).collect(Collectors.toList());
        //统计和备注
        countAndRemark(collect2);
        //销售方信息
        sellerDown(collect2);
    }

    //字段解析
    private String getStr(String key, OcrResModel item, List<OcrResModel> list, Integer index) {
        String value = "";
        String text = item.getText();
        if (text.contains(key)) {
            if (text.endsWith("：")) {
                if ((index + 1) < list.size() && list.get(index + 1) != null && list.get(index + 1).getY1() < item.getY2()) {
                    value = list.get(index + 1).getText();
                }
            } else {
                value = text.substring(text.indexOf("：") + 1);
            }
        }
        return value;
    }

    //上部分
    private void topInfo(List<OcrResModel> collect1) {
        int xMid = widthMax / 2;
        int[] hSet = collect1.stream().filter(t -> "名".equals(t.getText()) || t.getText().contains("称：") || t.getText().startsWith("名称"))
                .map(OcrResModel::getY1).collect(Collectors.toSet()).stream().mapToInt(Integer::intValue).toArray();
        if (hSet.length > 0) {
            //购买方名称以上部分
            List<OcrResModel> collect3 = collect1.stream().filter(t -> t.getY2() < hSet[0]).collect(Collectors.toList());
            OcrResModel ocrResModel1 = collect3.stream().filter(t -> t.getX1() < xMid && t.getX2() > xMid).findFirst().orElse(null);
            title = ocrResModel1 != null ? ocrResModel1.getText() : null;
            int i = 0;
            for (OcrResModel item : collect3) {
                if (StringUtil.isEmpty(code)) code = getStr("代码：", item, collect3, i);
                if (StringUtil.isEmpty(number)) number = getStr("号码：", item, collect3, i);
                if (StringUtil.isEmpty(number) && item.getText().startsWith("No")) {
                    number = item.getText().substring(2);
                }
                if (date == null) {
                    String dateStr = getStr("开票", item, collect3, i);
                    if (StringUtil.isNotEmpty(dateStr)) {
                        dateStr = dateStr.length() > 11 ? dateStr.substring(dateStr.length() - 11) : dateStr;
                        Date date1 = DateUtil.checkDate(dateStr, "yyyy年MM月dd日");
                        date = date1 != null ? date1.getTime() : null;
                    }
                }
                if (StringUtil.isEmpty(checkCode)) checkCode = getStr("校验码", item, collect3, i);
                i++;
            }

            //补充，如果识别不到代码和号码
            if (StringUtil.isEmpty(code)) {
                for (OcrResModel item : collect3) {
                    if (Pattern.compile("\\d{10,12}").matcher(item.getText()).matches()) {
                        code = item.getText();
                        return;
                    }
                }
            }
        }
    }

    //购买方
    private void buyerInfo(List<OcrResModel> collect1) {
        int xMid = widthMax / 2;
        OcrResModel namModel = collect1.stream().filter(t -> "名".equals(t.getText()) || t.getText().contains("称：")
                || "名称：".equals(t.getText())).findFirst().orElse(null);
        //购买方部分
        List<OcrResModel> collect4 = namModel != null ? collect1.stream().filter(t -> t.getY2() > namModel.getY1() - 5).collect(Collectors.toList()) : collect1;
        //购买方左边
        List<OcrResModel> collect5 = collect4.stream().filter(t -> t.getX1() < xMid).collect(Collectors.toList());
        int j = 0;
        for (OcrResModel item : collect5) {
            if (StringUtil.isEmpty(buyerName)) buyerName = getStr("称：", item, collect5, j);
            if (StringUtil.isEmpty(buyerTaxId)) buyerTaxId = getStr("识别号：", item, collect5, j);
            if (StringUtil.isEmpty(buyerContact)) buyerContact = getStr("电话：", item, collect5, j);
            if (StringUtil.isEmpty(buyerBankAccount)) buyerBankAccount = getStr("标识：", item, collect5, j);
            if (StringUtil.isEmpty(buyerBankAccount)) buyerBankAccount = getStr("账号：", item, collect5, j);
            j++;
        }
        //购买方右边
        List<OcrResModel> collect6 = collect4.stream().filter(t -> t.getX1() > xMid).collect(Collectors.toList());
        OcrResModel ocrResModel = collect6.stream().filter(t ->
                "密".equals(t.getText()) || "码".equals(t.getText()) || "区".equals(t.getText())).findFirst().orElse(null);
        if (ocrResModel != null) {
            password = collect6.stream().filter(t -> t.getX1() > ocrResModel.getX2()).map(OcrResModel::getText).collect(Collectors.joining());
        } else {
            password = collect6.stream().map(OcrResModel::getText).collect(Collectors.joining());
        }
    }

    //下部分
    private void countAndRemark(List<OcrResModel> collect2) {
        OcrResModel countModel = collect2.stream().filter(t -> "合".equals(t.getText()) || "计".equals(t.getText())).findFirst().orElse(null);
        if (countModel != null) {
            int indexCount = collect2.indexOf(countModel);
            if (collect2.get(indexCount + 1).getX1() < widthMax / 2) {
                indexCount = indexCount + 1;
            }
            totalAmount = collect2.get(indexCount + 1).getText();
            totalTaxAmount = collect2.get(indexCount + 2).getText();
            if (StringUtil.isNotEmpty(totalAmount) && !Character.isDigit(totalAmount.charAt(0))) {
                totalAmount = totalAmount.substring(1);
            }
            if (StringUtil.isNotEmpty(totalTaxAmount) && !Character.isDigit(totalTaxAmount.charAt(0))) {
                totalTaxAmount = totalTaxAmount.substring(1);
            }
            //中间部分数组
            List<OcrResModel> itemCollect = collect2.stream().filter(t -> t.getY2() < countModel.getY1()).collect(Collectors.toList());
            itemInfo(itemCollect);
        }

        OcrResModel countUpper = collect2.stream().filter(t -> "价税合计（大写）".equals(t.getText()) || t.getText().contains("税合")).findFirst().orElse(null);
        if (countUpper != null) {
            int indexCountUpper = collect2.indexOf(countUpper);
            totalAmountUpper = collect2.get(indexCountUpper + 1).getText();
            String tal = collect2.get(indexCountUpper + 2).getText();
            if (tal.endsWith("写）")) {
                totalAmountLower = collect2.get(indexCountUpper + 3).getText();
            } else {
                totalAmountLower = tal.contains("）") && tal.split("）").length > 1 ? tal.split("）")[1] : tal;
            }
            if (StringUtil.isNotEmpty(totalAmountLower) && !Character.isDigit(totalAmountLower.charAt(0))) {
                totalAmountLower = totalAmountLower.substring(1);
            }
        }

        int downIndex = countUpper != null ? countUpper.getY2() : heightMax / 2;
        //下面部分数组
        List<OcrResModel> downCollect = collect2.stream().filter(t -> t.getY2() > downIndex).collect(Collectors.toList());
        int i = 0;
        for (OcrResModel item : downCollect) {
            if (StringUtil.isEmpty(payee)) {
                String str = getStr("收款人：", item, downCollect, i);
                if (StringUtil.isNotEmpty(str) && !str.contains("复核")) {
                    payee = str;
                }
            }
            if (StringUtil.isEmpty(reviewer)) {
                String str = getStr("复核：", item, downCollect, i);
                if (StringUtil.isNotEmpty(str) && !str.contains("开票人")) {
                    reviewer = str;
                }
            }
            if (StringUtil.isEmpty(issuer)) issuer = getStr("开票人：", item, downCollect, i);
            i++;
        }
        //备注
        OcrResModel remarkM = collect2.stream().filter(t -> "备".equals(t.getText()) || "注".equals(t.getText())).findFirst().orElse(null);
        if (remarkM != null) {
            OcrResModel remarkLimit = collect2.stream().filter(t -> t.getText().contains("开票人")).findFirst().orElse(null);
            List<OcrResModel> collect = downCollect.stream().filter(t -> {
                if (remarkLimit != null) {
                    return t.getX1() > remarkM.getX2() && t.getY1() > downIndex && t.getY1() < remarkLimit.getY1();
                }
                return t.getX1() > remarkM.getX2() && t.getY1() > downIndex;
            }).collect(Collectors.toList());
            remarks = collect.stream().map(OcrResModel::getText).collect(Collectors.joining());
        }
    }

    //销售方
    private void sellerDown(List<OcrResModel> collect) {
        OcrResModel countUpper = collect.stream().filter(t -> "价税合计（大写）".equals(t.getText())
                || "税合".equals(t.getText())).findFirst().orElse(null);
        int downIndex = countUpper != null ? countUpper.getY2() : heightMax / 2;
        //下面部分数组
        List<OcrResModel> downCollect = collect.stream().filter(t -> t.getY2() > downIndex).collect(Collectors.toList());
        int i = 0;
        for (OcrResModel item : downCollect) {
            if (StringUtil.isEmpty(sellerName)) sellerName = getStr("称：", item, downCollect, i);
            if (StringUtil.isEmpty(sellerTaxId)) sellerTaxId = getStr("识别号：", item, downCollect, i);
            if (StringUtil.isEmpty(sellerContact)) sellerContact = getStr("电话：", item, downCollect, i);
            if (StringUtil.isEmpty(sellerBankAccount)) sellerBankAccount = getStr("标识：", item, downCollect, i);
            if (StringUtil.isEmpty(sellerBankAccount)) sellerBankAccount = getStr("账号：", item, downCollect, i);
            i++;
        }
    }

    //中间物资信息
    private void itemInfo(List<OcrResModel> itemCollect) {
        OcrResModel itemTitle = itemCollect.stream().filter(t ->
                "单位".equals(t.getText()) || "数量".equals(t.getText()) || "税率".equals(t.getText())).findFirst().orElse(null);

        OcrResModel nameT = itemCollect.stream().filter(t -> t.getText().contains("服务名称") || t.getText().contains("货物或")
                || t.getText().contains("项目名称")).findFirst().orElse(null);
        OcrResModel typeT = itemCollect.stream().filter(t -> t.getText().equals("规格型号") || t.getText().contains("型号")).findFirst().orElse(null);
        OcrResModel unitT = itemCollect.stream().filter(t -> t.getText().equals("单位")).findFirst().orElse(null);
        OcrResModel numT = itemCollect.stream().filter(t -> t.getText().equals("数量")).findFirst().orElse(null);
        if (numT == null) {
            numT = itemCollect.stream().filter(t -> t.getText().equals("数")).findFirst().orElse(null);
        }
        OcrResModel priceT = itemCollect.stream().filter(t -> t.getText().equals("单价")).findFirst().orElse(null);
        OcrResModel amountT = itemCollect.stream().filter(t -> t.getText().equals("金额")).findFirst().orElse(null);
        if (amountT == null) {
            amountT = itemCollect.stream().filter(t -> t.getText().equals("金")).findFirst().orElse(null);
        }
        OcrResModel rateT = itemCollect.stream().filter(t -> t.getText().contains("税率") || t.getText().equals("税单")).findFirst().orElse(null);
        OcrResModel taxAmountT = itemCollect.stream().filter(t -> t.getText().equals("税额")).findFirst().orElse(null);
        if (taxAmountT == null) {
            taxAmountT = itemCollect.stream().filter(t -> t.getText().equals("税")).findFirst().orElse(null);
        }
        if (nameT == null || typeT == null || unitT == null || numT == null || priceT == null
                || amountT == null || rateT == null || taxAmountT == null) {
            return;
        }
        List<OcrResModel> eachModel = new ArrayList<>();
        eachModel.add(0, nameT);
        eachModel.add(1, typeT);
        eachModel.add(2, unitT);
        eachModel.add(3, numT);
        eachModel.add(4, priceT);
        eachModel.add(5, amountT);
        eachModel.add(6, rateT);
        eachModel.add(7, taxAmountT);
        eachModel.add(8, itemTitle);
        itemInfoParse(itemCollect, eachModel);
    }

    private void itemInfoParse(List<OcrResModel> itemCollect, List<OcrResModel> eachModel) {
        List<OcrResModel> nameList = new ArrayList<>();//品名
        List<OcrResModel> typeList = new ArrayList<>();//类型
        List<OcrResModel> unitList = new ArrayList<>();//单位
        List<OcrResModel> numList = new ArrayList<>();//数量
        List<OcrResModel> priceList = new ArrayList<>();//单价
        List<OcrResModel> amountList = new ArrayList<>();//金额
        List<OcrResModel> rateList = new ArrayList<>();//税率
        List<OcrResModel> taxAmountList = new ArrayList<>();//税额
        List<OcrResModel> collect = itemCollect.stream().filter(t -> t.getY1() > eachModel.get(8).getY2() - 8).collect(Collectors.toList());//误差给8
        for (OcrResModel item : collect) {
            if (item.getX2() > eachModel.get(0).getX1() && item.getX1() < (eachModel.get(1).getX1() + eachModel.get(0).getX2()) / 2) {
                nameList.add(item);
            }
            if (item.getX1() > (eachModel.get(1).getX1() + eachModel.get(0).getX2()) / 2 && item.getX1() < (eachModel.get(2).getX1() + eachModel.get(1).getX2()) / 2) {
                typeList.add(item);
            }
            if (item.getX1() > (eachModel.get(2).getX1() + eachModel.get(1).getX2()) / 2 && item.getX1() < (eachModel.get(3).getX1() + eachModel.get(2).getX2()) / 2) {
                unitList.add(item);
            }
            if (item.getX1() > (eachModel.get(3).getX1() + eachModel.get(2).getX2()) / 2 && item.getX1() < (eachModel.get(4).getX1() + eachModel.get(3).getX2()) / 2) {
                numList.add(item);
            }
            if (item.getX1() > (eachModel.get(4).getX1() + eachModel.get(3).getX2()) / 2 && item.getX1() < (eachModel.get(5).getX1() + eachModel.get(4).getX2()) / 2) {
                priceList.add(item);
            }
            if (item.getX1() > (eachModel.get(5).getX1() + eachModel.get(4).getX2()) / 2 && item.getX1() < (eachModel.get(6).getX1() + eachModel.get(6).getX1()) / 2) {
                amountList.add(item);
            }
            if (item.getX1() > (eachModel.get(6).getX1() + eachModel.get(6).getX1()) / 2 && item.getX1() < eachModel.get(7).getX1()) {
                if (StringUtil.isNotEmpty(item.getText()) && !Character.isDigit(item.getText().length() - 1)) {
                    item.setText(item.getText().substring(0, item.getText().length() - 1));
                }
                rateList.add(item);
            }
            if (item.getX1() > eachModel.get(7).getX1()) {
                taxAmountList.add(item);
            }
        }
        Map<String, List<OcrResModel>> map = new HashMap<>();
        map.put("nameList", nameList);
        map.put("typeList", typeList);
        map.put("unitList", unitList);
        map.put("numList", numList);
        map.put("priceList", priceList);
        map.put("amountList", amountList);
        map.put("rateList", rateList);
        map.put("taxAmountList", taxAmountList);
        invoiceDetail = InvoiceDetail.getList(map);
    }

    //+++++++++++++++++++++++++++++++++++++++++第二种发票类型+++++++++++++++++++++++++++++++++++++++
    //第二种发票类型
    private void setInfoTwo(List<OcrResModel> param) {
        //购买方底部线
        Integer h1 = param.stream().filter(t -> "单位".equals(t.getText()) || "数量".equals(t.getText()) || "单价".equals(t.getText()))
                .collect(Collectors.groupingBy(OcrResModel::getY1, Collectors.counting()))
                .entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(0);
        //货物服务清单以上部分
        List<OcrResModel> collect1 = param.stream().filter(t -> t.getY2() < h1).collect(Collectors.toList());
        //头部信息
        topInfo(collect1);
        //购买方信息 和销售方信息
        buyerAndSeller(collect1);
        //货物服务清单以下部分
        List<OcrResModel> collect2 = param.stream().filter(t -> t.getY1() >= h1 - 10).collect(Collectors.toList());
        //统计和备注
        countAndRemark(collect2);
    }

    //第二种发票购买方和销售方
    private void buyerAndSeller(List<OcrResModel> upList) {
        int xMid = widthMax / 2;
        OcrResModel namModel = upList.stream().filter(t -> t.getText().startsWith("名称：")).findFirst().orElse(null);
        if (namModel == null) return;
        //购买方部分
        List<OcrResModel> collect4 = upList.stream().filter(t -> t.getY2() > namModel.getY1() - 10).collect(Collectors.toList());
        //购买方左边
        List<OcrResModel> collect5 = collect4.stream().filter(t -> t.getX1() < xMid).collect(Collectors.toList());
        int i = 0;
        for (OcrResModel item : collect5) {
            if (StringUtil.isEmpty(buyerName)) buyerName = getStr("名称：", item, collect5, i);
            if (StringUtil.isEmpty(buyerTaxId)) buyerTaxId = getStr("识别号：", item, collect5, i);
            i++;
        }
        //购买方右边
        List<OcrResModel> collect6 = collect4.stream().filter(t -> t.getX1() > xMid).collect(Collectors.toList());
        int j = 0;
        for (OcrResModel item : collect6) {
            if (StringUtil.isEmpty(sellerName)) sellerName = getStr("名称：", item, collect5, j);
            if (StringUtil.isEmpty(sellerTaxId)) sellerTaxId = getStr("识别号：", item, collect5, j);
            j++;
        }
    }
}