feat(zk_toolbox): Add prover generate-sk command

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

@@ -2,4 +2,5 @@ pub mod args;
 pub mod chain;
 pub mod containers;
 pub mod ecosystem;
+pub mod prover;
 pub mod server;

zk_toolbox/crates/zk_inception/src/commands/prover/args/init.rs

@@ -0,0 +1,5 @@
+use clap::Parser;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Clone, Serialize, Deserialize, Parser)]
+pub struct InitArgs {}

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

@@ -0,0 +1 @@
+pub mod init;

zk_toolbox/crates/zk_inception/src/commands/prover/init.rs

@@ -0,0 +1,29 @@
+use anyhow::Ok;
+use common::{cmd::Cmd, logger, spinner::Spinner};
+use config::EcosystemConfig;
+use xshell::{cmd, Shell};
+
+use super::args::init::InitArgs;
+use crate::messages::{MSG_GENERATING_VK_SPINNER, MSG_INITIALIZING_PROVER, MSG_PROVER_INITIALIZED};
+
+pub(crate) async fn run(_args: InitArgs, shell: &Shell) -> anyhow::Result<()> {
+    logger::info(MSG_INITIALIZING_PROVER);
+    let ecosystem_config = EcosystemConfig::from_file(shell)?;
+    let mut link_to_prover = ecosystem_config.link_to_code.into_os_string();
+    link_to_prover.push("/prover");
+    shell.change_dir(link_to_prover.clone());
+
+    let spinner = Spinner::new(MSG_GENERATING_VK_SPINNER);
+    let mut cmd = Cmd::new(cmd!(
+        shell,
+        "cargo run --features gpu --release --bin key_generator -- 
+            generate-sk all --recompute-if-missing 
+            --setup-path=vk_setup_data_generator_server_fri/data 
+            --path={link_to_prover}/vk_setup_data_generator_server_fri/data"
+    ));
+    cmd.run()?;
+    spinner.finish();
+    logger::outro(MSG_PROVER_INITIALIZED);
+
+    Ok(())
+}

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

@@ -0,0 +1,17 @@
+use args::init::InitArgs;
+use clap::Subcommand;
+use xshell::Shell;
+mod args;
+mod init;
+
+#[derive(Subcommand, Debug)]
+pub enum ProverCommands {
+    /// Initialize prover
+    Init(InitArgs),
+}
+
+pub(crate) async fn run(shell: &Shell, args: ProverCommands) -> anyhow::Result<()> {
+    match args {
+        ProverCommands::Init(args) => init::run(args, shell).await,
+    }
+}

zk_toolbox/crates/zk_inception/src/main.rs

@@ -7,7 +7,9 @@ use common::{
 use config::EcosystemConfig;
 use xshell::Shell;
 
-use crate::commands::{args::RunServerArgs, chain::ChainCommands, ecosystem::EcosystemCommands};
+use crate::commands::{
+    args::RunServerArgs, chain::ChainCommands, ecosystem::EcosystemCommands, prover::ProverCommands,
+};
 
 pub mod accept_ownership;
 mod commands;
@@ -35,6 +37,9 @@ pub enum InceptionSubcommands {
     /// Chain related commands
     #[command(subcommand)]
     Chain(ChainCommands),
+    /// Prover related commands
+    #[command(subcommand)]
+    Prover(ProverCommands),
     /// Run server
     Server(RunServerArgs),
     /// Run containers for local development
@@ -101,6 +106,7 @@ async fn run_subcommand(inception_args: Inception, shell: &Shell) -> anyhow::Res
     match inception_args.command {
         InceptionSubcommands::Ecosystem(args) => commands::ecosystem::run(shell, args).await?,
         InceptionSubcommands::Chain(args) => commands::chain::run(shell, args).await?,
+        InceptionSubcommands::Prover(args) => commands::prover::run(shell, args).await?,
         InceptionSubcommands::Server(args) => commands::server::run(shell, args)?,
         InceptionSubcommands::Containers => commands::containers::run(shell)?,
     }

zk_toolbox/crates/zk_inception/src/messages.rs

@@ -173,3 +173,8 @@ pub(super) fn msg_address_doesnt_have_enough_money_prompt(address: &H160) -> Str
         "Address {address:?} doesn't have enough money to deploy contracts do you want to try again?"
     )
 }
+
+/// Prover related messages
+pub(super) const MSG_INITIALIZING_PROVER: &str = "Initializing prover";
+pub(super) const MSG_PROVER_INITIALIZED: &str = "Prover initialized successfully";
+pub(super) const MSG_GENERATING_VK_SPINNER: &str = "Generating verification keys...";