yahoo_tick_grabber 0.1.3

A wrapper API around reqwest library that as of currently returns a string containing the information about a ticker off Yahoo Finance
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

use reqwest::header::COOKIE;
use reqwest::header::USER_AGENT;

use std::borrow::Borrow;

use reqwest::Result;

use std::sync::Arc;
use tokio::sync::RwLock;

pub struct YAHOOCONNECT {
    multiclient: reqwest::Client,
    cookie: Arc<RwLock<String>>,
    crumb: Arc<RwLock<String>>,
    cookie_url: Arc<String>,
    crumb_url: Arc<String>,
    user_agent: Arc<String>
}

impl YAHOOCONNECT {
    pub async fn new() -> Result<YAHOOCONNECT> {
        let multiclient = reqwest::Client::new();
        let cookie = Arc::new(RwLock::new(String::new()));
        let crumb = Arc::new(RwLock::new(String::new()));
        let cookie_url = Arc::new(String::from("https://fc.yahoo.com"));
        let crumb_url = Arc::new(String::from("https://query2.finance.yahoo.com/v7/finance/quote?symbols="));
        let user_agent = Arc::new(String::from("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"));
        let te = YAHOOCONNECT {
            multiclient,
            cookie,
            crumb,
            cookie_url,
            crumb_url,
            user_agent,
        };
        YAHOOCONNECT::update_crumb_n_cookie(te.borrow()).await?;
        return Ok(te);
    }

    async fn update_crumb_n_cookie(&self) -> Result<()>  {
        // get cookie first from link.
        let response = self.multiclient.get(self.cookie_url.as_str()).send().await?;
        let mut cookie_str = String::new();
        for sr in response.cookies()
        {
            cookie_str = format!("{}={}", sr.name(), sr.value());
        }

        {
            let mut re = self.cookie.write().await;
            *re = cookie_str.clone();
        }

        let user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36";
        let crumb_response = self.multiclient
        .get("https://query2.finance.yahoo.com/v1/test/getcrumb")
        .header(COOKIE, cookie_str)
        .header(USER_AGENT, user_agent)
        .send()
        .await?.text().await?;

        {
            let mut re = self.crumb.write().await;
            *re = crumb_response;
        }

        Ok(())
    } //if update doesnt work, return error with each step.

    pub async fn get_ticker(&self, name: &str) -> std::result::Result<String, String> {
        let ticker_info = self.get_tic_internal(name).await.unwrap();
        if ticker_info.contains("result\":[{")
        {
            return Ok(ticker_info)
        }
        if ticker_info.contains("Invalid Crumb")
        {
            self.update_crumb_n_cookie().await.unwrap();
            let ticker_info = self.get_tic_internal(name).await.unwrap();
            if ticker_info.contains("quoteResponse")
            {
            return Ok(ticker_info)
            } else {
                return Err("Search Error".to_string());
            }
        } else {
             return Err("Search Error".to_string());
        }

        }
        
        //test //we read the error, if it is
    
    async fn get_tic_internal(&self,name: &str) -> Result<String>
    {
        let final_get = format!("{}{}&crumb={}",self.crumb_url.as_str(),name,self.crumb.read().await.as_str());
        let ticker_info = self.multiclient.get(final_get)
        .header(COOKIE, self.cookie.read().await.as_str())
        .header(USER_AGENT, self.user_agent.as_str())
        .send()
        .await?
        .text()
        .await?;
        
        return Ok(ticker_info);
    }                                                            //crumb/cookie related udpate
                                                                     //that.
}


//error checking added, need to update version.

//botched error checking, minor change needed to correctly return error when no stock found with ticker.