package jnpf.base.controller;

import com.alibaba.fastjson.JSONObject;
import feign.Response;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jnpf.base.ActionResult;
import jnpf.base.model.ocr.OcrForm;
import jnpf.base.model.ocr.OcrModel;
import jnpf.config.ConfigValueUtil;
import jnpf.constant.FileTypeConstant;
import jnpf.constant.MsgCode;
import jnpf.exception.DataException;
import jnpf.file.FileUploadApi;
import jnpf.model.upload.UploadFileModel;
import jnpf.util.StringUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.Base64;
import java.util.List;

/**
 * ocr解析
 *
 * @author JNPF开发平台组
 * @version v6.1.0
 * @copyright 引迈信息技术有限公司
 * @date 2025/7/29 16:58:47
 */
@Slf4j
@Tag(name = "ocr解析", description = "ocr")
@RestController
@RequiredArgsConstructor
@RequestMapping("/ocr")
public class OcrController {

    private final ConfigValueUtil configValueUtil;
    private final FileUploadApi fileUploadApi;

    @Operation(summary = "ocr解析")
    @PostMapping
    public ActionResult<OcrModel> form(@RequestBody OcrForm form) {
        String url = form.getUrl();
        String substring = url.substring(url.lastIndexOf("/") + 1);
        //下载到本地
        Response response = fileUploadApi.downFile(new UploadFileModel(FileTypeConstant.ANNEXPIC, substring));
        try (InputStream ins = response.body().asInputStream()) {
            byte[] byteArray = IOUtils.toByteArray(ins);
            String str = Base64.getEncoder().encodeToString(byteArray);
            JSONObject body = new JSONObject();
            List<String> list = new ArrayList<>();
            list.add(str);
            body.put("images", list);
            String ocrUrl = configValueUtil.getOcrUrl();
            if (StringUtil.isEmpty(ocrUrl)) {
                return ActionResult.fail(MsgCode.SYS183.get());
            }
            if (ocrUrl.endsWith("/")) {
                ocrUrl = ocrUrl.substring(0, ocrUrl.length() - 1);
            }
            OcrModel ocrModel = OcrModel.get(form.getType());
            ocrModel.extract(ocrUrl, body);
            return ActionResult.success(ocrModel);
        } catch (DataException e) {
            log.error(e.getMessage(), e);
            return ActionResult.fail(e.getMessage());
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
        return ActionResult.fail(MsgCode.FA104.get());
    }
}