package com.ruoyi.web.controller.project;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.annotation.RepeatSubmit;
import com.ruoyi.common.config.RuoYiConfig;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.ChineseUtils;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.file.FileUploadUtils;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.config.ServerConfig;
import com.ruoyi.project.domain.ServiceInvoice;
import com.ruoyi.project.service.IServiceInvoiceService;
import com.ruoyi.web.controller.tool.HttpUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.http.HttpResponse;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.math.BigDecimal;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.*;

/**
 * 发票管理Controller
 *
 * @author ruoyi
 * @date 2026-10-21
 */
@Api("发票管理")
@RestController
@RequestMapping("/project/invoice")
public class ServiceInvoiceController extends BaseController {
    @Autowired
    private IServiceInvoiceService serviceInvoiceService;
    @Autowired
    private ServerConfig serverConfig;

    @Value("${appcode}")
    private String  appcode;
    /**
     * 查询发票管理列表
     */
    @ApiOperation("查询发票管理列表")
    @GetMapping("/list")
    public TableDataInfo list(ServiceInvoice serviceInvoice) {
        startPage();
        List<ServiceInvoice> list = serviceInvoiceService.queryList(serviceInvoice);
        return getDataTable(list);
    }

    /**
     * 导出发票管理列表
     */
    @ApiOperation("导出发票管理列表")
    @Log(title = "发票管理", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(ServiceInvoice serviceInvoice) {
        List<ServiceInvoice> list = serviceInvoiceService.queryList(serviceInvoice);
        ExcelUtil<ServiceInvoice> util = new ExcelUtil<ServiceInvoice>(ServiceInvoice.class);
        return util.exportExcel(list, "发票管理数据");
    }

    /**
     * 获取发票管理详细信息
     */
    @ApiOperation("获取发票管理详细信息")
    @GetMapping(value = "/getInfo/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id) {
        return AjaxResult.success(serviceInvoiceService.getById(id));
    }

    /**
     * 新增发票管理
     */
    @ApiOperation("新增发票管理")
    @Log(title = "发票管理", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @RepeatSubmit
    public AjaxResult add(@RequestBody ServiceInvoice serviceInvoice) {
        return toAjax(serviceInvoiceService.save(serviceInvoice));
    }

    /**
     * 修改发票管理
     */
    @ApiOperation("修改发票管理")
    @Log(title = "发票管理", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @RepeatSubmit
    public AjaxResult edit(@RequestBody ServiceInvoice serviceInvoice) {
        return toAjax(serviceInvoiceService.updateById(serviceInvoice));
    }

    /**
     * 删除发票管理
     */
    @ApiOperation("删除发票管理")
    @Log(title = "发票管理", businessType = BusinessType.DELETE)
    @GetMapping("/remove/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids) {
        return toAjax(serviceInvoiceService.removeByIds(Arrays.asList(ids)));
    }

    /**
     * 通用上传请求
     */
    @ApiOperation("testOCR")
    @PostMapping("/testOCR")
    public AjaxResult testOCR(@RequestBody String fileName) throws Exception {
        Path filePath= Paths.get(fileName);
        String base64= Base64.getEncoder().encodeToString( Files.readAllBytes(filePath));

        AjaxResult ajax = AjaxResult.success();
        ajax.put("OCRList", aliOCR(base64,0,0));
        return ajax;
    }

    /**
     * 通用上传请求
     */
    @PostMapping("/uploadOCR")
    public AjaxResult uploadOCR(@RequestParam("file") MultipartFile file, @RequestParam("userId") long userId, @RequestParam("caseId") long caseId) throws Exception {

        try {
            // 上传文件路径
            String filePath = RuoYiConfig.getUploadPath();
            //判断文件是否存在
            String filename = file.getOriginalFilename().trim();
            String felinamePath = FileUploadUtils.extractFilename(file);
            String newFilename="";
//            String pathFileName = FileUploadUtils.getPathFileName(filePath, s);
            for (int i=1;i<500;i++) {
                File file1 = new File(filePath + "/" + felinamePath);
                boolean exists = file1.exists();
                if (exists) {
                    int index = filename.lastIndexOf(".");
                    String startStr = filename.substring(0, index);
                    String endStr = filename.substring(index, filename.length());
                    newFilename = startStr + "-" + i + endStr; // your new filename
                    felinamePath = DateUtils.datePath() + "/" + newFilename;
                    String contentType = file.getContentType();
                    byte[] bytes = file.getBytes();
                    file = new MockMultipartFile(newFilename, newFilename, contentType, bytes);
                }else{
                    break;
                }
            }
            // 上传并返回新文件名称
            String fileName = FileUploadUtils.upload(filePath, file);
            String url1 = serverConfig.getUrl();
            String url = url1 + fileName;
            String base64= Base64.getEncoder().encodeToString(file.getBytes());
            List<Map<String,Object>> lists=aliOCR(base64,userId,caseId);
            for(Map<String,Object> row : lists){
                ServiceInvoice invoice=new ServiceInvoice();
                invoice.setUserId(userId);
                invoice.setCaseId(caseId);
                invoice.setInvoiceType(row.get("type")+"");
                invoice.setInvoiceNo(row.get("发票号码")+"");
                invoice.setOccurDate(row.get("开票日期")+"");
                invoice.setInvoiceAmount(row.get("发票金额")+"");
                invoice.setSellerName(row.get("受票方名称")+"");
                invoice.setInvoiceUrl(url);
                invoice.setInvoiceName(newFilename);
                invoice.setExtContent(JSON.toJSONString(row));
                serviceInvoiceService.save(invoice);
            }

            AjaxResult ajax = AjaxResult.success();
            ajax.put("fileName", fileName);
            ajax.put("url", url);
            ajax.put("OCRList", lists);
            return ajax;
        } catch (Exception e) {
            return AjaxResult.error(e.getMessage());
        }
    }

    public List<Map<String,Object>> aliOCR(String base64,long userId,long caseId) {
        String host = "https://mixedmulti.market.alicloudapi.com";
        String path = "/ocrservice/mixedMultiInvoice";
        String method = "POST";
        Map<String, String> headers = new HashMap<String, String>();
        //最后在header中的格式(中间是英文空格)为Authorization:APPCODE 83359fd73fe94948385f570e3c139105
        headers.put("Authorization", "APPCODE " + appcode);
        //根据API的要求，定义相对应的Content-Type
        headers.put("Content-Type", "application/json; charset=UTF-8");
        Map<String, String> querys = new HashMap<String, String>();
        String bodys = "{\"img\":\""+base64+"\"}";
        List<Map<String,Object>> lists=new ArrayList<>();
        try {
            HttpResponse response = HttpUtils.doPost(host, path, method, headers, querys, bodys);
            //获取response的body
            String content= EntityUtils.toString(response.getEntity());
            JSONObject jsonContent=JSONObject.parseObject(content);
            JSONArray subMsgs=jsonContent.getJSONArray("subMsgs");

            for(Object row: subMsgs){

                Map<String,Object> tempList=new HashMap<>();
                JSONObject rowObject=(JSONObject)row;
                tempList.put("发票分类",rowObject.getString("type"));
                JSONObject result=((JSONObject)rowObject.get("result")).getJSONObject("data");

                for(Map.Entry<String, Object> entry : result.entrySet()){
                    tempList.put(entry.getKey(),entry.getValue());
                }
                if(rowObject.getString("type").equals("银行承兑汇票")){
                    BigDecimal totalAmount= ChineseUtils.chineseToNumber(result.getString("票据金额大写"));
                    tempList.put("发票金额",totalAmount);
                }
                if(rowObject.getString("type").equals("火车票")){
                    tempList.put("发票金额",result.getString("票价"));
                }

                lists.add(tempList);
            }

            return lists;
        } catch (Exception e) {
            return lists;
        }
    }
}