2024-04-17 14:29:47 -04:00
|
|
|
/*
|
|
|
|
* Vencord, a modification for Discord's desktop app
|
|
|
|
* Copyright (c) 2023 Vendicated and contributors
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2024-11-18 22:19:10 -05:00
|
|
|
import { Flogger, settings } from ".";
|
|
|
|
import { addMessageIDB, db, DBMessageStatus, deleteMessagesBulkIDB, getOldestMessagesIDB } from "./db";
|
|
|
|
import { LoggedMessage, LoggedMessageJSON } from "./types";
|
|
|
|
import { cleanupMessage } from "./utils";
|
|
|
|
import { cacheMessageImages } from "./utils/saveImage";
|
|
|
|
|
|
|
|
export const addMessage = async (message: LoggedMessage | LoggedMessageJSON, status: DBMessageStatus) => {
|
|
|
|
if (settings.store.saveImages && status === DBMessageStatus.DELETED)
|
2024-04-17 14:29:47 -04:00
|
|
|
await cacheMessageImages(message);
|
|
|
|
const finalMessage = cleanupMessage(message);
|
|
|
|
|
2024-11-18 22:19:10 -05:00
|
|
|
await addMessageIDB(finalMessage, status);
|
2024-04-17 14:29:47 -04:00
|
|
|
|
2024-11-18 22:19:10 -05:00
|
|
|
if (settings.store.messageLimit > 0) {
|
|
|
|
const currentMessageCount = await db.count("messages");
|
|
|
|
if (currentMessageCount > settings.store.messageLimit) {
|
|
|
|
const messagesToDelete = currentMessageCount - settings.store.messageLimit;
|
|
|
|
if (messagesToDelete <= 0 || messagesToDelete >= settings.store.messageLimit) return;
|
2024-04-17 14:29:47 -04:00
|
|
|
|
2024-11-18 22:19:10 -05:00
|
|
|
const oldestMessages = await getOldestMessagesIDB(messagesToDelete);
|
2024-04-17 14:29:47 -04:00
|
|
|
|
2024-11-18 22:19:10 -05:00
|
|
|
Flogger.info(`Deleting ${messagesToDelete} oldest messages`);
|
|
|
|
await deleteMessagesBulkIDB(oldestMessages.map(m => m.message_id));
|
2024-04-17 14:29:47 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|