feat(zk_toolbox): Add update command

zk_toolbox/crates/config/src/chain.rs

@@ -104,6 +104,10 @@ impl ChainConfig {
         self.configs.join(GENERAL_FILE)
     }
 
+    pub fn path_to_genesis_config(&self) -> PathBuf {
+        self.configs.join(GENESIS_FILE)
+    }
+
     pub fn path_to_secrets_config(&self) -> PathBuf {
         self.configs.join(SECRETS_FILE)
     }

zk_toolbox/crates/zk_inception/src/commands/update.rs

@@ -6,11 +6,12 @@ use common::{
     logger,
     spinner::Spinner,
 };
-use config::{ChainConfig, EcosystemConfig, GENERAL_FILE};
+use config::{ChainConfig, EcosystemConfig, GENERAL_FILE, GENESIS_FILE};
 use xshell::Shell;
 
 use crate::messages::{
-    msg_updating_chain, MSG_CHAIN_NOT_FOUND_ERR, MSG_PULLING_ZKSYNC_CODE_SPINNER, MSG_SHOW_DIFF,
+    msg_diff_genesis_config, msg_updating_chain, MSG_CHAIN_NOT_FOUND_ERR,
+    MSG_PULLING_ZKSYNC_CODE_SPINNER, MSG_SHOW_DIFF, MSG_SHOW_DIFF_ADDED_FIELDS,
     MSG_UPDATING_GENERAL_CONFIG, MSG_UPDATING_SUBMODULES_SPINNER, MSG_UPDATING_ZKSYNC,
     MSG_ZKSYNC_UPDATED,
 };
@@ -33,15 +34,20 @@ pub fn run(shell: &Shell) -> anyhow::Result<()> {
     submodule_update(shell, link_to_code.clone())?;
     spinner.finish();
 
-    let updated_config_path = ecosystem.get_default_configs_path().join(GENERAL_FILE);
-    let general_config = serde_yaml::from_str(&shell.read_file(updated_config_path)?)?;
+    let general_config = serde_yaml::from_str(
+        &shell.read_file(ecosystem.get_default_configs_path().join(GENERAL_FILE))?,
+    )?;
+
+    let genesis_config = serde_yaml::from_str(
+        &shell.read_file(ecosystem.get_default_configs_path().join(GENESIS_FILE))?,
+    )?;
 
     for chain in ecosystem.list_of_chains() {
         logger::info(msg_updating_chain(&chain));
         let chain = ecosystem
             .load_chain(Some(chain))
             .context(MSG_CHAIN_NOT_FOUND_ERR)?;
-        update_chain(shell, &chain, &general_config)?;
+        update_chain(shell, &chain, &general_config, &genesis_config)?;
     }
 
     logger::outro(MSG_ZKSYNC_UPDATED);
@@ -82,6 +88,7 @@ fn merge_yaml(a: &mut serde_yaml::Value, b: serde_yaml::Value) -> anyhow::Result
 }
 
 fn save_updated_config(
+    shell: &Shell,
     config: serde_yaml::Value,
     path: &Path,
     diff: ConfigDiff,
@@ -90,14 +97,33 @@ fn save_updated_config(
         return Ok(());
     }
 
-    logger::info(MSG_SHOW_DIFF);
+    logger::info(MSG_SHOW_DIFF_ADDED_FIELDS);
     for (key, value) in diff.added_fields {
         let key = key.as_str().unwrap();
         logger::info(format!("{}: {:?}", key, value));
     }
 
     let general_config = serde_yaml::to_string(&config)?;
-    std::fs::write(path, general_config)?;
+    shell.write_file(path, general_config)?;
+
+    Ok(())
+}
+
+fn check_diff(diff: &ConfigDiff, msg: &str) -> anyhow::Result<()> {
+    if diff.value_diff.is_empty() && diff.added_fields.is_empty() {
+        return Ok(());
+    }
+
+    logger::warn(msg);
+    logger::info(MSG_SHOW_DIFF);
+    for (key, value) in diff.added_fields.iter() {
+        let key = key.as_str().unwrap();
+        logger::info(format!("{}: {:?}", key, value));
+    }
+    for (key, value) in diff.value_diff.iter() {
+        let key = key.as_str().unwrap();
+        logger::info(format!("{}: {:?}", key, value));
+    }
 
     Ok(())
 }
@@ -106,12 +132,24 @@ fn update_chain(
     shell: &Shell,
     chain: &ChainConfig,
     general: &serde_yaml::Value,
+    genesis: &serde_yaml::Value,
 ) -> anyhow::Result<()> {
     logger::info(MSG_UPDATING_GENERAL_CONFIG);
     let current_general_config_path = chain.path_to_general_config();
     let mut current_general_config =
         serde_yaml::from_str(&shell.read_file(current_general_config_path.clone())?)?;
     let diff = merge_yaml(&mut current_general_config, general.clone())?;
-    save_updated_config(current_general_config, &current_general_config_path, diff)?;
+    save_updated_config(
+        shell,
+        current_general_config,
+        &current_general_config_path,
+        diff,
+    )?;
+
+    let mut current_genesis_config =
+        serde_yaml::from_str(&shell.read_file(chain.path_to_genesis_config())?)?;
+    let diff = merge_yaml(&mut current_genesis_config, genesis.clone())?;
+    check_diff(&diff, &msg_diff_genesis_config(&chain.name))?;
+
     Ok(())
 }

zk_toolbox/crates/zk_inception/src/messages.rs

@@ -325,7 +325,15 @@ pub(super) const MSG_ZKSYNC_UPDATED: &str = "ZkSync updated successfully";
 pub(super) const MSG_PULLING_ZKSYNC_CODE_SPINNER: &str = "Pulling zksync-era repo...";
 pub(super) const MSG_UPDATING_SUBMODULES_SPINNER: &str = "Updating submodules...";
 pub(super) const MSG_UPDATING_GENERAL_CONFIG: &str = "Updating general config";
-pub(super) const MSG_SHOW_DIFF: &str = "Added the following values:";
+pub(super) const MSG_SHOW_DIFF_ADDED_FIELDS: &str = "Added the following values:";
+pub(super) const MSG_SHOW_DIFF: &str = "Found differences in the following fields:";
+
+pub(super) fn msg_diff_genesis_config(chain: &str) -> String {
+    format!(
+        "Found differences between chain {} and era genesis configs. Consider updating the chain {} genesis config and re-running genesis.",
+        chain, chain
+    )
+}
 
 pub(super) fn msg_updating_chain(chain: &str) -> String {
     format!("Updating chain: {}", chain)