psdevbot 0.6.0

Bot used to report GitHub changes in Pokémon Showdown chatrooms
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

use log::info;
use lru::LruCache;
use reqwest::{header, Client};
use serde::Deserialize;
use std::time::Duration;

pub struct GitHubApi {
    user: String,
    password: String,
    cache: LruCache<String, User>,
    client: Client,
}

impl GitHubApi {
    pub fn new(user: String, password: String) -> Self {
        Self {
            user,
            password,
            cache: LruCache::new(100),
            client: Client::builder()
                .timeout(Duration::from_secs(5))
                .user_agent("psdevbot-rust")
                .build()
                .unwrap(),
        }
    }

    pub async fn fetch_user(
        &mut self,
        #[allow(clippy::ptr_arg)] // due to LruCache limitations accepting &String is necessary.
        user_name: &String,
    ) -> Option<&User> {
        if !self.cache.contains(user_name) {
            info!("Fetching user `{}` from GitHub", user_name);
            let user = self
                .client
                .get(&format!("https://api.github.com/users/{}", user_name))
                .header(header::ACCEPT, "application/vnd.github.v3+json")
                .basic_auth(&self.user, Some(&self.password))
                .send()
                .await
                .ok()?
                .json()
                .await
                .ok()?;
            self.cache.put(user_name.clone(), user);
        }
        self.cache.get(user_name)
    }
}

#[derive(Deserialize)]
pub struct User {
    pub html_url: String,
}