From 54c42b6ef451a9b60144b05ec9b61b9e58eeb27e Mon Sep 17 00:00:00 2001
From: jack shelton <me@jackshelton.com>
Date: Mon, 22 Apr 2024 09:45:31 -0500
Subject: [PATCH] fix: popover flakiness

---
 .../popover/examples/programmatic.tsx         |  4 +--
 .../src/components/popover/popover-impl.tsx   |  2 +-
 .../components/popover/popover-trigger.tsx    | 25 +++++++++----------
 .../src/components/popover/popover.test.ts    |  9 +++++--
 4 files changed, 22 insertions(+), 18 deletions(-)

diff --git a/apps/website/src/routes/docs/headless/popover/examples/programmatic.tsx b/apps/website/src/routes/docs/headless/popover/examples/programmatic.tsx
index f80e1e7a3..03a29980b 100644
--- a/apps/website/src/routes/docs/headless/popover/examples/programmatic.tsx
+++ b/apps/website/src/routes/docs/headless/popover/examples/programmatic.tsx
@@ -7,9 +7,9 @@ export default component$(() => {
       <button
         class="popover-invoker"
         preventdefault:click
-        onKeyDown$={(e) => {
+        onKeyDown$={async (e) => {
           if (e.key === 'o') {
-            togglePopover();
+            await togglePopover();
           }
         }}
       >
diff --git a/packages/kit-headless/src/components/popover/popover-impl.tsx b/packages/kit-headless/src/components/popover/popover-impl.tsx
index d932cc364..1b39f8035 100644
--- a/packages/kit-headless/src/components/popover/popover-impl.tsx
+++ b/packages/kit-headless/src/components/popover/popover-impl.tsx
@@ -102,7 +102,7 @@ export const PopoverImpl = component$<PopoverImplProps>((props) => {
         hasTopLayerAncestorSig.value = true;
       }
 
-      document.dispatchEvent(new CustomEvent('showpopover'));
+      document.dispatchEvent(new CustomEvent('showpopoverpoly'));
 
       cleanup(() => popoverRef.value);
     }
diff --git a/packages/kit-headless/src/components/popover/popover-trigger.tsx b/packages/kit-headless/src/components/popover/popover-trigger.tsx
index 66c849c29..2a9f9378c 100644
--- a/packages/kit-headless/src/components/popover/popover-trigger.tsx
+++ b/packages/kit-headless/src/components/popover/popover-trigger.tsx
@@ -20,6 +20,7 @@ export function usePopover(popovertarget: string) {
 
   const didInteractSig = useSignal<boolean>(false);
   const popoverSig = useSignal<HTMLElement | null>(null);
+  const initialClickSig = useSignal<boolean>(false);
 
   const loadPolyfill$ = $(async () => {
     await import('@oddbird/popover-polyfill');
@@ -42,7 +43,7 @@ export function usePopover(popovertarget: string) {
     didInteractSig.value = true;
   });
 
-  useTask$(({ track }) => {
+  useTask$(async ({ track }) => {
     track(() => didInteractSig.value);
 
     if (!isBrowser) return;
@@ -50,22 +51,16 @@ export function usePopover(popovertarget: string) {
     // get popover
     if (!popoverSig.value) {
       popoverSig.value = document.getElementById(popovertarget);
-    }
-
-    // so it only runs once on click for supported browsers
-    if (isSupportedSig.value) {
-      if (!popoverSig.value) return;
 
-      if (popoverSig.value && popoverSig.value.hasAttribute('popover')) {
-        /* opens manual on any event */
-        popoverSig.value.showPopover();
+      if (!initialClickSig.value) {
+        popoverSig.value?.showPopover();
       }
     }
   });
 
   // event is created after teleported properly
   useOnDocument(
-    'showpopover',
+    'showpopoverpoly',
     $(() => {
       if (!didInteractSig.value) return;
 
@@ -74,7 +69,10 @@ export function usePopover(popovertarget: string) {
       // calls code in here twice for some reason, we think it's because of the client re-render, but it still works
 
       // so it only runs once on first click
-      if (!popoverSig.value.classList.contains(':popover-open')) {
+      if (
+        !popoverSig.value.classList.contains(':popover-open') &&
+        !isSupportedSig.value
+      ) {
         popoverSig.value.showPopover();
       }
     }),
@@ -101,12 +99,12 @@ export function usePopover(popovertarget: string) {
     popoverSig.value?.hidePopover();
   });
 
-  return { showPopover, togglePopover, hidePopover, initPopover$ };
+  return { showPopover, togglePopover, hidePopover, initPopover$, initialClickSig };
 }
 
 export const PopoverTrigger = component$<PopoverTriggerProps>(
   ({ popovertarget, disableClickInitPopover = false, ...rest }: PopoverTriggerProps) => {
-    const { initPopover$ } = usePopover(popovertarget);
+    const { initPopover$, initialClickSig } = usePopover(popovertarget);
 
     return (
       <button
@@ -116,6 +114,7 @@ export const PopoverTrigger = component$<PopoverTriggerProps>(
           rest.onClick$,
           !disableClickInitPopover
             ? $(async () => {
+                initialClickSig.value = true;
                 await initPopover$();
               })
             : undefined,
diff --git a/packages/kit-headless/src/components/popover/popover.test.ts b/packages/kit-headless/src/components/popover/popover.test.ts
index 8272577c8..a5623344f 100644
--- a/packages/kit-headless/src/components/popover/popover.test.ts
+++ b/packages/kit-headless/src/components/popover/popover.test.ts
@@ -236,9 +236,14 @@ test.describe('Keyboard Behavior', () => {
 
     await expect(d.getPopover()).toBeHidden();
 
+    const programmaticTrigger = page.getByRole('button', {
+      name: "Focus me and press the 'o'",
+    });
+
     // Using `page` here because driver is scoped to the popover
-    await page.getByRole('button', { name: "Focus me and press the 'o'" }).focus();
-    await page.keyboard.type('o');
+    await expect(programmaticTrigger).toBeVisible();
+    await programmaticTrigger.focus();
+    await programmaticTrigger.press('o');
 
     await expect(d.getPopover()).toBeVisible();
   });