upstream-rocketchat/apps/meteor/server/meteor-methods/messages/loadSurroundingMessages.ts

import type { IMessage } from '@rocket.chat/core-typings';
import type { ServerMethods } from '@rocket.chat/ddp-client';
import { Messages } from '@rocket.chat/models';
import { check } from 'meteor/check';
import { Meteor } from 'meteor/meteor';
import type { FindOptions } from 'mongodb';

import { canAccessRoomIdAsync } from '../../lib/authorization/canAccessRoom';
import { methodDeprecationLogger } from '../../lib/deprecationWarningLogger';
import { normalizeMessagesForUser } from '../../lib/utils/lib/normalizeMessagesForUser';
import { settings } from '../../settings';

declare module '@rocket.chat/ddp-client' {
	// eslint-disable-next-line @typescript-eslint/naming-convention
	interface ServerMethods {
		loadSurroundingMessages(
			message: Pick<IMessage, '_id' | 'rid'> & { ts?: Date },
			limit?: number,
			showThreadMessages?: boolean,
		):
			| {
					messages: IMessage[];
					moreBefore: boolean;
					moreAfter: boolean;
			  }
			| false;
	}
}

Meteor.methods<ServerMethods>({
	async loadSurroundingMessages(message, limit = 50, showThreadMessages = true) {
		methodDeprecationLogger.method('loadSurroundingMessages', '9.0.0', '/v1/rooms.history');
		check(message, Object);
		check(limit, Number);
		check(showThreadMessages, Boolean);

		const fromId = Meteor.userId() ?? undefined;

		if (!fromId && settings.get('Accounts_AllowAnonymousRead') === false) {
			throw new Meteor.Error('error-invalid-user', 'Invalid user', {
				method: 'loadSurroundingMessages',
			});
		}

		if (!message._id) {
			return false;
		}

		const mainMessage = await Messages.findOneById(message._id);

		if (!mainMessage?.rid) {
			return false;
		}

		if (!(await canAccessRoomIdAsync(mainMessage.rid, fromId))) {
			return false;
		}

		limit -= 1;

		const options: FindOptions<IMessage> = {
			sort: {
				ts: -1,
			},
			limit: Math.ceil(limit / 2),
		};

		const messages = await Messages.findVisibleByRoomIdBeforeTimestamp(
			mainMessage.rid,
			mainMessage.ts,
			showThreadMessages,
			options,
		).toArray();

		const moreBefore = messages.length === options.limit;

		messages.push(mainMessage);

		options.sort = {
			ts: 1,
		};

		options.limit = Math.floor(limit / 2);

		const afterMessages = await Messages.findVisibleByRoomIdAfterTimestamp(
			mainMessage.rid,
			mainMessage.ts,
			showThreadMessages,
			options,
		).toArray();

		const moreAfter = afterMessages.length === options.limit;

		messages.push(...afterMessages);

		return {
			messages: fromId ? await normalizeMessagesForUser(messages, fromId) : messages,
			moreBefore,
			moreAfter,
		};
	},
});