🐛 fix: String Interpolation in Messages Endpoint from #9155 (#9312)

* feat: move buildTree function for message hierarchy to data provider

* refactor: consolidate buildTree import from utils to data provider

* fix: correct string interpolation in messages function, which caused message search requests to fail
This commit is contained in:
Danny Avila 2025-08-27 13:48:48 -04:00 committed by GitHub
parent 78d735f35c
commit a362963017
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 63 additions and 94 deletions

View file

@ -0,0 +1,50 @@
import type { TFile } from './types/files';
import type { TMessage } from './types';
export type ParentMessage = TMessage & { children: TMessage[]; depth: number };
export function buildTree({
messages,
fileMap,
}: {
messages: (TMessage | undefined)[] | null;
fileMap?: Record<string, TFile>;
}) {
if (messages === null) {
return null;
}
const messageMap: Record<string, ParentMessage> = {};
const rootMessages: TMessage[] = [];
const childrenCount: Record<string, number> = {};
messages.forEach((message) => {
if (!message) {
return;
}
const parentId = message.parentMessageId ?? '';
childrenCount[parentId] = (childrenCount[parentId] || 0) + 1;
const extendedMessage: ParentMessage = {
...message,
children: [],
depth: 0,
siblingIndex: childrenCount[parentId] - 1,
};
if (message.files && fileMap) {
extendedMessage.files = message.files.map((file) => fileMap[file.file_id ?? ''] ?? file);
}
messageMap[message.messageId] = extendedMessage;
const parentMessage = messageMap[parentId];
if (parentMessage) {
parentMessage.children.push(extendedMessage);
extendedMessage.depth = parentMessage.depth + 1;
} else {
rootMessages.push(extendedMessage);
}
});
return rootMessages;
}