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

//! This crate exports some GitHub API bindings through [`GitHub`].

use std::collections::HashSet;

use derive_builder::Builder;
use futures::TryFutureExt;
use reqwest::{header, Client, Response, Result};
use serde::Deserialize;
use tracing::{debug, info, instrument, warn, Level};
use url::Url;

/// Asynchronous GitHub API bindings that wraps a [`reqwest::Client`] internally.
#[derive(Debug, Clone, Builder)]
pub struct GitHub {
    #[builder(
        setter(name = "token", into),
        field(
            ty = "String",
            build = r#"
                let mut headers = header::HeaderMap::new();
                headers.insert("User-Agent", header::HeaderValue::from_static("gfas"));
                headers.insert("Authorization", format!("token {}", self.client).parse().unwrap());
                Client::builder().default_headers(headers).build().unwrap()
            "#
        )
    )]
    client: Client,

    #[builder]
    endpoint: Url
}

impl GitHub {
    /// Alias for [`GitHubBuilder::create_empty()`].
    pub fn builder() -> GitHubBuilder {
        GitHubBuilder::create_empty()
    }

    /// Paginates through the given user profile link and returns
    /// discovered users collected in [`HashSet`].
    ///
    /// `role` should be either `"following"` or `"followers"`.
    ///
    /// # Errors
    ///
    /// Fails if an error occurs during sending requests.
    #[instrument(skip(self), ret(level = Level::TRACE), err)]
    pub async fn explore(&self, user: &str, role: &str) -> Result<HashSet<String>> {
        let mut res = HashSet::new();

        let url = self.endpoint.join(&format!("users/{user}/{role}")).unwrap();

        #[derive(Deserialize)]
        struct User {
            login: String
        }

        const PER_PAGE: usize = 100;

        for page in 1.. {
            debug!("page {page}");

            let users: Vec<_> = self
                .client
                .get(url.clone())
                .query(&[("page", page), ("per_page", PER_PAGE)])
                .send()
                .and_then(|r| r.json::<Vec<User>>())
                .await?
                .into_iter()
                .map(|u| u.login)
                .collect();

            let len = users.len();

            res.extend(users);

            info!("{}(+{len})", res.len());

            if len < PER_PAGE {
                break;
            }
        }

        Ok(res)
    }

    /// Follows a user.
    ///
    /// # Errors
    ///
    /// Fails if an error occurs during sending the request.
    #[instrument(skip(self), ret(level = Level::TRACE), err)]
    pub async fn follow(&self, user: &str) -> Result<Response> {
        warn!("");

        let url = self.endpoint.join(&format!("/user/following/{user}")).unwrap();
        self.client.put(url).send().await
    }

    /// Unfollows a user.
    ///
    /// # Errors
    ///
    /// Fails if an error occurs during sending the request.
    #[instrument(skip(self), ret(level = Level::TRACE), err)]
    pub async fn unfollow(&self, user: &str) -> Result<Response> {
        warn!("");

        let url = self.endpoint.join(&format!("/user/following/{user}")).unwrap();
        self.client.delete(url).send().await
    }
}