From a252ebae60eb780725999805dc4714b2ef003294 Mon Sep 17 00:00:00 2001
From: Bjarte Stien Karlsen <bjarte@bjartek.org>
Date: Wed, 1 May 2024 01:13:47 +0200
Subject: [PATCH] added method to set network host

---
 setup.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/setup.go b/setup.go
index 685c0d6..1f75dc2 100644
--- a/setup.go
+++ b/setup.go
@@ -91,6 +91,7 @@ type OverflowBuilder struct {
 	PrintOptions                        *[]OverflowPrinterOption
 	GlobalEventFilter                   OverflowEventFilter
 	Path                                string
+	NetworkHost                         string
 	Network                             string
 	ScriptFolderName                    string
 	ServiceSuffix                       string
@@ -183,6 +184,9 @@ func (o *OverflowBuilder) StartResult() *OverflowState {
 		overflow.Error = err
 		return overflow
 	}
+	if o.NetworkHost != "" {
+		network.Host = o.NetworkHost
+	}
 	overflow.Network = *network
 
 	logger := output.NewStdoutLogger(o.LogLevel)
@@ -503,6 +507,13 @@ func WithEmulatorOption(opt ...emulator.Option) OverflowOption {
 	}
 }
 
+// Set custom network host if different from the one in flow.json since we cannot env substs there
+func WithNetworkHost(host string) OverflowOption {
+	return func(o *OverflowBuilder) {
+		o.NetworkHost = host
+	}
+}
+
 func WithUnderflowOptions(opt underflow.Options) OverflowOption {
 	return func(o *OverflowBuilder) {
 		o.UnderflowOptions = opt