import type { ILivechatCustomField } from '@rocket.chat/core-typings';
import { LivechatCustomField, LivechatVisitors, LivechatRooms } from '@rocket.chat/models';
import type { PaginatedResult } from '@rocket.chat/rest-typings';
import { escapeRegExp } from '@rocket.chat/tools';
import type { UpdateResult, Document } from 'mongodb';

export async function findLivechatCustomFields({
	text,
	pagination: { offset, count, sort },
}: {
	text?: string;
	pagination: { offset: number; count: number; sort: Record<string, 1 | -1> };
}): Promise<PaginatedResult<{ customFields: Array<ILivechatCustomField> }>> {
	const query = {
		...(text && {
			$or: [{ label: new RegExp(escapeRegExp(text), 'i') }, { _id: new RegExp(escapeRegExp(text), 'i') }],
		}),
	};

	const { cursor, totalCount } = LivechatCustomField.findPaginated(query, {
		sort: sort || { label: 1 },
		skip: offset,
		limit: count,
	});

	const [customFields, total] = await Promise.all([cursor.toArray(), totalCount]);

	return {
		customFields,
		count: customFields.length,
		offset,
		total,
	};
}

export async function findCustomFieldById({
	customFieldId,
}: {
	customFieldId: string;
}): Promise<{ customField: ILivechatCustomField | null }> {
	return {
		customField: await LivechatCustomField.findOneById(customFieldId),
	};
}

export async function setCustomField(
	token: string,
	key: string,
	value: string,
	overwrite = true,
): Promise<boolean | UpdateResult | Document> {
	const customField = await LivechatCustomField.findOneById(key);
	if (customField) {
		if (customField.scope === 'room') {
			return LivechatRooms.updateDataByToken(token, key, value, overwrite);
		}
		return LivechatVisitors.updateLivechatDataByToken(token, key, value, overwrite);
	}

	return true;
}