From b252642725eda808d0674646ed0718fc6355da4b Mon Sep 17 00:00:00 2001
From: Richard Watts <richard@zilliqa.com>
Date: Mon, 3 Jun 2024 18:09:22 +0100
Subject: [PATCH] (feat) Track overridden zq2 source under its own name

---
 docgen/src/main.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/docgen/src/main.rs b/docgen/src/main.rs
index 6eaca86..0a903ea 100644
--- a/docgen/src/main.rs
+++ b/docgen/src/main.rs
@@ -29,7 +29,17 @@ async fn main() -> Result<()> {
     let root_path = PathBuf::from(&here);
     let versions: Zq2Spec =
         serde_yaml::from_str(&fs::read_to_string(format!("{}/zq2_spec.yaml", here))?)?;
-    for vrec in &versions.versions {
+    let overridden_zq2 = std::env::var("USE_ZQ2_FROM").is_ok();
+    let vtable = if overridden_zq2 {
+        vec![Version {
+            refspec: "use_zq2_from".to_string(),
+            name: Some("use_zq2_from".to_string()),
+        }]
+    } else {
+        versions.versions.clone()
+    };
+
+    for vrec in &vtable {
         let refspec = &vrec.refspec;
         let name: String = match vrec.name {
             None => {