set platform on login based on user permissions

Author: Panaetius

coman/src/config.rs

@@ -9,7 +9,7 @@ use color_eyre::{
 use directories::ProjectDirs;
 use lazy_static::lazy_static;
 use serde::{Deserialize, Serialize};
-use strum_macros::{EnumString, VariantNames};
+use strum_macros::{EnumIter, EnumString, VariantNames};
 use toml_edit::DocumentMut;
 
 const DEFAULT_CONFIG_TOML: &str = include_str!("../.config/config.toml");
@@ -33,7 +33,7 @@ pub struct SystemDescription {
     pub architecture: Vec<String>,
 }
 
-#[derive(Clone, Debug, Serialize, Deserialize, Default, strum::Display, EnumString, VariantNames)]
+#[derive(Clone, Debug, Serialize, Deserialize, Default, strum::Display, EnumString, VariantNames, EnumIter)]
 #[strum(serialize_all = "lowercase")]
 #[allow(clippy::upper_case_acronyms)]
 pub enum ComputePlatform {
@@ -331,6 +331,18 @@ impl Config {
         let _cfg: ComanConfig = builder.build()?.try_deserialize().wrap_err("invalid config")?;
         Ok(())
     }
+
+    // Returns tuple of bool saying whether a values is set in (default, global, project local) config
+    pub fn value_source(&self, key_path: &str) -> Result<(bool, bool, bool)> {
+        Ok((
+            self.default_layer.get(key_path).is_some(),
+            self.global_layer.get(key_path).unwrap_or_default().is_some(),
+            self.project_layer
+                .as_ref()
+                .map(|l| l.get(key_path).unwrap_or_default().is_some())
+                .unwrap_or(false),
+        ))
+    }
 }
 
 pub fn global_config_layer() -> Result<Layer> {

coman/src/cscs/handlers.rs

@@ -19,6 +19,7 @@ use itertools::Itertools;
 use regex::Regex;
 use reqwest::Url;
 use sha2::{Digest, Sha256};
+use strum::IntoEnumIterator;
 use tarpc::{client, context, serde_transport, tokio_serde::formats::Bincode};
 use tokio::{
     fs::File,
@@ -84,9 +85,23 @@ pub(crate) async fn cscs_login(client_id: String, client_secret: String) -> Resu
     store_secret(CLIENT_ID_SECRET_NAME, client_id_secret.clone()).await?;
     let client_secret_secret = Secret::new(client_secret);
     store_secret(CLIENT_SECRET_SECRET_NAME, client_secret_secret.clone()).await?;
-    client_credentials_login(client_id_secret, client_secret_secret)
-        .await
-        .map(|_| ())
+    let token = client_credentials_login(client_id_secret, client_secret_secret).await?;
+
+    // figure out what platform the user has access to and set it in config
+    let mut config = Config::new()?;
+    let source = config.value_source("cscs.current_platform")?;
+    if source.1 || source.2 {
+        // don't override setting if it's already provided
+        return Ok(());
+    }
+    for platform in ComputePlatform::iter() {
+        let api_client = CscsApi::new(token.0.0.clone(), Some(platform.clone())).unwrap();
+        if (api_client.list_systems().await).is_ok() {
+            config.set("cscs.current_platform", platform.to_string(), true)?;
+            break;
+        }
+    }
+    Ok(())
 }
 
 #[allow(dead_code)]