remove the option for standard building

Author: akundaz

src/args.rs

@@ -83,9 +83,9 @@ pub struct FlashblocksArgs {
     	name = "WS_ADDRESS",
 		env = "FLASHBLOCKS_WS_ADDRESS",
     	num_args = 0..=1,
-		default_missing_value = "0.0.0.0:10111"
+		default_value = "0.0.0.0:10111"
 	)]
-	enabled: Option<SocketAddr>,
+	pub ws_address: SocketAddr,
 
 	/// Should we calculate the state root for each flashblock
 	#[arg(
@@ -106,18 +106,6 @@ impl Default for BuilderArgs {
 	}
 }
 
-impl FlashblocksArgs {
-	/// Returns true if flashblocks are enabled.
-	pub fn enabled(&self) -> bool {
-		self.enabled.is_some()
-	}
-
-	/// Returns the WebSocket address for flashblocks publishing socket.
-	pub fn ws_address(&self) -> Option<SocketAddr> {
-		self.enabled
-	}
-}
-
 /// This type is used to store the builder's secret key for signing the last
 /// transaction in the block.
 #[derive(Debug, Clone, Deref, FromStr, Into, From)]
@@ -220,10 +208,12 @@ impl FlashblocksArgs {
 		Self {
 			interval: Duration::from_millis(250),
 			leeway_time: Duration::from_millis(75),
-			enabled: Some(
-				SocketAddrV4::new(Ipv4Addr::UNSPECIFIED, Self::get_available_port())
-					.into(),
-			),
+			ws_address: SocketAddrV4::new(
+				Ipv4Addr::UNSPECIFIED,
+				Self::get_available_port(),
+			)
+			.into(),
+
 			calculate_state_root: true,
 		}
 	}
@@ -237,10 +227,12 @@ impl FlashblocksArgs {
 		Self {
 			interval,
 			leeway_time,
-			enabled: Some(
-				SocketAddrV4::new(Ipv4Addr::UNSPECIFIED, Self::get_available_port())
-					.into(),
-			),
+			ws_address: SocketAddrV4::new(
+				Ipv4Addr::UNSPECIFIED,
+				Self::get_available_port(),
+			)
+			.into(),
+
 			calculate_state_root: true,
 		}
 	}

src/main.rs

@@ -80,54 +80,6 @@ fn build_pipeline(
 	cli_args: &BuilderArgs,
 	pool: &OrderPool<Flashblocks>,
 ) -> eyre::Result<Pipeline<Flashblocks>> {
-	let pipeline = if cli_args.flashblocks_args.enabled() {
-		build_flashblocks_pipeline(cli_args, pool)?
-	} else {
-		build_classic_pipeline(cli_args, pool)
-	};
-	pool.attach_pipeline(&pipeline);
-
-	Ok(pipeline)
-}
-
-/// Classic block builder
-///
-/// Block building strategy that builds blocks using the classic approach by
-/// producing one block payload per CL payload job.
-fn build_classic_pipeline(
-	cli_args: &BuilderArgs,
-	pool: &OrderPool<Flashblocks>,
-) -> Pipeline<Flashblocks> {
-	if cli_args.revert_protection {
-		Pipeline::<Flashblocks>::named("classic")
-			.with_prologue(OptimismPrologue)
-			.with_pipeline(
-				Loop,
-				(
-					AppendOrders::from_pool(pool),
-					OrderByPriorityFee::default(),
-					RemoveRevertedTransactions::default(),
-				),
-			)
-	} else {
-		Pipeline::<Flashblocks>::named("classic")
-			.with_prologue(OptimismPrologue)
-			.with_pipeline(
-				Loop,
-				(AppendOrders::from_pool(pool), OrderByPriorityFee::default()),
-			)
-	}
-}
-
-fn build_flashblocks_pipeline(
-	cli_args: &BuilderArgs,
-	pool: &OrderPool<Flashblocks>,
-) -> eyre::Result<Pipeline<Flashblocks>> {
-	let socket_address = cli_args
-		.flashblocks_args
-		.ws_address()
-		.expect("WebSocket address must be set for Flashblocks");
-
 	// how often a flashblock is published
 	let interval = cli_args.flashblocks_args.interval;
 
@@ -139,7 +91,7 @@ fn build_flashblocks_pipeline(
 	// building and the payload job deadline that is given by the CL.
 	let total_building_time = Minus(leeway_time);
 
-	let ws = Arc::new(WebSocketSink::new(socket_address)?);
+	let ws = Arc::new(WebSocketSink::new(cli_args.flashblocks_args.ws_address)?);
 
 	let flashblock_building_pipeline_steps = (
 		AppendOrders::from_pool(pool).with_ok_on_limit(),
@@ -191,7 +143,9 @@ fn build_flashblocks_pipeline(
 				.with_step(BreakAfterMaxFlashblocks::new(flashblock_number)),
 		)
 		.with_limits(Scaled::default().deadline(total_building_time));
+
 	ws.watch_shutdown(&pipeline);
+	pool.attach_pipeline(&pipeline);
 
 	Ok(pipeline)
 }

src/tests/args.rs

@@ -58,7 +58,7 @@ fn test_leeway_time_invalid_format() {
 }
 
 #[test]
-fn test_leeway_time_with_flashblocks_enabled() {
+fn test_leeway_time() {
 	let args = Cli::try_parse_from([
 		"flashblocks",
 		"node",
@@ -72,11 +72,7 @@ fn test_leeway_time_with_flashblocks_enabled() {
 	if let Commands::Node(node_command) = args.command {
 		let flashblocks_args = &node_command.ext.flashblocks_args;
 		assert_eq!(flashblocks_args.leeway_time, Duration::from_millis(150));
-		assert!(flashblocks_args.enabled());
-		assert_eq!(
-			flashblocks_args.ws_address().unwrap().to_string(),
-			"127.0.0.1:8080"
-		);
+		assert_eq!(flashblocks_args.ws_address.to_string(), "127.0.0.1:8080");
 	} else {
 		panic!("Expected Node command");
 	}

src/tests/bundle.rs

@@ -14,7 +14,7 @@ use {
 
 #[tokio::test]
 async fn one_valid_tx_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let bundle_with_one_tx = random_valid_bundle(1);
 	let bundle_hash = bundle_with_one_tx.hash();
@@ -40,7 +40,7 @@ async fn one_valid_tx_included() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn two_valid_txs_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let bundle_with_two_txs = random_valid_bundle(2);
 	let bundle_hash = bundle_with_two_txs.hash();
@@ -67,7 +67,7 @@ async fn two_valid_txs_included() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn min_block_timestamp_constraint() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let mut bundle_with_one_tx = random_valid_bundle(1);
 	bundle_with_one_tx.min_block_number = Some(3);

src/tests/flashblocks.rs

@@ -10,7 +10,7 @@ use {
 
 #[tokio::test]
 async fn empty_blocks_smoke() -> eyre::Result<()> {
-	let (node, ws_addr) = Flashblocks::test_node_with_flashblocks_on().await?;
+	let (node, ws_addr) = Flashblocks::test_node().await?;
 	let ws = WebSocketObserver::new(ws_addr).await?;
 
 	for i in 1..=5 {
@@ -38,7 +38,7 @@ async fn blocks_with_txs_smoke() -> eyre::Result<()> {
 	const BLOCKS: usize = 5;
 	const TXS_PER_BLOCK: usize = 60;
 
-	let (node, ws_addr) = Flashblocks::test_node_with_flashblocks_on().await?;
+	let (node, ws_addr) = Flashblocks::test_node().await?;
 	let ws = WebSocketObserver::new(ws_addr).await?;
 
 	for i in 1..=BLOCKS {
@@ -98,7 +98,7 @@ async fn flashblock_timings_2000ms_block_time_0ms_leeway_time()
 	const TXS_PER_BLOCK: usize = 60;
 
 	let (node, ws_addr) =
-		Flashblocks::test_node_with_flashblocks_on_and_custom_leeway_time_and_interval(
+		Flashblocks::test_node_with_custom_leeway_time_and_interval(
 			Duration::from_millis(0),
 			Duration::from_millis(250),
 		)
@@ -149,7 +149,7 @@ async fn flashblock_timings_2000ms_block_time_75ms_leeway_time()
 -> eyre::Result<()> {
 	const TXS_PER_BLOCK: usize = 60;
 
-	let (node, ws_addr) = Flashblocks::test_node_with_flashblocks_on().await?;
+	let (node, ws_addr) = Flashblocks::test_node().await?;
 	let ws = WebSocketObserver::new(ws_addr).await?;
 
 	// Create a block at the top of the timestamp second
@@ -197,7 +197,7 @@ async fn flashblock_timings_2000ms_block_time_500ms_leeway_time()
 	const TXS_PER_BLOCK: usize = 60;
 
 	let (node, ws_addr) =
-		Flashblocks::test_node_with_flashblocks_on_and_custom_leeway_time_and_interval(
+		Flashblocks::test_node_with_custom_leeway_time_and_interval(
 			Duration::from_millis(500),
 			Duration::from_millis(500),
 		)

src/tests/limits.rs

@@ -31,7 +31,7 @@ async fn run_test(
 	const TXS_PER_BLOCK: usize = 60;
 
 	let (node, ws_addr) =
-		Flashblocks::test_node_with_flashblocks_on_and_custom_leeway_time_and_interval(
+		Flashblocks::test_node_with_custom_leeway_time_and_interval(
 			leeway_time,
 			flashblocks_interval,
 		)

src/tests/mod.rs

@@ -1,5 +1,5 @@
 use {
-	crate::{args::BuilderArgs, bundle::FlashblocksBundle},
+	crate::bundle::FlashblocksBundle,
 	rand::{Rng, rng},
 	rblib::{
 		alloy::{

src/tests/revert.rs

@@ -19,7 +19,7 @@ use {
 
 #[tokio::test]
 async fn critical_reverted_tx_not_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let bundle_without_reverts = random_valid_bundle(1);
 	let bundle_with_reverts = random_bundle_with_reverts(0, 1);
@@ -54,7 +54,7 @@ async fn critical_reverted_tx_not_included() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn faliable_reverted_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	// create a bundle with one valid tx
 	let bundle_without_reverts = random_valid_bundle(1);
@@ -96,7 +96,7 @@ async fn faliable_reverted_included() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn faliable_optional_reverted_not_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	// create a bundle with one valid and one reverting tx
 	let mut bundle_with_reverts = random_bundle_with_reverts(1, 1);
@@ -127,11 +127,7 @@ async fn faliable_optional_reverted_not_included() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn when_disabled_reverted_txs_are_included() -> eyre::Result<()> {
-	let node = Flashblocks::test_node_with_cli_args(BuilderArgs {
-		revert_protection: false,
-		..Default::default()
-	})
-	.await?;
+	let (node, _) = Flashblocks::test_node_with_revert_protection_off().await?;
 
 	// create a bundle with one valid and one reverting tx
 	let mut bundle_with_reverts = random_bundle_with_reverts(1, 1);

src/tests/rpc.rs

@@ -29,7 +29,7 @@ macro_rules! assert_ineligible {
 /// should be rejected by the RPC before making it to the orders pool.
 #[tokio::test]
 async fn max_block_number_in_past() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let block = node.next_block().await?;
 	assert_eq!(block.number(), 1);
@@ -63,7 +63,7 @@ async fn max_block_number_in_past() -> eyre::Result<()> {
 #[tokio::test]
 async fn max_block_timestamp_in_past() -> eyre::Result<()> {
 	// node at genesis, block 0
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 	let genesis_timestamp = node.config().chain.genesis_timestamp();
 	let mut bundle = random_valid_bundle(1);
 	bundle.max_timestamp = Some(genesis_timestamp.saturating_sub(1));
@@ -81,7 +81,7 @@ async fn max_block_timestamp_in_past() -> eyre::Result<()> {
 
 #[tokio::test]
 async fn empty_bundle_rejected_by_rpc() -> eyre::Result<()> {
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	let empty_bundle = FlashblocksBundle::with_transactions(vec![]);
 	let result = BundlesApiClient::<Flashblocks>::send_bundle(

src/tests/standard.rs

@@ -18,7 +18,7 @@ use {
 #[tokio::test]
 async fn chain_produces_empty_blocks() -> eyre::Result<()> {
 	// builders signer is not configured, so won't produce a builder tx
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	for i in 1..5 {
 		let block = node.next_block().await?;
@@ -38,7 +38,7 @@ async fn chain_produces_blocks_with_txs() -> eyre::Result<()> {
 	const TXS_PER_BLOCK: usize = 5;
 
 	// builders signer is not configured, so won't produce a builder tx
-	let node = Flashblocks::test_node().await?;
+	let (node, _) = Flashblocks::test_node().await?;
 
 	for i in 1..=BLOCKS {
 		let mut txs = Vec::with_capacity(TXS_PER_BLOCK);
@@ -65,7 +65,7 @@ async fn chain_produces_blocks_with_txs() -> eyre::Result<()> {
 /// when the builder signer is provided in the CLI arguments.
 #[tokio::test]
 async fn blocks_have_builder_tx() -> eyre::Result<()> {
-	let node = Flashblocks::test_node_with_builder_signer().await?;
+	let (node, _) = Flashblocks::test_node_with_builder_signer().await?;
 
 	let block = node.next_block().await?;
 	debug!("produced block: {block:#?}");