create an object if given a string

Co-authored-by: Kristján Oddsson <koddsson@gmail.com>
github · Oct 23, 2024 · 3a8e33a · 3a8e33a
1 parent bd69f5d
commit 3a8e33a
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/src/lazy-define.ts b/src/lazy-define.ts
@@ -85,15 +85,12 @@ export function lazyDefine(object: Record<string, () => void>): void
 export function lazyDefine(tagName: string, callback: () => void): void
 export function lazyDefine(tagNameOrObj: string | Record<string, () => void>, singleCallback?: () => void) {
   if (typeof tagNameOrObj === 'string' && singleCallback) {
-    if (!dynamicElements.has(tagNameOrObj)) dynamicElements.set(tagNameOrObj, new Set<() => void>())
-    dynamicElements.get(tagNameOrObj)!.add(singleCallback)
-  } else {
-    for (const [tagName, callback] of Object.entries(tagNameOrObj)) {
-      if (!dynamicElements.has(tagName)) dynamicElements.set(tagName, new Set<() => void>())
-      dynamicElements.get(tagName)!.add(callback)
-    }
+    tagNameOrObj = {[tagNameOrObj]: singleCallback}
+  }
+  for (const [tagName, callback] of Object.entries(tagNameOrObj)) {
+    if (!dynamicElements.has(tagName)) dynamicElements.set(tagName, new Set<() => void>())
+    dynamicElements.get(tagName)!.add(callback)
   }
-
   observe(document)
 }