From bf2949722960831199e55b3feb976aab43d9b2a6 Mon Sep 17 00:00:00 2001
From: cdamian <17934949+cdamian@users.noreply.github.com>
Date: Wed, 1 Nov 2023 17:22:51 +0100
Subject: [PATCH] companion: Build relay block after onboarding

---
 core/src/builder/companion/src/expand/mod.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/core/src/builder/companion/src/expand/mod.rs b/core/src/builder/companion/src/expand/mod.rs
index 29ff9bd..3ecc5f5 100644
--- a/core/src/builder/companion/src/expand/mod.rs
+++ b/core/src/builder/companion/src/expand/mod.rs
@@ -123,6 +123,13 @@ pub fn expand(def: CompanionDef) -> SynResult<TokenStream> {
 
 				)*
 
+				{
+					__hidden_tracing::enter_span!(sp_tracing::Level::TRACE, std::stringify!(#relay_chain_name - BlockBuilding:));
+
+					companion.#relay_chain_name.build_block().map_err(|e| BuilderError::Relaychain(e.into())).map(|_| ())?;
+					companion.#relay_chain_name.import_block().map_err(|e| BuilderError::Relaychain(e.into())).map(|_| ())?;
+				}
+
 				Ok(companion)
 			}