@uppy/utils: improve types for `findDOMElement` (#4873)

packages/@uppy/utils/src/findDOMElement.ts

@@ -1,19 +1,19 @@
 import isDOMElement from './isDOMElement.ts'
 
-/**
- * Find a DOM element.
- */
-export default function findDOMElement(
-  element: Node | string,
-  context = document,
-): Element | null {
+export default function findDOMElement<T>(
+  element: T,
+  context: Document = document,
+): T extends Element ? T : T extends Node | string ? Element | null : null {
   if (typeof element === 'string') {
+    // @ts-expect-error ????
     return context.querySelector(element)
   }
 
   if (isDOMElement(element)) {
+    // @ts-expect-error ????
     return element
   }
 
+  // @ts-expect-error ????
   return null
 }