From 7c7f30beedc704d47251fc2de90450a3ddd68bdb Mon Sep 17 00:00:00 2001
From: Theo Sanderson <theo@sndrsn.co.uk>
Date: Mon, 26 Feb 2024 12:55:19 +0000
Subject: [PATCH] organism selector stays on same page

---
 .../components/Navigation/OrganismSelector.astro | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/website/src/components/Navigation/OrganismSelector.astro b/website/src/components/Navigation/OrganismSelector.astro
index 41d8e56fc..eeab27dbf 100644
--- a/website/src/components/Navigation/OrganismSelector.astro
+++ b/website/src/components/Navigation/OrganismSelector.astro
@@ -1,11 +1,15 @@
 ---
 import { cleanOrganism } from './cleanOrganism';
+import { type Organism } from '../../config';
 import { routes } from '../../routes';
 import IwwaArrowDown from '~icons/iwwa/arrow-down';
 
 const { knownOrganisms, organism } = cleanOrganism(Astro.params.organism);
 
 const label = organism === undefined ? 'Organisms' : organism.displayName;
+const firstBitOfUrl = Astro.url.pathname.split('/')[1];
+const isOrganismPage = knownOrganisms.some((knownOrganism: Organism) => knownOrganism.key === firstBitOfUrl);
+const restOfUrl = Astro.url.pathname.split('/').slice(2).join('/');
 ---
 
 <div class='dropdown dropdown-hover hidden sm:flex relative'>
@@ -15,9 +19,17 @@ const label = organism === undefined ? 'Organisms' : organism.displayName;
     </label>
     <ul tabindex='0' class='dropdown-content z-[1] menu p-1 shadow bg-base-100 rounded-btn absolute top-full -left-4'>
         {
-            knownOrganisms.map((knownOrganism) => (
+            knownOrganisms.map((knownOrganism: Organism) => (
                 <li>
-                    <a href={routes.organismStartPage(knownOrganism.key)}>{knownOrganism.displayName}</a>
+                    <a
+                        href={
+                            isOrganismPage
+                                ? `/${knownOrganism.key}/${restOfUrl}`
+                                : routes.organismStartPage(knownOrganism.key)
+                        }
+                    >
+                        {knownOrganism.displayName}
+                    </a>
                 </li>
             ))
         }