blivemsg 0.2.0

Bilibili Live Message Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use std::path::Path;
use serde_json::Value;
use crate::error::Error;

/// Cookie 来源抽象
#[derive(Debug, Clone)]
pub enum CookieSource {
    /// 从文件读取
    File(std::path::PathBuf),
    /// 直接使用字符串
    Inline(String),
}

impl CookieSource {
    /// 获取 Cookie 字符串（只使用 SESSDATA 和 buvid3）
    pub fn get_cookie(&self) -> Result<String, Error> {
        match self {
            CookieSource::File(path) => {
                if !path.exists() {
                    return Err(Error::CookieFileNotFound(path.clone()));
                }
                
                // 读取 JSON 文件并提取 SESSDATA 和 buvid3
                let content = std::fs::read_to_string(path)?;
                let json: Value = serde_json::from_str(&content)?;
                
                let sessdata = json["SESSDATA"]
                    .as_str()
                    .ok_or_else(|| Error::AuthenticationFailed)?;
                let buvid3 = json["buvid3"]
                    .as_str()
                    .unwrap_or("");
                
                Ok(format!("SESSDATA={}; buvid3={}", sessdata, buvid3))
            }
            CookieSource::Inline(cookie) => Ok(cookie.clone()),
        }
    }
}

impl From<&str> for CookieSource {
    fn from(s: &str) -> Self {
        // 判断是文件路径还是直接的 Cookie 字符串
        if s.contains("SESSDATA=") || s.contains("buvid3=") {
            CookieSource::Inline(s.to_string())
        } else {
            CookieSource::File(std::path::PathBuf::from(s))
        }
    }
}

impl From<String> for CookieSource {
    fn from(s: String) -> Self {
        s.as_str().into()
    }
}

impl From<&Path> for CookieSource {
    fn from(p: &Path) -> Self {
        CookieSource::File(p.to_path_buf())
    }
}