From 064df6b01ef3530e1ee06f4df912c520049b5c60 Mon Sep 17 00:00:00 2001
From: Yi-Te Huang <44385685+ytdHuang@users.noreply.github.com>
Date: Mon, 11 Nov 2024 20:53:56 +0900
Subject: [PATCH] Fix `VersionPicker.vue` (#297)

---
 docs/src/.vitepress/theme/VersionPicker.vue | 148 +++++++++++++++-----
 1 file changed, 113 insertions(+), 35 deletions(-)

diff --git a/docs/src/.vitepress/theme/VersionPicker.vue b/docs/src/.vitepress/theme/VersionPicker.vue
index ba744d8f..b6505402 100644
--- a/docs/src/.vitepress/theme/VersionPicker.vue
+++ b/docs/src/.vitepress/theme/VersionPicker.vue
@@ -1,64 +1,142 @@
 <!-- Adapted from https://github.com/MakieOrg/Makie.jl/blob/master/docs/src/.vitepress/theme/VersionPicker.vue -->
 
 <script setup lang="ts">
-import { computed, ref, onMounted } from 'vue'
-import { useRoute } from 'vitepress'
+import { ref, onMounted, computed } from 'vue'
+import { useData } from 'vitepress'
 import VPNavBarMenuGroup from 'vitepress/dist/client/theme-default/components/VPNavBarMenuGroup.vue'
 import VPNavScreenMenuGroup from 'vitepress/dist/client/theme-default/components/VPNavScreenMenuGroup.vue'
 
+// Extend the global Window interface to include DOC_VERSIONS and DOCUMENTER_CURRENT_VERSION
+declare global {
+  interface Window {
+    DOC_VERSIONS?: string[];
+    DOCUMENTER_CURRENT_VERSION?: string;
+  }
+}
+
 const props = defineProps<{
   screenMenu?: boolean
 }>()
 
-const route = useRoute()
-
-const versions = ref([]);
+const versions = ref<Array<{ text: string, link: string }>>([]);
 const currentVersion = ref('Versions');
+const isClient = ref(false);
+const { site } = useData()
 
-const waitForGlobalDocumenterVars = () => {
-    return new Promise((resolve) => {
+const isLocalBuild = () => {
+  return typeof window !== 'undefined' && (window.location.hostname === 'localhost' || window.location.hostname === '127.0.0.1');
+}
+
+const getBaseRepository = () => {
+  if (typeof window === 'undefined') return ''; // Handle server-side rendering (SSR)
+  const { origin, pathname } = window.location;
+  // Check if it's a GitHub Pages (or similar) setup
+  if (origin.includes('github.io')) {
+    // Extract the first part of the path as the repository name
+    const pathParts = pathname.split('/').filter(Boolean);
+    const baseRepo = pathParts.length > 0 ? `/${pathParts[0]}/` : '/';
+    return `${origin}${baseRepo}`;
+  } else {
+    // For custom domains, use just the origin (e.g., https://docs.makie.org)
+    return origin;
+  }
+};
+
+const waitForScriptsToLoad = () => {
+  return new Promise<boolean>((resolve) => {
+    if (isLocalBuild()) {
+      resolve(false);
+      return;
+    }
     const checkInterval = setInterval(() => {
-        if (window.DOC_VERSIONS && window.DOCUMENTER_CURRENT_VERSION) {
+      if (window.DOC_VERSIONS && window.DOCUMENTER_CURRENT_VERSION) {
         clearInterval(checkInterval);
-        resolve({
-            versions: window.DOC_VERSIONS,
-            currentVersion: window.DOCUMENTER_CURRENT_VERSION
-        });
-        }
-    }, 100); // Check every 100ms
-    });
+        resolve(true);
+      }
+    }, 100);
+    // Timeout after 5 seconds
+    setTimeout(() => {
+      clearInterval(checkInterval);
+      resolve(false);
+    }, 5000);
+  });
 };
 
-onMounted(async () => {
-    const globalvars = await waitForGlobalDocumenterVars();
-    versions.value = globalvars.versions.map((v) => {
-        return {text: v, link: `${window.location.origin}/${v}/`}
-    });
-    currentVersion.value = globalvars.currentVersion;
-});
+const loadVersions = async () => {
+  if (typeof window === 'undefined') return; // Guard for SSR
+
+  try {
+    if (isLocalBuild()) {
+      // Handle the local build scenario directly
+      const fallbackVersions = ['dev'];
+      versions.value = fallbackVersions.map(v => ({
+        text: v,
+        link: '/'
+      }));
+      currentVersion.value = 'dev';
+    } else {
+      // For non-local builds, wait for scripts to load
+      const scriptsLoaded = await waitForScriptsToLoad();
+      const getBaseRepositoryPath = computed(() => {
+        return getBaseRepository();
+      });
 
+      if (scriptsLoaded && window.DOC_VERSIONS && window.DOCUMENTER_CURRENT_VERSION) {
+        versions.value = window.DOC_VERSIONS.map((v: string) => ({
+          text: v,
+          link: `${getBaseRepositoryPath.value}/${v}/`
+        }));
+        currentVersion.value = window.DOCUMENTER_CURRENT_VERSION;
+      } else {
+        // Fallback logic if scripts fail to load or are not available
+        const fallbackVersions = ['dev'];
+        versions.value = fallbackVersions.map(v => ({
+          text: v,
+          link: `${getBaseRepositoryPath.value}/${v}/`
+        }));
+        currentVersion.value = 'dev';
+      }
+    }
+  } catch (error) {
+    console.warn('Error loading versions:', error);
+    // Use fallback logic in case of an error
+    const fallbackVersions = ['dev'];
+    const getBaseRepositoryPath = computed(() => {
+        return getBaseRepository();
+      });
+    versions.value = fallbackVersions.map(v => ({
+      text: v,
+      link: `${getBaseRepositoryPath.value}/${v}/`
+    }));
+    currentVersion.value = 'dev';
+  }
+  isClient.value = true;
+};
+
+onMounted(loadVersions);
 </script>
 
 <template>
-  <VPNavBarMenuGroup
-    v-if="!screenMenu"
-    :item="{ text: currentVersion, items: versions }"
-    class="VPVersionPicker"
-  />
-  <VPNavScreenMenuGroup
-    v-else
-    :text="currentVersion"
-    :items="versions"
-    class="VPVersionPicker"
-  />
+  <template v-if="isClient">
+    <VPNavBarMenuGroup
+      v-if="!screenMenu && versions.length > 0"
+      :item="{ text: currentVersion, items: versions }"
+      class="VPVersionPicker"
+    />
+    <VPNavScreenMenuGroup
+      v-else-if="screenMenu && versions.length > 0"
+      :text="currentVersion"
+      :items="versions"
+      class="VPVersionPicker"
+    />
+  </template>
 </template>
 
 <style scoped>
 .VPVersionPicker :deep(button .text) {
   color: var(--vp-c-text-1) !important;
 }
-
 .VPVersionPicker:hover :deep(button .text) {
   color: var(--vp-c-text-2) !important;
 }
-</style>
\ No newline at end of file
+</style>