package com.ruoyi.system.controller;

import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.utils.UrlDownloader;
import com.ruoyi.common.core.utils.orc.OcrUtils;
import com.ruoyi.system.api.model.BusinessLicense;
import com.ruoyi.system.api.model.IdCard;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.InputStream;

@RestController
@RequestMapping("/common")
@Tag(name = "通用接口")
public class CommonController {

    /**
     * ocr提取身份证信息
     */
    @GetMapping("/idCard")
    @Operation(summary = "ocr提取身份证信息", description = "ocr提取身份证信息")
    public R<IdCard> idCard(@Parameter(name = "url", description = "图片地址") String url) {
        InputStream inputStream;
        try {

            inputStream = UrlDownloader.downloadAsStream(url);
            String jsonStr = OcrUtils.idCard(inputStream,"IdCard");
            if (StringUtils.isEmpty(jsonStr)){
                return R.fail("身份证识别失败");
            }

            JSONObject jsonObject = JSONObject.parseObject(jsonStr);
            Integer statusCode = jsonObject.getInteger("statusCode");
            if (statusCode != 200) {
                return R.fail("识别失败");
            }
            JSONObject body = jsonObject.getJSONObject("body");
            JSONObject data = body.getJSONObject("Data");
            JSONArray subImages = data.getJSONArray("SubImages");
            if (subImages != null && !subImages.isEmpty()){
                JSONObject subImage = subImages.getJSONObject(0);
                JSONObject kvInfo = subImage.getJSONObject("KvInfo");
                IdCard idCard = kvInfo.getObject("Data", IdCard.class);
                return R.ok(idCard);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return R.fail();
    }

    /**
     * ocr提取营业执照信息
     */
    @GetMapping("/businessLicense")
    @Operation(summary = "ocr提取营业执照信息", description = "ocr提取营业执照信息")
    public R<BusinessLicense> businessLicense(@Parameter(name = "url", description = "图片地址") String url) {
        InputStream inputStream;
        try {
            inputStream = UrlDownloader.downloadAsStream(url);
            String jsonStr = OcrUtils.idCard(inputStream,"BusinessLicense");
            if (StringUtils.isEmpty(jsonStr)){
                return R.fail();
            }
            JSONObject jsonObject = JSONObject.parseObject(jsonStr);
            Integer statusCode = jsonObject.getInteger("statusCode");
            if (statusCode != 200) {
                return R.fail("识别失败");
            }
            JSONObject body = jsonObject.getJSONObject("body");
            JSONObject data = body.getJSONObject("Data");
            JSONArray subImages = data.getJSONArray("SubImages");
            if (subImages != null && !subImages.isEmpty()){
                JSONObject subImage = subImages.getJSONObject(0);
                JSONObject kvInfo = subImage.getJSONObject("KvInfo");
                BusinessLicense businessLicense = kvInfo.getObject("Data", BusinessLicense.class);
                return R.ok(businessLicense);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return R.fail();
    }
}