Add caching for new-env and when prompting if no wallet file exists

Author: stefan-mysten

crates/sui-cluster-test/src/cluster.rs

@@ -377,6 +377,7 @@ pub async fn new_wallet_context_from_cluster(
             rpc: fullnode_url.into(),
             ws: None,
             basic_auth: None,
+            chain_id: None,
         }],
         active_address: Some(address),
         active_env: Some("localnet".to_string()),

crates/sui-sdk/src/sui_client_config.rs

@@ -165,6 +165,10 @@ impl Display for SuiEnv {
             writeln!(writer)?;
             write!(writer, "Basic Auth: {}", basic_auth)?;
         }
+        if let Some(chain_id) = &self.chain_id {
+            writeln!(writer)?;
+            write!(writer, "Chain ID: {}", chain_id)?;
+        }
         write!(f, "{}", writer)
     }
 }

crates/sui/src/client_commands.rs

@@ -1821,7 +1821,7 @@ impl SuiClientCommands {
                         "Environment config with name [{alias}] already exists."
                     ));
                 }
-                let env = SuiEnv {
+                let mut env = SuiEnv {
                     alias,
                     rpc,
                     ws,
@@ -1833,7 +1833,8 @@ impl SuiClientCommands {
                 env.create_rpc_client(None, None).await?;
                 context.config.envs.push(env.clone());
                 context.config.save()?;
-                let _ = context.cache_chain_id(&context.get_client().await?).await?;
+                let chain_id = context.cache_chain_id(&context.get_client().await?).await?;
+                env.chain_id = Some(chain_id);
                 SuiClientCommandResult::NewEnv(env)
             }
             SuiClientCommands::ActiveEnv => SuiClientCommandResult::ActiveEnv(

crates/sui/src/sui_commands.rs

@@ -1625,6 +1625,9 @@ async fn prompt_if_no_config(
             }
             .persisted(wallet_conf_path)
             .save()?;
+
+            let context = WalletContext::new(wallet_conf_path)?;
+            let _ = context.cache_chain_id(&context.get_client().await?).await?;
         }
     }
     Ok(())