package com.kt.ai.controller;

import com.kt.ai.model.dto.response.HealthCheckResponse;
import com.kt.ai.model.enums.CircuitBreakerState;
import com.kt.ai.model.enums.ServiceStatus;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.time.LocalDateTime;

/**
 * 헬스체크 Controller
 *
 * @author AI Service Team
 * @since 1.0.0
 */
@Slf4j
@Tag(name = "Health Check", description = "서비스 상태 확인")
@RestController
public class HealthController {

    @Autowired(required = false)
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 서비스 헬스체크
     */
    @Operation(summary = "서비스 헬스체크", description = "AI Service 상태 및 외부 연동 확인")
    @GetMapping("/health")
    public ResponseEntity<HealthCheckResponse> healthCheck() {
        // Redis 상태 확인
        ServiceStatus redisStatus = checkRedis();

        // 전체 서비스 상태 (Redis가 DOWN이면 DEGRADED, UNKNOWN이면 UP으로 처리)
        ServiceStatus overallStatus;
        if (redisStatus == ServiceStatus.DOWN) {
            overallStatus = ServiceStatus.DEGRADED;
        } else {
            overallStatus = ServiceStatus.UP;
        }

        HealthCheckResponse.Services services = HealthCheckResponse.Services.builder()
                .kafka(ServiceStatus.UP)  // TODO: 실제 Kafka 상태 확인
                .redis(redisStatus)
                .claudeApi(ServiceStatus.UP)  // TODO: 실제 Claude API 상태 확인
                .gpt4Api(ServiceStatus.UP)  // TODO: 실제 GPT-4 API 상태 확인 (선택)
                .circuitBreaker(CircuitBreakerState.CLOSED)  // TODO: 실제 Circuit Breaker 상태 확인
                .build();

        HealthCheckResponse response = HealthCheckResponse.builder()
                .status(overallStatus)
                .timestamp(LocalDateTime.now())
                .services(services)
                .build();

        return ResponseEntity.ok(response);
    }

    /**
     * Redis 연결 상태 확인
     */
    private ServiceStatus checkRedis() {
        // RedisTemplate이 주입되지 않은 경우 (로컬 환경 등)
        if (redisTemplate == null) {
            log.warn("RedisTemplate이 주입되지 않았습니다. Redis 상태를 UNKNOWN으로 표시합니다.");
            return ServiceStatus.UNKNOWN;
        }

        try {
            log.debug("Redis 연결 테스트 시작...");
            String pong = redisTemplate.getConnectionFactory().getConnection().ping();
            log.info("✅ Redis 연결 성공! PING 응답: {}", pong);
            return ServiceStatus.UP;
        } catch (Exception e) {
            log.error("❌ Redis 연결 실패", e);
            log.error("상세 오류 정보:");
            log.error("  - 오류 타입: {}", e.getClass().getName());
            log.error("  - 오류 메시지: {}", e.getMessage());
            if (e.getCause() != null) {
                log.error("  - 원인: {}", e.getCause().getMessage());
            }
            return ServiceStatus.DOWN;
        }
    }
}
