yggdrasil/src/api/comments/helpers.rs

//! 评论模块的辅助函数：数据转换、校验、哈希与头像生成。
//!
//! 所有工具函数仅在 `feature = "server"` 启用的服务端构建中使用。

#![allow(clippy::unused_unit, deprecated)]

#[cfg(feature = "server")]
use crate::models::comment::{AdminComment, CommentStatus, PublicComment};

/// 计算字符串的 MD5 哈希，用于 Gravatar。
#[cfg(feature = "server")]
pub fn md5_hash(input: &str) -> String {
    use md5::Digest;
    let hash = md5::Md5::digest(input.as_bytes());
    hex::encode(hash)
}

/// 对用于 HTML 展示的文本做基础转义，防止 XSS。
#[cfg(feature = "server")]
pub fn escape_html(input: &str) -> String {
    input
        .replace('&', "&")
        .replace('<', "&lt;")
        .replace('>', "&gt;")
        .replace('"', "&quot;")
        .replace('\'', "&#x27;")
}

/// 根据邮箱生成 Cravatar（Gravatar 国内镜像）头像 URL。
#[cfg(feature = "server")]
pub fn gravatar_url(email: &str) -> String {
    let hash = md5_hash(&email.trim().to_lowercase());
    format!("https://cravatar.cn/avatar/{}?d=mp&s=80", hash)
}

/// 将数据库行转换为前端展示的公开评论结构。
#[cfg(feature = "server")]
pub fn row_to_public_comment(row: &tokio_postgres::Row) -> PublicComment {
    let email: String = row.get("author_email");
    let created_at_dt: chrono::DateTime<chrono::Utc> = row.get("created_at");
    let created_at_iso = created_at_dt.to_rfc3339();
    let created_at_relative = format_relative_time(created_at_dt);

    PublicComment {
        id: row.get("id"),
        parent_id: row.get("parent_id"),
        depth: row.get("depth"),
        author_name: row.get("author_name"),
        author_url: row.get("author_url"),
        avatar_url: gravatar_url(&email),
        content_html: row.get("content_html"),
        created_at: created_at_relative,
        created_at_iso,
    }
}

/// 将数据库行转换为后台管理使用的评论结构。
#[cfg(feature = "server")]
pub fn row_to_admin_comment(row: &tokio_postgres::Row) -> AdminComment {
    let status_str: String = row.get("status");
    let email: String = row.get("author_email");

    AdminComment {
        id: row.get("id"),
        post_id: row.get("post_id"),
        post_title: row.get("post_title"),
        post_slug: row.get("post_slug"),
        parent_id: row.get("parent_id"),
        depth: row.get("depth"),
        author_name: row.get("author_name"),
        author_email: email.clone(),
        author_url: row.get("author_url"),
        avatar_url: gravatar_url(&email),
        content_md: row.get("content_md"),
        status: CommentStatus::from_str(&status_str),
        created_at: row.get("created_at"),
    }
}

/// 将 UTC 时间格式化为相对时间（刚刚 / N 分钟前 / N 小时前 / N 天前 / 日期）。
#[cfg(feature = "server")]
pub fn format_relative_time(dt: chrono::DateTime<chrono::Utc>) -> String {
    let now = chrono::Utc::now();
    let diff = now.signed_duration_since(dt);

    if diff.num_seconds() < 60 {
        "刚刚".to_string()
    } else if diff.num_minutes() < 60 {
        format!("{} 分钟前", diff.num_minutes())
    } else if diff.num_hours() < 24 {
        format!("{} 小时前", diff.num_hours())
    } else if diff.num_days() < 30 {
        format!("{} 天前", diff.num_days())
    } else {
        dt.format("%Y-%m-%d").to_string()
    }
}

/// 校验评论作者昵称：非空且不超过 50 字符。
#[cfg(feature = "server")]
pub fn validate_comment_name(name: &str) -> Result<(), String> {
    let trimmed = name.trim();
    if trimmed.is_empty() {
        return Err("请输入昵称".to_string());
    }
    if trimmed.len() > 50 {
        return Err("昵称长度不能超过 50 个字符".to_string());
    }
    Ok(())
}

#[cfg(feature = "server")]
static EMAIL_REGEX: std::sync::LazyLock<regex::Regex> = std::sync::LazyLock::new(|| {
    regex::Regex::new(r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$").unwrap()
});

/// 校验评论作者邮箱格式。
#[cfg(feature = "server")]
pub fn validate_comment_email(email: &str) -> Result<(), String> {
    if !EMAIL_REGEX.is_match(email.trim()) {
        return Err("邮箱格式不正确".to_string());
    }
    Ok(())
}

/// 校验评论作者网址：为空时允许，非空时必须以 http:// 或 https:// 开头、
/// 不含 HTML 特殊字符与空白，且不超过 200 字符。
#[cfg(feature = "server")]
pub fn validate_comment_url(url: &str) -> Result<(), String> {
    let trimmed = url.trim();
    if trimmed.is_empty() {
        return Ok(());
    }
    let lower = trimmed.to_ascii_lowercase();
    if !lower.starts_with("http://") && !lower.starts_with("https://") {
        return Err("网址必须以 http:// 或 https:// 开头".to_string());
    }
    if trimmed.len() > 200 {
        return Err("网址长度不能超过 200 个字符".to_string());
    }
    if trimmed.chars().any(|c| matches!(c, '<' | '>' | '"' | '\'' | '&' | ' ' | '\t' | '\n' | '\r'))
    {
        return Err("网址包含非法字符".to_string());
    }
    Ok(())
}

/// 校验评论内容：非空且不超过 10000 字符。
#[cfg(feature = "server")]
pub fn validate_comment_content(content: &str) -> Result<(), String> {
    let trimmed = content.trim();
    if trimmed.is_empty() {
        return Err("请输入评论内容".to_string());
    }
    if trimmed.len() > 10000 {
        return Err("评论内容不能超过 10000 个字符".to_string());
    }
    Ok(())
}

/// 计算评论内容哈希，用于检测短时间内的重复提交。
#[cfg(feature = "server")]
pub fn compute_content_hash(
    post_id: i32,
    parent_id: Option<i64>,
    name: &str,
    content: &str,
) -> String {
    use sha2::Digest;
    let input = format!(
        "{}:{}:{}:{}",
        post_id,
        parent_id.map(|id| id.to_string()).unwrap_or_default(),
        name.trim(),
        content.trim()
    );
    let hash = sha2::Sha256::digest(input.as_bytes());
    hex::encode(hash)
}

#[cfg(all(test, feature = "server"))]
mod tests {
    use super::*;

    #[test]
    fn escape_html_escapes_special_chars() {
        assert_eq!(
            escape_html("<script>alert(1)</script>"),
            "&lt;script&gt;alert(1)&lt;/script&gt;"
        );
        assert_eq!(
            escape_html("\"quoted' & ampersand"),
            "&quot;quoted&#x27; &amp; ampersand"
        );
    }

    #[test]
    fn md5_hash_known_value() {
        assert_eq!(md5_hash("hello"), "5d41402abc4b2a76b9719d911017c592");
    }

    #[test]
    fn md5_hash_empty() {
        assert_eq!(md5_hash(""), "d41d8cd98f00b204e9800998ecf8427e");
    }

    #[test]
    fn gravatar_url_format() {
        let url = gravatar_url("test@example.com");
        assert!(url.starts_with("https://cravatar.cn/avatar/"));
        assert!(url.contains("?d=mp&s=80"));
    }

    #[test]
    fn gravatar_url_normalizes_email() {
        let url1 = gravatar_url("Test@Example.com");
        let url2 = gravatar_url("test@example.com");
        assert_eq!(url1, url2);
    }

    #[test]
    fn gravatar_url_trims_whitespace() {
        let url1 = gravatar_url(" test@example.com ");
        let url2 = gravatar_url("test@example.com");
        assert_eq!(url1, url2);
    }

    #[test]
    fn format_relative_time_just_now() {
        let now = chrono::Utc::now();
        assert_eq!(format_relative_time(now), "刚刚");
    }

    #[test]
    fn format_relative_time_minutes() {
        let dt = chrono::Utc::now() - chrono::Duration::minutes(5);
        assert_eq!(format_relative_time(dt), "5 分钟前");
    }

    #[test]
    fn format_relative_time_hours() {
        let dt = chrono::Utc::now() - chrono::Duration::hours(3);
        assert_eq!(format_relative_time(dt), "3 小时前");
    }

    #[test]
    fn format_relative_time_days() {
        let dt = chrono::Utc::now() - chrono::Duration::days(7);
        assert_eq!(format_relative_time(dt), "7 天前");
    }

    #[test]
    fn format_relative_time_one_minute() {
        let dt = chrono::Utc::now() - chrono::Duration::minutes(1);
        assert_eq!(format_relative_time(dt), "1 分钟前");
    }

    #[test]
    fn format_relative_time_one_hour() {
        let dt = chrono::Utc::now() - chrono::Duration::hours(1);
        assert_eq!(format_relative_time(dt), "1 小时前");
    }

    #[test]
    fn format_relative_time_one_day() {
        let dt = chrono::Utc::now() - chrono::Duration::days(1);
        assert_eq!(format_relative_time(dt), "1 天前");
    }

    #[test]
    fn format_relative_time_old_date() {
        let dt = chrono::Utc::now() - chrono::Duration::days(60);
        let result = format_relative_time(dt);
        assert!(result.contains('-'));
        assert_eq!(result.len(), 10);
    }

    #[test]
    fn validate_comment_name_valid() {
        assert!(validate_comment_name("Alice").is_ok());
        assert!(validate_comment_name("张三").is_ok());
    }

    #[test]
    fn validate_comment_name_empty() {
        assert!(validate_comment_name("").is_err());
        assert!(validate_comment_name("   ").is_err());
    }

    #[test]
    fn validate_comment_name_too_long() {
        assert!(validate_comment_name(&"a".repeat(51)).is_err());
    }

    #[test]
    fn validate_comment_name_max_length() {
        assert!(validate_comment_name(&"a".repeat(50)).is_ok());
    }

    #[test]
    fn validate_comment_email_valid() {
        assert!(validate_comment_email("user@example.com").is_ok());
        assert!(validate_comment_email("a.b+c@domain.co").is_ok());
    }

    #[test]
    fn validate_comment_email_invalid() {
        assert!(validate_comment_email("notanemail").is_err());
        assert!(validate_comment_email("@domain.com").is_err());
        assert!(validate_comment_email("user@").is_err());
    }

    #[test]
    fn validate_comment_url_valid() {
        assert!(validate_comment_url("http://example.com").is_ok());
        assert!(validate_comment_url("https://example.com/path").is_ok());
    }

    #[test]
    fn validate_comment_url_empty_is_ok() {
        assert!(validate_comment_url("").is_ok());
        assert!(validate_comment_url("   ").is_ok());
    }

    #[test]
    fn validate_comment_url_invalid_scheme() {
        assert!(validate_comment_url("ftp://example.com").is_err());
        assert!(validate_comment_url("javascript:alert(1)").is_err());
    }

    #[test]
    fn validate_comment_url_uppercase_scheme() {
        assert!(validate_comment_url("HTTP://example.com").is_ok());
        assert!(validate_comment_url("HTTPS://example.com").is_ok());
        assert!(validate_comment_url("Http://example.com").is_ok());
    }

    #[test]
    fn validate_comment_url_fragment() {
        assert!(validate_comment_url("https://example.com#section").is_ok());
    }

    #[test]
    fn validate_comment_url_relative_path_rejected() {
        assert!(validate_comment_url("/path/to/page").is_err());
        assert!(validate_comment_url("path/to/page").is_err());
    }

    #[test]
    fn validate_comment_url_too_long() {
        let long_url = format!("https://example.com/{}", "a".repeat(200));
        assert!(validate_comment_url(&long_url).is_err());
    }

    #[test]
    fn validate_comment_content_valid() {
        assert!(validate_comment_content("Hello world").is_ok());
    }

    #[test]
    fn validate_comment_content_empty() {
        assert!(validate_comment_content("").is_err());
        assert!(validate_comment_content("   ").is_err());
    }

    #[test]
    fn validate_comment_content_too_long() {
        assert!(validate_comment_content(&"a".repeat(10001)).is_err());
    }

    #[test]
    fn validate_comment_content_max_length() {
        assert!(validate_comment_content(&"a".repeat(10000)).is_ok());
    }

    #[test]
    fn compute_content_hash_deterministic() {
        let h1 = compute_content_hash(1, None, "Alice", "Hello");
        let h2 = compute_content_hash(1, None, "Alice", "Hello");
        assert_eq!(h1, h2);
    }

    #[test]
    fn compute_content_hash_different_inputs() {
        let h1 = compute_content_hash(1, None, "Alice", "Hello");
        let h2 = compute_content_hash(2, None, "Alice", "Hello");
        assert_ne!(h1, h2);
    }

    #[test]
    fn compute_content_hash_trims_whitespace() {
        let h1 = compute_content_hash(1, None, "Alice", "Hello");
        let h2 = compute_content_hash(1, None, " Alice ", " Hello ");
        assert_eq!(h1, h2);
    }

    #[test]
    fn compute_content_hash_64_hex_chars() {
        let h = compute_content_hash(1, None, "Alice", "Hello");
        assert_eq!(h.len(), 64);
        assert!(h.chars().all(|c| c.is_ascii_hexdigit()));
    }
}