update dom utils

Author: chaance

packages/core/primitive/src/index.ts

@@ -1 +1,2 @@
 export * from './primitive';
+export type * from './types';

packages/core/primitive/src/primitive.tsx

@@ -1,3 +1,5 @@
+/* eslint-disable no-restricted-properties */
+
 /* eslint-disable no-restricted-globals */
 export const canUseDOM = !!(
   typeof window !== 'undefined' &&
@@ -20,18 +22,54 @@ export function composeEventHandlers<E extends { defaultPrevented: boolean }>(
   };
 }
 
-export function getOwnerWindow(element: Element | null | undefined) {
+export function getOwnerWindow(element: Node | null | undefined) {
   if (!canUseDOM) {
     throw new Error('Cannot access window outside of the DOM');
   }
   // eslint-disable-next-line no-restricted-globals
   return element?.ownerDocument?.defaultView ?? window;
 }
 
-export function getOwnerDocument(element: Element | null | undefined) {
+export function getOwnerDocument(element: Node | null | undefined) {
   if (!canUseDOM) {
     throw new Error('Cannot access document outside of the DOM');
   }
   // eslint-disable-next-line no-restricted-globals
   return element?.ownerDocument ?? document;
 }
+
+/**
+ * Lifted from https://github.com/ariakit/ariakit/blob/main/packages/ariakit-core/src/utils/dom.ts#L37
+ * MIT License, Copyright (c) AriaKit.
+ */
+export function getActiveElement(
+  node: Node | null | undefined,
+  activeDescendant = false
+): HTMLElement | null {
+  const { activeElement } = getOwnerDocument(node);
+  if (!activeElement?.nodeName) {
+    // `activeElement` might be an empty object if we're interacting with elements
+    // inside of an iframe.
+    return null;
+  }
+
+  if (isFrame(activeElement) && activeElement.contentDocument) {
+    return getActiveElement(activeElement.contentDocument.body, activeDescendant);
+  }
+
+  if (activeDescendant) {
+    const id = activeElement.getAttribute('aria-activedescendant');
+    if (id) {
+      const element = getOwnerDocument(activeElement).getElementById(id);
+      if (element) {
+        return element;
+      }
+    }
+  }
+
+  return activeElement as HTMLElement | null;
+}
+
+export function isFrame(element: Element): element is HTMLIFrameElement {
+  return element.tagName === 'IFRAME';
+}

packages/core/primitive/src/types.ts

@@ -0,0 +1,3 @@
+export type Timeout = ReturnType<typeof setTimeout>;
+export type Interval = ReturnType<typeof setInterval>;
+export type Immediate = ReturnType<typeof setImmediate>;