feat(zkstack): Generating keys for compressor (#4301)

## What ❔
Added zkstack feature to generate keys for compressor
<!-- What are the changes this PR brings about? -->
<!-- Example: This PR adds a PR template to the repo. -->
<!-- (For bigger PRs adding more context is appreciated) -->

## Why ❔
There is no functionality in zkstack to generate keys for compressor.
<!-- Why are these changes done? What goal do they contribute to? What
are the principles behind them? -->
<!-- The `Why` has to be clear to non-Matter Labs entities running their
own ZK Chain -->
<!-- Example: PR templates ensure PR reviewers, observers, and future
iterators are in context about the evolution of repos. -->

## Is this a breaking change?
- [ ] Yes
- [ ] No

## Operational changes
<!-- Any config changes? Any new flags? Any changes to any scripts? -->
<!-- Please add anything that non-Matter Labs entities running their own
ZK Chain may need to know -->

## Checklist

<!-- Check your PR fulfills the following items. -->
<!-- For draft PRs check the boxes as you complete them. -->

- [ ] PR title corresponds to the body of PR (we generate changelog
entries from PRs).
- [ ] Tests for the changes have been added / updated.
- [ ] Documentation comments have been added / updated.
- [ ] Code has been formatted via `zkstack dev fmt` and `zkstack dev
lint`.

Author: AlexOnTheStorm

zkstack_cli/crates/config/src/general.rs

@@ -211,6 +211,10 @@ impl GeneralConfigPatch {
             .get("proof_compressor.universal_setup_download_url")
     }
 
+    pub fn proof_compressor_setup_path(&self) -> anyhow::Result<PathBuf> {
+        self.0.base().get("proof_compressor.universal_setup_path")
+    }
+
     pub fn set_proof_compressor_setup_path(&mut self, path: &Path) -> anyhow::Result<()> {
         self.0
             .insert_path("proof_compressor.universal_setup_path", path)

zkstack_cli/crates/zkstack/src/commands/prover/setup_keys.rs

@@ -1,13 +1,16 @@
-use anyhow::Ok;
+use anyhow::{Context, Ok};
 use xshell::{cmd, Shell};
 use zkstack_cli_common::{
     check_prerequisites, cmd::Cmd, logger, spinner::Spinner, GCLOUD_PREREQUISITE, GPU_PREREQUISITES,
 };
 use zkstack_cli_config::{get_link_to_prover, EcosystemConfig};
 
 use crate::{
-    commands::prover::args::setup_keys::{Mode, Region, SetupKeysArgs},
-    messages::{MSG_GENERATING_SK_SPINNER, MSG_SK_GENERATED},
+    commands::prover::{
+        args::setup_keys::{Mode, Region, SetupKeysArgs},
+        compressor_keys::download_compressor_key,
+    },
+    messages::{MSG_CHAIN_NOT_FOUND_ERR, MSG_GENERATING_SK_SPINNER, MSG_SK_GENERATED},
 };
 
 pub(crate) async fn run(args: SetupKeysArgs, shell: &Shell) -> anyhow::Result<()> {
@@ -16,6 +19,27 @@ pub(crate) async fn run(args: SetupKeysArgs, shell: &Shell) -> anyhow::Result<()
 
     if args.mode == Mode::Generate {
         check_prerequisites(shell, &GPU_PREREQUISITES, false);
+        let chain_config = ecosystem_config
+            .load_current_chain()
+            .context(MSG_CHAIN_NOT_FOUND_ERR)?;
+        let mut general_config = chain_config.get_general_config().await?.patched();
+        let proof_compressor_setup_path = general_config
+            .proof_compressor_setup_path()
+            .context("proof_compressor_setup_path")?;
+        if proof_compressor_setup_path.exists() {
+            logger::info(format!(
+                "Initial proof compressor setup already present at {}",
+                proof_compressor_setup_path.display()
+            ));
+        } else {
+            logger::info(format!(
+                "Initial proof compressor setup not found at {}, downloading",
+                proof_compressor_setup_path.display()
+            ));
+            download_compressor_key(shell, &mut general_config, &proof_compressor_setup_path)?;
+        }
+        std::env::set_var("COMPACT_CRS_FILE", &proof_compressor_setup_path);
+
         let link_to_prover = get_link_to_prover(&ecosystem_config);
         shell.change_dir(&link_to_prover);
 
@@ -28,6 +52,12 @@ pub(crate) async fn run(args: SetupKeysArgs, shell: &Shell) -> anyhow::Result<()
             --path={link_to_prover}/data/keys"
         ));
         cmd.run()?;
+        let cmd = Cmd::new(cmd!(
+            shell,
+            "cargo run --features gpu --release --bin key_generator -- 
+            generate-compressor-data"
+        ));
+        cmd.run()?;
         spinner.finish();
         logger::outro(MSG_SK_GENERATED);
     } else {