From 4493dcfac5171f839b1b0e39444a15ce696adee4 Mon Sep 17 00:00:00 2001
From: Chris Swithinbank <swithinbank@gmail.com>
Date: Wed, 10 Apr 2024 22:10:25 +0200
Subject: [PATCH] Refactor `<ThemeSelect>` script to avoid double media query
 event listener (#1734)

---
 .changeset/sweet-vans-carry.md                |  5 ++
 .../starlight/components/ThemeSelect.astro    | 87 ++++++++-----------
 packages/starlight/global.d.ts                |  2 +-
 3 files changed, 42 insertions(+), 52 deletions(-)
 create mode 100644 .changeset/sweet-vans-carry.md

diff --git a/.changeset/sweet-vans-carry.md b/.changeset/sweet-vans-carry.md
new file mode 100644
index 00000000000..3005a1dcf23
--- /dev/null
+++ b/.changeset/sweet-vans-carry.md
@@ -0,0 +1,5 @@
+---
+'@astrojs/starlight': patch
+---
+
+Refactors `<ThemeSelect>` custom element logic to improve performance
diff --git a/packages/starlight/components/ThemeSelect.astro b/packages/starlight/components/ThemeSelect.astro
index 8467a43b35e..970156028ee 100644
--- a/packages/starlight/components/ThemeSelect.astro
+++ b/packages/starlight/components/ThemeSelect.astro
@@ -28,65 +28,50 @@ const { labels } = Astro.props;
 <script>
 	type Theme = 'auto' | 'dark' | 'light';
 
-	class StarlightThemeSelect extends HTMLElement {
-		/** Key in `localStorage` to store color theme preference at. */
-		#key = 'starlight-theme';
+	/** Key in `localStorage` to store color theme preference at. */
+	const storageKey = 'starlight-theme';
 
-		constructor() {
-			super();
-			this.#onThemeChange(this.#loadTheme());
-			const select = this.querySelector('select');
-			if (select) {
-				select.addEventListener('change', (e) => {
-					if (e.currentTarget instanceof HTMLSelectElement) {
-						this.#onThemeChange(this.#parseTheme(e.currentTarget.value));
-					}
-				});
-				matchMedia(`(prefers-color-scheme: light)`).addEventListener('change', () => {
-					if (this.#loadTheme() === 'auto') this.#onThemeChange('auto');
-				});
-			}
-		}
+	/** Get a typesafe theme string from any JS value (unknown values are coerced to `'auto'`). */
+	const parseTheme = (theme: unknown): Theme =>
+		theme === 'auto' || theme === 'dark' || theme === 'light' ? theme : 'auto';
 
-		/** Get a typesafe theme string from any JS value (unknown values are coerced to `'auto'`). */
-		#parseTheme(theme: unknown): Theme {
-			if (theme === 'auto' || theme === 'dark' || theme === 'light') {
-				return theme;
-			} else {
-				return 'auto';
-			}
-		}
+	/** Load the user’s preference from `localStorage`. */
+	const loadTheme = (): Theme =>
+		parseTheme(typeof localStorage !== 'undefined' && localStorage.getItem(storageKey));
 
-		/** Get the preferred system color scheme. */
-		#getPreferredColorScheme(): Theme {
-			return matchMedia('(prefers-color-scheme: light)').matches ? 'light' : 'dark';
+	/** Store the user’s preference in `localStorage`. */
+	function storeTheme(theme: Theme): void {
+		if (typeof localStorage !== 'undefined') {
+			localStorage.setItem(storageKey, theme === 'light' || theme === 'dark' ? theme : '');
 		}
+	}
 
-		/** Update select menu UI, document theme, and local storage state. */
-		#onThemeChange(theme: Theme): void {
-			StarlightThemeProvider.updatePickers(theme);
-			document.documentElement.dataset.theme =
-				theme === 'auto' ? this.#getPreferredColorScheme() : theme;
-			this.#storeTheme(theme);
-		}
+	/** Get the preferred system color scheme. */
+	const getPreferredColorScheme = (): Theme =>
+		matchMedia('(prefers-color-scheme: light)').matches ? 'light' : 'dark';
 
-		/** Store the user’s preference in `localStorage`. */
-		#storeTheme(theme: Theme): void {
-			if (typeof localStorage !== 'undefined') {
-				if (theme === 'light' || theme === 'dark') {
-					localStorage.setItem(this.#key, theme);
-				} else {
-					localStorage.removeItem(this.#key);
-				}
-			}
-		}
+	/** Update select menu UI, document theme, and local storage state. */
+	function onThemeChange(theme: Theme): void {
+		StarlightThemeProvider.updatePickers(theme);
+		document.documentElement.dataset.theme = theme === 'auto' ? getPreferredColorScheme() : theme;
+		storeTheme(theme);
+	}
 
-		/** Load the user’s preference from `localStorage`. */
-		#loadTheme(): Theme {
-			const theme = typeof localStorage !== 'undefined' && localStorage.getItem(this.#key);
-			return this.#parseTheme(theme);
+	// React to changes in system color scheme.
+	matchMedia(`(prefers-color-scheme: light)`).addEventListener('change', () => {
+		if (loadTheme() === 'auto') onThemeChange('auto');
+	});
+
+	class StarlightThemeSelect extends HTMLElement {
+		constructor() {
+			super();
+			onThemeChange(loadTheme());
+			this.querySelector('select')?.addEventListener('change', (e) => {
+				if (e.currentTarget instanceof HTMLSelectElement) {
+					onThemeChange(parseTheme(e.currentTarget.value));
+				}
+			});
 		}
 	}
-
 	customElements.define('starlight-theme-select', StarlightThemeSelect);
 </script>
diff --git a/packages/starlight/global.d.ts b/packages/starlight/global.d.ts
index 6b31ebeac65..13121ecaffa 100644
--- a/packages/starlight/global.d.ts
+++ b/packages/starlight/global.d.ts
@@ -1,4 +1,4 @@
-declare global {
+export declare global {
 	var StarlightThemeProvider: {
 		updatePickers(theme?: string): void;
 	};