Pass reference

shockpkg · Oct 15, 2023 · 8375959 · 8375959
1 parent 7363e50
commit 8375959
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/projector/html.ts b/src/projector/html.ts
@@ -164,8 +164,11 @@ export class ProjectorHtml extends Projector {
 	/**
 	 * Custom HTML to use instead of generated HTML.
 	 */
-	public html: string | (() => string) | (() => Promise<string>) | null =
-		null;
+	public html:
+		| string
+		| ((self: this) => string)
+		| ((self: this) => Promise<string>)
+		| null = null;
 
 	/**
 	 * ProjectorHtml constructor.
@@ -191,7 +194,7 @@ export class ProjectorHtml extends Projector {
 	public async getHtml() {
 		const {html} = this;
 		if (html) {
-			return typeof html === 'function' ? html() : html;
+			return typeof html === 'function' ? html(this) : html;
 		}
 		return this.getHtmlDefault();
 	}