2019-09-16 20:26:32 +00:00
|
|
|
import { sanitizedRaw } from '@nozbe/watermelondb/RawRecord';
|
|
|
|
|
2018-05-18 17:55:08 +00:00
|
|
|
import messagesStatus from '../../constants/messagesStatus';
|
2019-09-16 20:26:32 +00:00
|
|
|
import database from '../database';
|
2018-05-18 17:55:08 +00:00
|
|
|
import log from '../../utils/log';
|
2018-12-05 20:52:08 +00:00
|
|
|
import random from '../../utils/random';
|
2020-09-11 14:31:38 +00:00
|
|
|
import { Encryption } from '../encryption';
|
|
|
|
import { E2E_MESSAGE_TYPE, E2E_STATUS } from '../encryption/constants';
|
2018-04-24 19:34:03 +00:00
|
|
|
|
2020-03-02 20:12:41 +00:00
|
|
|
const changeMessageStatus = async(id, tmid, status, message) => {
|
2019-12-13 13:35:12 +00:00
|
|
|
const db = database.active;
|
2021-03-15 20:16:34 +00:00
|
|
|
const msgCollection = db.get('messages');
|
|
|
|
const threadMessagesCollection = db.get('thread_messages');
|
2019-12-13 13:35:12 +00:00
|
|
|
const successBatch = [];
|
|
|
|
const messageRecord = await msgCollection.find(id);
|
|
|
|
successBatch.push(
|
|
|
|
messageRecord.prepareUpdate((m) => {
|
|
|
|
m.status = status;
|
2020-03-05 16:30:13 +00:00
|
|
|
if (message) {
|
|
|
|
m.mentions = message.mentions;
|
|
|
|
m.channels = message.channels;
|
|
|
|
}
|
2019-12-13 13:35:12 +00:00
|
|
|
})
|
|
|
|
);
|
|
|
|
|
|
|
|
if (tmid) {
|
|
|
|
const threadMessageRecord = await threadMessagesCollection.find(id);
|
|
|
|
successBatch.push(
|
|
|
|
threadMessageRecord.prepareUpdate((tm) => {
|
|
|
|
tm.status = status;
|
2020-03-05 16:30:13 +00:00
|
|
|
if (message) {
|
|
|
|
tm.mentions = message.mentions;
|
|
|
|
tm.channels = message.channels;
|
|
|
|
}
|
2019-12-13 13:35:12 +00:00
|
|
|
})
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
|
|
|
await db.action(async() => {
|
|
|
|
await db.batch(...successBatch);
|
|
|
|
});
|
|
|
|
} catch (error) {
|
|
|
|
// Do nothing
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2019-10-08 12:36:15 +00:00
|
|
|
export async function sendMessageCall(message) {
|
2020-09-11 14:31:38 +00:00
|
|
|
const { _id, tmid } = message;
|
2018-05-23 13:39:18 +00:00
|
|
|
try {
|
2019-12-04 16:41:37 +00:00
|
|
|
const sdk = this.shareSDK || this.sdk;
|
2019-10-08 12:36:15 +00:00
|
|
|
// RC 0.60.0
|
2020-09-11 14:31:38 +00:00
|
|
|
const result = await sdk.post('chat.sendMessage', { message });
|
2020-03-05 16:30:13 +00:00
|
|
|
if (result.success) {
|
|
|
|
return changeMessageStatus(_id, tmid, messagesStatus.SENT, result.message);
|
|
|
|
}
|
|
|
|
} catch {
|
|
|
|
// do nothing
|
2018-05-23 13:39:18 +00:00
|
|
|
}
|
2020-03-05 16:30:13 +00:00
|
|
|
return changeMessageStatus(_id, tmid, messagesStatus.ERROR);
|
2018-04-24 19:34:03 +00:00
|
|
|
}
|
|
|
|
|
2020-09-11 14:31:38 +00:00
|
|
|
export async function resendMessage(message, tmid) {
|
|
|
|
const db = database.active;
|
|
|
|
try {
|
|
|
|
await db.action(async() => {
|
|
|
|
await message.update((m) => {
|
|
|
|
m.status = messagesStatus.TEMP;
|
|
|
|
});
|
|
|
|
});
|
|
|
|
let m = {
|
|
|
|
_id: message.id,
|
|
|
|
rid: message.subscription.id,
|
|
|
|
msg: message.msg
|
|
|
|
};
|
|
|
|
if (tmid) {
|
|
|
|
m = {
|
|
|
|
...m,
|
|
|
|
tmid
|
|
|
|
};
|
|
|
|
}
|
|
|
|
m = await Encryption.encryptMessage(m);
|
|
|
|
await sendMessageCall.call(this, m);
|
|
|
|
} catch (e) {
|
|
|
|
log(e);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-10-30 17:35:07 +00:00
|
|
|
export default async function(rid, msg, tmid, user, tshow) {
|
2018-04-24 19:34:03 +00:00
|
|
|
try {
|
2019-09-16 20:26:32 +00:00
|
|
|
const db = database.active;
|
2021-03-15 20:16:34 +00:00
|
|
|
const subsCollection = db.get('subscriptions');
|
|
|
|
const msgCollection = db.get('messages');
|
|
|
|
const threadCollection = db.get('threads');
|
|
|
|
const threadMessagesCollection = db.get('thread_messages');
|
2019-10-08 12:36:15 +00:00
|
|
|
const messageId = random(17);
|
|
|
|
const batch = [];
|
2020-09-11 14:31:38 +00:00
|
|
|
|
|
|
|
let message = {
|
2020-10-30 17:35:07 +00:00
|
|
|
_id: messageId, rid, msg, tmid, tshow
|
2019-10-08 12:36:15 +00:00
|
|
|
};
|
2020-09-11 14:31:38 +00:00
|
|
|
message = await Encryption.encryptMessage(message);
|
|
|
|
|
2019-10-08 12:36:15 +00:00
|
|
|
const messageDate = new Date();
|
|
|
|
let tMessageRecord;
|
|
|
|
|
|
|
|
// If it's replying to a thread
|
|
|
|
if (tmid) {
|
|
|
|
try {
|
|
|
|
// Find thread message header in Messages collection
|
|
|
|
tMessageRecord = await msgCollection.find(tmid);
|
|
|
|
batch.push(
|
|
|
|
tMessageRecord.prepareUpdate((m) => {
|
|
|
|
m.tlm = messageDate;
|
|
|
|
m.tcount += 1;
|
|
|
|
})
|
|
|
|
);
|
|
|
|
|
|
|
|
try {
|
|
|
|
// Find thread message header in Threads collection
|
|
|
|
await threadCollection.find(tmid);
|
|
|
|
} catch (error) {
|
|
|
|
// If there's no record, create one
|
|
|
|
batch.push(
|
|
|
|
threadCollection.prepareCreate((tm) => {
|
|
|
|
tm._raw = sanitizedRaw({ id: tmid }, threadCollection.schema);
|
|
|
|
tm.subscription.id = rid;
|
|
|
|
tm.tmid = tmid;
|
|
|
|
tm.msg = tMessageRecord.msg;
|
|
|
|
tm.ts = tMessageRecord.ts;
|
|
|
|
tm._updatedAt = messageDate;
|
|
|
|
tm.status = messagesStatus.SENT; // Original message was sent already
|
|
|
|
tm.u = tMessageRecord.u;
|
2020-09-11 14:31:38 +00:00
|
|
|
tm.t = message.t;
|
|
|
|
if (message.t === E2E_MESSAGE_TYPE) {
|
|
|
|
tm.e2e = E2E_STATUS.DONE;
|
|
|
|
}
|
2019-10-08 12:36:15 +00:00
|
|
|
})
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Create the message sent in ThreadMessages collection
|
|
|
|
batch.push(
|
|
|
|
threadMessagesCollection.prepareCreate((tm) => {
|
|
|
|
tm._raw = sanitizedRaw({ id: messageId }, threadMessagesCollection.schema);
|
|
|
|
tm.subscription.id = rid;
|
|
|
|
tm.rid = tmid;
|
|
|
|
tm.msg = msg;
|
|
|
|
tm.ts = messageDate;
|
|
|
|
tm._updatedAt = messageDate;
|
|
|
|
tm.status = messagesStatus.TEMP;
|
|
|
|
tm.u = {
|
|
|
|
_id: user.id || '1',
|
2021-03-15 20:16:34 +00:00
|
|
|
username: user.username,
|
|
|
|
name: user.name
|
2019-10-08 12:36:15 +00:00
|
|
|
};
|
2020-09-11 14:31:38 +00:00
|
|
|
tm.t = message.t;
|
|
|
|
if (message.t === E2E_MESSAGE_TYPE) {
|
|
|
|
tm.e2e = E2E_STATUS.DONE;
|
|
|
|
}
|
2019-10-08 12:36:15 +00:00
|
|
|
})
|
|
|
|
);
|
|
|
|
} catch (e) {
|
|
|
|
log(e);
|
|
|
|
}
|
2019-09-16 20:26:32 +00:00
|
|
|
}
|
2018-04-24 19:34:03 +00:00
|
|
|
|
2019-10-08 12:36:15 +00:00
|
|
|
// Create the message sent in Messages collection
|
|
|
|
batch.push(
|
|
|
|
msgCollection.prepareCreate((m) => {
|
|
|
|
m._raw = sanitizedRaw({ id: messageId }, msgCollection.schema);
|
|
|
|
m.subscription.id = rid;
|
|
|
|
m.msg = msg;
|
|
|
|
m.ts = messageDate;
|
|
|
|
m._updatedAt = messageDate;
|
|
|
|
m.status = messagesStatus.TEMP;
|
|
|
|
m.u = {
|
|
|
|
_id: user.id || '1',
|
2021-03-15 20:16:34 +00:00
|
|
|
username: user.username,
|
|
|
|
name: user.name
|
2019-10-08 12:36:15 +00:00
|
|
|
};
|
2020-03-05 16:31:21 +00:00
|
|
|
if (tmid && tMessageRecord) {
|
2019-10-08 12:36:15 +00:00
|
|
|
m.tmid = tmid;
|
2020-10-30 17:35:07 +00:00
|
|
|
// m.tlm = messageDate; // I don't think this is necessary... leaving it commented just in case...
|
2019-10-08 12:36:15 +00:00
|
|
|
m.tmsg = tMessageRecord.msg;
|
2020-10-30 17:35:07 +00:00
|
|
|
m.tshow = tshow;
|
2019-10-08 12:36:15 +00:00
|
|
|
}
|
2020-09-11 14:31:38 +00:00
|
|
|
m.t = message.t;
|
|
|
|
if (message.t === E2E_MESSAGE_TYPE) {
|
|
|
|
m.e2e = E2E_STATUS.DONE;
|
|
|
|
}
|
2019-10-08 12:36:15 +00:00
|
|
|
})
|
|
|
|
);
|
|
|
|
|
2019-09-16 20:26:32 +00:00
|
|
|
try {
|
2019-10-08 12:36:15 +00:00
|
|
|
const room = await subsCollection.find(rid);
|
|
|
|
if (room.draftMessage) {
|
|
|
|
batch.push(
|
|
|
|
room.prepareUpdate((r) => {
|
|
|
|
r.draftMessage = null;
|
|
|
|
})
|
|
|
|
);
|
|
|
|
}
|
2019-09-16 20:26:32 +00:00
|
|
|
} catch (e) {
|
|
|
|
// Do nothing
|
2019-04-17 17:01:03 +00:00
|
|
|
}
|
2018-04-24 19:34:03 +00:00
|
|
|
|
2018-11-16 11:06:29 +00:00
|
|
|
try {
|
2019-09-16 20:26:32 +00:00
|
|
|
await db.action(async() => {
|
2019-10-08 12:36:15 +00:00
|
|
|
await db.batch(...batch);
|
2018-11-16 11:06:29 +00:00
|
|
|
});
|
|
|
|
} catch (e) {
|
2019-10-08 12:36:15 +00:00
|
|
|
log(e);
|
|
|
|
return;
|
2018-11-16 11:06:29 +00:00
|
|
|
}
|
2019-10-08 12:36:15 +00:00
|
|
|
|
|
|
|
await sendMessageCall.call(this, message);
|
2018-04-24 19:34:03 +00:00
|
|
|
} catch (e) {
|
2019-08-23 13:18:47 +00:00
|
|
|
log(e);
|
2018-04-24 19:34:03 +00:00
|
|
|
}
|
|
|
|
}
|