feat(loadResource): 支持 hook

fjc0k · Jul 30, 2021 · 2473759 · 2473759
1 parent 95dac32
commit 2473759
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/utils/loadResource.test.ts b/src/utils/loadResource.test.ts
@@ -1,6 +1,6 @@
 import { loadResource, LoadResourceUrlType } from './loadResource'
 
-describe(loadResource.name, () => {
+describe('loadResource', () => {
   const createElement = document.createElement.bind(document)
   beforeAll(() => {
     jest.spyOn(document, 'createElement').mockImplementation((tag: string) => {
@@ -120,9 +120,23 @@ describe(loadResource.name, () => {
     expect(document.documentElement.outerHTML).toInclude(src)
   })
 
-  test('样式资源不插入 DOM', async () => {
+  test('图片资源不插入 DOM', async () => {
     const src = 'http://foo.bar/this-is-an-image.jpg'
     await loadResource(src)
     expect(document.documentElement.outerHTML).not.toInclude(src)
   })
+
+  test('支持 hook', async () => {
+    const src = 'http://foo.bar/this-is-an-image.js'
+    await loadResource({
+      type: LoadResourceUrlType.js,
+      path: src,
+      hook: el => {
+        el.dataset.hooked = '1'
+      },
+    })
+    expect(
+      document.documentElement.querySelector(`[data-hooked="1"]`)!.tagName,
+    ).toBe('SCRIPT')
+  })
 })
diff --git a/src/utils/loadResource.ts b/src/utils/loadResource.ts
@@ -30,6 +30,9 @@ export interface LoadResourceUrl {
 
   /** 备用资源路径 */
   alternatePath?: string
+
+  /** 钩子 */
+  hook?: (el: HTMLScriptElement | HTMLLinkElement | HTMLImageElement) => any
 }
 
 function loadSpecificResource(
@@ -55,6 +58,9 @@ function loadSpecificResource(
       default:
         break
     }
+    if (url.hook) {
+      url.hook(el)
+    }
     el.onload = () => resolve(el)
     el.onerror = () => {
       if (url.alternatePath) {