import type { IDirectMessageRoom, IRoom } from '@rocket.chat/core-typings';
import { Analytics } from '@rocket.chat/models';
import moment from 'moment';

import { convertDateToInt, diffBetweenDaysInclusive } from './date';
import { roomCoordinator } from '../../../../server/lib/rooms/roomCoordinator';

export const findChannelsWithNumberOfMessages = async ({
	start,
	end,
	options = {},
}: {
	start: Date;
	end: Date;
	options: {
		offset?: number;
		count?: number;
	};
}): Promise<{
	channels: {
		room: {
			_id: IRoom['_id'];
			name: IRoom['name'] | IRoom['fname'];
			ts: IRoom['ts'];
			t: IRoom['t'];
			_updatedAt: IRoom['_updatedAt'];
			usernames?: IDirectMessageRoom['usernames'];
		};
		messages: number;
		lastWeekMessages: number;
		diffFromLastWeek: number;
	}[];
	total: number;
}> => {
	const daysBetweenDates = diffBetweenDaysInclusive(end, start);
	const endOfLastWeek = moment(start).subtract(1, 'days').toDate();
	const startOfLastWeek = moment(endOfLastWeek).subtract(daysBetweenDates, 'days').toDate();
	const roomTypes = roomCoordinator.getTypesToShowOnDashboard() as Array<IRoom['t']>;

	const aggregationResult = await Analytics.findRoomsByTypesWithNumberOfMessagesBetweenDate({
		types: roomTypes,
		start: convertDateToInt(start),
		end: convertDateToInt(end),
		startOfLastWeek: convertDateToInt(startOfLastWeek),
		endOfLastWeek: convertDateToInt(endOfLastWeek),
		options,
	}).toArray();

	// The aggregation result may be undefined if there are no matching analytics or corresponding rooms in the period
	if (!aggregationResult.length) {
		return { channels: [], total: 0 };
	}

	const [{ channels, total }] = aggregationResult;
	return {
		channels,
		total,
	};
};