mitao
2025-01-21 19ce15ddb66af0fb0b2e447bd70dd39e08bb30fd
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java
@@ -1,9 +1,12 @@
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;
@@ -13,7 +16,6 @@
import org.springframework.web.bind.annotation.RestController;
import java.io.InputStream;
import java.net.URLEncoder;
@RestController
@RequestMapping("/common")
@@ -28,19 +30,63 @@
    public R<IdCard> idCard(@Parameter(name = "url", description = "图片地址") String url) {
        InputStream inputStream;
        try {
            // 找到最后一个斜杠的位置
            int lastSlashIndex = url.lastIndexOf("/");
            // 从斜杠后提取文件名
            String baseUrl = url.substring(0, lastSlashIndex);
            String fileName = url.substring(lastSlashIndex + 1);
            String encode = URLEncoder.encode(fileName, "UTF-8");
            url = baseUrl + "/" + encode;
            inputStream = UrlDownloader.downloadAsStream(url);
            JSONObject jsonObject = OcrUtils.idCard(inputStream);
            IdCard idCard = JSONObject.parseObject(jsonObject.toJSONString(), IdCard.class);
            return R.ok(idCard);
            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();
    }
}