packages/cross-storage/lib/host.ts

   1 import { getSecondLevelDomain } from '@proton/shared/lib/helpers/url';
   2
   3 import type { CrossStorageMessage, ResponseMessage } from './interface';
   4 import { getTestKeyValue, setTestKeyValue } from './support';
   5
   6 const createHost = <Message, MessageResponse>(handler: (message: Message) => Promise<MessageResponse>) => {
   7     const isEmbedded = window.location !== window.parent.location;
   8     const hostSecondLevelDomain = getSecondLevelDomain(window.location.hostname);
   9
  10     if (!isEmbedded) {
  11         throw new Error('Window not embedded');
  12     }
  13
  14     const postMessage = (message: CrossStorageMessage, origin: string) => {
  15         return window.parent.postMessage(message, origin);
  16     };
  17
  18     postMessage({ type: 'init', payload: { value: getTestKeyValue(window) } }, '*');
  19
  20     const reply = (origin: string, payload: ResponseMessage<MessageResponse>) => {
  21         postMessage(payload, origin);
  22     };
  23
  24     window.addEventListener('message', (event: MessageEvent<CrossStorageMessage>) => {
  25         const { source, data, origin } = event;
  26         if (
  27             source !== window.parent ||
  28             getSecondLevelDomain(new URL(origin).hostname) !== hostSecondLevelDomain ||
  29             data?.type !== 'message'
  30         ) {
  31             return;
  32         }
  33         handler(data.payload)
  34             .then((response) => {
  35                 reply(origin, { id: data.id, type: 'response', status: 'success', payload: response });
  36             })
  37             .catch((e) => {
  38                 reply(origin, {
  39                     id: data.id,
  40                     type: 'response',
  41                     status: 'error',
  42                     payload: e,
  43                 });
  44             });
  45     });
  46 };
  47
  48 export const initMainHost = () => {
  49     setTestKeyValue(window);
  50 };
  51
  52 export default createHost;