package kr.re.etri.autoflow.controllers;

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.Content;
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 kr.re.etri.autoflow.service.DatasetService;
import lombok.RequiredArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.util.Map;

@Tag(name = "Dataset API", description = "외부 데이터셋 목록 조회 API (Python requests 코드 기반)")
@RestController
@RequestMapping("/api/datasets")
@RequiredArgsConstructor
public class ExternalDataSetController {

    private final DatasetService datasetService;

    @Operation(
            summary = "데이터셋 목록 조회",
            description = "외부 API(/export/dataset_list)를 호출하여 데이터셋 목록을 조회합니다."
    )
    @ApiResponses({
            @ApiResponse(responseCode = "200", description = "성공적으로 데이터셋을 조회했습니다."),
            @ApiResponse(responseCode = "400", description = "잘못된 요청입니다.", content = @Content),
            @ApiResponse(responseCode = "500", description = "서버 내부 오류", content = @Content)
    })
    @GetMapping("/list")
    public ResponseEntity<?> getDatasetList(
            @Parameter(description = "프로젝트 인덱스", example = "10")
            @RequestParam(required = false) Integer ds_prj_idx,
            @Parameter(description = "검색 키워드", example = "traffic")
            @RequestParam(required = false) String search_keyword,
            @Parameter(description = "그룹 이름", example = "AIGroup")
            @RequestParam(required = false) String grp_name
    ) {
        Map<String, Object> result = datasetService.getDatasetList(ds_prj_idx, search_keyword, grp_name);
        return ResponseEntity.ok(result);
    }

    @PostMapping("/download")
    @Operation(
            summary = "데이터셋 다운로드",
            description = "외부 API 서버로 POST 요청을 보내 ZIP 파일을 다운로드합니다.",
            parameters = {
                    @Parameter(name = "datasetName", description = "다운로드할 데이터셋 이름", in = ParameterIn.QUERY, required = true)
            }
    )
    public ResponseEntity<?> downloadDataset(
            @RequestParam("datasetName") String datasetName
    ) {
        return datasetService.downloadDataset(datasetName);
    }
}