package org.jeecg.modules.ai.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.aspect.annotation.AutoLog;
import org.jeecg.common.system.base.controller.JeecgController;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.StrUtils;
import org.jeecg.modules.ai.entity.FileImgInfo;
import org.jeecg.modules.ai.entity.FilePdfInfo;
import org.jeecg.modules.ai.service.IFileImgInfoService;
import org.jeecg.modules.ai.service.IFilePdfInfoService;
import org.jeecg.modules.ai.vo.FilePdfSelectVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author clown
 * * @date 2024/8/9
 */
@Slf4j
@Api(tags = "PDF图片文件管理")
@RestController
@RequestMapping("/ai/fileImg")
public class FileImgInfoController extends JeecgController<FileImgInfo, IFileImgInfoService> {

    @Autowired
    private IFilePdfInfoService filePdfInfoService;

    @AutoLog(value = "分页列表查询")
    @ApiOperation(value = "分页列表查询", notes = "分页列表查询")
    @GetMapping(value = "/list")
    public Result<?> queryPageList(FilePdfSelectVo vo, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
                                   @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize){

        Page page = new Page<>(pageNo, pageSize);
        FilePdfInfo info = null;
        if (StrUtils.isNotBlankOrNull(vo.getPdfFileId())) {
            info = filePdfInfoService.getById(vo.getPdfFileId());
        } else if (StrUtils.isNotBlankOrNull(vo.getPdfName())) {
            info = filePdfInfoService.findNewFilePDF(vo.getPdfName());
        }

        if (info == null) {
            return Result.error("无此文件");
        }
        IPage<FileImgInfo> imgInfos = service.queryPageList(info.getId(), page);
        return Result.OK(imgInfos);
    }




}