snapshot of https://github.com/RocketChat/Rocket.Chat.git (develop, 2026-09-16; blobs > 131072B omitted)

apps/meteor/client/navbar/NavBarSearch/hooks/useAISearchRooms.ts

@@ -0,0 +1,115 @@
+import { useDebouncedValue } from '@rocket.chat/fuselage-hooks';
+import { escapeRegExp, isTruthy } from '@rocket.chat/tools';
+import type { SubscriptionWithRoom } from '@rocket.chat/ui-contexts';
+import { useEndpoint, useUserSubscriptions } from '@rocket.chat/ui-contexts';
+import { useQuery } from '@tanstack/react-query';
+import { useMemo } from 'react';
+
+import { getConfig } from '../../../lib/utils/getConfig';
+
+const LIMIT = parseInt(String(getConfig('Sidebar_Search_Spotlight_LIMIT', 20)));
+
+const options = {
+	sort: {
+		lm: -1,
+		name: 1,
+	},
+	limit: LIMIT,
+} as const;
+
+type AISearchRoomResult = {
+	_id: string;
+	name: string;
+	fname?: string;
+	t: string;
+	rid?: string;
+	status?: string;
+	avatarETag?: string;
+	outside?: boolean;
+	uids?: string[];
+};
+
+export const useAISearchRooms = (filterText = ''): { items: SubscriptionWithRoom[]; isLoading: boolean } => {
+	const [, mention, name] = useMemo(() => filterText.match(/(@|#)?(.*)/i) || [], [filterText]);
+
+	const query = useMemo(() => {
+		const filterRegex = new RegExp(escapeRegExp(name), 'i');
+
+		return {
+			...(mention && {
+				t: mention === '@' ? 'd' : { $ne: 'd' },
+			}),
+			$or: [{ name: filterRegex }, { fname: filterRegex }],
+		};
+	}, [name, mention]);
+
+	const localRooms = useUserSubscriptions(query, options);
+	const debouncedName = useDebouncedValue(name, 500);
+	const usernamesFromClient = localRooms.map(({ t, name }) => (t === 'd' ? name : null)).filter(isTruthy);
+	const searchForChannels = mention === '#';
+	const searchForDMs = mention === '@';
+	const type = useMemo(() => {
+		if (searchForChannels) {
+			return { users: false, rooms: true, includeFederatedRooms: true };
+		}
+
+		if (searchForDMs) {
+			return { users: true, rooms: false };
+		}
+
+		return { users: true, rooms: true, includeFederatedRooms: true };
+	}, [searchForChannels, searchForDMs]);
+
+	const getSpotlight = useEndpoint('GET', '/v1/spotlight');
+	const { data: serverResults, isFetching } = useQuery({
+		queryKey: ['sidebar/ai-search/spotlight', debouncedName, mention, type],
+		enabled: localRooms.length < LIMIT,
+		queryFn: async () => {
+			const spotlight = await getSpotlight({
+				query: debouncedName,
+				usernames: usernamesFromClient.join(','),
+				type: JSON.stringify(type),
+			});
+
+			const filterUsersUnique = ({ _id }: { _id: string }, index: number, users: { _id: string }[]): boolean =>
+				index === users.findIndex((user) => _id === user._id);
+			const userMap = (user: { _id: string; name: string; username: string; status?: string }): AISearchRoomResult => ({
+				_id: user._id,
+				name: user.username,
+				fname: user.name,
+				t: 'd',
+				rid: user._id,
+				status: user.status,
+			});
+			const results: AISearchRoomResult[] = spotlight.users.filter(filterUsersUnique).map(userMap);
+			results.push(...spotlight.rooms);
+
+			return results;
+		},
+		staleTime: 60_000,
+	});
+
+	const items = useMemo(() => {
+		const filterRegex = new RegExp(escapeRegExp(name), 'i');
+		const matchesFilter = ({ name, fname }: { name?: string; fname?: string }) =>
+			(name && filterRegex.test(name)) || (fname && filterRegex.test(fname));
+		const isLocalDuplicate = (item: { _id: string; t?: string; uids?: string[]; name: string }): boolean =>
+			localRooms.some((room) => {
+				const sameRoom = [room.rid, room._id].includes(item._id);
+				const sameGroupDM = item.t === 'd' && !!item.uids && item.uids.length > 1 && item.uids.includes(room._id);
+				const sameDirectDM = item.t === 'd' && room.t === 'd' && !!room.uids && room.uids.length === 2 && room.uids.includes(item._id);
+				const sameUserDM = item.t === 'd' && room.t === 'd' && item.name === room.name;
+				return sameRoom || sameGroupDM || sameDirectDM || sameUserDM;
+			});
+
+		const candidates = name === debouncedName && localRooms.length < LIMIT ? (serverResults ?? []) : [];
+		const fromServer = candidates.filter((item) => matchesFilter(item) && !isLocalDuplicate(item));
+		const exact = fromServer.filter((item) => [item.name, item.fname].includes(name));
+
+		return [...exact, ...localRooms, ...fromServer.filter((item) => !exact.includes(item))] as SubscriptionWithRoom[];
+	}, [debouncedName, localRooms, name, serverResults]);
+
+	const isLoading = isFetching && serverResults === undefined;
+
+	return { items, isLoading };
+};