From c32348d3cc523a2971efb94e379d35e7ca8a31d2 Mon Sep 17 00:00:00 2001
From: Shir Serlui <70711723+shirser121@users.noreply.github.com>
Date: Mon, 15 Aug 2022 16:11:29 +0300
Subject: [PATCH] Use quick reply (#1651)

* Use template only if url or call buttons exist

* fix typo

* fix for tests

* fix for test

* Update src/util/Injected.js

Co-authored-by: Rajeh Taher <rajeh@reforward.dev>

* Add quick reply models

Co-authored-by: Rajeh Taher <rajeh@reforward.dev>
---
 src/util/Injected.js | 106 +++++++++++++++++++++++++++++++------------
 1 file changed, 77 insertions(+), 29 deletions(-)

diff --git a/src/util/Injected.js b/src/util/Injected.js
index 17e17c9..2ce9fec 100644
--- a/src/util/Injected.js
+++ b/src/util/Injected.js
@@ -110,6 +110,10 @@ exports.ExposeStore = (moduleRaidStr) => {
     // Find button models
     window.Store.TemplateButtonModel = window.findProxyModel('TemplateButtonModel');
     window.Store.TemplateButtonCollection = window.mR.findModule('TemplateButtonCollection')[0].TemplateButtonCollection;
+    
+    // Find quick reply models
+    window.Store.ReplyButtonModel = window.findProxyModel('ReplyButtonModel');
+    window.Store.ButtonCollection = window.mR.findModule('ButtonCollection')[0].ButtonCollection;
 
     // Modify functions 
     window.injectToFunction({
@@ -248,6 +252,14 @@ exports.ExposeStore = (moduleRaidStr) => {
             }
             return 'text';
         }
+        
+        if (
+            proto.buttonsMessage?.headerType === 1 ||
+            proto.buttonsMessage?.headerType === 2
+        ) {
+            return 'text';
+        }
+        
         return func(...args);
     });
 
@@ -307,6 +319,18 @@ exports.ExposeStore = (moduleRaidStr) => {
 
         return func(...args);
     });
+    
+    window.injectToFunction({
+        index: 0,
+        name: 'encodeMaybeMediaType',
+        property: 'encodeMaybeMediaType',
+    }, (func, args) => {
+        const [type] = args;
+        if (type === 'button') {
+            return window.mR.findModule('DROP_ATTR')[0].DROP_ATTR;
+        }
+        return func(...args);
+    });
 };
 
 exports.LoadUtils = () => {
@@ -328,40 +352,64 @@ exports.LoadUtils = () => {
         if (!buttonsOptions.buttons) {
             return returnObject;
         }
-
+        
+        if (typeof buttonsOptions.useTemplateButtons === 'undefined' || buttonsOptions.useTemplateButtons === null) {
+            buttonsOptions.useTemplateButtons = buttonsOptions.buttons.some((button) => {
+                return 'callButton' in button || 'urlButton' in button;
+            });
+        }
+        
         returnObject.title = buttonsOptions.title;
         returnObject.footer = buttonsOptions.footer;
-        returnObject.isFromTemplate = true;
-        returnObject.buttons = new window.Store.TemplateButtonCollection;
+    
+        if (buttonsOptions.useTemplateButtons) {
+            returnObject.isFromTemplate = true;
+            returnObject.hydratedButtons = buttonsOptions.buttons;
+            returnObject.buttons = new window.Store.TemplateButtonCollection;
 
-        returnObject.hydratedButtons = buttonsOptions.buttons;
+            returnObject.buttons.add(returnObject.hydratedButtons.map((button, index) => {
+                const buttonIndex = button.index ? button.index : index;
+                if (button.urlButton) {
+                    return new window.Store.TemplateButtonModel({
+                        id: buttonIndex,
+                        displayText: button.urlButton?.displayText || '',
+                        url: button.urlButton?.url,
+                        subtype: 'url'
+                    });
+                } else if (button.callButton) {
+                    return new window.Store.TemplateButtonModel({
+                        id: buttonIndex,
+                        displayText: button.callButton?.displayText,
+                        phoneNumber: button.callButton?.phoneNumber,
+                        subtype: 'call'
+                    });
+                } else {
+                    return new window.Store.TemplateButtonModel({
+                        id: buttonIndex,
+                        displayText: button.quickReplyButton?.displayText,
+                        selectionId: button.quickReplyButton?.id,
+                        subtype: 'quick_reply'
+                    });
+                }
+            }));
+        }
+        else {
+            returnObject.isDynamicReplyButtonsMsg = true;
 
-        returnObject.buttons.add(returnObject.hydratedButtons.map((button, index) => {
-            const buttonIndex = button.index ? button.index : index;
-            if (button.urlButton) {
-                return new window.Store.TemplateButtonModel({
-                    id: buttonIndex,
-                    displayText: button.urlButton?.displayText || '',
-                    url: button.urlButton?.url,
-                    subtype: 'url'
-                });
-            } else if (button.callButton) {
-                return new window.Store.TemplateButtonModel({
-                    id: buttonIndex,
-                    displayText: button.callButton?.displayText,
-                    phoneNumber: button.callButton?.phoneNumber,
-                    subtype: 'call'
-                });
-            } else {
-                return new window.Store.TemplateButtonModel({
-                    id: buttonIndex,
-                    displayText: button.quickReplyButton?.displayText,
-                    selectionId: button.quickReplyButton?.id,
-                    subtype: 'quick_reply'
-                });
-            }
-        }));
+            returnObject.dynamicReplyButtons = buttonsOptions.buttons.map((button, index) => ({
+                buttonId: button.index || `${index}`,
+                buttonText: {displayText: button.quickReplyButton?.displayText},
+                type: 1,
+            }));
 
+            // For UI only
+            returnObject.replyButtons = new window.Store.ButtonCollection();
+            returnObject.replyButtons.add(returnObject.dynamicReplyButtons.map((button) => new window.Store.ReplyButtonModel({
+                id: button.buttonId,
+                displayText: button.buttonText?.displayText || undefined,
+            })));
+
+        }
         return returnObject;
     };