88 lines
2.9 KiB
TypeScript
88 lines
2.9 KiB
TypeScript
|
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';
|
||
|
|
import { TThreadMessageModel } from '../../definitions';
|
||
|
|
import sdk from '../rocketchat/services/sdk';
|
||
|
|
|
||
|
|
async function load({ tmid }: { tmid: string }) {
|
||
|
|
try {
|
||
|
|
// RC 1.0
|
||
|
|
const result = await sdk.methodCallWrapper('getThreadMessages', { tmid });
|
||
|
|
if (!result) {
|
||
|
|
return [];
|
||
|
|
}
|
||
|
|
return EJSON.fromJSONValue(result);
|
||
|
|
} catch (error) {
|
||
|
|
console.log(error);
|
||
|
|
return [];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default function loadThreadMessages({ tmid, rid }: { tmid: string; rid: string }) {
|
||
|
|
return new Promise<void>(async (resolve, reject) => {
|
||
|
|
try {
|
||
|
|
let data = await load({ tmid });
|
||
|
|
if (data && data.length) {
|
||
|
|
try {
|
||
|
|
data = data.filter((m: TThreadMessageModel) => m.tmid).map((m: TThreadMessageModel) => 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();
|
||
|
|
const filterThreadMessagesToCreate = data.filter(
|
||
|
|
(i1: TThreadMessageModel) => !allThreadMessagesRecords.find(i2 => i1._id === i2.id)
|
||
|
|
);
|
||
|
|
const filterThreadMessagesToUpdate = allThreadMessagesRecords.filter(i1 =>
|
||
|
|
data.find((i2: TThreadMessageModel) => i1.id === i2._id)
|
||
|
|
);
|
||
|
|
|
||
|
|
const threadMessagesToCreate = filterThreadMessagesToCreate.map((threadMessage: TThreadMessageModel) =>
|
||
|
|
threadMessagesCollection.prepareCreate(
|
||
|
|
protectedFunction((tm: TThreadMessageModel) => {
|
||
|
|
tm._raw = sanitizedRaw({ id: threadMessage._id }, threadMessagesCollection.schema);
|
||
|
|
Object.assign(tm, threadMessage);
|
||
|
|
if (tm.subscription) {
|
||
|
|
tm.subscription.id = rid;
|
||
|
|
}
|
||
|
|
if (threadMessage.tmid) {
|
||
|
|
tm.rid = threadMessage.tmid;
|
||
|
|
}
|
||
|
|
delete threadMessage.tmid;
|
||
|
|
})
|
||
|
|
)
|
||
|
|
);
|
||
|
|
|
||
|
|
const threadMessagesToUpdate = filterThreadMessagesToUpdate.map(threadMessage => {
|
||
|
|
const newThreadMessage = data.find((t: TThreadMessageModel) => t._id === threadMessage.id);
|
||
|
|
return threadMessage.prepareUpdate(
|
||
|
|
protectedFunction((tm: TThreadMessageModel) => {
|
||
|
|
Object.assign(tm, newThreadMessage);
|
||
|
|
if (threadMessage.tmid) {
|
||
|
|
tm.rid = threadMessage.tmid;
|
||
|
|
}
|
||
|
|
delete threadMessage.tmid;
|
||
|
|
})
|
||
|
|
);
|
||
|
|
});
|
||
|
|
|
||
|
|
await db.write(async () => {
|
||
|
|
await db.batch(...threadMessagesToCreate, ...threadMessagesToUpdate);
|
||
|
|
});
|
||
|
|
} catch (e) {
|
||
|
|
log(e);
|
||
|
|
}
|
||
|
|
return resolve(data);
|
||
|
|
}
|
||
|
|
return resolve();
|
||
|
|
} catch (e) {
|
||
|
|
reject(e);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|