Rocket.Chat/apps/meteor/app/api/server/v1/autotranslate.ts

import { Messages } from '@rocket.chat/models';
import {
	isAutotranslateSaveSettingsParamsPOST,
	isAutotranslateTranslateMessageParamsPOST,
	isAutotranslateGetSupportedLanguagesParamsGET,
} from '@rocket.chat/rest-typings';

import { getSupportedLanguages } from '../../../autotranslate/server/functions/getSupportedLanguages';
import { saveAutoTranslateSettings } from '../../../autotranslate/server/functions/saveSettings';
import { translateMessage } from '../../../autotranslate/server/functions/translateMessage';
import { settings } from '../../../settings/server';
import { API } from '../api';

API.v1.addRoute(
	'autotranslate.getSupportedLanguages',
	{
		authRequired: true,
		validateParams: isAutotranslateGetSupportedLanguagesParamsGET,
	},
	{
		async get() {
			if (!settings.get('AutoTranslate_Enabled')) {
				return API.v1.failure('AutoTranslate is disabled.');
			}
			const { targetLanguage } = this.queryParams;
			const languages = await getSupportedLanguages(this.userId, targetLanguage);

			return API.v1.success({ languages: languages || [] });
		},
	},
);

API.v1.addRoute(
	'autotranslate.saveSettings',
	{
		authRequired: true,
		validateParams: isAutotranslateSaveSettingsParamsPOST,
	},
	{
		async post() {
			const { roomId, field, value, defaultLanguage } = this.bodyParams;
			if (!settings.get('AutoTranslate_Enabled')) {
				return API.v1.failure('AutoTranslate is disabled.');
			}

			if (!roomId) {
				return API.v1.failure('The bodyParam "roomId" is required.');
			}
			if (!field) {
				return API.v1.failure('The bodyParam "field" is required.');
			}
			if (value === undefined) {
				return API.v1.failure('The bodyParam "value" is required.');
			}
			if (field === 'autoTranslate' && typeof value !== 'boolean') {
				return API.v1.failure('The bodyParam "autoTranslate" must be a boolean.');
			}

			if (field === 'autoTranslateLanguage' && (typeof value !== 'string' || !Number.isNaN(Number.parseInt(value)))) {
				return API.v1.failure('The bodyParam "autoTranslateLanguage" must be a string.');
			}

			await saveAutoTranslateSettings(this.userId, roomId, field, value === true ? '1' : String(value).valueOf(), {
				defaultLanguage: defaultLanguage || '',
			});

			return API.v1.success();
		},
	},
);

API.v1.addRoute(
	'autotranslate.translateMessage',
	{
		authRequired: true,
		validateParams: isAutotranslateTranslateMessageParamsPOST,
	},
	{
		async post() {
			const { messageId, targetLanguage } = this.bodyParams;
			if (!settings.get('AutoTranslate_Enabled')) {
				return API.v1.failure('AutoTranslate is disabled.');
			}
			if (!messageId) {
				return API.v1.failure('The bodyParam "messageId" is required.');
			}
			const message = await Messages.findOneById(messageId);
			if (!message) {
				return API.v1.failure('Message not found.');
			}

			const translatedMessage = await translateMessage(targetLanguage, message);

			return API.v1.success({ message: translatedMessage });
		},
	},
);