lyrics-search 0.5.0

//! HTTP 请求辅助。
//!
//! 对应 C# `BaseApi`：每个请求都单独构造请求头（`CreateRequest`），
//! 请求头不会在 Provider 之间共享或残留。
//!
//! 请求分两步：先由 [`send`] / [`send_json`] / [`send_form`] 发送并拿回原始响应，
//! 再由 [`json`] / [`text`] 解码响应体。所有失败都返回 [`SearchError`]，
//! 不再像 0.2.0 那样把 reqwest / serde 的错误压成 `None`。

use reqwest::{Client, RequestBuilder};
use serde::{Serialize, de::DeserializeOwned};
use std::sync::LazyLock;
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use crate::error::SearchError;

// 公开签名里出现的 reqwest 类型一并再导出，调用方不必自己依赖 reqwest。
pub use reqwest::{Method, Response, StatusCode};

/// 默认 User-Agent，对应 C# `BaseApi.UserAgent`。
///
/// 该值作为 HTTP 客户端的默认请求头，Provider 可在单次请求中自行覆盖。
pub const USER_AGENT: &str = "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36";

/// 网易云音乐默认 Cookie，对应 C# `BaseApi.Cookie`。
pub const COOKIE: &str = "os=pc;osver=Microsoft-Windows-10-Professional-build-16299.125-64bit;appver=2.0.3.131777;channel=netease;__remember_me=true";

/// 单次请求的总时长上限。
///
/// reqwest 默认不设上限，卡住的连接会一直挂着；AMLL TTML DB 下载索引时还持有锁，
/// 一条挂住的请求会让之后的搜索全部跟着等。一分钟够国内直连 GitHub 下完 1.6 MB 的索引。
const REQUEST_TIMEOUT: Duration = Duration::from_secs(60);

static HTTP_CLIENT: LazyLock<Client> = LazyLock::new(|| {
    Client::builder()
        .user_agent(USER_AGENT)
        .timeout(REQUEST_TIMEOUT)
        .build()
        .expect("Failed to create HTTP client")
});

/// 当前 Unix 时间戳（秒）。
pub fn unix_seconds() -> u64 {
    since_unix_epoch().as_secs()
}

/// 当前 Unix 时间戳（毫秒）。
pub fn unix_millis() -> u128 {
    since_unix_epoch().as_millis()
}

/// 系统时钟早于 1970 年时签名与时间戳参数全都无效，静默返回 0 只会让请求在服务端
/// 莫名失败，因此直接视为环境错误。
fn since_unix_epoch() -> Duration {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .expect("系统时钟早于 Unix 纪元")
}

/// 发送不带请求体的请求，返回原始响应。
pub async fn send(
    method: Method,
    url: &str,
    headers: &[(&str, &str)],
) -> Result<Response, SearchError> {
    send_request(method, url, headers, |request| request).await
}

/// 发送 JSON POST 请求，返回原始响应。
pub async fn send_json(
    url: &str,
    body: &impl Serialize,
    headers: &[(&str, &str)],
) -> Result<Response, SearchError> {
    send_request(Method::POST, url, headers, |request| request.json(body)).await
}

/// 发送表单 POST 请求，返回原始响应。
pub async fn send_form(
    url: &str,
    form: &[(&str, &str)],
    headers: &[(&str, &str)],
) -> Result<Response, SearchError> {
    send_request(Method::POST, url, headers, |request| request.form(form)).await
}

/// 将响应体反序列化为指定类型。
///
/// 状态码非 2xx 时返回 [`SearchError::Status`]，不会去解析错误页正文。
pub async fn json<T: DeserializeOwned>(response: Response) -> Result<T, SearchError> {
    ensure_success(&response)?;
    Ok(response.json::<T>().await?)
}

/// 读取响应体文本。
///
/// 状态码非 2xx 时返回 [`SearchError::Status`]；部分平台（如 QQ 音乐）会在 2xx 里
/// 返回带业务错误码的 JSON，这类正文仍会原样交给调用方。
pub async fn text(response: Response) -> Result<String, SearchError> {
    ensure_success(&response)?;
    Ok(response.text().await?)
}

/// 构造请求、附加请求头并发送。
async fn send_request(
    method: Method,
    url: &str,
    headers: &[(&str, &str)],
    body: impl FnOnce(RequestBuilder) -> RequestBuilder,
) -> Result<Response, SearchError> {
    let mut request = HTTP_CLIENT.request(method, url);
    for (key, value) in headers {
        request = request.header(*key, *value);
    }
    Ok(body(request).send().await?)
}

/// 状态码非 2xx 时构造 [`SearchError::Status`]。
fn ensure_success(response: &Response) -> Result<(), SearchError> {
    let status = response.status();
    if status.is_success() {
        Ok(())
    } else {
        Err(SearchError::Status(status.as_u16()))
    }
}