<?php
namespace app\middleware;

use Closure;
use think\App;
use think\Cache;
use think\facade\Request;
use think\facade\Response;

class RateLimit
{
    protected $app;
    protected $cache;

    /**
     * 构造函数：注入应用实例和缓存实例
     * @param App $app
     * @param Cache $cache
     */
    public function __construct(App $app, Cache $cache)
    {
        $this->app = $app;
        $this->cache = $cache;
    }

    /**
     * 中间件处理逻辑
     * @param $request 请求对象
     * @param Closure $next 下一个中间件/控制器
     * @param int $window 时间窗口（秒），默认60秒
     * @param int $limit 窗口内最大请求数，默认60次
     * @return Response
     */
    public function handle($request, Closure $next, int $window = 60, int $limit = 60)
    {
        // 获取限流唯一标识（用户ID优先，无则用IP）
        $key = $this->getLimitKey();
        
        // 从缓存获取当前请求计数（默认0）
        $count = $this->cache->get($key, 0);
        
        // 超过阈值则返回限流响应
        if ($count >= $limit) {
            return Response::create([
                'code' => 429,
                'msg' => "请求过于频繁，请 {$window} 秒后再试",
                'data' => []
            ])->header(['Retry-After' => $window]); // 告知客户端重试时间
        }

        // 首次请求：初始化缓存并设置过期时间
        if ($count === 0) {
            $this->cache->set($key, 1, $window);
        } else {
            // 非首次请求：计数+1
            $this->cache->inc($key);
        }

        // 放行请求到下一个中间件/控制器
        return $next($request);
    }

    /**
     * 生成限流唯一标识（用户维度优先，保障登录用户体验）
     * @return string
     */
    protected function getLimitKey()
    {
        $ip = Request::ip(); // 获取客户端IP（支持XFF头解析）
        $userId = $this->app->session->get('user_id'); // 获取登录用户ID

        return $userId ? "rate_limit:user:{$userId}" : "rate_limit:ip:{$ip}";
    }
}