import type { IUser } from '@rocket.chat/core-typings';
import { Settings, Users } from '@rocket.chat/models';
import semver from 'semver';

import { Info } from '../../../../../app/utils/rocketchat.info';
import { settings } from '../../../../settings';
import { updateAuditedBySystem } from '../../../../settings/lib/auditedSettingUpdates';
import { i18n } from '../../../i18n';
import { notifyOnSettingChangedById } from '../../../notifyListener';
import { sendMessagesToAdmins } from '../../../sendMessagesToAdmins';

const cleanupOutdatedVersionUpdateBanners = async (): Promise<void> => {
	const admins = Users.findUsersInRolesWithQuery('admin', { banners: { $exists: true } }, { projection: { _id: 1, banners: 1 } });

	const updates: { userId: IUser['_id']; banners: NonNullable<IUser['banners']> }[] = [];

	for await (const admin of admins) {
		if (!admin.banners) {
			continue;
		}

		const filteredBanners = Object.fromEntries(
			Object.entries(admin.banners).filter(([bannerId]) => {
				if (!bannerId.startsWith('versionUpdate-')) {
					return true;
				}
				const version = bannerId.replace('versionUpdate-', '').replace(/_/g, '.');
				if (!semver.valid(version) || semver.lte(version, Info.version)) {
					return false;
				}
				return true;
			}),
		);

		if (Object.keys(filteredBanners).length !== Object.keys(admin.banners).length) {
			updates.push({ userId: admin._id, banners: filteredBanners });
		}
	}

	if (updates.length > 0) {
		await Users.setBannersInBulk(updates);
	}
};

export const buildVersionUpdateMessage = async (
	versions: {
		version: string;
		security: boolean;
		infoUrl: string;
	}[] = [],
) => {
	if (process.env.TEST_MODE) {
		return;
	}

	const lastCheckedVersion = settings.get<string>('Update_LatestAvailableVersion');

	if (!lastCheckedVersion) {
		return;
	}

	const sortedVersions = [...versions].sort((a, b) => semver.rcompare(a.version, b.version));

	await cleanupOutdatedVersionUpdateBanners();

	for await (const version of sortedVersions) {
		if (semver.prerelease(version.version)) {
			continue;
		}

		if (semver.lte(version.version, lastCheckedVersion)) {
			continue;
		}

		if (semver.lte(version.version, Info.version)) {
			continue;
		}

		(
			await updateAuditedBySystem({
				reason: 'buildVersionUpdateMessage',
			})(Settings.updateValueById, 'Update_LatestAvailableVersion', version.version)
		).modifiedCount && void notifyOnSettingChangedById('Update_LatestAvailableVersion');

		await sendMessagesToAdmins({
			msgs: async ({ adminUser }) => [
				{
					msg: `*${i18n.t('Update_your_RocketChat', { ...(adminUser.language && { lng: adminUser.language }) })}*\n${i18n.t(
						'New_version_available_(s)',
						{
							postProcess: 'sprintf',
							sprintf: [version.version],
						},
					)}\n${version.infoUrl}`,
				},
			],
			banners: [
				{
					id: `versionUpdate-${version.version}`.replace(/\./g, '_'),
					priority: 10,
					title: 'Update_your_RocketChat',
					text: 'New_version_available_(s)',
					textArguments: [version.version],
					link: version.infoUrl,
					modifiers: [],
				},
			],
		});
		break;
	}
};