From e1ffe582c3adc45ee6316a1d937a060fed1f38a1 Mon Sep 17 00:00:00 2001
From: sachaaaaa <sacha@loki.network>
Date: Thu, 25 Oct 2018 15:18:32 +1100
Subject: [PATCH] Prompt user when an incoming friend request is received

---
 background.html                      |  3 +++
 js/background.js                     |  6 ++++++
 js/views/app_view.js                 | 11 ++++++++++
 js/views/confirmation_dialog_view.js |  3 +++
 libtextsecure/message_receiver.js    | 32 ++++++++++++++++++++++++++++
 5 files changed, 55 insertions(+)

diff --git a/background.html b/background.html
index 483f588cf..ffdd9fc77 100644
--- a/background.html
+++ b/background.html
@@ -152,6 +152,9 @@
   </script>
   <script type='text/x-tmpl-mustache' id='confirmation-dialog'>
     <div class="content">
+      {{ #title }}
+        <h4>{{ title }}</h4>
+      {{ /title }}
       <div class='message'>{{ message }}</div>
       <div class='buttons'>
         {{ #showCancel }}
diff --git a/js/background.js b/js/background.js
index 75e2b9b6f..2c62d39c1 100644
--- a/js/background.js
+++ b/js/background.js
@@ -595,6 +595,12 @@
         });
       }
     });
+
+    Whisper.events.on('showFriendRequest', friendRequest => {
+      if (appView) {
+        appView.showFriendRequest(friendRequest);
+      }
+    });
   }
 
   window.getSyncRequest = () =>
diff --git a/js/views/app_view.js b/js/views/app_view.js
index 1db6314e3..6304698ec 100644
--- a/js/views/app_view.js
+++ b/js/views/app_view.js
@@ -178,5 +178,16 @@
         });
       }
     },
+    showFriendRequest({ pubKey, message, accept, decline }) {
+      const dialog = new Whisper.ConfirmationDialogView({
+        title: `${pubKey} sent you a friend request:`,
+        message: message,
+        okText: 'Accept',
+        cancelText: 'Decline',
+        resolve: accept,
+        reject: decline,
+      });
+      this.el.append(dialog.el);
+    },
   });
 })();
diff --git a/js/views/confirmation_dialog_view.js b/js/views/confirmation_dialog_view.js
index c2081d733..a62e1024c 100644
--- a/js/views/confirmation_dialog_view.js
+++ b/js/views/confirmation_dialog_view.js
@@ -19,6 +19,8 @@
       this.reject = options.reject;
       this.cancelText = options.cancelText || i18n('cancel');
 
+      this.title = options.title;
+
       this.render();
     },
     events: {
@@ -32,6 +34,7 @@
         showCancel: !this.hideCancel,
         cancel: this.cancelText,
         ok: this.okText,
+        title: this.title,
       };
     },
     ok() {
diff --git a/libtextsecure/message_receiver.js b/libtextsecure/message_receiver.js
index 1499a647b..24059ecf9 100644
--- a/libtextsecure/message_receiver.js
+++ b/libtextsecure/message_receiver.js
@@ -836,9 +836,41 @@ MessageReceiver.prototype.extend({
       }
     });
   },
+  async promptUserToAcceptFriendRequest(pubKey, message) {
+    pubKey = pubKey.slice(0,30)+"...";
+    let p = new Promise(resolve => {
+      window.Whisper.events.trigger('showFriendRequest', {
+        pubKey,
+        message,
+        accept: () => {
+          resolve(true);
+        },
+        decline: () => {
+          resolve(false);
+        }
+      });
+    });
+    return await p;
+  },
   async innerHandleContentMessage(envelope, plaintext) {
     const content = textsecure.protobuf.Content.decode(plaintext);
 
+    if (envelope.type == textsecure.protobuf.Envelope.Type.FRIEND_REQUEST) {
+      // only prompt friend request if there is no conversation yet
+      let conversation;
+      try {
+        conversation = ConversationController.get(envelope.source);
+      } catch(e) {
+      }
+      if (!conversation) {
+        const accepted = await this.promptUserToAcceptFriendRequest(envelope.source, content.dataMessage.body);
+        if (!accepted) {
+          console.log('friend request declined!');
+          return;
+        }
+      }
+    }
+
     if (content.preKeyBundleMessage) {
       await this.handlePreKeyBundleMessage(envelope, content.preKeyBundleMessage);
     }