From db2ffadb43fad5c3c48eb4fba559d0b203d870a0 Mon Sep 17 00:00:00 2001
From: Tiago vasconcelos <talvasconcelos@gmail.com>
Date: Fri, 30 Sep 2022 17:28:35 +0100
Subject: [PATCH] retrieve chat messages from db

---
 lnbits/extensions/diagonalley/crud.py         | 16 ++++++++
 .../templates/diagonalley/chat.html           | 41 ++++++++++++++++---
 lnbits/extensions/diagonalley/views_api.py    | 14 +++++++
 3 files changed, 65 insertions(+), 6 deletions(-)

diff --git a/lnbits/extensions/diagonalley/crud.py b/lnbits/extensions/diagonalley/crud.py
index 9e3f2013..6bde37fa 100644
--- a/lnbits/extensions/diagonalley/crud.py
+++ b/lnbits/extensions/diagonalley/crud.py
@@ -9,6 +9,7 @@ from lnbits.settings import WALLET
 
 from . import db
 from .models import (
+    ChatMessage,
     CreateChatMessage,
     CreateMarket,
     CreateMarketStalls,
@@ -420,3 +421,18 @@ async def create_chat_message(data: CreateChatMessage):
             data.room_name,
         ),
     )
+
+async def get_diagonalley_latest_chat_messages(room_name: str):
+    rows = await db.fetchall(
+        "SELECT * FROM diagonalley.messages WHERE id_conversation = ? ORDER BY timestamp DESC LIMIT 20", (room_name,)
+    )
+
+    return [ChatMessage(**row) for row in rows]
+
+async def get_diagonalley_chat_messages(room_name: str):
+    rows = await db.fetchall(
+        "SELECT * FROM diagonalley.messages WHERE id_conversation = ? ORDER BY timestamp DESC", (room_name,)
+    )
+
+    return [ChatMessage(**row) for row in rows]
+    
diff --git a/lnbits/extensions/diagonalley/templates/diagonalley/chat.html b/lnbits/extensions/diagonalley/templates/diagonalley/chat.html
index 21f59361..adbdb047 100644
--- a/lnbits/extensions/diagonalley/templates/diagonalley/chat.html
+++ b/lnbits/extensions/diagonalley/templates/diagonalley/chat.html
@@ -125,8 +125,13 @@
 </div>
 {% endblock %} {% block scripts %}
 <script>
-  const mapChatMsg = obj => {
-    return
+  const mapChatMsg = msg => {
+    let obj = {}
+    obj.timestamp = {
+      msg: msg,
+      pubkey: pubkey
+    }
+    return obj
   }
   Vue.component(VueQrcode.name, VueQrcode)
   new Vue({
@@ -186,6 +191,29 @@
             LNbits.utils.notifyApiError(error)
           })
       },
+      async getMessages(room_name, all = false) {
+        await LNbits.api
+          .request(
+            'GET',
+            `/diagonalley/api/v1/chat/messages/${room_name}${
+              all ? '?all_messages=true' : ''
+            }`
+          )
+          .then(response => {
+            if (response.data) {
+              console.log(response.data)
+              response.data.reverse().map(m => {
+                this.$set(this.messages, m.timestamp * 1000, {
+                  msg: m.msg,
+                  pubkey: m.pubkey
+                })
+              })
+            }
+          })
+          .catch(function (error) {
+            LNbits.utils.notifyApiError(error)
+          })
+      },
       changeOrder() {
         console.log(this.selectedOrder)
       },
@@ -212,7 +240,6 @@
           let event_data = JSON.parse(event.data)
 
           this.$set(this.messages, Date.now(), event_data)
-          console.log(this.messages)
         }
 
         this.ws = ws
@@ -230,8 +257,8 @@
         if (data) {
           this.user = data
           //add chat key (merchant pubkey) if not set
-          if (!this.user.chats[`${this.stall.publickey}`]) {
-            this.$set(this.user.chats, this.stall.publickey, [])
+          if (!this.user.chats[`${order_id}`]) {
+            this.$set(this.user.chats, order_id, [])
           }
           //this.$q.localStorage.set(`lnbits.diagonalley.data`, this.user)
         } else {
@@ -239,7 +266,7 @@
           await this.generateKeys()
           // populate user data
           this.user.chats = {
-            [`${this.stall.publickey}`]: []
+            [`${order_id}`]: []
           }
           this.user.orders = []
         }
@@ -248,6 +275,8 @@
         this.user.orders = [...new Set([...this.user.orders, order_id])]
         this.selectedOrder = order_id
 
+        await this.getMessages(order_id)
+
         this.$q.localStorage.set(`lnbits.diagonalley.data`, this.user)
         this.startChat(order_id)
       } catch (e) {
diff --git a/lnbits/extensions/diagonalley/views_api.py b/lnbits/extensions/diagonalley/views_api.py
index aa8d338d..2dcd16aa 100644
--- a/lnbits/extensions/diagonalley/views_api.py
+++ b/lnbits/extensions/diagonalley/views_api.py
@@ -34,6 +34,8 @@ from .crud import (
     delete_diagonalley_product,
     delete_diagonalley_stall,
     delete_diagonalley_zone,
+    get_diagonalley_chat_messages,
+    get_diagonalley_latest_chat_messages,
     get_diagonalley_market,
     get_diagonalley_market_stalls,
     get_diagonalley_markets,
@@ -481,3 +483,15 @@ async def api_diagonalley_generate_keys():
         private_key = PrivateKey()
         public_key = private_key.pubkey.serialize().hex()
     return {"privkey": private_key.serialize(), "pubkey": public_key[2:]}
+
+
+## MESSAGES/CHAT
+
+@diagonalley_ext.get("/api/v1/chat/messages/{room_name}")
+async def api_get_latest_chat_msg(room_name: str, all_messages: bool = Query(False)):
+    if(all_messages):
+        messages = await get_diagonalley_chat_messages(room_name)
+    else:
+        messages = await get_diagonalley_latest_chat_messages(room_name)
+
+    return messages