Fix NPE (#59)

martrapp · Apr 2, 2024 · 9e8532a · 9e8532a
1 parent 9c5b2c6
commit 9e8532a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/.changeset/tidy-pots-turn.md b/.changeset/tidy-pots-turn.md
@@ -0,0 +1,5 @@
+---
+"astro-vtbot": patch
+---
+
+Fixes a bug accessing an attribute of null when linking to view transition pages outside the starlight site.
diff --git a/components/starlight/StarlightConnector.astro b/components/starlight/StarlightConnector.astro
@@ -17,9 +17,7 @@ export interface Props {}
 	function closeMobileMenu() {
 		if (document.body.hasAttribute('data-mobile-menu-expanded')) {
 			document.body
-				.querySelector('starlight-menu-button')!
-				.closest('nav')!
-				.dispatchEvent(
+				.querySelector('starlight-menu-button')?.closest('nav')?.dispatchEvent(
 					new KeyboardEvent('keyup', {
 						key: 'Escape',
 						code: 'Escape',
@@ -35,7 +33,7 @@ export interface Props {}
 	}
 
 	function markMainFrameForReplacementSwap(doc: Document) {
-		doc.body.querySelector('div.main-frame')!.setAttribute('data-vtbot-replace', 'main');
+		doc.body.querySelector('div.main-frame')?.setAttribute('data-vtbot-replace', 'main');
 	}
 
 	function setTransitionScope(e: TransitionBeforePreparationEvent) {