1 files changed, 124 insertions, 0 deletions
diff --git a/public/src/app/sdc/base-pubsub.ts b/public/src/app/sdc/base-pubsub.ts
new file mode 100644
index 0000000..f7fcadc
--- /dev/null
+++ b/public/src/app/sdc/base-pubsub.ts
@@ -0,0 +1,124 @@
+declare const window: Window;
+
+export interface IPubSubEvent {
+  type: string;
+  originId: string;
+  data: any;
+}
+
+export interface ISubscriber {
+  window: Window;
+  locationUrl: string;
+}
+
+export class BasePubSub {
+  subscribers: Map<string, ISubscriber>;
+  eventsCallbacks: Array<Function>;
+  clientId: string;
+  eventsToWait: Map<string, Array<string>>;
+
+  constructor(pluginId: string) {
+    this.subscribers = new Map<string, ISubscriber>();
+    this.eventsCallbacks = [];
+    this.eventsToWait = new Map<string, Array<string>>();
+    this.clientId = pluginId;
+    this.onMessage = this.onMessage.bind(this);
+
+    window.addEventListener('message', this.onMessage);
+  }
+
+  public register(
+    subscriberId: string,
+    subscriberWindow: Window,
+    subscriberUrl: string
+  ) {
+    const subscriber = {
+      window: subscriberWindow,
+      locationUrl: subscriberUrl || subscriberWindow.location.href
+    };
+
+    this.subscribers.set(subscriberId, subscriber);
+  }
+
+  public unregister(subscriberId: string) {
+    this.subscribers.delete(subscriberId);
+  }
+
+  public on(callback: Function) {
+    const functionExists = this.eventsCallbacks.find((func: Function) => {
+      return callback.toString() === func.toString();
+    });
+
+    if (!functionExists) {
+      this.eventsCallbacks.push(callback);
+    }
+  }
+
+  public off(callback: Function) {
+    const index = this.eventsCallbacks.indexOf(callback);
+    this.eventsCallbacks.splice(index, 1);
+  }
+
+  public notify(eventType: string, eventData?: any) {
+    const eventObj = {
+      type: eventType,
+      data: eventData,
+      originId: this.clientId
+    };
+
+    this.subscribers.forEach(
+      (subscriber: ISubscriber, subscriberId: string) => {
+        subscriber.window.postMessage(eventObj, subscriber.locationUrl);
+      }
+    );
+
+    return {
+      subscribe: function(callbackFn) {
+        if (this.subscribers.size !== 0) {
+          const subscribersToNotify = Array.from(this.subscribers.keys());
+
+          const checkNotifyComplete = (subscriberId: string) => {
+            const index = subscribersToNotify.indexOf(subscriberId);
+            subscribersToNotify.splice(index, 1);
+
+            if (subscribersToNotify.length === 0) {
+              callbackFn();
+            }
+          };
+
+          this.subscribers.forEach(
+            (subscriber: ISubscriber, subscriberId: string) => {
+              if (
+                this.eventsToWait.has(subscriberId) &&
+                this.eventsToWait.get(subscriberId).indexOf(eventType) !== -1
+              ) {
+                const actionCompletedFunction = (
+                  eventData,
+                  subId = subscriberId
+                ) => {
+                  if (eventData.type === 'ACTION_COMPLETED') {
+                    checkNotifyComplete(subId);
+                  }
+                  this.off(actionCompletedFunction);
+                };
+                this.on(actionCompletedFunction);
+              } else {
+                checkNotifyComplete(subscriberId);
+              }
+            }
+          );
+        } else {
+          callbackFn();
+        }
+      }.bind(this)
+    };
+  }
+
+  protected onMessage(event: any) {
+    if (this.subscribers.has(event.data.originId)) {
+      this.eventsCallbacks.forEach((callback: Function) => {
+        callback(event.data, event);
+      });
+    }
+  }
+}