package com.ai.da.service;

import com.ai.da.mapper.primary.entity.Generate;
import com.ai.da.mapper.primary.entity.GenerateDetail;
import com.ai.da.model.dto.GenerateLikeDTO;
import com.ai.da.model.dto.GenerateThroughImageTextDTO;
import com.ai.da.model.vo.*;
import com.baomidou.mybatisplus.extension.service.IService;

import java.util.List;

public interface GenerateService extends IService<Generate> {

    GenerateCaptionVO generateCaption(Long sketchElementId);

    void generateThroughImageText(GenerateThroughImageTextDTO generateThroughImageTextDTO);

    void processGenerateResult(String taskId, String url, String category);

    GenerateLikeVO generateLike(GenerateLikeDTO generateLikeDTO);

    Boolean generateDislike(Long generateDetailId, String timeZone);

    void updateLikeStatusBatch(List<Long> generateDetailIdList, Byte hasLike, Long libraryId, String timeZone);

    List<GenerateDetail> selectBatchByLibraryId(List<Long> libraryId);

    GenerateCollectionVO getGenerateResult(String uniqueId);

    List<GenerateResultVO> getGenerateResultList(List<String> taskIdList);

    PrepareForGenerateVO prepareForGenerate(GenerateThroughImageTextDTO generateThroughImageTextDTO);

    Long getRankPosition(String uniqueId);

    void cancelGenerate(Long userId, List<String> uniqueId, String timeZone);

}