fix: use useState for userSavedProxy to make it reactive

blink-so[bot] · kylecarbs · blink-so[bot] · commit ffbeed9d25c0 · 2025-07-17T22:05:07.000Z
- Change userSavedProxy from useMemo to useState to make it reactive to changes
- Update setUserSavedProxy when auto-selecting proxy or manually setting/clearing
- This ensures userProxy in context reflects current localStorage state
- Matches main branch behavior where userSavedProxy was a useState

Co-authored-by: kylecarbs &lt;7122116+kylecarbs@users.noreply.github.com&gt;
diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -93,10 +93,9 @@ export const ProxyContext = createContext<ProxyContextValue | undefined>(
 export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 	const queryClient = useQueryClient();
 
-	// Get user saved proxy from localStorage
-	const userSavedProxy = useMemo(() => {
-		return loadUserSelectedProxy();
-	}, []);
+	// Using a useState so the caller always has the latest user saved
+	// proxy.
+	const [userSavedProxy, setUserSavedProxy] = useState(loadUserSelectedProxy());
 
 	// Load the initial state from localStorage only
 	// Let useEffect handle proper initialization when data is available
@@ -182,6 +181,7 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 
 		if (best?.proxy) {
 			saveUserSelectedProxy(best.proxy);
+			setUserSavedProxy(best.proxy);
 			updateProxy();
 		}
 	}, [latenciesLoaded, proxiesResp, proxyLatencies]);
@@ -203,12 +203,14 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 				setProxy: (proxy: Region) => {
 					// Save to localStorage
 					saveUserSelectedProxy(proxy);
+					setUserSavedProxy(proxy);
 					// Update the selected proxy
 					updateProxy();
 				},
 				clearProxy: () => {
 					// Clear from localStorage
 					clearUserSelectedProxy();
+					setUserSavedProxy(undefined);
 					updateProxy();
 				},
 			}}
