Add support for differing username between cluster and local machine

1 files changed, 9 insertions, 0 deletions

diff --git a/src/context.rs b/src/context.rs
index f67c302..4c1b639 100644
--- a/src/context.rs
+++ b/src/context.rs
@@ -15,6 +15,7 @@ pub struct Context {
     job_id: Option<u32>,
     infer_job_id: bool,
     frontend_hostname: Option<String>,
+    cluster_username: Option<String>
 }
 
 impl Context {
@@ -22,12 +23,14 @@ impl Context {
         job_id: Option<u32>,
         infer_job_id: bool,
         frontend_hostname: Option<String>,
+        cluster_username: Option<String>
     ) -> Result<Self> {
         Ok(Self {
             node: get_execution_node().await?,
             job_id,
             infer_job_id,
             frontend_hostname,
+            cluster_username
         })
     }
 
@@ -57,6 +60,12 @@ impl Context {
             .as_deref()
             .ok_or_else(|| eyre::eyre!("missing frontend hostname"))
     }
+
+    pub fn cluster_username(&self) -> Result<&str> {
+        self.cluster_username
+            .as_deref()
+            .ok_or_else(|| eyre::eyre!("missing cluster username"))
+    }
 }
 
 async fn get_execution_node() -> Result<ExecutionNode> {