capricorn86#1099@minor: Add support for currentScript.

ckhampus · Sep 27, 2023 · 1c0174a · 1c0174a
1 parent c06f50d
commit 1c0174a
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 6 deletions.
diff --git a/packages/happy-dom/src/nodes/document/Document.ts b/packages/happy-dom/src/nodes/document/Document.ts
@@ -68,6 +68,7 @@ export default class Document extends Node implements IDocument {
 	public readonly _readyStateManager: DocumentReadyStateManager;
 	public readonly _children: IHTMLCollection<IElement> = new HTMLCollection<IElement>();
 	public _activeElement: IHTMLElement = null;
+	public _currentScript: IHTMLScriptElement = null;
 
 	// Used as an unique identifier which is updated whenever the DOM gets modified.
 	public _cacheID = 0;
@@ -501,6 +502,15 @@ export default class Document extends Node implements IDocument {
 		return true;
 	}
 
+	/**
+	 * Gets the currently executing script element.
+	 * 
+	 * @returns the currently executing script element.
+	 */
+	public get currentScript(): IHTMLScriptElement {
+		return this._currentScript;
+	}
+
 	/**
 	 * Inserts a set of Node objects or DOMString objects after the last child of the ParentNode. DOMString objects are inserted as equivalent Text nodes.
 	 *

diff --git a/packages/happy-dom/src/nodes/document/IDocument.ts b/packages/happy-dom/src/nodes/document/IDocument.ts
@@ -47,6 +47,7 @@ export default interface IDocument extends IParentNode {
 	readonly hidden: boolean;
 	readonly links: IHTMLCollection<IHTMLElement>;
 	readonly referrer: string;
+	readonly currentScript: IHTMLScriptElement;
 	cookie: string;
 	title: string;
 

diff --git a/packages/happy-dom/src/nodes/html-script-element/HTMLScriptElement.ts b/packages/happy-dom/src/nodes/html-script-element/HTMLScriptElement.ts
@@ -189,12 +189,18 @@ export default class HTMLScriptElement extends HTMLElement implements IHTMLScrip
 						type === 'application/x-javascript' ||
 						type.startsWith('text/javascript'))
 				) {
-					if (this.ownerDocument.defaultView.happyDOM.settings.disableErrorCapturing) {
-						this.ownerDocument.defaultView.eval(textContent);
-					} else {
-						WindowErrorUtility.captureError(this.ownerDocument.defaultView, () =>
-							this.ownerDocument.defaultView.eval(textContent)
-						);
+					try {
+						this.ownerDocument['_currentScript'] = this;
+
+						if (this.ownerDocument.defaultView.happyDOM.settings.disableErrorCapturing) {
+							this.ownerDocument.defaultView.eval(textContent);
+						} else {
+							WindowErrorUtility.captureError(this.ownerDocument.defaultView, () =>
+								this.ownerDocument.defaultView.eval(textContent)
+							);
+						}
+					} finally {
+						this.ownerDocument['_currentScript'] = null;
 					}
 				}
 			}

diff --git a/packages/happy-dom/test/nodes/document/Document.test.ts b/packages/happy-dom/test/nodes/document/Document.test.ts
@@ -1290,4 +1290,20 @@ describe('Document', () => {
 			}
 		});
 	});
+
+	describe('currentScript', () => {
+		it('Returns the currently executing script element.', () => {
+			expect(document.currentScript).toBe(null);
+			const script1 = document.createElement('script');
+			script1.textContent = 'window.test = document.currentScript;';
+			document.body.appendChild(script1);
+			expect(window['test']).toBe(script1);
+			expect(document.currentScript).toBe(null);
+			const script2 = document.createElement('script');
+			script2.textContent = 'window.test = document.currentScript;';
+			document.body.appendChild(script2);
+			expect(window['test']).toBe(script2);
+			expect(document.currentScript).toBe(null);
+		});
+	});
 });