gcookie 0.1.2

An tool for getting site cookie from your browser
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use super::utils::get_site;
use std::{error::Error, path::PathBuf};

// cookies to string
fn gen_cookies_string(cookies: Vec<rookie::enums::Cookie>) -> String {
    let mut cookie_str = String::new();
    cookies.iter().for_each(|cookie| {
        cookie_str.push_str(&format!("{}={}; ", cookie.name, cookie.value));
    });
    cookie_str.pop();
    cookie_str.pop();
    cookie_str
}

/// Get cookie from site
///
/// # Examples
///
/// ```no_run
/// let site = "http://bing.com";
///  
/// let cookie =  match gcookie::get_cookies("chrome", site) {
///     Ok(cookie) => cookie,
///     Err(err) => panic!("An error occurred when get cookie '{}': {}", site, err),
/// };
/// ```
pub fn get_cookies(browser: &str, site: &str) -> Result<String, Box<dyn Error>> {
    let site = get_site(site)?;
    let domains = Some(vec![site]);
    let cookies = match browser.to_lowercase().as_str() {
        "firefox" => rookie::firefox(domains),
        "chrome" => rookie::chrome(domains),
        "chromium" => rookie::chromium(domains),
        "edge" => rookie::edge(domains),
        _ => {
            return Err(Box::from(format!(
                "Unsupported browser: {}; please use firefox, chrome, chromium or edge",
                browser
            )))
        }
    }?;

    Ok(gen_cookies_string(cookies))
}

/// Get cookie from site by Chrome with path.
///
/// # Examples
///
/// ```no_run
/// let site = "https://google.com";
///
/// let mut path = PathBuf::new();
/// path.push(r"C:\my_chrome\user data\default");
///
/// let cookie =  match gcookie::get_chrome_cookies_by_path(site, &path) {
///     Ok(cookie) => cookie,
///     Err(err) => panic!("An error occurred when get cookie '{}': {}", site, err),
/// };
/// ```
pub fn get_chrome_cookies_by_path(site: &str, path: &PathBuf) -> Result<String, Box<dyn Error>> {
    let cookies_path = path.join("Network/Cookies");
    if !cookies_path.exists() {
        panic!("{} not exists", cookies_path.display());
    }
    let key_path = path.join("../Local State");
    let site = get_site(site)?;
    let domains = Some(vec![site]);
    let cookies = rookie::any_browser(
        cookies_path.to_str().unwrap(),
        domains,
        Some(key_path.to_str().unwrap()),
    )?;

    Ok(gen_cookies_string(cookies))
}

/// Get cookie from site by Firefox with path.
///
/// # Examples
///
/// ```no_run
/// let site = "https://www.mozilla.org/";
///
/// let mut path = PathBuf::new();
/// path.push(r"C:\my_firefox\profile");
///
/// let cookie =  match gcookie::get_firefox_cookies_by_path(site, &path) {
///     Ok(cookie) => cookie,
///     Err(err) => panic!("An error occurred when get cookie '{}': {}", site, err),
/// };
/// ```
pub fn get_firefox_cookies_by_path(site: &str, path: &PathBuf) -> Result<String, Box<dyn Error>> {
    let site = get_site(site)?;
    let cookies_path = path.join("cookies.sqlite");
    if !cookies_path.exists() {
        panic!("{} not exists", cookies_path.display());
    }
    let domains = Some(vec![site]);
    let cookies = rookie::any_browser(cookies_path.to_str().unwrap(), domains, None)?;

    Ok(gen_cookies_string(cookies))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn rookie_chrome_ok() {
        let site = "google.com";
        let browser = "chrome";

        let cookie = get_cookies(browser, site);
        assert!(cookie.is_ok());
    }

    #[test]
    #[cfg(target_os = "windows")]
    fn rookie_edge_path_ok() {
        let site = "bing.com";
        let home_dir = dirs::home_dir().unwrap();
        let profile_path = home_dir.join("AppData/Local/Microsoft/Edge/User Data/Default/");
        let cookie = get_chrome_cookies_by_path(site, &profile_path);
        assert!(cookie.is_ok());
        // println!("{}", cookie.unwrap());
    }

    #[test]
    #[cfg(target_os = "windows")]
    fn rookie_firefox_windows_ok() {
        let site = "https://www.mozilla.org/";
        use std::fs;
        let profiles_path = dirs::home_dir()
            .unwrap()
            .join("AppData/Roaming/Mozilla/Firefox/Profiles/");
        let mut profile = profiles_path.clone();
        for (i, p) in fs::read_dir(profiles_path).unwrap().enumerate() {
            if i == 0 {
                profile = p.unwrap().path();
                break;
            }
        }
        let cookie = get_firefox_cookies_by_path(site, &profile);
        assert!(cookie.is_ok());
    }
}