<?php
function naive_retry_api_call($url, $args, $retries = 3) {
    for ($i = 0; $i < $retries; $i++) {
        $response = wp_remote_get($url, $args);
        if (!is_wp_error($response) && wp_remote_retrieve_response_code($response) === 200) {
            return $response; // 成功了，直接回傳
        }

        // 如果是 429 錯誤，就等一下
        if (wp_remote_retrieve_response_code($response) === 429) {
            sleep(5); // 不管三七二十一，就是睡 5 秒
        }
    }
    return new WP_Error('api_failed', 'API request failed after multiple retries.');
}
?>

<?php
/**
 * 使用指數退讓和抖動策略來進行穩健的 API 請求。
 *
 * @param string $url API 的 URL。
 * @param array $args wp_remote_get 的參數。
 * @param int $max_retries 最大重試次數。
 * @param int $initial_delay 初始延遲時間（毫秒）。
 * @return array|WP_Error 成功時回傳 response，失敗時回傳 WP_Error。
 */
function robust_api_call($url, $args = [], $max_retries = 5, $initial_delay = 1000) {
    for ($attempt = 0; $attempt < $max_retries; $attempt++) {
        $response = wp_remote_get($url, $args);

        if (is_wp_error($response)) {
            // 處理 cURL 錯誤等網路問題
            // ... log error ...
            continue; // 繼續嘗試
        }

        $status_code = wp_remote_retrieve_response_code($response);

        if ($status_code >= 200 && $status_code < 300) {
            return $response; // 請求成功！
        }

        if ($status_code === 429 || $status_code >= 500) {
            // 遇到速率限制或伺服器端錯誤，準備重試
            $headers = wp_remote_retrieve_headers($response);
            
            if (isset($headers['retry-after'])) {
                // 如果 API 有明確指示，就聽它的
                $delay_seconds = (int) $headers['retry-after'];
                sleep($delay_seconds);
                continue;
            }
            
            // 計算指數退讓時間
            $delay = $initial_delay * pow(2, $attempt);
            
            // 加入抖動 (Jitter)，這裡用的是 Full Jitter
            $jitter = mt_rand(0, $delay);
            
            // 總等待時間（轉為微秒 usleep）
            $wait_time = ($delay + $jitter) * 1000; 
            usleep($wait_time);

        } else {
            // 其他客戶端錯誤 (如 400, 401, 403)，通常重試也沒用
            return new WP_Error('client_error', 'Client error: ' . $status_code, ['status' => $status_code]);
        }
    }

    return new WP_Error('max_retries_exceeded', 'API request failed after ' . $max_retries . ' attempts.', ['status' => $status_code]);
}
?>

<?php
add_filter('rest_pre_dispatch', 'my_custom_api_rate_limiter', 10, 3);

function my_custom_api_rate_limiter($result, $server, $request) {
    // 只針對我們自訂的某個 API 路由
    if (strpos($request->get_route(), '/my-custom-namespace/v1/') === false) {
        return $result;
    }

    $limit = 100; // 每小時最多 100 次請求
    $ip = $_SERVER['REMOTE_ADDR'];
    $transient_key = 'rate_limit_' . $ip;

    $request_count = get_transient($transient_key);

    if ($request_count === false) {
        // 第一次請求，設定計數器為 1，過期時間為 1 小時
        set_transient($transient_key, 1, HOUR_IN_SECONDS);
    } elseif ($request_count >= $limit) {
        // 超過限制，回傳 429 錯誤
        return new WP_Error(
            'rest_too_many_requests',
            'Too many requests. Please try again later.',
            ['status' => 429, 'headers' => ['Retry-After' => HOUR_IN_SECONDS - (time() - (get_option('_transient_timeout_' . $transient_key) - HOUR_IN_SECONDS))]]
        );
    } else {
        // 還在額度內，計數器 +1
        set_transient($transient_key, $request_count + 1, HOUR_IN_SECONDS);
    }

    return $result;
}
?>