tdm_server_rust/utils/
cos_cdn.rs

1//! 腾讯云 CDN TypeD 鉴权链接生成
2//!
3//! 对齐 Java `OssServiceImpl.generateCdnUrl`。
4//! 生成带时间戳和 MD5 签名的 CDN 鉴权下载链接，防止盗链。
5//!
6//! ## 鉴权算法
7//!
8//! TypeD 鉴权：`sign = md5(key + encoded_path + timestamp)`
9//! URL 格式：`https://{domain}{path}?sign={sign}&t={ts}`
10//!
11//! ## 配置要求
12//!
13//! 需要 `tencent.cdn_domain` 和 `tencent.cdn_key` 均已配置且非空，
14//! 否则返回 `None`（降级为无鉴权访问）。
15
16use std::time::{SystemTime, UNIX_EPOCH};
17
18/// 生成腾讯云 CDN TypeD 鉴权下载链接
19///
20/// 对指定对象生成有时效的 CDN 下载链接，防止盗链。
21/// 签名有效期为长期（时间戳生成后永久可访问），
22/// 若需更严格的控制可修改为添加过期参数。
23///
24/// # 参数
25///
26/// - `cdn_domain`: CDN 加速域名，如 `"ossdev.yuriful.top"`
27/// - `cdn_key`: CDN 鉴权密钥（从配置加载）
28/// - `object_key`: COS 对象 Key，如 `"manga_1/episode_2/file.7z"`
29///
30/// # 返回值
31///
32/// - `Some(url)` — 生成成功，返回完整的鉴权下载链接
33/// - `None` — CDN 域名或密钥为空，无法生成鉴权 URL
34///
35/// # 路径编码
36///
37/// 对象 Key 中的特殊字符（中文、空格、`!`、`*` 等）会被 URL 编码，
38/// 保留 `/` 分隔符以保持路径结构。
39///
40/// # 示例
41///
42/// ```rust,ignore
43/// let url = generate_cdn_url("cdn.example.com", "mykey", "manga/cover.jpg");
44/// // => "https://cdn.example.com/manga/cover.jpg?sign=abc123&t=1710000000"
45/// ```
46pub fn generate_cdn_url(cdn_domain: &str, cdn_key: &str, object_key: &str) -> Option<String> {
47    let domain = cdn_domain.trim();
48    let key = cdn_key.trim();
49    if domain.is_empty() || key.is_empty() {
50        return None;
51    }
52    let raw_path = if object_key.starts_with('/') {
53        object_key.to_string()
54    } else {
55        format!("/{object_key}")
56    };
57    let encoded_path = encode_cdn_path(&raw_path);
58    let ts = SystemTime::now().duration_since(UNIX_EPOCH).ok()?.as_secs();
59    let sign_content = format!("{key}{encoded_path}{ts}");
60    let sign = format!("{:x}", md5::compute(sign_content.as_bytes()));
61    let domain = domain.trim_end_matches('/');
62    let protocol = if domain.starts_with("http://") || domain.starts_with("https://") {
63        ""
64    } else {
65        "https://"
66    };
67    Some(format!(
68        "{protocol}{domain}{encoded_path}?sign={sign}&t={ts}"
69    ))
70}
71
72/// CDN 路径编码：保留 `/`，编码中文及特殊字符
73///
74/// 对路径中每一段分别进行 URL 编码，同时对 `!` `'` `(` `)` `*` 做额外百分号编码，
75/// 对齐 Java `OssServiceImpl` 的编码行为。
76fn encode_cdn_path(path: &str) -> String {
77    path.split('/')
78        .map(|seg| {
79            if seg.is_empty() {
80                String::new()
81            } else {
82                urlencoding::encode(seg)
83                    .replace('!', "%21")
84                    .replace('\'', "%27")
85                    .replace('(', "%28")
86                    .replace(')', "%29")
87                    .replace('*', "%2A")
88            }
89        })
90        .collect::<Vec<_>>()
91        .join("/")
92}
93
94#[cfg(test)]
95mod tests {
96    use super::*;
97
98    #[test]
99    fn test_generate_cdn_url_has_sign_and_t() {
100        let url = generate_cdn_url(
101            "ossdev.yuriful.top",
102            "testkey",
103            "manga_1/episode_2/manga-raw/file.7z",
104        )
105        .unwrap();
106        assert!(url.contains("?sign="));
107        assert!(url.contains("&t="));
108        assert!(url.starts_with("https://ossdev.yuriful.top/"));
109    }
110}
tdm_server_rust/utils/cos_cdn.rs

tdm_server_rust/utils/
cos_cdn.rs
