From cf26e8aa9c2c575634ce319e5a9b93be4c7f72ab Mon Sep 17 00:00:00 2001
From: Brad Nelson <flagxor@gmail.com>
Date: Sat, 6 Mar 2021 00:44:21 -0800
Subject: [PATCH] More menu tweaks.

---
 ueforth/site/menu.html | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/ueforth/site/menu.html b/ueforth/site/menu.html
index 46f0b1e..f900b42 100644
--- a/ueforth/site/menu.html
+++ b/ueforth/site/menu.html
@@ -7,23 +7,26 @@
 </div>
 <script>
 (function() {
+  function Behead(full, link) {
+    if (full) {
+      link = link.replace('https://eforth.appspot.com/', './');
+      link = link.replace('https://esp32forth.appspot.com/', './');
+    }
+    return link.split('?')[0];
+  }
   var anchors = document.getElementsByTagName('a');
   if (window.location.href.indexOf('file://') == 0) {
     for (var i = 0; i < anchors.length; ++i) {
-      anchors[i].href = anchors[i].href.replace('https://eforth.appspot.com/', './');
-      anchors[i].href = anchors[i].href.replace('https://esp32forth.appspot.com/', './');
+      anchors[i].href = Behead(true, anchors[i].href);
     }
   }
   for (var i = 0; i < anchors.length; ++i) {
-    var link = anchors[i].href;
-    link = link.replace('https://eforth.appspot.com/', './');
-    link = link.replace('https://esp32forth.appspot.com/', './');
-    if (link.split('?')[0] == window.location.href.split('?')[0]) {
-      if (anchors[i].href.split('?')[0] == window.location.href.split('?')[0]) {
+    if (Behead(true, anchors[i].href) == Behead(true, window.location.href)) {
+      if (Behead(false, anchors[i].href) == Behead(false, window.location.href)) {
         anchors[i].parentElement.classList.add('picked');
       } else {
         // Redirect if on the wrong host.
-        window.location.replace(anchors[i].ref);
+        window.location.replace(anchors[i].href);
       }
     }
   }