From 5a23dbb687fa9a6875531a5d62cca7e67dd140d5 Mon Sep 17 00:00:00 2001
From: Beaudan <beau@loki.network>
Date: Fri, 1 Mar 2019 13:49:59 +1100
Subject: [PATCH] Review comments and add local server listener before starting

---
 libtextsecure/message_receiver.js | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/libtextsecure/message_receiver.js b/libtextsecure/message_receiver.js
index 3530a6008..83d75c497 100644
--- a/libtextsecure/message_receiver.js
+++ b/libtextsecure/message_receiver.js
@@ -83,6 +83,7 @@ MessageReceiver.prototype.extend({
         this.onEmpty();
       }
     });
+    localLokiServer.on('message', this.handleP2pMessage.bind(this));
     this.startLocalServer();
 
     // TODO: Rework this socket stuff to work with online messaging
@@ -115,14 +116,14 @@ MessageReceiver.prototype.extend({
     this.incoming = [this.pending];
   },
   async startLocalServer() {
-    let myLokiIp;
-    let myServerPort;
     try {
-      myLokiIp = await window.lokiSnodeAPI.getMyLokiIp();
-      myServerPort = await localLokiServer.start(localServerPort, myLokiIp);
+      const myLokiIp = await window.lokiSnodeAPI.getMyLokiIp();
+      const myServerPort = await localLokiServer.start(
+        localServerPort,
+        myLokiIp
+      );
       window.log.info(`Local Server started at ${myLokiIp}:${myServerPort}`);
       libloki.api.broadcastOnlineStatus();
-      localLokiServer.on('message', this.handleP2pMessage.bind(this));
     } catch (e) {
       if (e instanceof textsecure.LokiIpError) {
         window.log.warn(