packages/extension-app/src/content/index.ts
// Copyright 2019-2020 @polkadot/extension authors & contributors
// This software may be modified and distributed under the terms
// of the Apache-2.0 license. See the LICENSE file for details.
import { PORT_CONTENT } from '@polkadot/extension-base/defaults';
import chrome from '@polkadot/extension-inject/chrome';
// connect to the extension
const port = chrome.runtime.connect({ name: PORT_CONTENT });
// send any messages from the extension back to the page
port.onMessage.addListener((data): void => {
window.postMessage({ ...data, origin: 'content' }, '*');
});
// all messages from the page, pass them to the extension
window.addEventListener('message', ({ data, source }): void => {
// only allow messages from our window, by the inject
if (source !== window || data.origin !== 'page') {
return;
}
port.postMessage(data);
});
// inject our data injector
const script = document.createElement('script');
script.src = chrome.extension.getURL('page/index.js');
script.onload = (): void => {
// remove the injecting tag when loaded
if (script.parentNode) {
script.parentNode.removeChild(script);
}
};
(document.head || document.documentElement).appendChild(script);