yup-oauth2/examples/auth.rs

use chrono::Local;
use getopts::{Fail, HasArg, Occur, Options};
use hyper_tls::HttpsConnector;
use std::default::Default;
use std::env;
use std::thread::sleep;
use std::time::Duration;
use yup_oauth2::{self as oauth2, GetToken};

fn usage(program: &str, opts: &Options, err: Option<Fail>) -> ! {
    if err.is_some() {
        println!("{}", err.unwrap());
        std::process::exit(1);
    }
    println!("{}", opts.short_usage(program) + " SCOPE [SCOPE ...]");
    println!(
        "{}",
        opts.usage(
            "A program to authenticate against oauthv2 services.\n\
             See https://developers.google.com/youtube/registering_an_application\n\
             and https://developers.google.com/youtube/v3/guides/authentication#devices"
        )
    );

    std::process::exit(0);
}

fn main() {
    let args: Vec<String> = env::args().collect();
    let prog = args[0].clone();

    let mut opts = Options::new();
    opts.opt(
        "c",
        "id",
        "oauthv2 ID of your application",
        "CLIENT_ID",
        HasArg::Yes,
        Occur::Req,
    )
    .opt(
        "s",
        "secret",
        "oauthv2 secret of your application",
        "CLIENT_SECRET",
        HasArg::Yes,
        Occur::Req,
    );

    let m = match opts.parse(&args[1..]) {
        Ok(m) => m,
        Err(e) => {
            usage(&prog, &opts, Some(e));
        }
    };

    if m.free.len() == 0 {
        let msg = Fail::ArgumentMissing(
            "you must provide one or more authorization scopes as free options".to_string(),
        );
        usage(&prog, &opts, Some(msg));
    }

    let secret = oauth2::ApplicationSecret {
        client_id: m.opt_str("c").unwrap(),
        client_secret: m.opt_str("s").unwrap(),
        token_uri: Default::default(),
        auth_uri: Default::default(),
        redirect_uris: Default::default(),
        ..Default::default()
    };

    println!("THIS PROGRAM PRINTS ALL COMMUNICATION TO STDERR !!!");

    struct StdoutHandler;
    impl oauth2::AuthenticatorDelegate for StdoutHandler {
        fn present_user_code(&mut self, pi: &oauth2::PollInformation) {
            println!(
                "Please enter '{}' at {} and authenticate the application for the\n\
                      given scopes. This is not a test !\n\
                      You have time until {} to do that.
                      Do not terminate the program until you deny or grant access !",
                pi.user_code,
                pi.verification_url,
                pi.expires_at.with_timezone(&Local)
            );
            let delay = Duration::from_secs(5);
            println!("Browser opens automatically in {:?} seconds", delay);
            sleep(delay);
            open::that(&pi.verification_url).ok();
            println!("DONE - waiting for authorization ...");
        }
    }

    let https = HttpsConnector::new(4).unwrap();
    let client = hyper::Client::builder().build(https);

    match oauth2::Authenticator::new(&secret, StdoutHandler, client, oauth2::NullStorage, None)
        .token(&m.free)
    {
        Ok(t) => {
            println!("Authentication granted !");
            println!("You should store the following information for use, or revoke it.");
            println!("All dates are given in UTC.");
            println!("{:?}", t);
        }
        Err(err) => {
            println!("Access token wasn't obtained: {}", err);
            std::process::exit(10);
        }
    };
}