From a5a04b2671d46a52ba17f32248e8c84da3be6106 Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Tue, 4 Feb 2025 16:22:54 +0800
Subject: [PATCH] fix: set default docker image - set nomos-cli command

---
 src/env_vars.py       | 6 ++++++
 src/node/node_vars.py | 8 +++++---
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/env_vars.py b/src/env_vars.py
index 78c051d..5e28e58 100644
--- a/src/env_vars.py
+++ b/src/env_vars.py
@@ -19,10 +19,16 @@ NOMOS = "nomos"
 NOMOS_EXECUTOR = "nomos_executor"
 CFGSYNC = "cfgsync"
 
+DEFAULT_IMAGE = "ghcr.io/logos-co/nomos-node:testnet"
+
 NODE_1 = get_env_var("NODE_1", NOMOS)
 NODE_2 = get_env_var("NODE_2", NOMOS_EXECUTOR)
 NODE_3 = get_env_var("NODE_3", CFGSYNC)
 
+NOMOS_IMAGE = get_env_var("NOMOS_IMAGE", DEFAULT_IMAGE)
+
+NOMOS_CLI = ["docker", "run", "--rm", "--entrypoint", "", NOMOS_IMAGE]
+
 ADDITIONAL_NODES = get_env_var("ADDITIONAL_NODES", f"{NOMOS},{NOMOS}")
 # more nodes need to follow the NODE_X pattern
 DOCKER_LOG_DIR = get_env_var("DOCKER_LOG_DIR", "./log/docker")
diff --git a/src/node/node_vars.py b/src/node/node_vars.py
index 333a749..a381084 100644
--- a/src/node/node_vars.py
+++ b/src/node/node_vars.py
@@ -1,18 +1,20 @@
+from src.env_vars import NOMOS_IMAGE
+
 nomos_nodes = {
     "nomos": {
-        "image": "nomos:latest",
+        "image": NOMOS_IMAGE,
         "volumes": ["cluster_config:/etc/nomos", "./kzgrs/kzgrs_test_params:/kzgrs_test_params:z"],
         "ports": ["3000/udp", "18080/tcp"],
         "entrypoint": "/etc/nomos/scripts/run_nomos_node.sh",
     },
     "nomos_executor": {
-        "image": "nomos:latest",
+        "image": NOMOS_IMAGE,
         "volumes": ["cluster_config:/etc/nomos", "./kzgrs/kzgrs_test_params:/kzgrs_test_params:z"],
         "ports": ["3000/udp", "18080/tcp"],
         "entrypoint": "/etc/nomos/scripts/run_nomos_executor.sh",
     },
     "cfgsync": {
-        "image": "nomos:latest",
+        "image": NOMOS_IMAGE,
         "volumes": ["cluster_config:/etc/nomos"],
         "ports": "",
         "entrypoint": "/etc/nomos/scripts/run_cfgsync.sh",