package com.example.mailservice.controller;

import com.example.mailservice.dto.ApiResponse;
import com.example.mailservice.dto.MailRequest;
import com.example.mailservice.service.MailLimitService;
import com.example.mailservice.service.MailService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
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.RestController;

import javax.servlet.http.HttpServletRequest;

/**
 * 邮件发送控制器
 */
@RestController
@RequestMapping("/api/mail")
public class MailController {

    @Autowired
    private MailService mailService;

    @Autowired
    private MailLimitService mailLimitService;

    /**
     * 发送邮件接口
     */
    @PostMapping("/send")
    public ApiResponse sendMail(@Validated @RequestBody MailRequest mailRequest, HttpServletRequest request) {
        // 获取客户端IP地址
        String ipAddress = getClientIpAddress(request);
        
        // 检查是否达到发送限制
        if (!mailLimitService.canSendMail(ipAddress)) {
            return ApiResponse.fail(403, "您的IP今日发送邮件已达上限，请明天再试");
        }
        
        // 发送邮件
        boolean sendSuccess = mailService.sendSimpleMail(
                mailRequest.getName(),
                mailRequest.getEmail(),
                mailRequest.getSubject(), 
                mailRequest.getMessage()
        );
        
        if (sendSuccess) {
            // 增加发送计数
            long count = mailLimitService.incrementSendCount(ipAddress);
            int remaining = mailLimitService.getRemainingCount(ipAddress);
            return ApiResponse.success("邮件发送成功，今日剩余发送次数: " + remaining);
        } else {
            return ApiResponse.fail(500, "邮件发送失败，请稍后重试");
        }
    }

    /**
     * 获取客户端IP地址
     */
    private String getClientIpAddress(HttpServletRequest request) {
        String xForwardedForHeader = request.getHeader("X-Forwarded-For");
        if (xForwardedForHeader != null) {
            // X-Forwarded-For: client, proxy1, proxy2
            return xForwardedForHeader.split(",")[0].trim();
        }
        return request.getRemoteAddr();
    }
}