ENG-1279 Discourse Context Overlay doesn't update number (#693)

* eng-1279 refreshing the discourse context should also refresh the overlay

Author: maparent

apps/roam/src/components/DiscourseContext.tsx

@@ -31,6 +31,7 @@ export type DiscourseContextResults = Awaited<
 type Props = {
   uid: string;
   results?: DiscourseContextResults;
+  overlayRefresh?: () => void;
 };
 
 const ExtraColumnRow = (r: Result) => {
@@ -279,7 +280,7 @@ const ContextTab = ({
             <CreateRelationButton
               sourceNodeUid={parentUid}
               onClose={() => {
-                window.setTimeout(onRefresh, 250);
+                window.setTimeout(onRefresh, 450, true);
               }}
             />
             <Switch
@@ -316,7 +317,7 @@ const ContextTab = ({
   );
 };
 
-export const ContextContent = ({ uid, results }: Props) => {
+export const ContextContent = ({ uid, results, overlayRefresh }: Props) => {
   const [rawQueryResults, setRawQueryResults] = useState<
     Record<string, DiscourseContextResults[number]>
   >({});
@@ -350,13 +351,16 @@ export const ContextContent = ({ uid, results }: Props) => {
         uid,
         onResult: addLabels,
         ignoreCache,
-      }).finally(() => setLoading(false));
+      }).finally(() => {
+        if (overlayRefresh) overlayRefresh();
+        setLoading(false);
+      });
     },
-    [uid, setRawQueryResults, setLoading, addLabels],
+    [uid, setRawQueryResults, setLoading, addLabels, overlayRefresh],
   );
 
   const delayedRefresh = () => {
-    window.setTimeout(onRefresh, 250);
+    window.setTimeout(onRefresh, 450, true);
   };
 
   useEffect(() => {

apps/roam/src/components/DiscourseContextOverlay.tsx

@@ -148,7 +148,11 @@ const DiscourseContextOverlay = ({
             results.length === 0 ? "flex items-center justify-center" : ""
           }`}
         >
-          <ContextContent uid={tagUid} results={results} />
+          <ContextContent
+            uid={tagUid}
+            results={results}
+            overlayRefresh={refresh}
+          />
         </div>
       }
       target={

apps/roam/src/components/results-view/ResultsTable.tsx

@@ -350,7 +350,7 @@ const ResultRow = ({
               content: "Relation deleted",
               intent: "success",
             });
-            onRefresh();
+            onRefresh(true);
           })
           .catch((e) => {
             // this one should be an internalError