micron 0.1.0

Build web apps fast. Repeat.
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
114
115
116
117
118
119
120
121
122
123
124

// GitHub: "https://github.com/login/oauth/authorize", "https://github.com/login/oauth/access_token",

use std::sync::Arc;

use cookie::Cookie;
use http::header::{ACCEPT, USER_AGENT};
use mime::Mime;
use oauth2::basic::BasicClient;
use oauth2::reqwest::async_http_client;
use oauth2::{
    AuthUrl, AuthorizationCode, ClientId, ClientSecret, CsrfToken, RedirectUrl, Scope,
    TokenResponse, TokenUrl,
};
use tracing::debug;

use serde::Deserialize;
use uuid::Uuid;

use crate::auth::login::log_in_user_id;
use crate::auth::TokenMeta;
use crate::config;
use crate::db::{decode, encode, Database};
use crate::error::Result;
use crate::user::User;
use crate::Config;

use super::UserInfo;

pub const AUTH_URL: &str = "https://github.com/login/oauth/authorize";
pub const TOKEN_URL: &str = "https://github.com/login/oauth/access_token";

/// User information to be retrieved from the GitHub API.
#[derive(Clone, Debug, Default, Deserialize)]
#[serde(default)]
pub struct GitHubUserInfo {
    login: String,
    email: Option<String>,
    name: Option<String>,
    location: Option<String>,
    avatar_url: Option<String>,
}

impl Into<UserInfo> for GitHubUserInfo {
    fn into(self) -> UserInfo {
        UserInfo {
            // TODO: don't unwrap like this
            email: self.email.unwrap(),
            handle: Some(self.login),
            full_name: self.name,
            location: self.location,
            avatar_url: self.avatar_url,
            ..Default::default()
        }
    }
}

/// Constructs common user info using auth code sent by the service provider.
///
/// Includes additional calls to target service to get user information.
pub async fn get_user_info<'c>(
    auth_code: String,
    config: &Config,
    db: &Database,
) -> Result<UserInfo> {
    // Get an auth token
    let client = crate::oauth::client(
        &config.oauth.github,
        config.domain.clone(),
        "github".to_string(),
        AUTH_URL.to_string(),
        TOKEN_URL.to_string(),
    )?;
    let token = client
        .exchange_code(AuthorizationCode::new(auth_code.clone()))
        .request_async(async_http_client)
        .await
        .map_err(|e| crate::ErrorKind::AuthFailed(e.to_string()))?;

    let mime: Mime = "application/vnd.github.v3+json"
        .parse()
        .expect("parse GitHub MIME type");

    // Fetch user data from the `/user` endpoint
    let client = reqwest::Client::new();
    let response = client
        .get("https://api.github.com/user")
        .bearer_auth(token.access_token().secret())
        .header(ACCEPT, mime.essence_str())
        .header(USER_AGENT, format!("{}", config.name))
        .send()
        .await?;

    let mut user_info: GitHubUserInfo = response.json::<GitHubUserInfo>().await?;
    println!("user_info: {:?}", user_info);

    // Fetch user email, if it was not provided with the `/user` response
    // NOTE: for that email to be returned with `/user` response it needs to be
    // set to public on the user github profile
    if user_info.email.is_none() {
        #[derive(Deserialize)]
        struct Email {
            email: String,
            verified: bool,
            primary: bool,
            visibility: Option<String>,
        }
        let response = client
            .get("https://api.github.com/user/emails")
            .bearer_auth(token.access_token().secret())
            .header(ACCEPT, mime.essence_str())
            .header(USER_AGENT, format!("{}", config.name))
            .send()
            .await?;

        let emails = response.json::<Vec<Email>>().await.unwrap();
        for email in emails {
            if email.verified && email.primary {
                user_info.email = Some(email.email);
            }
        }
    }

    Ok(user_info.into())
}