From c38b81c254b28e8b7fe6f4509f7ca1389b0b8fc4 Mon Sep 17 00:00:00 2001
From: Anton <anton@northernforest.nl>
Date: Fri, 9 Feb 2024 10:28:45 +0200
Subject: [PATCH] feat: Add suave-enabled geth support (#489)

Very minor addition to enable running suave-enabled geth (a.k.a.
`suave-execution-geth`) instead of the stock geths.
---
 src/el/geth/geth_launcher.star | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/el/geth/geth_launcher.star b/src/el/geth/geth_launcher.star
index 727818f7c..7b0ab6941 100644
--- a/src/el/geth/geth_launcher.star
+++ b/src/el/geth/geth_launcher.star
@@ -66,6 +66,7 @@ VERBOSITY_LEVELS = {
 }
 
 BUILDER_IMAGE_STR = "builder"
+SUAVE_ENABLED_GETH_IMAGE_STR = "suave"
 
 
 def launch(
@@ -293,6 +294,13 @@ def get_config(
             if "--ws.api" in arg:
                 cmd[index] = "--ws.api=admin,engine,net,eth,web3,debug,mev,flashbots"
 
+    if SUAVE_ENABLED_GETH_IMAGE_STR in image:
+        for index, arg in enumerate(cmd):
+            if "--http.api" in arg:
+                cmd[index] = "--http.api=admin,engine,net,eth,web3,debug,suavex"
+            if "--ws.api" in arg:
+                cmd[index] = "--ws.api=admin,engine,net,eth,web3,debug,suavex"
+
     if network == constants.NETWORK_NAME.kurtosis:
         if len(existing_el_clients) > 0:
             cmd.append(