Rocket.Chat/apps/meteor/app/cloud/server/functions/saveRegistrationData.ts

import { applyLicense } from '@rocket.chat/license';
import { Settings } from '@rocket.chat/models';

import { syncCloudData } from './syncWorkspace/syncCloudData';
import { updateAuditedBySystem } from '../../../../server/settings/lib/auditedSettingUpdates';
import { notifyOnSettingChangedById } from '../../../lib/server/lib/notifyListener';
import { settings } from '../../../settings/server';

type SaveRegistrationDataDTO = {
	workspaceId: string;
	client_name: string;
	client_id: string;
	client_secret: string;
	client_secret_expires_at: number;
	publicKey: string;
	registration_client_uri: string;
};

type ManualSaveRegistrationDataDTO = SaveRegistrationDataDTO & { licenseData: { license: string } };

export async function saveRegistrationData({
	workspaceId,
	client_name,
	client_id,
	client_secret,
	client_secret_expires_at,
	publicKey,
	registration_client_uri,
}: SaveRegistrationDataDTO) {
	await saveRegistrationDataBase({
		workspaceId,
		client_name,
		client_id,
		client_secret,
		client_secret_expires_at,
		publicKey,
		registration_client_uri,
	});

	await syncCloudData();
}

async function saveRegistrationDataBase({
	workspaceId,
	client_name,
	client_id,
	client_secret,
	client_secret_expires_at,
	publicKey,
	registration_client_uri,
}: SaveRegistrationDataDTO) {
	const settingsData = [
		{ _id: 'Register_Server', value: true },
		{ _id: 'Cloud_Workspace_Id', value: workspaceId },
		{ _id: 'Cloud_Workspace_Name', value: client_name },
		{ _id: 'Cloud_Workspace_Client_Id', value: client_id },
		{ _id: 'Cloud_Workspace_Client_Secret', value: client_secret },
		{ _id: 'Cloud_Workspace_Client_Secret_Expires_At', value: client_secret_expires_at },
		{ _id: 'Cloud_Workspace_PublicKey', value: publicKey },
		{ _id: 'Cloud_Workspace_Registration_Client_Uri', value: registration_client_uri },
	];

	const promises = [
		...settingsData.map(({ _id, value }) =>
			updateAuditedBySystem({
				reason: 'saveRegistrationDataBase',
			})(Settings.updateValueById, _id, value),
		),
	];

	(await Promise.all(promises)).forEach((value, index) => {
		if (value?.modifiedCount) {
			void notifyOnSettingChangedById(settingsData[index]._id);
		}
	});

	// Question: Why is this taking so long that needs a timeout?
	// Answer: we use cache that requires a 'roundtrip' through the db and the application
	// we need to make sure that the cache is updated before we continue the procedures
	// we don't actually need to wait a whole second for this, but look this is just a retry mechanism it doesn't mean that actually takes all this time
	for await (const retry of Array.from({ length: 10 })) {
		const isSettingsUpdated =
			settings.get('Register_Server') === true &&
			settings.get('Cloud_Workspace_Id') === workspaceId &&
			settings.get('Cloud_Workspace_Name') === client_name &&
			settings.get('Cloud_Workspace_Client_Id') === client_id &&
			settings.get('Cloud_Workspace_Client_Secret') === client_secret &&
			settings.get('Cloud_Workspace_Client_Secret_Expires_At') === client_secret_expires_at &&
			settings.get('Cloud_Workspace_PublicKey') === publicKey &&
			settings.get('Cloud_Workspace_Registration_Client_Uri') === registration_client_uri;

		if (isSettingsUpdated) {
			return;
		}

		if (retry === 9) {
			throw new Error('Failed to save registration data');
		}

		await new Promise((resolve) => setTimeout(resolve, 1000));
	}
}

export async function saveRegistrationDataManual({
	workspaceId,
	client_name,
	client_id,
	client_secret,
	client_secret_expires_at,
	publicKey,
	registration_client_uri,
	licenseData,
}: ManualSaveRegistrationDataDTO) {
	await saveRegistrationDataBase({
		workspaceId,
		client_name,
		client_id,
		client_secret,
		client_secret_expires_at,
		publicKey,
		registration_client_uri,
	});
	await applyLicense(licenseData.license, true);
}