ghtool 0.9.1

A command-line tool for interacting with Github API with some specialized features oriented around Checks
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

use std::time::Duration;

use eyre::{eyre, Context, Result};
use http::StatusCode;
use indicatif::ProgressBar;
use tracing::info;

use crate::{
    github::{
        AccessToken, AccessTokenResponse, CodeResponse, CurrentUser, GithubApiError,
        GithubAuthClient, GithubClient,
    },
    spinner::make_spinner_style,
    term::{bold, prompt_for_user_to_continue, read_stdin},
    token_store::{self, get_token},
};

pub async fn login(use_stdin_token: bool) -> Result<()> {
    // Assume hostname github.com for now
    let hostname = "github.com";
    if let Some(current_user) = validate_existing_token(hostname).await? {
        println!("Already logged in as {}", bold(&current_user.viewer.login));
        println!("To log out, run {}", bold("ght logout"));
        return Ok(());
    }

    let access_token = if use_stdin_token {
        read_stdin()?
    } else {
        acquire_token_from_github().await?
    };

    token_store::set_token(hostname, &access_token)
        .map_err(|e| eyre!(e).wrap_err("Failed to store token"))?;

    let client = GithubClient::new(&access_token)?;
    let current_user = client.get_current_user().await?;

    println!(
        "Logged in to {} as {}",
        hostname,
        bold(&current_user.viewer.login)
    );
    Ok(())
}

async fn validate_existing_token(hostname: &str) -> Result<Option<CurrentUser>> {
    let token = match get_token(hostname) {
        Ok(t) => t,
        Err(keyring::Error::NoEntry) => {
            info!("No token stored, continuing");
            return Ok(None);
        }
        Err(err) => {
            return Err(eyre!(err).wrap_err("Failed to get token from keyring"));
        }
    };

    let client = GithubClient::new(&token)?;
    match client.get_current_user().await {
        Ok(current_user) => Ok(Some(current_user)),
        Err(GithubApiError::ErrorResponse(StatusCode::UNAUTHORIZED, _)) => {
            info!("Token is invalid, continuing");
            Ok(None)
        }
        Err(err) => Err(eyre!(err).wrap_err("Failed to get current user")),
    }
}

async fn acquire_token_from_github() -> Result<String> {
    let auth_client = GithubAuthClient::new()?;
    let code_response = auth_client
        .get_device_code()
        .await
        .wrap_err("Failed to get device code")?;

    println!(
        "First copy your one-time code: {}",
        bold(&code_response.user_code)
    );

    prompt_for_user_to_continue("Press Enter to open github.com in your browser...")?;

    info!("Opening {} in browser", code_response.verification_uri);
    open::that(&code_response.verification_uri)?;

    let pb = create_progress_bar();
    let token = await_authorization(&auth_client, &code_response).await?;
    pb.finish_and_clear();
    Ok(token.access_token)
}

fn create_progress_bar() -> ProgressBar {
    let pb = ProgressBar::new_spinner();
    pb.enable_steady_tick(Duration::from_millis(100));
    pb.set_style(make_spinner_style());
    pb.set_message("Waiting for authorization...");
    pb
}

async fn await_authorization(
    auth_client: &GithubAuthClient,
    code_response: &CodeResponse,
) -> Result<AccessToken> {
    loop {
        let token_response = auth_client
            .get_access_token(&code_response.device_code)
            .await?;

        match token_response {
            AccessTokenResponse::AuthorizationPending(_) => {
                tokio::time::sleep(Duration::from_secs(code_response.interval.into())).await;
            }
            AccessTokenResponse::AccessToken(token) => {
                return Ok(token);
            }
        };
    }
}