package com.kamco.cd.training.model;

import com.kamco.cd.training.config.api.ApiResponseDto;
import com.kamco.cd.training.dataset.dto.DatasetDto;
import com.kamco.cd.training.dataset.dto.DatasetDto.SelectDataSet;
import com.kamco.cd.training.model.dto.ModelConfigDto;
import com.kamco.cd.training.model.dto.ModelTrainMngDto;
import com.kamco.cd.training.model.dto.ModelTrainMngDto.Basic;
import com.kamco.cd.training.model.service.ModelTrainMngService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.media.Content;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.responses.ApiResponse;
import io.swagger.v3.oas.annotations.responses.ApiResponses;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import java.util.List;
import java.util.UUID;
import lombok.RequiredArgsConstructor;
import org.springframework.data.domain.Page;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
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.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequiredArgsConstructor
@Tag(name = "모델학습 관리", description = "어드민 홈 > 모델학습관리 > 모델관리 > 목록")
@RequestMapping("/api/models")
public class ModelTrainMngApiController {
  private final ModelTrainMngService modelTrainMngService;

  @Operation(summary = "모델학습 목록 조회", description = "모델학습 목록 조회 API")
  @ApiResponses(
      value = {
        @ApiResponse(
            responseCode = "200",
            description = "검색 성공",
            content =
                @Content(
                    mediaType = "application/json",
                    schema = @Schema(implementation = Page.class))),
        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @GetMapping("/list")
  public ApiResponseDto<Page<Basic>> findByModelList(
      @Parameter(
              description = "상태코드",
              example = "IN_PROGRESS",
              schema = @Schema(allowableValues = {"", "IN_PROGRESS", "COMPLETED"}))
          @RequestParam(required = false)
          String status,
      @Parameter(
              description = "모델",
              example = "G1",
              schema = @Schema(allowableValues = {"G1", "G2", "G3"}))
          @RequestParam(required = false)
          String modelNo,
      @Parameter(description = "페이지 번호") @RequestParam(defaultValue = "0") int page,
      @Parameter(description = "페이지 크기") @RequestParam(defaultValue = "20") int size) {
    ModelTrainMngDto.SearchReq searchReq =
        new ModelTrainMngDto.SearchReq(status, modelNo, page, size);
    return ApiResponseDto.ok(modelTrainMngService.getModelList(searchReq));
  }

  @Operation(summary = "학습 모델 삭제", description = "학습 모델 삭제 API")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "삭제 성공", content = @Content),
        @ApiResponse(responseCode = "409", description = "HPs_0001 삭제 불가", content = @Content)
      })
  @DeleteMapping("/{uuid}")
  public ApiResponseDto<Void> deleteModelTrain(
      @Parameter(description = "학습 모델 uuid", example = "f2b02229-90f2-45f5-92ea-c56cf1c29f79")
          @PathVariable
          UUID uuid) {
    modelTrainMngService.deleteModelTrain(uuid);
    return ApiResponseDto.ok(null);
  }

  @Operation(summary = "학습 모델 등록", description = "학습 모델 등록 API")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "등록 성공", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @PostMapping
  public ApiResponseDto<String> createModelTrain(@Valid @RequestBody ModelTrainMngDto.AddReq req) {
    modelTrainMngService.createModelTrain(req);
    return ApiResponseDto.ok("ok");
  }

  @Operation(summary = "모델학습 config 정보 조회", description = "모델학습 config 정보 조회 API")
  @ApiResponses(
      value = {
        @ApiResponse(
            responseCode = "200",
            description = "검색 성공",
            content =
                @Content(
                    mediaType = "application/json",
                    schema = @Schema(implementation = ModelConfigDto.Basic.class))),
        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @GetMapping("/config/{uuid}")
  public ApiResponseDto<ModelConfigDto.Basic> updateModelTrain(@PathVariable UUID uuid) {
    return ApiResponseDto.ok(modelTrainMngService.getModelConfigByModelId(uuid));
  }

  @Operation(summary = "모델별 데이터셋 목록 조회", description = "모델별 데이터셋 목록 조회 API")
  @ApiResponses(
      value = {
        @ApiResponse(
            responseCode = "200",
            description = "조회 성공",
            content =
                @Content(
                    mediaType = "application/json",
                    schema = @Schema(implementation = DatasetDto.Basic.class))),
        @ApiResponse(responseCode = "404", description = "데이터셋을 찾을 수 없음", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @GetMapping("/select-dataset-list")
  public ApiResponseDto<List<SelectDataSet>> getDatasetSelectList(
      @Parameter(
              description = "모델 구분",
              example = "",
              schema = @Schema(allowableValues = {"G1", "G2", "G3"}))
          @RequestParam
          String modelType,
      @Parameter(
              description = "선택 구분",
              example = "",
              schema = @Schema(allowableValues = {"CURRENT", "DELIVER", "PRODUCTION"}))
          @RequestParam
          String selectType) {
    return ApiResponseDto.ok(modelTrainMngService.getDatasetSelectList(modelType, selectType));
  }
}