feat(zk_toolbox): Add update command

zk_toolbox/crates/common/src/git.rs

@@ -29,3 +29,9 @@ pub fn submodule_update(shell: &Shell, link_to_code: PathBuf) -> anyhow::Result<
     .run()?;
     Ok(())
 }
+
+pub fn pull(shell: &Shell, link_to_code: PathBuf) -> anyhow::Result<()> {
+    let _dir_guard = shell.push_dir(link_to_code);
+    Cmd::new(cmd!(shell, "git pull")).run()?;
+    Ok(())
+}

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

@@ -6,3 +6,4 @@ pub mod ecosystem;
 pub mod external_node;
 pub mod prover;
 pub mod server;
+pub mod update;

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

@@ -0,0 +1,116 @@
+use std::path::Path;
+
+use anyhow::Context;
+use common::{
+    git::{pull, submodule_update},
+    logger,
+    spinner::Spinner,
+};
+use config::{ChainConfig, EcosystemConfig};
+use xshell::Shell;
+
+use crate::{
+    consts::GENERAL_FILE,
+    messages::{
+        msg_updating_chain, MSG_CHAIN_NOT_FOUND_ERR, MSG_PULLING_ZKSYNC_CODE_SPINNER,
+        MSG_SHOW_DIFF, MSG_UPDATING_GENERAL_CONFIG, MSG_UPDATING_SUBMODULES_SPINNER,
+        MSG_UPDATING_ZKSYNC, MSG_ZKSYNC_UPDATED,
+    },
+};
+
+#[derive(Default)]
+struct ConfigDiff {
+    pub value_diff: serde_yaml::Mapping,
+    pub added_fields: serde_yaml::Mapping,
+}
+
+pub fn run(shell: &Shell) -> anyhow::Result<()> {
+    logger::info(MSG_UPDATING_ZKSYNC);
+    let ecosystem = EcosystemConfig::from_file(shell)?;
+    let link_to_code = ecosystem.link_to_code.clone();
+
+    let spinner = Spinner::new(MSG_PULLING_ZKSYNC_CODE_SPINNER);
+    pull(shell, link_to_code.clone())?;
+    spinner.finish();
+    let spinner = Spinner::new(MSG_UPDATING_SUBMODULES_SPINNER);
+    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_reader(std::fs::File::open(updated_config_path)?)?;
+
+    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(&chain, &general_config)?;
+    }
+
+    logger::outro(MSG_ZKSYNC_UPDATED);
+
+    Ok(())
+}
+
+fn merge_yaml_internal(
+    a: &mut serde_yaml::Value,
+    b: serde_yaml::Value,
+    current_key: serde_yaml::Value,
+    diff: &mut ConfigDiff,
+) -> anyhow::Result<()> {
+    match (a, b) {
+        (serde_yaml::Value::Mapping(a), serde_yaml::Value::Mapping(b)) => {
+            for (key, value) in b {
+                if a.contains_key(&key) {
+                    merge_yaml_internal(a.get_mut(&key).unwrap(), value, key, diff)?;
+                } else {
+                    a.insert(key.clone(), value.clone());
+                    diff.added_fields.insert(key, value);
+                }
+            }
+        }
+        (a, b) => {
+            if a != &b {
+                diff.value_diff.insert(current_key, b);
+            }
+        }
+    }
+    Ok(())
+}
+
+fn merge_yaml(a: &mut serde_yaml::Value, b: serde_yaml::Value) -> anyhow::Result<ConfigDiff> {
+    let mut diff = ConfigDiff::default();
+    merge_yaml_internal(a, b, "".into(), &mut diff)?;
+    Ok(diff)
+}
+
+fn save_updated_config(
+    config: serde_yaml::Value,
+    path: &Path,
+    diff: ConfigDiff,
+) -> anyhow::Result<()> {
+    if diff.added_fields.is_empty() {
+        return Ok(());
+    }
+
+    logger::info(MSG_SHOW_DIFF);
+    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)?;
+
+    Ok(())
+}
+
+fn update_chain(chain: &ChainConfig, general_config: &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_reader(std::fs::File::open(current_general_config_path.clone())?)?;
+    let diff = merge_yaml(&mut current_general_config, general_config.clone())?;
+    save_updated_config(current_general_config, &current_general_config_path, diff)?;
+    Ok(())
+}

zk_toolbox/crates/zk_inception/src/consts.rs

@@ -7,3 +7,5 @@ pub const PROVER_STORE_MAX_RETRIES: u16 = 10;
 pub const DEFAULT_CREDENTIALS_FILE: &str = "~/.config/gcloud/application_default_credentials.json";
 pub const DEFAULT_PROOF_STORE_DIR: &str = "artifacts";
 pub const BELLMAN_CUDA_DIR: &str = "era-bellman-cuda";
+
+pub const GENERAL_FILE: &str = "general.yaml";

zk_toolbox/crates/zk_inception/src/main.rs

@@ -53,6 +53,8 @@ pub enum InceptionSubcommands {
     /// Run contract verifier
     #[command(subcommand)]
     ContractVerifier(ContractVerifierCommands),
+    /// Update zkSync
+    Update,
 }
 
 #[derive(Parser, Debug)]
@@ -110,6 +112,7 @@ async fn run_subcommand(inception_args: Inception, shell: &Shell) -> anyhow::Res
         InceptionSubcommands::ContractVerifier(args) => {
             commands::contract_verifier::run(shell, args).await?
         }
+        InceptionSubcommands::Update => commands::update::run(shell)?,
     }
     Ok(())
 }

zk_toolbox/crates/zk_inception/src/messages.rs

@@ -310,3 +310,16 @@ pub(super) fn msg_binary_already_exists(name: &str, version: &str) -> String {
 pub(super) fn msg_downloading_binary_spinner(name: &str, version: &str) -> String {
     format!("Downloading {} {} binary", name, version)
 }
+
+/// Update related messages
+
+pub(super) const MSG_UPDATING_ZKSYNC: &str = "Updating ZkSync";
+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) fn msg_updating_chain(chain: &str) -> String {
+    format!("Updating chain: {}", chain)
+}