Add bproxy and websocket proxy

playground/catalog.go

@@ -25,6 +25,8 @@ func init() {
 	register(&OpRbuilder{})
 	register(&FlashblocksRPC{})
 	register(&Contender{})
+	register(&BProxy{})
+	register(&WebsocketProxy{})
 }
 
 func FindComponent(name string) ServiceGen {

playground/components.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/ethereum/go-ethereum/common/hexutil"
@@ -100,13 +101,31 @@ func (o *OpRbuilder) Name() string {
 
 type FlashblocksRPC struct {
 	FlashblocksWSService string
+	BaseOverlay bool
 }
 
 func (f *FlashblocksRPC) Run(service *Service, ctx *ExContext) {
-	service.WithImage("flashbots/flashblocks-rpc").
-		WithTag("sha-7caffb9").
-		WithArgs(
-			"node",
+	if f.BaseOverlay {
+		// Base doesn't have built image, so we use mikawamp/base-reth-node
+		service.WithImage("docker.io/mikawamp/base-reth-node").
+			WithTag("latest").
+			WithEntrypoint("/app/base-reth-node").
+			WithArgs(
+				"node",
+				// We use websocket proxy to connect to rollup-boost, so we need to add /ws to the url
+				"--websocket-url", ConnectWs(f.FlashblocksWSService, "flashblocks") + "/ws",
+			)
+	} else {
+		service.WithImage("flashbots/flashblocks-rpc").
+			WithTag("sha-7caffb9").
+			WithArgs(
+				"node",
+				"--flashblocks.enabled",
+				// We use websocket proxy to connect to rollup-boost, so we need to add /ws to the url
+				"--flashblocks.websocket-url", ConnectWs(f.FlashblocksWSService, "flashblocks") + "/ws",
+			)
+	}
+	service.WithArgs(
 			"--authrpc.port", `{{Port "authrpc" 8551}}`,
 			"--authrpc.addr", "0.0.0.0",
 			"--authrpc.jwtsecret", "/data/jwtsecret",
@@ -119,8 +138,6 @@ func (f *FlashblocksRPC) Run(service *Service, ctx *ExContext) {
 			"--color", "never",
 			"--metrics", `0.0.0.0:{{Port "metrics" 9090}}`,
 			"--port", `{{Port "rpc" 30303}}`,
-			"--flashblocks.enabled",
-			"--flashblocks.websocket-url", ConnectWs(f.FlashblocksWSService, "flashblocks"),
 		).
 		WithArtifact("/data/jwtsecret", "jwtsecret").
 		WithArtifact("/data/l2-genesis.json", "l2-genesis.json").
@@ -137,6 +154,73 @@ func (f *FlashblocksRPC) Name() string {
 	return "flashblocks-rpc"
 }
 
+type BProxy struct {
+	TargetAuthrpc string
+	Peers []string
+	Flashblocks bool
+	FlashblocksBuilderURL string
+}
+
+func (f* BProxy) Run(service *Service, ctx *ExContext) {
+	peers := []string{}
+	for _, peer := range f.Peers {
+		peers = append(peers, Connect(peer, "authrpc"))
+	}
+	service.WithImage("ghcr.io/flashbots/bproxy").
+		WithTag("v0.0.91").
+		WithArgs(
+			"serve",
+			"--authrpc-backend", f.TargetAuthrpc,
+			"--authrpc-backend-timeout", "5s",
+			"--authrpc-client-idle-connection-timeout", "15m",
+			"--authrpc-deduplicate-fcus",
+			"--authrpc-enabled",
+			"--authrpc-listen-address", `0.0.0.0:{{Port "authrpc" 8651}}`,
+			"--authrpc-log-requests",
+			"--authrpc-log-responses",
+			"--authrpc-max-backend-connections-per-host", "1",
+			"--authrpc-max-request-size", "150",
+			"--authrpc-max-response-size", "1150",
+			"--authrpc-peers", strings.Join(peers, ","),
+			"--authrpc-remove-backend-from-peers",
+			"--authrpc-use-priority-queue",
+		).
+		WithArtifact("/data/jwtsecret", "jwtsecret")
+
+	if f.Flashblocks {
+		service.WithArgs(
+			"--flashblocks-backend", f.FlashblocksBuilderURL,
+			"--flashblocks-enabled",
+			"--flashblocks-listen-address", `0.0.0.0:{{Port "flashblocks" 1114}}`,
+			"--flashblocks-log-messages",
+		)
+	}
+
+}
+
+func (f *BProxy) Name() string {
+	return "bproxy"
+}
+
+type WebsocketProxy struct {
+	Upstream string
+}
+
+func (w *WebsocketProxy) Run(service *Service, ctx *ExContext) {
+	service.WithImage("docker.io/mikawamp/websocket-rpc").
+		WithTag("latest").
+		WithArgs(
+			"--listen-addr", `0.0.0.0:{{Port "flashblocks" 1115}}`,
+			"--upstream-ws", ConnectWs(w.Upstream, "flashblocks"),
+			"--enable-compression",
+			"--client-ping-enabled",
+		)
+}
+
+func (w *WebsocketProxy) Name() string {
+	return "websocket-proxy"
+}
+
 type OpBatcher struct {
 	L1Node             string
 	L2Node             string

playground/recipe_opstack.go

@@ -30,6 +30,9 @@ type OpRecipe struct {
 	// flashblocksBuilderURL is the URL of the builder that returns the flashblocks. This is meant to be used
 	// for external builders.
 	flashblocksBuilderURL string
+
+	// Indicates that flashblocks-rpc should use base image
+	baseOverlay bool 
 }
 
 func (o *OpRecipe) Name() string {
@@ -47,6 +50,7 @@ func (o *OpRecipe) Flags() *flag.FlagSet {
 	flags.Uint64Var(&o.blockTime, "block-time", defaultOpBlockTimeSeconds, "Block time to use for the rollup")
 	flags.Uint64Var(&o.batcherMaxChannelDuration, "batcher-max-channel-duration", 2, "Maximum channel duration to use for the batcher")
 	flags.BoolVar(&o.flashblocks, "flashblocks", false, "Whether to enable flashblocks")
+	flags.BoolVar(&o.baseOverlay, "base-overlay", false, "Whether to use base implementation for flashblocks-rpc")
 	flags.StringVar(&o.flashblocksBuilderURL, "flashblocks-builder", "", "External URL of builder flashblocks stream")
 	return flags
 }
@@ -70,6 +74,7 @@ func (o *OpRecipe) Apply(ctx *ExContext, artifacts *Artifacts) *Manifest {
 
 	flashblocksBuilderURLRef := o.flashblocksBuilderURL
 	externalBuilderRef := o.externalBuilder
+	peers := []string{}
 
 	opGeth := &OpGeth{}
 	svcManager.AddService("op-geth", opGeth)
@@ -96,21 +101,38 @@ func (o *OpRecipe) Apply(ctx *ExContext, artifacts *Artifacts) *Manifest {
 		flashblocksBuilderURLRef = ConnectWs("op-rbuilder", "flashblocks")
 	}
 
+	if o.flashblocks {
+		peers = append(peers, "flashblocks-rpc")
+	}
+
+	svcManager.AddService("bproxy", &BProxy{
+		TargetAuthrpc: externalBuilderRef,
+		Peers: peers,
+		Flashblocks: o.flashblocks,
+		FlashblocksBuilderURL: flashblocksBuilderURLRef,
+	})
+
+	svcManager.AddService("websocket-proxy", &WebsocketProxy{
+		Upstream: "rollup-boost",
+	})
+
 	elNode := "op-geth"
 	if o.externalBuilder != "" {
 		elNode = "rollup-boost"
 
 		svcManager.AddService("rollup-boost", &RollupBoost{
 			ELNode:                "op-geth",
-			Builder:               externalBuilderRef,
+			Builder:               Connect("bproxy", "authrpc"),
 			Flashblocks:           o.flashblocks,
-			FlashblocksBuilderURL: flashblocksBuilderURLRef,
+			FlashblocksBuilderURL: ConnectWs("bproxy", "flashblocks"),
 		})
 	}
 
+
 	if o.flashblocks {
 		svcManager.AddService("flashblocks-rpc", &FlashblocksRPC{
-			FlashblocksWSService: "rollup-boost", // rollup-boost provides the websocket stream
+			FlashblocksWSService: "websocket-proxy", // rollup-boost provides the websocket stream
+			BaseOverlay: o.baseOverlay,
 		})
 	}