summaryrefslogtreecommitdiffstats
path: root/remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
commit6bf0a5cb5034a7e684dcc3500e841785237ce2dd (patch)
treea68f146d7fa01f0134297619fbe7e33db084e0aa /remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts
parentInitial commit. (diff)
downloadthunderbird-6bf0a5cb5034a7e684dcc3500e841785237ce2dd.tar.xz
thunderbird-6bf0a5cb5034a7e684dcc3500e841785237ce2dd.zip
Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts')
-rw-r--r--remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts49
1 files changed, 49 insertions, 0 deletions
diff --git a/remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts b/remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts
new file mode 100644
index 0000000000..cb0c039530
--- /dev/null
+++ b/remote/test/puppeteer/packages/puppeteer-core/src/common/ScriptInjector.ts
@@ -0,0 +1,49 @@
+import {source as injectedSource} from '../generated/injected.js';
+
+class ScriptInjector {
+ #updated = false;
+ #amendments = new Set<string>();
+
+ // Appends a statement of the form `(PuppeteerUtil) => {...}`.
+ append(statement: string): void {
+ this.#update(() => {
+ this.#amendments.add(statement);
+ });
+ }
+
+ pop(statement: string): void {
+ this.#update(() => {
+ this.#amendments.delete(statement);
+ });
+ }
+
+ inject(inject: (script: string) => void, force = false) {
+ if (this.#updated || force) {
+ inject(this.#get());
+ }
+ this.#updated = false;
+ }
+
+ #update(callback: () => void): void {
+ callback();
+ this.#updated = true;
+ }
+
+ #get(): string {
+ return `(() => {
+ const module = {};
+ ${injectedSource}
+ ${[...this.#amendments]
+ .map(statement => {
+ return `(${statement})(module.exports.default);`;
+ })
+ .join('')}
+ return module.exports.default;
+ })()`;
+ }
+}
+
+/**
+ * @internal
+ */
+export const scriptInjector = new ScriptInjector();