codedlabdev
diff --git a/‎app/utils/toolMentionParser.ts‎
Lines changed: 25 additions & 0 deletions b/‎app/utils/toolMentionParser.ts‎
Lines changed: 25 additions & 0 deletions
@@ -4,6 +4,8 @@ export interface AutocompleteState {
   atPosition: number;
 }
 
+export type ReferenceType = 'file' | 'tool';
+
 export function shouldShowAutocomplete(text: string, cursorPos: number): AutocompleteState {
   const textBeforeCursor = text.slice(0, cursorPos);
   const lastAtIndex = textBeforeCursor.lastIndexOf('@');
@@ -34,6 +36,10 @@ export function extractSearchQuery(text: string, cursorPos: number): string {
   return state.searchQuery;
 }
 
+export function detectReferenceType(searchQuery: string): ReferenceType {
+  return /[\/\.]/.test(searchQuery) ? 'file' : 'tool';
+}
+
 function getTextWidth(text: string, font: string): number {
   const canvas = document.createElement('canvas');
   const context = canvas.getContext('2d');
@@ -94,3 +100,22 @@ export function insertToolMention(
 
   return { newText, newCursorPos };
 }
+
+export function insertFileReference(
+  text: string,
+  cursorPos: number,
+  filePath: string,
+): { newText: string; newCursorPos: number } {
+  const state = shouldShowAutocomplete(text, cursorPos);
+
+  if (!state.isOpen) {
+    return { newText: text, newCursorPos: cursorPos };
+  }
+
+  const beforeAt = text.slice(0, state.atPosition);
+  const afterCursor = text.slice(cursorPos);
+  const newText = `${beforeAt}@${filePath} ${afterCursor}`;
+  const newCursorPos = state.atPosition + filePath.length + 2;
+
+  return { newText, newCursorPos };
+}