package com.youlai.boot.mini.controller;

import com.youlai.boot.common.annotation.Log;
import com.youlai.boot.common.annotation.RepeatSubmit;
import com.youlai.boot.common.enums.ActionTypeEnum;
import com.youlai.boot.common.enums.LogModuleEnum;
import com.youlai.boot.common.result.Result;
import com.youlai.boot.framework.security.util.SecurityUtils;
import com.youlai.boot.mini.model.dto.PhotoToComicRequest;
import com.youlai.boot.mini.model.vo.AiTaskCallbackVO;
import com.youlai.boot.mini.service.AiGenerationService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.enums.ParameterIn;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import lombok.RequiredArgsConstructor;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.Collections;
import java.util.List;

@Tag(name = "AI生成图片视频相关接口")
@RestController
@RequestMapping("/api/v1/mini/ai/generation")
@RequiredArgsConstructor
@Valid
public class AiGenerationController {

    private final AiGenerationService aiGenerationService;

    @Operation(summary = "上传AI生成参考文件", operationId = "AiReferenceSaveFile")
    @PostMapping(value = "/upload-reference", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    @RepeatSubmit
    @Log(module = LogModuleEnum.AI_TASK_MEDIA, value = ActionTypeEnum.INSERT)
    public Result<List<String>> uploadReferenceFile(
            @RequestPart(name = "images", required = false) List<MultipartFile> images,
            @RequestPart(name = "videos", required = false) List<MultipartFile> videos
    ) {
        Long userId = SecurityUtils.getUserId();
        List<String> urlList = aiGenerationService.uploadReferenceFile(images, videos, userId);
        return Result.success(urlList);
    }

    @Operation(summary = "提交单图漫画生成任务")
    @PostMapping("/generate-single-image")
    @Log(module = LogModuleEnum.AI_GENERATION_TASK, value = ActionTypeEnum.INSERT)
    public Result<String> generateSingleImage(@Valid @RequestBody PhotoToComicRequest request) {
        Long userId = SecurityUtils.getUserId();
        String taskUuid = aiGenerationService.createAndGenerateImage(request, userId);
        return Result.success(taskUuid);
    }

    @Operation(summary = "AI生成任务回调接口")
    @PostMapping("/task/callback")
    @Log(module = LogModuleEnum.AI_GENERATION_TASK, value = ActionTypeEnum.UPDATE)
    public Result<Boolean> taskCallback(@Valid @RequestBody AiTaskCallbackVO vo) {
        boolean success = aiGenerationService.handleTaskCallback(vo.getUuid(), vo.getStatus(), vo.getResultUrl());
        return Result.success(success);
    }
}