package com.kamco.cd.kamcoback.label;

import com.kamco.cd.kamcoback.config.api.ApiResponseDto;
import com.kamco.cd.kamcoback.label.dto.LabelAllocateDto;
import com.kamco.cd.kamcoback.label.dto.LabelAllocateDto.InferenceDetail;
import com.kamco.cd.kamcoback.label.dto.LabelAllocateDto.LabelerDetail;
import com.kamco.cd.kamcoback.label.dto.LabelAllocateDto.LabelingStatDto;
import com.kamco.cd.kamcoback.label.dto.LabelAllocateDto.WorkHistoryDto;
import com.kamco.cd.kamcoback.label.dto.WorkerStatsDto.UpdateClosedRequest;
import com.kamco.cd.kamcoback.label.dto.WorkerStatsDto.WorkerListResponse;
import com.kamco.cd.kamcoback.label.service.LabelAllocateService;
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 lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.domain.Page;
import org.springframework.web.bind.annotation.GetMapping;
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;

@Slf4j
@Tag(name = "라벨링 작업 관리", description = "라벨링 작업 배정 및 통계 조회 API")
@RequestMapping({"/api/training-data/stage"})
@RequiredArgsConstructor
@RestController
public class LabelAllocateApiController {

  private final LabelAllocateService labelAllocateService;

  @Operation(summary = "배정 가능한 사용자 목록 조회", description = "라벨링 작업 배정을 위한 활성 상태의 사용자 목록을 조회합니다.")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "조회 성공"),
        @ApiResponse(responseCode = "404", description = "코드를 찾을 수 없음"),
        @ApiResponse(responseCode = "500", description = "서버 오류")
      })
  @GetMapping("/avail-user")
  public ApiResponseDto<List<LabelAllocateDto.UserList>> availUserList(
      @Parameter(
              description = "사용자 역할",
              example = "LABELER",
              schema = @Schema(allowableValues = {"LABELER", "REVIEWER"}))
          @RequestParam
          String role) {
    return ApiResponseDto.ok(labelAllocateService.availUserList(role));
  }

  @Operation(
      summary = "작업현황 관리 > 프로젝트 및 진행 상황 정보",
      description = "작업현황 관리 > 프로젝트 및 진행 상황 정보. UUID를 입력하면 해당 프로젝트 정보를 조회하고, 미입력 시 최신 프로젝트를 조회합니다.")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "조회 성공"),
        @ApiResponse(responseCode = "404", description = "데이터를 찾을 수 없음"),
        @ApiResponse(responseCode = "500", description = "서버 오류")
      })
  @GetMapping("/projectinfo")
  public ApiResponseDto<WorkerListResponse> getWorkerStatistics(
      @Parameter(
              description = "프로젝트 UUID (선택) - 미입력 시 최신 프로젝트 조회",
              example = "f97dc186-e6d3-4645-9737-3173dde8dc64")
          @RequestParam(required = false)
          String uuid,
      @Parameter(
              description = "작업자 유형 (선택) - 미입력 시 전체 조회",
              example = "LABELER",
              schema = @Schema(allowableValues = {"LABELER", "REVIEWER"}))
          @RequestParam(required = false)
          String type,
      @Parameter(description = "검색어 (작업자 이름 또는 사번으로 검색, 부분 일치) - 미입력 시 전체 조회", example = "김라벨")
          @RequestParam(required = false)
          String search,
      @Parameter(
              description = "정렬 조건 (선택) - 미입력 시 이름 오름차순",
              example = "REMAINING_DESC",
              schema =
                  @Schema(
                      allowableValues = {
                        "REMAINING_DESC",
                        "REMAINING_ASC",
                        "COMPLETED_DESC",
                        "COMPLETED_ASC",
                        "NAME_ASC",
                        "NAME_DESC"
                      },
                      defaultValue = "NAME_ASC"))
          @RequestParam(required = false)
          String sort) {

    return ApiResponseDto.ok(
        labelAllocateService.getWorkerStatisticsByUuid(uuid, type, search, sort));
  }

  @Operation(summary = "라벨링작업 관리 > 작업 배정", description = "라벨링작업 관리 > 작업 배정")
  @ApiResponses(
      value = {
        @ApiResponse(
            responseCode = "201",
            description = "등록 성공",
            content =
                @Content(
                    mediaType = "application/json",
                    schema = @Schema(implementation = Long.class))),
        @ApiResponse(responseCode = "400", description = "잘못된 요청 데이터", content = @Content),
        @ApiResponse(responseCode = "404", description = "코드를 찾을 수 없음", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @PostMapping("/allocate")
  public ApiResponseDto<ApiResponseDto.ResponseObj> labelAllocate(
      @RequestBody @Valid LabelAllocateDto.AllocateDto dto) {

    int compareYyyy = Integer.parseInt(dto.getYyyy().split("-")[0]);
    int targetYyyy = Integer.parseInt(dto.getYyyy().split("-")[1]);
    return ApiResponseDto.okObject(
        labelAllocateService.allocateAsc(
            dto.getStage(), dto.getLabelers(), dto.getInspectors(), compareYyyy, targetYyyy));
  }

  @Operation(summary = "작업현황 관리 > 변화탐지 회차 정보", description = "작업현황 관리 > 변화탐지 회차 정보")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "조회 성공"),
        @ApiResponse(responseCode = "404", description = "데이터를 찾을 수 없음"),
        @ApiResponse(responseCode = "500", description = "서버 오류")
      })
  @GetMapping("/stage-detail")
  public ApiResponseDto<InferenceDetail> findInferenceDetail(
      @Parameter(
              description = "회차 마스터 key",
              required = true,
              example = "8584e8d4-53b3-4582-bde2-28a81495a626")
          @RequestParam
          String uuid) {
    return ApiResponseDto.ok(labelAllocateService.findInferenceDetail(uuid));
  }

  @Operation(
      summary = "작업현황 관리 > 라벨러/검수자 상세 정보, 작업이관 팝업 내 라벨러 상세 정보 동일",
      description = "작업현황 관리 > 라벨러/검수자 상세 정보, 작업이관 팝업 내 라벨러 상세 정보 동일")
  @GetMapping("/user-detail")
  public ApiResponseDto<LabelerDetail> findUserDetail(
      @RequestParam(defaultValue = "01022223333", required = true) String userId,
      @Parameter(
              description = "회차 마스터 key",
              required = true,
              example = "8584e8d4-53b3-4582-bde2-28a81495a626")
          @RequestParam
          String uuid,
      @Schema(
              allowableValues = {"LABELER", "REVIEWER"},
              defaultValue = "LABELER")
          @Parameter(description = "라벨러/검수자(LABELER/REVIEWER)", required = true)
          @RequestParam
          String type) {
    return ApiResponseDto.ok(labelAllocateService.findUserDetail(userId, uuid, type));
  }

  @Operation(summary = "작업현황 관리 > 상세 > 작업 이관", description = "작업현황 관리 > 상세 > 작업 이관")
  @ApiResponses(
      value = {
        @ApiResponse(
            responseCode = "201",
            description = "등록 성공",
            content =
                @Content(
                    mediaType = "application/json",
                    schema = @Schema(implementation = Long.class))),
        @ApiResponse(responseCode = "400", description = "잘못된 요청 데이터", content = @Content),
        @ApiResponse(responseCode = "404", description = "코드를 찾을 수 없음", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @PostMapping("/allocate-move")
  public ApiResponseDto<ApiResponseDto.ResponseObj> labelAllocateMove(
      @io.swagger.v3.oas.annotations.parameters.RequestBody(
              description = "라벨링 이관",
              required = true,
              content =
                  @Content(
                      mediaType = "application/json",
                      schema = @Schema(implementation = LabelAllocateDto.AllocateMoveDto.class)))
          @RequestBody
          LabelAllocateDto.AllocateMoveDto dto) {

    return ApiResponseDto.okObject(
        labelAllocateService.allocateMove(
            dto.getTotalCnt(), dto.getUuid(), dto.getLabelers(), dto.getUserId()));
  }

  @Operation(
      summary = "라벨링작업 관리 > 상세 > 라벨러/검수자 일별 작업량 목록",
      description = "라벨링작업 관리 > 상세 > 라벨러/검수자 일별 작업량 목록")
  @GetMapping("/daily-list")
  public ApiResponseDto<Page<LabelingStatDto>> findDaliyList(
      @RequestParam(defaultValue = "0", required = true) int page,
      @RequestParam(defaultValue = "20", required = true) int size,
      @Parameter(
              description = "회차 마스터 key",
              required = true,
              example = "8584e8d4-53b3-4582-bde2-28a81495a626")
          @RequestParam
          String uuid,
      @Parameter(description = "사번", required = true, example = "123456") @RequestParam
          String userId,
      @Schema(
              allowableValues = {"LABELER", "REVIEWER"},
              defaultValue = "LABELER")
          @Parameter(description = "라벨러/검수자(LABELER/REVIEWER)", required = true)
          @RequestParam
          String type) {
    LabelAllocateDto.searchReq searchReq = new LabelAllocateDto.searchReq(page, size, "");
    return ApiResponseDto.ok(labelAllocateService.findDaliyList(searchReq, uuid, userId, type));
  }

  @Operation(summary = "이관 가능한 사용자 목록 조회", description = "이관 가능한 사용자 목록 조회")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "조회 성공"),
        @ApiResponse(responseCode = "404", description = "코드를 찾을 수 없음"),
        @ApiResponse(responseCode = "500", description = "서버 오류")
      })
  @GetMapping("/move-user")
  public ApiResponseDto<LabelAllocateDto.MoveInfo> availMoveUserList(
      @Parameter(description = "해당 사용자 사번", example = "01022223333") @RequestParam String userId,
      @Parameter(description = "회차 마스터 key", example = "f97dc186-e6d3-4645-9737-3173dde8dc64")
          @RequestParam
          String uuid) {
    return ApiResponseDto.ok(labelAllocateService.moveAvailUserList(userId, uuid));
  }

  @Operation(
      summary = "작업현황 관리 > 라벨링/검수 종료 여부 업데이트",
      description = "라벨링/검수 종료 여부를 업데이트합니다. uuid 생략 시 최신 프로젝트 대상")
  @ApiResponses(
      value = {
        @ApiResponse(responseCode = "200", description = "업데이트 성공"),
        @ApiResponse(responseCode = "400", description = "잘못된 요청 데이터", content = @Content),
        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
      })
  @PostMapping("/projectinfo/closed")
  public ApiResponseDto<ApiResponseDto.ResponseObj> updateClosedYn(
      @io.swagger.v3.oas.annotations.parameters.RequestBody(
              description = "종료 여부 업데이트 요청",
              required = true,
              content =
                  @Content(
                      mediaType = "application/json",
                      schema = @Schema(implementation = UpdateClosedRequest.class),
                      examples = {
                        @io.swagger.v3.oas.annotations.media.ExampleObject(
                            name = "특정 프로젝트 라벨링 전체 종료",
                            value =
                                """
                {"uuid": "f97dc186-e6d3-4645-9737-3173dde8dc64", "closedType": "LABELING", "closedYn": "Y"}
                """),
                        @io.swagger.v3.oas.annotations.media.ExampleObject(
                            name = "특정 프로젝트 검수 전체 종료",
                            value =
                                """
                {"uuid": "f97dc186-e6d3-4645-9737-3173dde8dc64", "closedType": "INSPECTION", "closedYn": "Y"}
            """)
                      }))
          @RequestBody
          @Valid
          UpdateClosedRequest request) {

    labelAllocateService.updateClosedYn(
        request.getUuid(), request.getClosedType(), request.getClosedYn());

    String typeLabel = "LABELING".equals(request.getClosedType()) ? "라벨링" : "검수";
    String statusMessage =
        "Y".equals(request.getClosedYn())
            ? typeLabel + "이(가) 종료되었습니다."
            : typeLabel + "이(가) 재개되었습니다.";

    return ApiResponseDto.okObject(
        new ApiResponseDto.ResponseObj(ApiResponseDto.ApiResponseCode.OK, statusMessage));
  }

  @Operation(summary = "라벨링작업 관리 > 상세 > 작업이력", description = "라벨링작업 관리 > 상세 > 작업이력")
  @GetMapping("/work-history-list")
  public ApiResponseDto<Page<WorkHistoryDto>> findWorkHistoryList(
      @RequestParam(defaultValue = "0", required = true) int page,
      @RequestParam(defaultValue = "20", required = true) int size,
      @Parameter(description = "사번", required = true, example = "123456") @RequestParam
          String userId,
      @Schema(
              allowableValues = {"LABELER", "REVIEWER"},
              defaultValue = "LABELER")
          @Parameter(description = "라벨러/검수자(LABELER/REVIEWER)", required = true)
          @RequestParam
          String type) {
    LabelAllocateDto.searchReq searchReq = new LabelAllocateDto.searchReq(page, size, "");
    return ApiResponseDto.ok(labelAllocateService.findWorkHistoryList(searchReq, userId, type));
  }
}