From 6392c779523dc494b033b10baf276c56f2d685e0 Mon Sep 17 00:00:00 2001
From: Beaudan Brown <beau@loki.network>
Date: Thu, 29 Aug 2019 14:35:01 +1000
Subject: [PATCH] Fix database helpers

---
 app/sql.js         | 9 ++++++---
 js/modules/data.js | 4 ++--
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/app/sql.js b/app/sql.js
index 77bde9694..bd1e4f1e6 100644
--- a/app/sql.js
+++ b/app/sql.js
@@ -790,7 +790,7 @@ async function updateToLokiSchemaVersion1(currentVersion, instance) {
 
   await instance.run(
     `ALTER TABLE messages
-     ADD COLUMN serverId STRING;`
+     ADD COLUMN serverId INTEGER;`
   );
 
   await instance.run(
@@ -2060,11 +2060,14 @@ async function removeMessage(id) {
   );
 }
 
-async function getMessageByServerId(serverId) {
+async function getMessageByServerId(serverId, conversationId) {
   const row = await db.get(
-    'SELECT * FROM messages WHERE serverId = $serverId;',
+    `SELECT * FROM messages WHERE
+      serverId = $serverId AND
+      conversationId = $conversationId;`,
     {
       $serverId: serverId,
+      $conversationId: conversationId,
     }
   );
 
diff --git a/js/modules/data.js b/js/modules/data.js
index 80c21bdee..c410f9464 100644
--- a/js/modules/data.js
+++ b/js/modules/data.js
@@ -908,8 +908,8 @@ async function _removeMessages(ids) {
   await channels.removeMessage(ids);
 }
 
-async function getMessageByServerId(id, { Message }) {
-  const message = await channels.getMessageByServerId(id);
+async function getMessageByServerId(serverId, conversationId, { Message }) {
+  const message = await channels.getMessageByServerId(serverId, conversationId);
   if (!message) {
     return null;
   }