-
Notifications
You must be signed in to change notification settings - Fork 394
/
iframe-wrapper.js
72 lines (62 loc) · 2.26 KB
/
iframe-wrapper.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
/**
* Helper class arround the iframe HTML Element.
*/
export default class IframeWrapper {
/**
* @param {HTMLElement} parent - The parent element to inject the iframe into
* @param {Function} errorHandler - Callback calls if an error occurs in the iframe context
*/
constructor(parent, errorHandler = () => {}) {
this.el = document.createElement('iframe');
parent.appendChild(this.el);
this.onError = (msg, err, lin, col, error) => errorHandler(error);
this.onUnhandledRejection = promise => errorHandler(promise.reason);
this.attachEventListeners();
}
/**
* Properly destroy the iframe element
*/
destroy() {
this.detachEventListeners();
this.el.parentNode.removeChild(this.el);
}
/**
* Inject a script tag in the iframe.
* @param {String} src Script tag source attribute value
* @returns a Promise resolving once the script has loaded. If the script fails
* to load, the promise is rejected.
*/
injectScript(src) {
let scriptTag = this.document.createElement('script');
scriptTag.type = 'text/javascript';
scriptTag.src = src;
this.document.body.appendChild(scriptTag);
return new Promise((resolve, reject) => {
const onLoad = () => {
// Remove attached event listeners to avoid the script tag to leak
removeEventListeners();
resolve();
}
const removeEventListeners = () => {
scriptTag.removeEventListener('error', reject);
scriptTag.removeEventListener('load', onLoad);
}
scriptTag.addEventListener('error', reject);
scriptTag.addEventListener('load', onLoad);
});
}
get window() {
return this.el.contentWindow;
}
get document() {
return this.el.contentDocument;
}
attachEventListeners() {
this.el.addEventListener('error', this.onError);
this.el.addEventListener('unhandledrejection', this.onUnhandledRejection);
}
detachEventListeners() {
this.el.removeEventListener('error', this.onError);
this.el.removeEventListener('unhandledrejection', this.onUnhandledRejection);
}
}