From 6a9b4e188f2abe1ea19e002347b4f4899aad07ec Mon Sep 17 00:00:00 2001
From: harlan <harlan@harlanzw.com>
Date: Sun, 8 Dec 2024 13:26:17 +1100
Subject: [PATCH] fix(useScript): allow `load()` to be called again once
 `remove()` is called

---
 packages/unhead/src/composables/useScript.ts |  1 +
 test/unhead/dom/useScript.test.ts            | 28 ++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/packages/unhead/src/composables/useScript.ts b/packages/unhead/src/composables/useScript.ts
index 38846342..2ab51ec8 100644
--- a/packages/unhead/src/composables/useScript.ts
+++ b/packages/unhead/src/composables/useScript.ts
@@ -109,6 +109,7 @@ export function useScript<T extends Record<symbol | string, any> = Record<symbol
       script._triggerPromises = [] // clear any pending promises
       if (script.entry) {
         script.entry.dispose()
+        script.entry = undefined
         syncStatus('removed')
         delete head._scripts?.[id]
         return true
diff --git a/test/unhead/dom/useScript.test.ts b/test/unhead/dom/useScript.test.ts
index e9eb7bf6..8c3337c6 100644
--- a/test/unhead/dom/useScript.test.ts
+++ b/test/unhead/dom/useScript.test.ts
@@ -59,4 +59,32 @@ describe('dom useScript', () => {
 
     expect(await instance.proxy.test('hello-world')).toEqual('hello-world')
   })
+  it('remove & re-add', async () => {
+    useDOMHead()
+
+    const instance = useScript<{ test: (foo: string) => void }>({
+      src: 'https://cdn.example.com/script.js',
+    })
+
+    let dom = await useDelayedSerializedDom()
+    expect(dom.split('\n').filter(l => l.trim().startsWith('<script'))).toMatchInlineSnapshot(`
+      [
+        "<script data-onload="" data-onerror="" defer="" fetchpriority="low" crossorigin="anonymous" referrerpolicy="no-referrer" src="https://cdn.example.com/script.js" data-hid="c5c65b0"></script></head>",
+      ]
+    `)
+    instance.remove()
+    // wait
+    await new Promise((r) => setTimeout(r, 100))
+    dom = await useDelayedSerializedDom()
+    expect(dom.split('\n').filter(l => l.trim().startsWith('<script'))).toMatchInlineSnapshot(`[]`)
+    // reload
+    instance.load()
+    await new Promise((r) => setTimeout(r, 100))
+    dom = await useDelayedSerializedDom()
+    expect(dom.split('\n').filter(l => l.trim().startsWith('<script'))).toMatchInlineSnapshot(`
+      [
+        "<script data-onload="" data-onerror="" defer="" fetchpriority="low" crossorigin="anonymous" referrerpolicy="no-referrer" src="https://cdn.example.com/script.js" data-hid="c5c65b0"></script></head>",
+      ]
+    `)
+  })
 })