ticker-extension/background.js at main · anna-xing/ticker-extension · GitHub

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
chrome.runtime.onInstalled.addListener(() => {
    console.log("Extension installed.");
    chrome.tabs.create(
        {url: 'chrome-extension://' + chrome.runtime.id + '/options.html'},
        () => { console.log("Opened options page."); }
    );
});

// Sends response as object with all stock information, or false if none exists
chrome.runtime.onMessage.addListener((request, sender, send_response) => {
    if (request.cmd === "get_info") {
		get_info(request, send_response);
    }
    return true;
});

async function get_info(request, send_response) {
	let stock = false;
	let respond = () => {
		send_response({ status: "done", stock: stock });
	}

    // Check for ticker validity (NYSE dictates 16-char max)
	let ticker = request.ticker.split(" ")[0].slice(0, 16);
    if (ticker === "" || ticker != request.ticker) {
        // Nothing highlighted
        respond();
    } else {
        function get_key() {
            let curr_key_index = 0;
            if (request.api_keys[curr_key_index]) {
                let key = request.api_keys[curr_key_index];
                curr_key_index = (curr_key_index + 1) % 2;
                return key;
            } else {
                return request.api_keys[(curr_key_index + 1) % 2];
            }
        }

        let overview;
        await fetch("https://www.alphavantage.co/query?function=OVERVIEW&symbol=" + ticker + "&apikey=" + get_key())
			.then((result) => result.json())
            .then((result) => {
				overview = result;
                console.log("Fetching overview succeeded.");
                console.log(get_key())
            });

        let global_quote;
        await fetch("https://www.alphavantage.co/query?function=GLOBAL_QUOTE&symbol=" + ticker + "&apikey=" + get_key())
            .then((result) => result.json())
            .then((result) => {
				global_quote = result['Global Quote'];
                console.log("Fetching global quote succeeded.");
			});

        if (!overview || !global_quote) {
            send_response({ status: "overload", stock: stock });
            console.log("No data was returned. The API has been called too frequently.")
            return;
        } else if (Object.keys(overview).length === 0 || Object.keys(global_quote).length === 0) {
			respond();
			console.log("No data was found.")
			return;
        }

        // Get data for graphs
        let graph_w_pts = [];

        await fetch("https://www.alphavantage.co/query?function=TIME_SERIES_INTRADAY&symbol=" + ticker + "&interval=60min&apikey=" + get_key())
            .then((result) => result.json())
            .then((result) => {
                let time_series = result["Time Series (60min)"];
                let last_five_days = {}
                for (let time in time_series) {
                    // Trim to last five days (excluding closing values for current day)
                    if (Object.keys(last_five_days).length === 6) {
                        graph_w_pts.pop(); // Remove single stray value
                        break;
                    }
                    let datetime = new Date(time.replace(' ', 'T'));
                    last_five_days[datetime.getDate()] = null;
                    graph_w_pts.push({
                        x: datetime,
                        y: time_series[time]["4. close"]
                    });
                }
                console.log("Fetching intraday time series succeeded.");
            });

        let stock_name = overview["Name"];
        let exchange = overview["Exchange"];

        let current_price = global_quote["05. price"];
        let currency_change = global_quote["09. change"];
        let percent_change = global_quote["10. change percent"].replace('%', '');

        let high = overview["52WeekHigh"];
        let low = overview["52WeekLow"];

        let day_volume = global_quote["06. volume"];
        let div_yield = overview["DividendYield"];
        let pe_ratio = overview["PERatio"];
        let market_cap = overview["MarketCapitalization"];

        stock = {
            stock_name: stock_name,
            exchange: exchange,
            ticker: ticker,
            current_price: current_price,
            currency_change: currency_change,
            percent_change: percent_change,
            high: high,
            low: low,
            day_volume: day_volume,
            div_yield: div_yield,
            pe_ratio: pe_ratio,
            market_cap: market_cap,

			graph_w_pts: graph_w_pts,
        };

        respond();
    }
}