feat(zkstack_cli): Add the ability to run the server from Docker without building it

zkstack_cli/Cargo.toml

@@ -53,6 +53,8 @@ lazy_static = "1.4.0"
 once_cell = "1.19.0"
 prost = "0.12.1"
 rand = "0.8.5"
+reqwest = { version = "0.12.8", features = ["json"] }
+semver = "1.0.23"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 serde_yaml = "0.9"

zkstack_cli/crates/common/Cargo.toml

@@ -30,3 +30,5 @@ xshell.workspace = true
 thiserror.workspace = true
 strum.workspace = true
 git_version_macro.workspace = true
+reqwest.workspace = true
+semver.workspace = true

zkstack_cli/crates/common/src/github.rs

@@ -0,0 +1,112 @@
+use anyhow::{Context, Result};
+use reqwest::{
+    header::{HeaderMap, HeaderValue},
+    Client,
+};
+use semver::Version;
+use serde::{Deserialize, Serialize};
+
+const GITHUB_API_REPO_URL: &str = "https://api.github.com/repos/matter-labs/zksync-era";
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct Tag {
+    pub name: String,
+    pub commit: Commit,
+    pub zipball_url: String,
+    pub tarball_url: String,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct Commit {
+    pub sha: String,
+    pub url: String,
+}
+
+/// Fetches and sorts GitHub repository tags by semantic version
+pub struct GitHubTagFetcher {
+    client: Client,
+    auth_token: Option<String>,
+}
+
+impl GitHubTagFetcher {
+    pub fn new(auth_token: Option<String>) -> Result<Self> {
+        let client = Client::new();
+        Ok(Self { client, auth_token })
+    }
+
+    pub async fn get_newest_core_tags(&self, limit: Option<usize>) -> Result<Vec<Tag>> {
+        let mut all_tags = Vec::new();
+        let mut page = 1;
+
+        // Set up headers
+        let mut headers = HeaderMap::new();
+        headers.insert(
+            "Accept",
+            HeaderValue::from_static("application/vnd.github+json"),
+        );
+        headers.insert("User-Agent", HeaderValue::from_static("zkstack"));
+        headers.insert(
+            "X-GitHub-Api-Version",
+            HeaderValue::from_static("2022-11-28"),
+        );
+
+        if let Some(token) = &self.auth_token {
+            headers.insert(
+                "Authorization",
+                HeaderValue::from_str(&format!("Bearer {}", token))
+                    .context("Invalid authorization header")?,
+            );
+        }
+
+        // Fetch all pages
+        loop {
+            let url = format!("{}/tags?page={}&per_page=100", GITHUB_API_REPO_URL, page);
+
+            let response = self
+                .client
+                .get(&url)
+                .headers(headers.clone())
+                .send()
+                .await?;
+
+            if !response.status().is_success() {
+                return Err(anyhow::anyhow!(
+                    "GitHub API request failed with status: {}",
+                    response.status()
+                ));
+            }
+
+            let page_tags: Vec<Tag> = response.json().await?;
+            if page_tags.is_empty() {
+                break;
+            }
+
+            all_tags.extend(page_tags);
+            page += 1;
+        }
+
+        // filter tag names containing "core"
+        all_tags.retain(|t| t.name.contains("core"));
+
+        // Sort tags by semantic version
+        all_tags.sort_by(|a, b| {
+            let version_a = clean_version(&a.name);
+            let version_b = clean_version(&b.name);
+            version_b.cmp(&version_a) // Reverse order (newest first)
+        });
+
+        // Apply limit if specified
+        Ok(if let Some(limit) = limit {
+            all_tags.into_iter().take(limit).collect()
+        } else {
+            all_tags
+        })
+    }
+}
+
+/// Cleans and parses version strings into semver::Version
+fn clean_version(tag_name: &str) -> Version {
+    // Remove "core-v" prefix and parse the version string
+    let cleaned = tag_name.trim_start_matches("core-v");
+    Version::parse(cleaned).unwrap_or_else(|_| Version::new(0, 0, 0))
+}

zkstack_cli/crates/common/src/lib.rs

@@ -12,6 +12,7 @@ pub mod external_node;
 pub mod files;
 pub mod forge;
 pub mod git;
+pub mod github;
 pub mod server;
 pub mod version;
 pub mod wallets;

zkstack_cli/crates/common/src/server.rs

@@ -1,8 +1,9 @@
 use std::{ffi::OsStr, path::PathBuf};
 
+use anyhow::Context;
 use xshell::{cmd, Shell};
 
-use crate::cmd::Cmd;
+use crate::{cmd::Cmd, github::GitHubTagFetcher, PromptSelect};
 
 /// Allows to perform server operations.
 #[derive(Debug)]
@@ -19,6 +20,15 @@ pub enum ServerMode {
     Genesis,
 }
 
+/// Possible execution modes.
+#[derive(Debug, Default)]
+pub enum ExecutionMode {
+    #[default]
+    Release,
+    Debug,
+    Docker,
+}
+
 impl Server {
     /// Creates a new instance of the server.
     pub fn new(components: Option<Vec<String>>, code_path: PathBuf, uring: bool) -> Self {
@@ -31,16 +41,20 @@ impl Server {
 
     /// Runs the server.
     #[allow(clippy::too_many_arguments)]
-    pub fn run<P>(
+    pub async fn run<P>(
         &self,
         shell: &Shell,
+        execution_mode: ExecutionMode,
         server_mode: ServerMode,
+        configs_folder: P,
+        chains_folder: P,
         genesis_path: P,
         wallets_path: P,
         general_path: P,
         secrets_path: P,
         contracts_path: P,
         mut additional_args: Vec<String>,
+        tag: Option<String>,
     ) -> anyhow::Result<()>
     where
         P: AsRef<OsStr>,
@@ -56,28 +70,21 @@ impl Server {
 
         let uring = self.uring.then_some("--features=rocksdb/io-uring");
 
-        let mut cmd = Cmd::new(
-            cmd!(
-                shell,
-                "cargo run --release --bin zksync_server {uring...} --
-                --genesis-path {genesis_path}
-                --wallets-path {wallets_path}
-                --config-path {general_path}
-                --secrets-path {secrets_path}
-                --contracts-config-path {contracts_path}
-                "
-            )
-            .args(additional_args)
-            .env_remove("RUSTUP_TOOLCHAIN"),
-        );
-
-        // If we are running server in normal mode
-        // we need to get the output to the console
-        if let ServerMode::Normal = server_mode {
-            cmd = cmd.with_force_run();
-        }
-
-        cmd.run()?;
+        run_server(
+            shell,
+            uring,
+            chains_folder,
+            genesis_path,
+            wallets_path,
+            general_path,
+            secrets_path,
+            contracts_path,
+            additional_args,
+            execution_mode,
+            server_mode,
+            tag,
+        )
+        .await?;
 
         Ok(())
     }
@@ -99,3 +106,120 @@ impl Server {
         })
     }
 }
+
+#[allow(clippy::too_many_arguments)]
+async fn run_server<P>(
+    shell: &Shell,
+    uring: Option<&str>,
+    chains_folder: P,
+    genesis_path: P,
+    wallets_path: P,
+    general_path: P,
+    secrets_path: P,
+    contracts_path: P,
+    additional_args: Vec<String>,
+    execution_mode: ExecutionMode,
+    server_mode: ServerMode,
+    tag: Option<String>,
+) -> anyhow::Result<()>
+where
+    P: AsRef<OsStr>,
+{
+    let mut cmd = match execution_mode {
+        ExecutionMode::Release => cargo_run(
+            shell,
+            true,
+            uring,
+            genesis_path,
+            wallets_path,
+            general_path,
+            secrets_path,
+            contracts_path,
+            additional_args,
+        ),
+        ExecutionMode::Debug => cargo_run(
+            shell,
+            false,
+            uring,
+            genesis_path,
+            wallets_path,
+            general_path,
+            secrets_path,
+            contracts_path,
+            additional_args,
+        ),
+        ExecutionMode::Docker => {
+            let tag = tag.unwrap_or(select_tag().await?.to_owned());
+
+            Cmd::new(cmd!(
+                shell,
+                "docker run
+                --platform linux/amd64
+                --net=host
+                -v {chains_folder}:/chains
+                matterlabs/server-v2:{tag}
+                --genesis-path {genesis_path}
+                --wallets-path {wallets_path}
+                --config-path {general_path}
+                --secrets-path {secrets_path}
+                --contracts-config-path {contracts_path}
+                {additional_args...}"
+            ))
+        }
+    };
+
+    // If we are running server in normal mode
+    // we need to get the output to the console
+    if let ServerMode::Normal = server_mode {
+        cmd = cmd.with_force_run();
+    }
+
+    cmd.run().context("Failed to run server")
+}
+
+async fn select_tag() -> anyhow::Result<String> {
+    let fetcher = GitHubTagFetcher::new(None)?;
+    let gh_tags = fetcher.get_newest_core_tags(Some(5)).await?;
+
+    let tags: Vec<String> = std::iter::once("latest".to_string())
+        .chain(
+            gh_tags
+                .iter()
+                .map(|r| r.name.trim_start_matches("core-").to_string()),
+        )
+        .collect();
+
+    Ok(PromptSelect::new("Select image", tags).ask().into())
+}
+
+fn cargo_run<'a, P>(
+    shell: &'a Shell,
+    release: bool,
+    uring: Option<&str>,
+    genesis_path: P,
+    wallets_path: P,
+    general_path: P,
+    secrets_path: P,
+    contracts_path: P,
+    additional_args: Vec<String>,
+) -> Cmd<'a>
+where
+    P: AsRef<OsStr>,
+{
+    let compilation_mode: &str = release.then_some("--release").unwrap_or("");
+
+    Cmd::new(
+        cmd!(
+            shell,
+            "cargo run {compilation_mode} --bin zksync_server {uring...} --
+            --genesis-path {genesis_path}
+            --wallets-path {wallets_path}
+            --config-path {general_path}
+            --secrets-path {secrets_path}
+            --contracts-config-path {contracts_path}
+            "
+        )
+        .args(additional_args)
+        .env_remove("RUSTUP_TOOLCHAIN"),
+    )
+}

zkstack_cli/crates/zkstack/Cargo.toml

@@ -44,7 +44,7 @@ zksync_consensus_crypto.workspace = true
 zksync_protobuf.workspace = true
 zksync_protobuf_config.workspace = true
 prost.workspace = true
-reqwest = "0.12.8"
+reqwest.workspace = true
 
 [dev-dependencies]
 rand.workspace = true

zkstack_cli/crates/zkstack/completion/_zkstack.zsh

@@ -2006,9 +2006,11 @@ esac
 ;;
 (server)
 _arguments "${_arguments_options[@]}" : \
+'--mode=[]:MODE:(release debug docker)' \
 '*--components=[Components of server to run]:COMPONENTS:_default' \
 '*-a+[Additional arguments that can be passed through the CLI]:ADDITIONAL_ARGS:_default' \
 '*--additional-args=[Additional arguments that can be passed through the CLI]:ADDITIONAL_ARGS:_default' \
+'--tag=[]:TAG:_default' \
 '--chain=[Chain to use]:CHAIN:_default' \
 '--genesis[Run server in genesis mode]' \
 '--uring[Enables uring support for RocksDB]' \
@@ -2039,9 +2041,11 @@ _arguments "${_arguments_options[@]}" : \
 ;;
 (run)
 _arguments "${_arguments_options[@]}" : \
+'--mode=[]:MODE:(release debug docker)' \
 '*--components=[Components of server to run]:COMPONENTS:_default' \
 '*-a+[Additional arguments that can be passed through the CLI]:ADDITIONAL_ARGS:_default' \
 '*--additional-args=[Additional arguments that can be passed through the CLI]:ADDITIONAL_ARGS:_default' \
+'--tag=[]:TAG:_default' \
 '--chain=[Chain to use]:CHAIN:_default' \
 '--genesis[Run server in genesis mode]' \
 '--uring[Enables uring support for RocksDB]' \