Rocket.Chat.ReactNative/app/lib/methods/loadThreadMessages.js

77 lines
2.4 KiB
JavaScript

import { Q } from '@nozbe/watermelondb';
import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';
import EJSON from 'ejson';
import database from '../database';
import log from '../../utils/log';
import { Encryption } from '../encryption';
import protectedFunction from './helpers/protectedFunction';
import buildMessage from './helpers/buildMessage';
async function load({ tmid }) {
try {
// RC 1.0
const result = await this.methodCallWrapper('getThreadMessages', { tmid });
if (!result) {
return [];
}
return EJSON.fromJSONValue(result);
} catch (error) {
console.log(error);
return [];
}
}
export default function loadThreadMessages({ tmid, rid }) {
return new Promise(async (resolve, reject) => {
try {
let data = await load.call(this, { tmid });
if (data && data.length) {
try {
data = data.filter(m => m.tmid).map(m => buildMessage(m));
data = await Encryption.decryptMessages(data);
const db = database.active;
const threadMessagesCollection = db.get('thread_messages');
const allThreadMessagesRecords = await threadMessagesCollection.query(Q.where('rid', tmid)).fetch();
let threadMessagesToCreate = data.filter(i1 => !allThreadMessagesRecords.find(i2 => i1._id === i2.id));
let threadMessagesToUpdate = allThreadMessagesRecords.filter(i1 => data.find(i2 => i1.id === i2._id));
threadMessagesToCreate = threadMessagesToCreate.map(threadMessage =>
threadMessagesCollection.prepareCreate(
protectedFunction(tm => {
tm._raw = sanitizedRaw({ id: threadMessage._id }, threadMessagesCollection.schema);
Object.assign(tm, threadMessage);
tm.subscription.id = rid;
tm.rid = threadMessage.tmid;
delete threadMessage.tmid;
})
)
);
threadMessagesToUpdate = threadMessagesToUpdate.map(threadMessage => {
const newThreadMessage = data.find(t => t._id === threadMessage.id);
return threadMessage.prepareUpdate(
protectedFunction(tm => {
Object.assign(tm, newThreadMessage);
tm.rid = threadMessage.tmid;
delete threadMessage.tmid;
})
);
});
await db.action(async () => {
await db.batch(...threadMessagesToCreate, ...threadMessagesToUpdate);
});
} catch (e) {
log(e);
}
return resolve(data);
} else {
return resolve([]);
}
} catch (e) {
reject(e);
}
});
}