Receive support for View Once photos

js/modules/data.js

@@ -122,6 +122,9 @@ module.exports = {
   getOutgoingWithoutExpiresAt,
   getNextExpiringMessage,
   getMessagesByConversation,
+  getNextTapToViewMessageToExpire,
+  getNextTapToViewMessageToAgeOut,
+  getTapToViewMessagesNeedingErase,
 
   getUnprocessedCount,
   getAllUnprocessed,
@@ -674,7 +677,7 @@ async function getMessageCount() {
 
 async function saveMessage(data, { forceSave, Message } = {}) {
   const id = await channels.saveMessage(_cleanData(data), { forceSave });
-  Message.refreshExpirationTimer();
+  Message.updateTimers();
   return id;
 }
 
@@ -839,6 +842,27 @@ async function getNextExpiringMessage({ MessageCollection }) {
   return new MessageCollection(messages);
 }
 
+async function getNextTapToViewMessageToExpire({ Message }) {
+  const message = await channels.getNextTapToViewMessageToExpire();
+  if (!message) {
+    return null;
+  }
+
+  return new Message(message);
+}
+async function getNextTapToViewMessageToAgeOut({ Message }) {
+  const message = await channels.getNextTapToViewMessageToAgeOut();
+  if (!message) {
+    return null;
+  }
+
+  return new Message(message);
+}
+async function getTapToViewMessagesNeedingErase({ MessageCollection }) {
+  const messages = await channels.getTapToViewMessagesNeedingErase();
+  return new MessageCollection(messages);
+}
+
 // Unprocessed
 
 async function getUnprocessedCount() {