init commit

src/main.rs

@@ -0,0 +1,112 @@
+use std::collections::HashMap;
+use std::env;
+use std::process::exit;
+use std::time::Duration;
+use clap::{App, Arg};
+use reqwest::{ClientBuilder, Proxy};
+use rust_util::{debugging, information, opt_result, opt_value_result, XResult};
+use rust_util::util_msg::MessageType;
+use crate::oidc::{OpenIdClientConfiguration, OpenIdConfiguration};
+
+mod oidc;
+
+
+const NAME: &str = env!("CARGO_PKG_NAME");
+const VERSION: &str = env!("CARGO_PKG_VERSION");
+const AUTHORS: &str = env!("CARGO_PKG_AUTHORS");
+const DESCRIPTION: &str = env!("CARGO_PKG_DESCRIPTION");
+
+#[tokio::main]
+async fn main() -> XResult<()> {
+    let matches = App::new(NAME)
+        .version(VERSION)
+        .about(DESCRIPTION)
+        .author(AUTHORS)
+        .arg(Arg::new("version").short('V').long("version").help("Print version"))
+        .arg(Arg::new("verbose").short('v').long("verbose").help("Verbose"))
+        .arg(Arg::new("issuer").short('i').long("issuer").takes_value(true).help("Issuer"))
+        .arg(Arg::new("proxy").short('p').long("proxy").takes_value(true).help("Proxy"))
+        .arg(Arg::new("skip-certificate-check").short('k').long("skip-certificate-check").help("Skip certificate check"))
+        .get_matches();
+
+    if matches.is_present("verbose") {
+        env::set_var("LOGGER_LEVEL", "*");
+    }
+    if matches.is_present("version") {
+        information!("{} v{}", NAME, VERSION);
+        exit(1);
+    }
+    let issuer = opt_value_result!(matches.value_of("issuer"), "Issuer is required");
+    let openid_configuration_url = format!("{}/.well-known/openid-configuration", issuer);
+    debugging!("Open id configuration url: {}", openid_configuration_url);
+
+    let mut client_builder = ClientBuilder::new()
+        .timeout(Duration::from_secs(10))
+        .connect_timeout(Duration::from_secs(3));
+    if let Some(proxy) = matches.value_of("proxy") {
+        let proxy = opt_result!( Proxy::all(proxy), "Parse proxy: {} failed: {}", proxy);
+        client_builder = client_builder.proxy(proxy);
+    }
+    if (matches.is_present("skip-certificate-check")) {
+        client_builder = client_builder.danger_accept_invalid_certs(true);
+    }
+    let client = opt_result!( client_builder.build(), "Build reqwest client failed: {}");
+
+    debugging!("Sending request...");
+    let get_openid_configuration_result = opt_result!(
+        client.get(&openid_configuration_url).send().await, "Get url: {}, failed: {}", openid_configuration_url);
+    let openid_configuration = opt_result!(
+        get_openid_configuration_result.json::<OpenIdConfiguration>().await, "Parse open id configuration failed: {}");
+
+    rust_util::util_msg::when(MessageType::DEBUG, || {
+        debugging!("Return open id configuration: {}", serde_json::to_string_pretty(&openid_configuration).unwrap());
+    });
+
+    let device_authorization_endpoint = opt_value_result!(
+        openid_configuration.device_authorization_endpoint, "Open id configuration has not configured device authorization endpoint");
+    let token_endpoint = openid_configuration.token_endpoint;
+    let userinfo_endpoint = openid_configuration.userinfo_endpoint;
+
+    information!("device_authorization_endpoint: {}", &device_authorization_endpoint);
+    information!("token_endpoint: {}", &token_endpoint);
+    information!("userinfo_endpoint: {}", &userinfo_endpoint);
+
+    let openid_client_configuration = OpenIdClientConfiguration {
+        issuer: "".into(),
+        client_id: "".into(),
+        client_secret: "".into(),
+        scope: "".into(),
+    };
+
+    // xh POST https://oauth2.googleapis.com/device/code
+    // client_id==364023986785-o5mqsqd1ej2d1vmvcqai108b7m7v7vc9.apps.googleusercontent.com scope=openid
+    //{
+    //     "device_code": "***",
+    //     "user_code": "STLB-RNKS",
+    //     "expires_in": 1800,
+    //     "interval": 5,
+    //     "verification_url": "https://www.google.com/device"
+    // }
+    let mut params = HashMap::new();
+    params.insert("client_id", &openid_client_configuration.issuer);
+    params.insert("scope", &openid_client_configuration.scope);
+    let token_endpoint_builder = opt_result!(client.post(&device_authorization_endpoint)
+        .form(params).send().await, "Get token: {}, failed: {}", device_authorization_endpoint);
+    // token_endpoint_builder.
+
+    // xh POST https://oauth2.googleapis.com/token
+    // client_id==364023986785-o5mqsqd1ej2d1vmvcqai108b7m7v7vc9.apps.googleusercontent.com
+    // client_secret==***
+    // device_code==***
+    // grant_type==urn:ietf:params:oauth:grant-type:device_code
+    // {
+    //     "access_token": "***",
+    //     "expires_in": 3599,
+    //     "refresh_token": "***",
+    //     "scope": "openid",
+    //     "token_type": "Bearer",
+    //     "id_token": "***"
+    // }
+
+    Ok(())
+}