package com.example.xiaoshiweixinback.controller;

import com.alibaba.fastjson2.JSONObject;
import com.example.xiaoshiweixinback.business.common.Constants;
import com.example.xiaoshiweixinback.business.common.Response;
import com.example.xiaoshiweixinback.business.common.base.Records;
import com.example.xiaoshiweixinback.entity.dto.esPicture.EsPictureNoDTO;
import com.example.xiaoshiweixinback.entity.dto.esPicture.EsPatentVectorDTO;
import com.example.xiaoshiweixinback.entity.dto.patent.ImportTaskAMVO;
import com.example.xiaoshiweixinback.entity.product.ProductIdDTO;
import com.example.xiaoshiweixinback.entity.vo.esPicture.EsPictureNoVo;
import com.example.xiaoshiweixinback.service.common.EsDenseVectorService;
import com.example.xiaoshiweixinback.service.common.FileManagerService;
import com.example.xiaoshiweixinback.service.importPatent.ImportFromWebToEsService;
import io.swagger.v3.oas.annotations.Operation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.io.File;
import java.util.List;


@RequestMapping(Constants.XIAOSHI_WEIXINBACK + "/patent")
@RestController
public class PatentController {

    @Autowired
    private ImportFromWebToEsService importFromWebToEsService;

    @Autowired
    private FileManagerService fileManagerService;

    @Autowired
    private EsDenseVectorService esDenseVectorService;

    @Operation(summary = "导入专利")
    @GetMapping(value = "/importPatent")
    public Response queryPatent(@RequestBody ImportTaskAMVO importTaskAMVO) {
        try {
           importFromWebToEsService.ImportTask(importTaskAMVO);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return Response.success("");
    }

    @Operation(summary = "根据关键词获取列表(图片用于排序)--zero")
    @PostMapping(value = "/getPatentVectors")
    public Response getPatentVectors(String vectorDTO, File file) throws Exception {
        EsPatentVectorDTO esPatentVectorDTO = JSONObject.parseObject(vectorDTO, EsPatentVectorDTO.class);
        Records records = esDenseVectorService.getPatentVectors(esPatentVectorDTO, file);
        return Response.success(records);
    }

    @Operation(summary = "根据专利号获取相关图片--zero")
    @PostMapping(value = "/getPictureByNo")
    public Response getPictureByNo(@RequestBody EsPictureNoDTO pictureNoDTO) throws Exception {
        List<EsPictureNoVo> pictureByNo = esDenseVectorService.getPictureByNo(pictureNoDTO);
        return Response.success(pictureByNo);
    }
}