503 lines
23 KiB
TypeScript
503 lines
23 KiB
TypeScript
import { useContext, useEffect } from "react";
|
||
import { Attachment, AttachmentType, PacketMessage } from "../ProtocolProvider/protocol/packets/packet.message";
|
||
import { usePacket } from "../ProtocolProvider/usePacket";
|
||
import { BlacklistContext } from "../BlacklistProvider/BlacklistProvider";
|
||
import { useLogger } from "@/app/hooks/useLogger";
|
||
import { useMemory } from "../MemoryProvider/useMemory";
|
||
import { useIdle } from "@mantine/hooks";
|
||
import { useNotification } from "@/app/hooks/useNotification";
|
||
import { useWindowFocus } from "@/app/hooks/useWindowFocus";
|
||
import { MESSAGE_MAX_LOADED, TIME_TO_INACTIVE_FOR_MESSAGES_UNREAD } from "@/app/constants";
|
||
import { useDialogsCache } from "./useDialogsCache";
|
||
import { useDatabase } from "@/app/providers/DatabaseProvider/useDatabase";
|
||
import { usePrivatePlain } from "../AccountProvider/usePrivatePlain";
|
||
import { usePublicKey } from "../AccountProvider/usePublicKey";
|
||
import { chacha20Decrypt, decodeWithPassword, decrypt, encodeWithPassword, generateMd5 } from "@/app/crypto/crypto";
|
||
import { DeliveredMessageState, Message } from "./DialogProvider";
|
||
import { PacketRead } from "../ProtocolProvider/protocol/packets/packet.read";
|
||
import { PacketDelivery } from "../ProtocolProvider/protocol/packets/packet.delivery";
|
||
import { useConsoleLogger } from "@/app/hooks/useConsoleLogger";
|
||
import { useViewPanelsState, ViewPanelsState } from "@/app/hooks/useViewPanelsState";
|
||
import { generateRandomKeyFormSeed } from "@/app/utils/utils";
|
||
import { useFileStorage } from "@/app/hooks/useFileStorage";
|
||
import { useDialogsList } from "../DialogListProvider/useDialogsList";
|
||
import { useGroups } from "./useGroups";
|
||
import { useDialogState } from "../DialogStateProvider.tsx/useDialogState";
|
||
import { useUserInformation } from "../InformationProvider/useUserInformation";
|
||
import { useMentions } from "../DialogStateProvider.tsx/useMentions";
|
||
|
||
/**
|
||
* При вызове будет запущен "фоновый" обработчик
|
||
* входящих пакетов сообщений, который будет обрабатывать их и сохранять
|
||
* в базу данных в кэше или в базе данных
|
||
*/
|
||
export function useDialogFiber() {
|
||
const { blocked } = useContext(BlacklistContext);
|
||
const { runQuery } = useDatabase();
|
||
const privatePlain = usePrivatePlain();
|
||
const publicKey = usePublicKey();
|
||
const log = useLogger('useDialogFiber');
|
||
const [currentDialogPublicKeyView, _] = useMemory("current-dialog-public-key-view", "", true);
|
||
const idle = useIdle(TIME_TO_INACTIVE_FOR_MESSAGES_UNREAD * 1000);
|
||
const notify = useNotification();
|
||
const focused = useWindowFocus();
|
||
const { getDialogCache, addOrUpdateDialogCache } = useDialogsCache();
|
||
const {info, error} = useConsoleLogger('useDialogFiber');
|
||
const [viewState] = useViewPanelsState();
|
||
const {writeFile} = useFileStorage();
|
||
const {updateDialog} = useDialogsList();
|
||
const {hasGroup, getGroupKey, normalize} = useGroups();
|
||
const {muted} = useDialogState();
|
||
const [userInfo] = useUserInformation(publicKey);
|
||
const {pushMention} = useMentions();
|
||
|
||
/**
|
||
* Лог
|
||
*/
|
||
useEffect(() => {
|
||
info("Starting passive fiber for dialog packets");
|
||
}, []);
|
||
|
||
|
||
/**
|
||
* Нам приходят сообщения от себя самих же при синхронизации
|
||
* нужно обрабатывать их особым образом соотвественно
|
||
*/
|
||
usePacket(0x06, async (packet: PacketMessage) => {
|
||
const fromPublicKey = packet.getFromPublicKey();
|
||
const toPublicKey = packet.getToPublicKey();
|
||
const aesChachaKey = packet.getAesChachaKey();
|
||
const content = packet.getContent();
|
||
const timestamp = packet.getTimestamp();
|
||
const messageId = packet.getMessageId();
|
||
|
||
|
||
if(fromPublicKey != publicKey){
|
||
/**
|
||
* Игнорируем если это не сообщение от нас
|
||
*/
|
||
return;
|
||
}
|
||
|
||
const chachaDecryptedKey = Buffer.from(await decodeWithPassword(privatePlain, aesChachaKey), "binary");
|
||
const key = chachaDecryptedKey.slice(0, 32);
|
||
const nonce = chachaDecryptedKey.slice(32);
|
||
const decryptedContent = await chacha20Decrypt(content, nonce.toString('hex'), key.toString('hex'));
|
||
|
||
let attachmentsMeta: any[] = [];
|
||
let messageAttachments: Attachment[] = [];
|
||
for (let i = 0; i < packet.getAttachments().length; i++) {
|
||
const attachment = packet.getAttachments()[i];
|
||
log("Attachment received id " + attachment.id + " type " + attachment.type);
|
||
|
||
let nextLength = messageAttachments.push({
|
||
...attachment,
|
||
blob: ""
|
||
});
|
||
|
||
if(attachment.type == AttachmentType.MESSAGES){
|
||
/**
|
||
* Этот тип вложения приходит сразу в blob и не нуждается
|
||
* в последующем скачивании
|
||
*/
|
||
const decryptedBlob = await decodeWithPassword(chachaDecryptedKey.toString('utf-8'), attachment.blob);
|
||
writeFile(`m/${await generateMd5(attachment.id + publicKey)}`,
|
||
Buffer.from(await encodeWithPassword(privatePlain, decryptedBlob)).toString('binary'));
|
||
messageAttachments[nextLength - 1].blob = decryptedBlob;
|
||
}
|
||
|
||
attachmentsMeta.push({
|
||
id: attachment.id,
|
||
type: attachment.type,
|
||
preview: attachment.preview
|
||
});
|
||
}
|
||
|
||
const newMessage: Message = {
|
||
from_public_key: fromPublicKey,
|
||
to_public_key: toPublicKey,
|
||
content: content,
|
||
timestamp: timestamp,
|
||
readed: idle ? 0 : 1,
|
||
chacha_key: chachaDecryptedKey.toString('utf-8'),
|
||
from_me: fromPublicKey == publicKey ? 1 : 0,
|
||
plain_message: (decryptedContent as string),
|
||
delivered: DeliveredMessageState.DELIVERED,
|
||
message_id: messageId,
|
||
attachments: messageAttachments
|
||
};
|
||
|
||
await runQuery(`
|
||
INSERT INTO messages
|
||
(from_public_key, to_public_key, content, timestamp, read, chacha_key, from_me, plain_message, account, message_id, delivered, attachments) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||
`, [fromPublicKey,
|
||
toPublicKey,
|
||
content,
|
||
timestamp,
|
||
(currentDialogPublicKeyView == fromPublicKey && !idle && viewState != ViewPanelsState.DIALOGS_PANEL_ONLY) ? 1 : 0,
|
||
'',
|
||
0,
|
||
await encodeWithPassword(privatePlain, decryptedContent),
|
||
publicKey,
|
||
messageId,
|
||
DeliveredMessageState.DELIVERED,
|
||
JSON.stringify(attachmentsMeta)]);
|
||
|
||
updateDialog(fromPublicKey);
|
||
|
||
let dialogCache = getDialogCache(toPublicKey);
|
||
if (currentDialogPublicKeyView !== toPublicKey && dialogCache.length > 0) {
|
||
addOrUpdateDialogCache(toPublicKey, [...dialogCache, newMessage].slice(-MESSAGE_MAX_LOADED));
|
||
}
|
||
}, [privatePlain, currentDialogPublicKeyView]);
|
||
|
||
/**
|
||
* Обработчик сообщений для группы
|
||
*/
|
||
usePacket(0x06, async (packet: PacketMessage) => {
|
||
const fromPublicKey = packet.getFromPublicKey();
|
||
const toPublicKey = packet.getToPublicKey();
|
||
const content = packet.getContent();
|
||
const timestamp = packet.getTimestamp();
|
||
const messageId = packet.getMessageId();
|
||
if(!hasGroup(toPublicKey)){
|
||
/**
|
||
* Если это личное сообщение, то игнорируем его здесь
|
||
* для него есть отдельный слушатель usePacket (снизу)
|
||
*/
|
||
return;
|
||
}
|
||
if(fromPublicKey == publicKey){
|
||
/**
|
||
* Игнорируем свои же сообщения,
|
||
* такое получается при пакете синхронизации
|
||
*/
|
||
return;
|
||
}
|
||
const groupKey = await getGroupKey(toPublicKey);
|
||
if(!groupKey){
|
||
log("Group key not found for group " + toPublicKey);
|
||
error("Message dropped because group key not found for group " + toPublicKey);
|
||
return;
|
||
}
|
||
info("New group message packet received from " + fromPublicKey);
|
||
|
||
let decryptedContent = '';
|
||
|
||
try{
|
||
decryptedContent = await decodeWithPassword(groupKey, content);
|
||
}catch(e) {
|
||
decryptedContent = '';
|
||
}
|
||
|
||
let attachmentsMeta: any[] = [];
|
||
let messageAttachments: Attachment[] = [];
|
||
for (let i = 0; i < packet.getAttachments().length; i++) {
|
||
const attachment = packet.getAttachments()[i];
|
||
log("Attachment received id " + attachment.id + " type " + attachment.type);
|
||
|
||
let nextLength = messageAttachments.push({
|
||
...attachment,
|
||
blob: ""
|
||
});
|
||
|
||
if(attachment.type == AttachmentType.MESSAGES){
|
||
/**
|
||
* Этот тип вложения приходит сразу в blob и не нуждается
|
||
* в последующем скачивании
|
||
*/
|
||
const decryptedBlob = await decodeWithPassword(groupKey, attachment.blob);
|
||
writeFile(`m/${await generateMd5(attachment.id + publicKey)}`,
|
||
Buffer.from(await encodeWithPassword(privatePlain, decryptedBlob)).toString('binary'));
|
||
messageAttachments[nextLength - 1].blob = decryptedBlob;
|
||
}
|
||
|
||
attachmentsMeta.push({
|
||
id: attachment.id,
|
||
type: attachment.type,
|
||
preview: attachment.preview
|
||
});
|
||
}
|
||
|
||
const newMessage: Message = {
|
||
from_public_key: fromPublicKey,
|
||
to_public_key: toPublicKey,
|
||
content: content,
|
||
timestamp: timestamp,
|
||
readed: idle ? 0 : 1,
|
||
chacha_key: groupKey,
|
||
from_me: fromPublicKey == publicKey ? 1 : 0,
|
||
plain_message: decryptedContent,
|
||
delivered: DeliveredMessageState.DELIVERED,
|
||
message_id: messageId,
|
||
attachments: messageAttachments
|
||
};
|
||
|
||
await runQuery(`
|
||
INSERT INTO messages
|
||
(from_public_key, to_public_key, content, timestamp, read, chacha_key, from_me, plain_message, account, message_id, delivered, attachments) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||
`, [fromPublicKey,
|
||
toPublicKey,
|
||
content,
|
||
timestamp,
|
||
/**если текущий открытый диалог == беседе (которая приходит в toPublicKey) */
|
||
(currentDialogPublicKeyView == toPublicKey && !idle && viewState != ViewPanelsState.DIALOGS_PANEL_ONLY) ? 1 : 0,
|
||
'',
|
||
0,
|
||
await encodeWithPassword(privatePlain, decryptedContent),
|
||
publicKey,
|
||
messageId,
|
||
DeliveredMessageState.DELIVERED,
|
||
JSON.stringify(attachmentsMeta)]);
|
||
|
||
/**
|
||
* Так как у нас в toPublicKey приходит ID группы,
|
||
* то обновляем диалог по этому ID, а не по fromPublicKey
|
||
* как это сделано в личных сообщениях
|
||
*/
|
||
updateDialog(toPublicKey);
|
||
|
||
if (((normalize(currentDialogPublicKeyView) !== normalize(toPublicKey) || viewState == ViewPanelsState.DIALOGS_PANEL_ONLY) &&
|
||
(timestamp + TIME_TO_INACTIVE_FOR_MESSAGES_UNREAD) > (Date.now() / 1000)) || !focused) {
|
||
/**
|
||
* Условие со временем нужно для того,
|
||
* чтобы когда приходит пачка сообщений с сервера в момент того как
|
||
* пользователь был неактивен, не слать уведомления по всем этим сообщениям
|
||
*/
|
||
let mentionFlag = false;
|
||
if((newMessage.from_public_key != publicKey) && (decryptedContent.includes(`@${userInfo.username}`) || decryptedContent.includes(`@all`))){
|
||
/**
|
||
* Если в сообщении есть упоминание текущего пользователя или @all,
|
||
* при этом сообщение отправляли не мы,
|
||
* то добавляем упоминание в состояние диалога.
|
||
*
|
||
* TODO: сделать чтобы all работал только для админов группы
|
||
*/
|
||
mentionFlag = true;
|
||
}
|
||
|
||
if(!muted.includes(toPublicKey) || mentionFlag){
|
||
/**
|
||
* Если группа не в мутие или есть упоминание - отправляем уведомление
|
||
*/
|
||
notify("New message", "You have a new message");
|
||
}
|
||
if(mentionFlag){
|
||
/**
|
||
* Если в сообщении есть упоминание текущего пользователя или @all,
|
||
* то добавляем упоминание в состояние диалога
|
||
*
|
||
* TODO: сделать чтобы all работал только для админов группы
|
||
*/
|
||
pushMention({
|
||
dialog_id: toPublicKey,
|
||
message_id: messageId
|
||
});
|
||
}
|
||
}
|
||
let dialogCache = getDialogCache(toPublicKey);
|
||
if (currentDialogPublicKeyView !== toPublicKey && dialogCache.length > 0) {
|
||
addOrUpdateDialogCache(toPublicKey, [...dialogCache, newMessage].slice(-MESSAGE_MAX_LOADED));
|
||
}
|
||
}, [blocked, muted, updateDialog, focused, currentDialogPublicKeyView, viewState, idle]);
|
||
/**
|
||
* Обработчик личных сообщений
|
||
*/
|
||
usePacket(0x06, async (packet: PacketMessage) => {
|
||
const fromPublicKey = packet.getFromPublicKey();
|
||
if(fromPublicKey == publicKey){
|
||
/**
|
||
* Игнорируем свои же сообщения,
|
||
* такое получается при пакете синхронизации
|
||
*/
|
||
return;
|
||
}
|
||
const toPublicKey = packet.getToPublicKey();
|
||
const content = packet.getContent();
|
||
const chachaKey = packet.getChachaKey();
|
||
const timestamp = packet.getTimestamp();
|
||
const messageId = generateRandomKeyFormSeed(16, fromPublicKey + toPublicKey + timestamp.toString());
|
||
if(hasGroup(toPublicKey)){
|
||
/**
|
||
* Если это групповое сообщение, то игнорируем его здесь
|
||
* для него есть отдельный слушатель usePacket
|
||
*/
|
||
return;
|
||
}
|
||
info("New message packet received from " + fromPublicKey);
|
||
if (blocked.includes(fromPublicKey)) {
|
||
/**
|
||
* Если пользователь заблокирован и это не групповое сообщение,
|
||
* то игнорируем сообщение
|
||
*/
|
||
log("Message from blocked user, ignore " + fromPublicKey);
|
||
return;
|
||
}
|
||
|
||
if (privatePlain == "") {
|
||
return;
|
||
}
|
||
|
||
const chachaDecryptedKey = Buffer.from(await decrypt(chachaKey, privatePlain), "binary");
|
||
const key = chachaDecryptedKey.slice(0, 32);
|
||
const nonce = chachaDecryptedKey.slice(32);
|
||
const decryptedContent = await chacha20Decrypt(content, nonce.toString('hex'), key.toString('hex'));
|
||
|
||
let attachmentsMeta: any[] = [];
|
||
let messageAttachments: Attachment[] = [];
|
||
for (let i = 0; i < packet.getAttachments().length; i++) {
|
||
const attachment = packet.getAttachments()[i];
|
||
log("Attachment received id " + attachment.id + " type " + attachment.type);
|
||
|
||
let nextLength = messageAttachments.push({
|
||
...attachment,
|
||
blob: ""
|
||
});
|
||
|
||
if(attachment.type == AttachmentType.MESSAGES){
|
||
/**
|
||
* Этот тип вложения приходит сразу в blob и не нуждается
|
||
* в последующем скачивании
|
||
*/
|
||
const decryptedBlob = await decodeWithPassword(chachaDecryptedKey.toString('utf-8'), attachment.blob);
|
||
writeFile(`m/${await generateMd5(attachment.id + publicKey)}`,
|
||
Buffer.from(await encodeWithPassword(privatePlain, decryptedBlob)).toString('binary'));
|
||
messageAttachments[nextLength - 1].blob = decryptedBlob;
|
||
}
|
||
|
||
attachmentsMeta.push({
|
||
id: attachment.id,
|
||
type: attachment.type,
|
||
preview: attachment.preview
|
||
});
|
||
}
|
||
|
||
const newMessage: Message = {
|
||
from_public_key: fromPublicKey,
|
||
to_public_key: toPublicKey,
|
||
content: content,
|
||
timestamp: timestamp,
|
||
readed: idle ? 0 : 1,
|
||
chacha_key: chachaDecryptedKey.toString('utf-8'),
|
||
from_me: fromPublicKey == publicKey ? 1 : 0,
|
||
plain_message: (decryptedContent as string),
|
||
delivered: DeliveredMessageState.DELIVERED,
|
||
message_id: messageId,
|
||
attachments: messageAttachments
|
||
};
|
||
|
||
|
||
await runQuery(`
|
||
INSERT INTO messages
|
||
(from_public_key, to_public_key, content, timestamp, read, chacha_key, from_me, plain_message, account, message_id, delivered, attachments) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||
`, [fromPublicKey,
|
||
toPublicKey,
|
||
content,
|
||
timestamp,
|
||
(currentDialogPublicKeyView == fromPublicKey && !idle && viewState != ViewPanelsState.DIALOGS_PANEL_ONLY) ? 1 : 0,
|
||
chachaKey,
|
||
0,
|
||
await encodeWithPassword(privatePlain, decryptedContent),
|
||
publicKey,
|
||
messageId,
|
||
DeliveredMessageState.DELIVERED,
|
||
JSON.stringify(attachmentsMeta)]);
|
||
|
||
log("New message received from " + fromPublicKey);
|
||
|
||
updateDialog(fromPublicKey);
|
||
if (((currentDialogPublicKeyView !== fromPublicKey || viewState == ViewPanelsState.DIALOGS_PANEL_ONLY) &&
|
||
(timestamp + TIME_TO_INACTIVE_FOR_MESSAGES_UNREAD) > (Date.now() / 1000)) || !focused) {
|
||
/**
|
||
* Условие со временем нужно для того,
|
||
* чтобы когда приходит пачка сообщений с сервера в момент того как
|
||
* пользователь был неактивен, не слать уведомления по всем этим сообщениям
|
||
*/
|
||
if(!muted.includes(fromPublicKey)){
|
||
/**
|
||
* Если пользователь в муте - не отправляем уведомление
|
||
*/
|
||
notify("New message", "You have a new message");
|
||
}
|
||
}
|
||
let dialogCache = getDialogCache(fromPublicKey);
|
||
if (currentDialogPublicKeyView !== fromPublicKey && dialogCache.length > 0) {
|
||
addOrUpdateDialogCache(fromPublicKey, [...dialogCache, newMessage].slice(-MESSAGE_MAX_LOADED));
|
||
}
|
||
}, [blocked, muted, updateDialog, focused, currentDialogPublicKeyView, viewState, idle]);
|
||
|
||
/**
|
||
* Обработчик прочтения личных сообщений
|
||
*/
|
||
usePacket(0x07, async (packet: PacketRead) => {
|
||
if(hasGroup(packet.getToPublicKey())){
|
||
/**
|
||
* Если это относится к группам, то игнорируем здесь,
|
||
* для этого есть отдельный слушатель usePacket ниже
|
||
*/
|
||
return;
|
||
}
|
||
const fromPublicKey = packet.getFromPublicKey();
|
||
const toPublicKey = packet.getToPublicKey();
|
||
await runQuery(`UPDATE messages SET read = 1 WHERE from_public_key = ? AND to_public_key = ? AND account = ?`, [toPublicKey, fromPublicKey, publicKey]);
|
||
updateDialog(fromPublicKey);
|
||
log("Read packet received from " + fromPublicKey + " for " + toPublicKey);
|
||
addOrUpdateDialogCache(fromPublicKey, getDialogCache(fromPublicKey).map((message) => {
|
||
if (message.from_public_key == toPublicKey && !message.readed) {
|
||
console.info("Marking message as read in cache for dialog with " + fromPublicKey);
|
||
console.info({fromPublicKey, toPublicKey});
|
||
return {
|
||
...message,
|
||
readed: 1
|
||
}
|
||
}
|
||
return message;
|
||
}));
|
||
}, [updateDialog]);
|
||
/**
|
||
* Обработчик прочтения групповых сообщений
|
||
*/
|
||
usePacket(0x07, async (packet: PacketRead) => {
|
||
if(!hasGroup(packet.getToPublicKey())){
|
||
/**
|
||
* Если это не относится к группам, то игнорируем здесь,
|
||
* для этого есть отдельный слушатель usePacket выше
|
||
*/
|
||
return;
|
||
}
|
||
const fromPublicKey = packet.getFromPublicKey();
|
||
const toPublicKey = packet.getToPublicKey();
|
||
await runQuery(`UPDATE messages SET read = 1 WHERE to_public_key = ? AND from_public_key = ? AND account = ?`, [toPublicKey, publicKey, publicKey]);
|
||
updateDialog(toPublicKey);
|
||
addOrUpdateDialogCache(toPublicKey, getDialogCache(toPublicKey).map((message) => {
|
||
if (!message.readed) {
|
||
console.info("Marking message as read in cache for dialog with " + fromPublicKey);
|
||
console.info({fromPublicKey, toPublicKey});
|
||
return {
|
||
...message,
|
||
readed: 1
|
||
}
|
||
}
|
||
return message;
|
||
}));
|
||
}, [updateDialog]);
|
||
/**
|
||
* Обработчик доставки сообщений
|
||
*/
|
||
usePacket(0x08, async (packet: PacketDelivery) => {
|
||
const messageId = packet.getMessageId();
|
||
await runQuery(`UPDATE messages SET delivered = ?, timestamp = ? WHERE message_id = ? AND account = ?`, [DeliveredMessageState.DELIVERED, Date.now(), messageId, publicKey]);
|
||
updateDialog(packet.getToPublicKey());
|
||
log("Delivery packet received msg id " + messageId);
|
||
addOrUpdateDialogCache(packet.getToPublicKey(), getDialogCache(packet.getToPublicKey()).map((message) => {
|
||
if (message.message_id == messageId) {
|
||
return {
|
||
...message,
|
||
delivered: DeliveredMessageState.DELIVERED,
|
||
timestamp: Date.now()
|
||
}
|
||
}
|
||
return message;
|
||
}));
|
||
}, [updateDialog]);
|
||
} |