upstream-rocketchat/packages/apps/base-runtime/src/handlers/app/handleUploadEvents.ts

import { Buffer } from 'node:buffer';
import { readFile } from 'node:fs/promises';

import type { App } from '@rocket.chat/apps-engine/definition/App';
import { AppsEngineException } from '@rocket.chat/apps-engine/definition/exceptions/AppsEngineException';
import type { IFileUploadContext } from '@rocket.chat/apps-engine/definition/uploads/IFileUploadContext';
import type { IUploadDetails } from '@rocket.chat/apps-engine/definition/uploads/IUploadDetails';
import type { Defined } from 'jsonrpc-lite';
import { JsonRpcError } from 'jsonrpc-lite';

import { AppObjectRegistry } from '../../AppObjectRegistry';
import { AppAccessorsInstance } from '../../lib/accessors/mod';
import type { RequestContext } from '../../lib/requestContext';
import { wrapAppForRequest } from '../../lib/wrapAppForRequest';
import { assertAppAvailable, assertHandlerFunction, isPlainObject } from '../lib/assertions';

export const uploadEvents = ['executePreFileUpload'] as const;

function assertIsUpload(v: unknown): asserts v is IUploadDetails {
	if (isPlainObject(v) && !!v.rid && (!!v.userId || !!v.visitorToken)) return;

	throw JsonRpcError.invalidParams({ err: `Invalid 'file' parameter. Expected IUploadDetails, got`, value: v });
}

function assertString(v: unknown): asserts v is string {
	if (v && typeof v === 'string') return;

	throw JsonRpcError.invalidParams({ err: `Invalid 'path' parameter. Expected string, got`, value: v });
}

export default async function handleUploadEvents(request: RequestContext): Promise<Defined | JsonRpcError> {
	const { method: rawMethod, params } = request as {
		method: `app:${(typeof uploadEvents)[number]}`;
		params: [{ file?: IUploadDetails; path?: string }];
	};
	const [, method] = rawMethod.split(':') as ['app', (typeof uploadEvents)[number]];

	try {
		const [{ file, path }] = params;

		const app = AppObjectRegistry.get<App>('app');
		const handlerFunction = app?.[method as keyof App] as unknown;

		assertAppAvailable(app);
		assertHandlerFunction(handlerFunction);
		assertIsUpload(file);
		assertString(path);

		let context: IFileUploadContext;

		switch (method) {
			case 'executePreFileUpload': {
				const fileContents = await readFile(path);
				context = { file, content: Buffer.from(fileContents) };
				break;
			}
		}

		return await handlerFunction.call(
			wrapAppForRequest(app, request),
			context,
			AppAccessorsInstance.getReader(),
			AppAccessorsInstance.getHttp(),
			AppAccessorsInstance.getPersistence(),
			AppAccessorsInstance.getModifier(),
		);
	} catch (e) {
		if (e?.name === AppsEngineException.name) {
			return new JsonRpcError(e.message, AppsEngineException.JSONRPC_ERROR_CODE, { name: e.name });
		}

		if (e instanceof JsonRpcError) {
			return e;
		}

		return JsonRpcError.internalError({
			err: e.message,
			...(e.code && { code: e.code }),
		});
	}
}