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

use reqwest::header::{HeaderMap, HeaderValue, SET_COOKIE};

#[derive(Debug, Default, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub struct Cookie {
    pub nid: String,
}

impl Cookie {
    pub fn new() -> Self {
        Self {
            nid: Self::get_new_cookie(),
        }
    }

    pub fn get_new_cookie() -> String {
        const COOKIE_HANDSHAKE: &str =
            "https://consent.google.com/s?continue=https://www.google.com/";

        let response = reqwest::blocking::get(COOKIE_HANDSHAKE).unwrap();
        let cookie = response.headers().get(SET_COOKIE).unwrap();

        cookie
            .to_str()
            .unwrap()
            .to_string()
            .split(' ')
            .collect::<Vec<&str>>()[0]
            .to_string()
    }

    pub fn add_to_header(&self, mut header: HeaderMap) -> HeaderMap {
        header.insert("Cookie", HeaderValue::from_str(self.nid.as_str()).unwrap());
        header
    }
}