Start listening in 0-indent scope

src/customAppSdk.ts

@@ -1,4 +1,4 @@
-import { sendMessage, startListening } from "./iframeMessenger";
+import { sendMessage } from "./iframeMessenger";
 import { ErrorMessage } from "./iframeSchema";
 import { matchesSchema } from "./matchesSchema";
 
@@ -26,10 +26,8 @@ export type CustomAppContext =
       readonly description: string;
     };
 
-export const getCustomAppContext = (): Promise<CustomAppContext> => {
-  const stopListening = startListening();
-
-  return new Promise((resolve, reject) => {
+export const getCustomAppContext = (): Promise<CustomAppContext> =>
+  new Promise((resolve, reject) => {
     try {
       sendMessage<"[email protected]">(
         {
@@ -47,8 +45,5 @@ export const getCustomAppContext = (): Promise<CustomAppContext> => {
       );
     } catch (error) {
       reject(error);
-    } finally {
-      stopListening();
     }
   });
-};

src/iframeMessenger.ts

@@ -4,18 +4,6 @@ import type { AllClientResponses } from "./utilityTypes";
 
 let callbacks: Readonly<Record<string, (data: unknown) => void>> = {};
 
-export const startListening = (): (() => void) => {
-  if (window.self === window.top) {
-    throw new Error("Custom app is not hosted in an IFrame.");
-  }
-
-  window.addEventListener("message", processMessage, true);
-
-  return () => {
-    window.removeEventListener("message", processMessage, true);
-  };
-};
-
 export const sendMessage = <TMessageType extends keyof Schema["client"]>(
   message: Omit<Schema["client"][TMessageType]["request"], "requestId">,
   callback: (data: Schema["client"][TMessageType]["response"]) => void,
@@ -33,3 +21,9 @@ const processMessage = (event: MessageEvent<AllClientResponses>): void => {
   );
   callback?.(message);
 };
+
+if (window.self === window.top) {
+  throw new Error("Custom app is not hosted in an IFrame.");
+}
+
+window.addEventListener("message", processMessage, true);