refresh views after assistant changes

apps/nextjs/src/components/assistant/assistant-view-refresh.tsx

@@ -0,0 +1,52 @@
+"use client";
+
+import type { PropsWithChildren } from "react";
+import { createContext, useCallback, useContext, useMemo, useRef, useState } from "react";
+import { useRouter } from "next/navigation";
+import { useQueryClient } from "@tanstack/react-query";
+
+interface AssistantViewRefreshContextValue {
+  isRefreshing: boolean;
+  refreshCurrentView: () => Promise<void>;
+}
+
+const AssistantViewRefreshContext = createContext<AssistantViewRefreshContextValue | null>(null);
+
+export const AssistantViewRefreshProvider = ({ children }: PropsWithChildren) => {
+  const router = useRouter();
+  const queryClient = useQueryClient();
+  const [isRefreshing, setIsRefreshing] = useState(false);
+  const refreshPromiseRef = useRef<Promise<void> | null>(null);
+  const refreshAgainRef = useRef(false);
+
+  const refreshCurrentView = useCallback(() => {
+    if (refreshPromiseRef.current) {
+      refreshAgainRef.current = true;
+      return refreshPromiseRef.current;
+    }
+
+    setIsRefreshing(true);
+    const refreshPromise = (async () => {
+      do {
+        refreshAgainRef.current = false;
+        router.refresh();
+        await queryClient.invalidateQueries({ refetchType: "active" });
+      } while (refreshAgainRef.current);
+    })().finally(() => {
+      refreshPromiseRef.current = null;
+      setIsRefreshing(false);
+    });
+    refreshPromiseRef.current = refreshPromise;
+    return refreshPromise;
+  }, [queryClient, router]);
+
+  const value = useMemo(() => ({ isRefreshing, refreshCurrentView }), [isRefreshing, refreshCurrentView]);
+
+  return <AssistantViewRefreshContext.Provider value={value}>{children}</AssistantViewRefreshContext.Provider>;
+};
+
+export const useAssistantViewRefresh = () => {
+  const value = useContext(AssistantViewRefreshContext);
+  if (!value) throw new Error("useAssistantViewRefresh must be used within AssistantViewRefreshProvider");
+  return value;
+};