upstream-rocketchat/apps/meteor/client/lib/autotranslate/autotranslate.ts

import type { IRoom, ISubscription, ISupportedLanguage, ITranslatedMessage, MessageAttachmentDefault } from '@rocket.chat/core-typings';
import { isTranslatedMessageAttachment } from '@rocket.chat/core-typings';
import mem from 'mem';

import { PermissionsCachedStore } from '../../cachedStores';
import { Messages, Subscriptions, Users } from '../../stores';
import { hasTranslationLanguageInAttachments, hasTranslationLanguageInMessage } from '../../views/room/MessageList/lib/autoTranslate';
import { sdk } from '../SDKClient';
import { hasPermission } from '../authorization';
import { settings } from '../settings';
import { getUserId, userIdStore } from '../user';

let userLanguage = 'en';
let username = '';

const refreshUserCache = () => {
	const uid = userIdStore.getState();
	const user = uid ? Users.use.getState().get(uid) : undefined;
	if (!user) return;
	userLanguage = user.language || 'en';
	username = user.username || '';
};
refreshUserCache();
userIdStore.subscribe(refreshUserCache);
Users.use.subscribe(refreshUserCache);

export const AutoTranslate = {
	initialized: false,
	providersMetadata: {} as { [providerNamer: string]: { name: string; displayName: string } },
	messageIdsToWait: {} as { [messageId: string]: boolean },
	supportedLanguages: [] as ISupportedLanguage[] | undefined,

	findSubscriptionByRid: mem((rid) => Subscriptions.state.find((record) => record.rid === rid)),

	getLanguage(rid: IRoom['_id']): string {
		let subscription: ISubscription | undefined;
		if (rid) {
			subscription = this.findSubscriptionByRid(rid);
		}
		const language = (subscription?.autoTranslateLanguage || userLanguage || window.defaultUserLanguage?.()) as string;
		if (language.indexOf('-') !== -1) {
			if (!(this.supportedLanguages || []).some((supportedLanguage) => supportedLanguage.language === language)) {
				return language.slice(0, 2);
			}
		}
		return language;
	},

	translateAttachments(
		attachments: MessageAttachmentDefault[],
		language: string,
		autoTranslateShowInverse: boolean,
	): MessageAttachmentDefault[] {
		if (!isTranslatedMessageAttachment(attachments)) {
			return attachments;
		}
		for (const attachment of attachments) {
			if (attachment.author_name !== username) {
				if (attachment.text && attachment.translations && attachment.translations[language]) {
					attachment.translations.original = attachment.text;

					if (autoTranslateShowInverse) {
						attachment.text = attachment.translations.original;
					} else {
						attachment.text = attachment.translations[language];
					}
				}

				if (attachment.description && attachment.translations && attachment.translations[language]) {
					attachment.translations.original = attachment.description;

					if (autoTranslateShowInverse) {
						attachment.description = attachment.translations.original;
					} else {
						attachment.description = attachment.translations[language];
					}
				}

				if (attachment.attachments && attachment.attachments.length > 0) {
					// @ts-expect-error - not sure what to do with this
					attachment.attachments = this.translateAttachments(attachment.attachments, language);
				}
			}
		}
		return attachments;
	},

	init(): void {
		if (this.initialized) {
			return;
		}

		const loadProviders = async () => {
			try {
				const [providersMetadata, supportedLanguagesResponse] = await Promise.all([
					sdk.call('autoTranslate.getProviderUiMetadata'),
					sdk.rest.get('/v1/autotranslate.getSupportedLanguages', { targetLanguage: 'en' }),
				]);
				this.providersMetadata = providersMetadata;
				this.supportedLanguages = supportedLanguagesResponse.languages;
			} catch (e: unknown) {
				// Avoid unwanted error message on UI when autotranslate is disabled while fetching data
				console.error((e as Error).message);
			}
		};

		let loaded = false;
		const unsubs: Array<() => void> = [];
		const tryLoad = async () => {
			if (loaded) return;
			if (!settings.peek('AutoTranslate_Enabled') || !userIdStore.getState() || !hasPermission('auto-translate')) {
				return;
			}
			loaded = true;
			unsubs.splice(0).forEach((unsubscribe) => unsubscribe());
			await loadProviders();
		};

		unsubs.push(userIdStore.subscribe(() => void tryLoad()));
		unsubs.push(settings.observe('AutoTranslate_Enabled', () => void tryLoad()));
		unsubs.push(PermissionsCachedStore.useReady.subscribe(() => void tryLoad()));

		void tryLoad();

		Subscriptions.use.subscribe(() => {
			mem.clear(this.findSubscriptionByRid);
		});

		this.initialized = true;
	},
};

export const createAutoTranslateMessageStreamHandler = (): ((message: ITranslatedMessage) => void) => {
	AutoTranslate.init();

	return (message: ITranslatedMessage): void => {
		if (message.u && message.u._id !== getUserId()) {
			const subscription = AutoTranslate.findSubscriptionByRid(message.rid);
			const language = AutoTranslate.getLanguage(message.rid);
			if (
				subscription &&
				subscription.autoTranslate === true &&
				message.msg &&
				(!message.translations ||
					(!hasTranslationLanguageInMessage(message, language) && !hasTranslationLanguageInAttachments(message.attachments, language)))
			) {
				Messages.state.update(
					(record) => record._id === message._id,
					(record) => ({
						...record,
						autoTranslateFetching: true,
					}),
				);
			} else if (AutoTranslate.messageIdsToWait[message._id] !== undefined && subscription && subscription.autoTranslate !== true) {
				Messages.state.update(
					(record) => record._id === message._id,
					({ autoTranslateFetching: _, ...record }) => ({
						...record,
					}),
				);
				delete AutoTranslate.messageIdsToWait[message._id];
			} else if (message.autoTranslateFetching === true) {
				Messages.state.update(
					(record) => record._id === message._id,
					({ autoTranslateFetching: _, ...record }) => record,
				);
			}
		}
	};
};