Adding multiple symbolic ref : Fixed

Author: keerthnn

src/components/ReferenceSuggestionDialog.tsx

@@ -78,6 +78,13 @@ export function ReferenceSuggestionDialog({
     }
   }, [opened, definitionId]);
 
+  useEffect(() => {
+    setIndex((i) => Math.min(i, Math.max(0, suggestions.length - 1)));
+    setSelectedCandidate(null);
+    setError(null);
+    setSearchQuery("");
+  }, [suggestions]);
+
   const current = suggestions[index];
   const sniffyCandidateKeys = useMemo(
     () =>
@@ -128,7 +135,6 @@ export function ReferenceSuggestionDialog({
     setError(null);
     try {
       await onAccept(current, selectedCandidate);
-      goNext();
     } catch (err) {
       const message =
         err instanceof Error ? err.message : "Could not insert reference.";

src/components/StexCuration.tsx

@@ -234,6 +234,42 @@ export function StexCuration({ identity }: { identity: FileIdentity }) {
     setSuggestOpen(true);
   }
 
+  async function reloadSniffySession(definitionId: string) {
+    await queryClient.invalidateQueries({
+      queryKey: ["definitionsByIdentity", identity],
+      refetchType: "none",
+    });
+
+    const updatedData = await queryClient.fetchQuery({
+      queryKey: ["definitionsByIdentity", identity],
+      queryFn: () =>
+        getDefinitionsByIdentity({
+          data: identity,
+        }),
+    });
+    const updatedDef = updatedData.definitions.find(
+      (definition) => definition.id === definitionId,
+    );
+
+    if (!updatedDef) {
+      setActiveDefText("");
+      setActiveDefStatement(null);
+      setSuggestions([]);
+      setSuggestCandidateSymRefs({});
+      return;
+    }
+
+    const session = suggestRefsForDefinition(updatedDef, sniffyCatalog);
+
+    setActiveDefText(extractPlainText(updatedDef.statement));
+    setActiveDefStatement(updatedDef.statement);
+    setSuggestions(session.suggestions);
+    setSuggestCandidateSymRefs({
+      ...buildCandidateSymRefMap(sniffyCatalog, definitionId),
+      ...session.candidateSymRefs,
+    });
+  }
+
   async function handleAcceptSuggestion(
     s: SuggestedReference,
     candidate: SuggestedReferenceCandidate,
@@ -255,9 +291,7 @@ export function StexCuration({ identity }: { identity: FileIdentity }) {
       },
     });
 
-    await queryClient.invalidateQueries({
-      queryKey: ["definitionsByIdentity", identity],
-    });
+    await reloadSniffySession(activeDefId);
   }
 
   async function handleReplaceNode(

src/routes/files/$documentId.tsx

@@ -420,6 +420,39 @@ function RouteComponent() {
     setSuggestOpen(true);
   }
 
+  async function reloadSniffySession(definitionId: string) {
+    await queryClient.invalidateQueries({
+      queryKey: ["definitions", documentId],
+      refetchType: "none",
+    });
+
+    const updatedDefinitions = await queryClient.fetchQuery({
+      queryKey: ["definitions", documentId],
+      queryFn: () => listDefinition({ data: { documentId } }),
+    });
+    const updatedDef = updatedDefinitions.find(
+      (definition) => definition.id === definitionId,
+    );
+
+    if (!updatedDef) {
+      setActiveDefText("");
+      setActiveDefStatement(null);
+      setSuggestions([]);
+      setSuggestCandidateSymRefs({});
+      return;
+    }
+
+    const session = suggestRefsForDefinition(updatedDef, sniffyCatalog);
+
+    setActiveDefText(extractPlainText(updatedDef.statement));
+    setActiveDefStatement(updatedDef.statement);
+    setSuggestions(session.suggestions);
+    setSuggestCandidateSymRefs({
+      ...buildCandidateSymRefMap(sniffyCatalog, definitionId),
+      ...session.candidateSymRefs,
+    });
+  }
+
   async function handleAcceptSuggestion(
     s: SuggestedReference,
     candidate: SuggestedReferenceCandidate,
@@ -441,9 +474,7 @@ function RouteComponent() {
       },
     });
 
-    await queryClient.invalidateQueries({
-      queryKey: ["definitions", documentId],
-    });
+    await reloadSniffySession(activeDefId);
   }
 
   function handleEditDefinitionMeta(item: ExtractedItem) {