From f71e2f16d2cd69e26cc78aab63fb3b2c1d57a612 Mon Sep 17 00:00:00 2001
From: Nikos Kontakis <wirednkod@gmail.com>
Date: Mon, 31 Jul 2023 19:44:04 +0300
Subject: [PATCH] Add extra binaries in the setup command

---
 README.md                                    |  2 +-
 javascript/packages/cli/src/actions/setup.ts | 11 +++++++++--
 javascript/packages/cli/src/cli.ts           |  2 +-
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 776f27d9a..7f6b4915e 100644
--- a/README.md
+++ b/README.md
@@ -337,7 +337,7 @@ This allows to use the `setup` script, making everything ready for a ZombieNet d
 You can use the following arguments:
 
 `--help` shows the different options and commands for using the Zombienet CLI.
-`--binaries` or `-b`: enables providing the binaries that you want to be downloaded and installed during the setup. Possible options: `polkadot`, `polkadot-parachain`.
+`--binaries` or `-b`: enables providing the binaries that you want to be downloaded and installed during the setup. Possible options: `polkadot`, `polkadot-parachain`, `polkadot-prepare-worker`, `polkadot-execute-worker`
 
 For example:
 
diff --git a/javascript/packages/cli/src/actions/setup.ts b/javascript/packages/cli/src/actions/setup.ts
index 76b054582..3608e63e5 100644
--- a/javascript/packages/cli/src/actions/setup.ts
+++ b/javascript/packages/cli/src/actions/setup.ts
@@ -16,7 +16,12 @@ const options: OptIf = {};
  * @returns
  */
 export async function setup(params: any, opts?: any) {
-  const POSSIBLE_BINARIES = ["polkadot", "polkadot-parachain"];
+  const POSSIBLE_BINARIES = [
+    "polkadot",
+    "polkadot-parachain",
+    "polkadot-prepare-worker",
+    "polkadot-execute-worker",
+  ];
 
   console.log(decorators.green("\n\n🧟🧟🧟 ZombieNet Setup 🧟🧟🧟\n\n"));
   if (!["linux", "darwin"].includes(process.platform)) {
@@ -79,7 +84,9 @@ export async function setup(params: any, opts?: any) {
         decorators.red(
           `"${a}" is not one of the possible options for this setup and will be skipped;`,
         ),
-        decorators.green(` Valid options: polkadot polkadot-parachain`),
+        decorators.green(
+          ` Valid options: polkadot polkadot-parachain polkadot-prepare-worker polkadot-execute-worker`,
+        ),
       );
       return;
     }
diff --git a/javascript/packages/cli/src/cli.ts b/javascript/packages/cli/src/cli.ts
index 7760d5709..a5534a923 100644
--- a/javascript/packages/cli/src/cli.ts
+++ b/javascript/packages/cli/src/cli.ts
@@ -124,7 +124,7 @@ program
   )
   .argument(
     "<binaries...>",
-    `the binaries that you want to be downloaded, provided in a row without any separators;\nThey are downloaded in current directory and appropriate executable permissions are assigned.\nPossible options: 'polkadot', 'polkadot-parachain'\n${decorators.blue(
+    `the binaries that you want to be downloaded, provided in a row without any separators;\nThey are downloaded in current directory and appropriate executable permissions are assigned.\nPossible options: 'polkadot', 'polkadot-parachain', 'polkadot-prepare-worker', 'polkadot-execute-worker'\n${decorators.blue(
       "zombienet setup polkadot polkadot-parachain",
     )}`,
   )