Implement wasmer whoami

lib/cli/src/cli.rs

@@ -10,7 +10,9 @@ use crate::commands::CreateExe;
 use crate::commands::CreateObj;
 #[cfg(feature = "wast")]
 use crate::commands::Wast;
-use crate::commands::{Add, Cache, Config, Inspect, List, Login, Run, SelfUpdate, Validate};
+use crate::commands::{
+    Add, Cache, Config, Inspect, List, Login, Run, SelfUpdate, Validate, Whoami,
+};
 use crate::error::PrettyError;
 use clap::{CommandFactory, ErrorKind, Parser};
 use std::{fmt, str::FromStr};
@@ -151,6 +153,10 @@ enum WasmerCLIOptions {
     #[clap(name = "binfmt")]
     Binfmt(Binfmt),
 
+    /// Shows the current logged in user for the current active registry
+    #[clap(name = "whoami")]
+    Whoami(Whoami),
+
     /// Add a WAPM package's bindings to your application.
     Add(Add),
 }
@@ -176,6 +182,7 @@ impl WasmerCLIOptions {
             Self::Wast(wast) => wast.execute(),
             #[cfg(target_os = "linux")]
             Self::Binfmt(binfmt) => binfmt.execute(),
+            Self::Whoami(whoami) => whoami.execute(),
             Self::Add(install) => install.execute(),
         }
     }

lib/cli/src/commands.rs

@@ -18,6 +18,7 @@ mod self_update;
 mod validate;
 #[cfg(feature = "wast")]
 mod wast;
+mod whoami;
 
 #[cfg(target_os = "linux")]
 pub use binfmt::*;
@@ -30,7 +31,8 @@ pub use create_obj::*;
 #[cfg(feature = "wast")]
 pub use wast::*;
 pub use {
-    add::*, cache::*, config::*, inspect::*, list::*, login::*, run::*, self_update::*, validate::*,
+    add::*, cache::*, config::*, inspect::*, list::*, login::*, run::*, self_update::*,
+    validate::*, whoami::*,
 };
 
 /// The kind of object format to emit.

lib/cli/src/commands/whoami.rs

@@ -0,0 +1,18 @@
+use clap::Parser;
+
+#[derive(Debug, Parser)]
+/// The options for the `wasmer whoami` subcommand
+pub struct Whoami {
+    /// Which registry to check the logged in username for
+    #[clap(long, name = "registry")]
+    pub registry: Option<String>,
+}
+
+impl Whoami {
+    /// Execute `wasmer whoami`
+    pub fn execute(&self) -> Result<(), anyhow::Error> {
+        let (registry, username) = wasmer_registry::whoami(self.registry.as_deref())?;
+        println!("logged into registry {registry:?} as user {username:?}");
+        Ok(())
+    }
+}

lib/registry/src/graphql.rs

@@ -114,6 +114,14 @@ pub(crate) mod proxy {
 )]
 pub(crate) struct GetPackageVersionQuery;
 
+#[derive(GraphQLQuery)]
+#[graphql(
+    schema_path = "graphql/schema.graphql",
+    query_path = "graphql/queries/whoami.graphql",
+    response_derives = "Debug"
+)]
+pub(crate) struct WhoAmIQuery;
+
 #[derive(GraphQLQuery)]
 #[graphql(
     schema_path = "graphql/schema.graphql",

lib/registry/src/lib.rs

@@ -894,6 +894,48 @@ pub fn install_package(
     ))
 }
 
+pub fn whoami(
+    #[cfg(test)] test_name: &str,
+    registry: Option<&str>,
+) -> Result<(String, String), anyhow::Error> {
+    use crate::graphql::{who_am_i_query, WhoAmIQuery};
+    use graphql_client::GraphQLQuery;
+
+    #[cfg(test)]
+    let config = PartialWapmConfig::from_file(test_name);
+    #[cfg(not(test))]
+    let config = PartialWapmConfig::from_file();
+
+    let config = config
+        .map_err(|e| anyhow::anyhow!("{e}"))
+        .context(anyhow::anyhow!("{registry:?}"))?;
+
+    let registry = match registry {
+        Some(s) => format_graphql(s),
+        None => config.registry.get_current_registry(),
+    };
+
+    let login_token = config
+        .registry
+        .get_login_token_for_registry(&registry)
+        .ok_or_else(|| anyhow::anyhow!("not logged into registry {:?}", registry))?;
+
+    let q = WhoAmIQuery::build_query(who_am_i_query::Variables {});
+    let response: who_am_i_query::ResponseData =
+        crate::graphql::execute_query(&registry, &login_token, &q)
+            .map_err(|e| anyhow::anyhow!("{e}"))
+            .context(anyhow::anyhow!("{registry:?}"))?;
+
+    let username = response
+        .viewer
+        .as_ref()
+        .ok_or_else(|| anyhow::anyhow!("not logged into registry {:?}", registry))?
+        .username
+        .to_string();
+
+    Ok((registry, username))
+}
+
 pub fn test_if_registry_present(registry: &str) -> Result<bool, String> {
     use crate::graphql::{test_if_registry_present, TestIfRegistryPresent};
     use graphql_client::GraphQLQuery;

tests/integration/cli/tests/run.rs

@@ -81,6 +81,53 @@ fn test_cross_compile_python_windows() -> anyhow::Result<()> {
     Ok(())
 }
 
+#[test]
+fn run_whoami_works() -> anyhow::Result<()> {
+    let ciuser_token = std::env::var("WAPM_DEV_TOKEN").expect("no CIUSER / WAPM_DEV_TOKEN token");
+
+    let output = Command::new(get_wasmer_path())
+        .arg("login")
+        .arg("--registry")
+        .arg("wapm.dev")
+        .arg(ciuser_token)
+        .output()?;
+
+    if !output.status.success() {
+        bail!(
+            "wasmer login failed with: stdout: {}\n\nstderr: {}",
+            std::str::from_utf8(&output.stdout)
+                .expect("stdout is not utf8! need to handle arbitrary bytes"),
+            std::str::from_utf8(&output.stderr)
+                .expect("stderr is not utf8! need to handle arbitrary bytes")
+        );
+    }
+
+    let output = Command::new(get_wasmer_path())
+        .arg("whoami")
+        .arg("--registry")
+        .arg("wapm.dev")
+        .output()?;
+
+    let stdout = std::str::from_utf8(&output.stdout)
+        .expect("stdout is not utf8! need to handle arbitrary bytes");
+
+    if !output.status.success() {
+        bail!(
+            "linking failed with: stdout: {}\n\nstderr: {}",
+            stdout,
+            std::str::from_utf8(&output.stderr)
+                .expect("stderr is not utf8! need to handle arbitrary bytes")
+        );
+    }
+
+    assert_eq!(
+        stdout,
+        "logged into registry \"https://registry.wapm.dev/graphql\" as user \"ciuser\"\n"
+    );
+
+    Ok(())
+}
+
 #[test]
 fn run_wasi_works() -> anyhow::Result<()> {
     let output = Command::new(get_wasmer_path())