From 7de8427a2854a33f535af29d9682148ef525c091 Mon Sep 17 00:00:00 2001
From: John Yani <361985+Vanuan@users.noreply.github.com>
Date: Tue, 21 Jan 2025 00:06:24 +0000
Subject: [PATCH] [Bonsai docs] Update version switching logic (#6005)

* Create brand.html

* Update custom.css for version switcher

* Update brand.html

* Update brand.html

Improve select box logic
---
 src/bonsai/docs/_templates/sidebar/brand.html | 33 +++++++++++++++----
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/src/bonsai/docs/_templates/sidebar/brand.html b/src/bonsai/docs/_templates/sidebar/brand.html
index f6c5c1ed0a..2952fdecf8 100644
--- a/src/bonsai/docs/_templates/sidebar/brand.html
+++ b/src/bonsai/docs/_templates/sidebar/brand.html
@@ -31,17 +31,38 @@ Hope your day's going well. :)
   <select class="doc-version-switcher" onchange="switchDocs()">
     <option value="stable">Stable</option>
     <option value="unstable">Unstable</option>
+    <!-- Add more versions here as needed -->
   </select>
 </a>
 
 <script>
+  // Define the mapping of versions to URLs
+  const versionURLs = {
+    stable: 'http://docs.bonsaibim.org/',
+    unstable: 'http://docs-unstable.bonsaibim.org/',
+    // Add more versions here as needed
+  };
+
+  // Set the select box value based on the current URL
+  document.addEventListener('DOMContentLoaded', (event) => {
+    const selectBox = document.querySelector('.doc-version-switcher');
+    const currentURL = window.location.href;
+
+    for (const [version, url] of Object.entries(versionURLs)) {
+      if (currentURL.includes(url)) {
+        selectBox.value = version;
+        break;
+      }
+    }
+  });
+
   function switchDocs() {
-    var selectBox = document.querySelector('.doc-version-switcher');
-    var selectedValue = selectBox.options[selectBox.selectedIndex].value;
-    if (selectedValue === 'unstable') {
-      window.location.href = 'http://docs-unstable.bonsaibim.org/';
-    } else {
-      window.location.href = 'http://docs.bonsaibim.org/';
+    const selectBox = document.querySelector('.doc-version-switcher');
+    const selectedValue = selectBox.value;
+    const currentURL = window.location.href;
+
+    if (!currentURL.includes(versionURLs[selectedValue])) {
+      window.location.href = versionURLs[selectedValue];
     }
   }
 </script>