From ba178736b1423afa099980e39b98e0450b7692f2 Mon Sep 17 00:00:00 2001
From: jfldde <168934971+jfldde@users.noreply.github.com>
Date: Mon, 6 Jan 2025 14:09:40 +0000
Subject: [PATCH] Allow passing network arg

---
 src/node.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/node.rs b/src/node.rs
index 0662002..2ef4a80 100644
--- a/src/node.rs
+++ b/src/node.rs
@@ -283,8 +283,13 @@ where
     let node_config_args = config.get_node_config_args().unwrap_or_default();
     let rollup_config_args = config.get_rollup_config_args();
 
+    let network_arg = match std::env::var("CITREA_NETWORK") {
+        Ok(network) => vec!["--network".to_string(), network],
+        _ => vec!["--dev".to_string()],
+    };
+
     [
-        vec!["--dev".to_string()],
+        network_arg,
         vec!["--da-layer".to_string(), config.da_layer().to_string()],
         node_config_args,
         rollup_config_args,