update

Author: raj-subhankar

ui/desktop/src/components/ProviderGuard.tsx

@@ -40,6 +40,7 @@ export default function ProviderGuard({ didSelectProvider, children }: ProviderG
   const [isTetrateSetupInProgress, setIsTetrateSetupInProgress] = useState(false);
   const onboardingTracked = useRef(false);
   const tetrateSetupRunId = useRef(0);
+  const tetrateSuccessTimerId = useRef<ReturnType<typeof setTimeout> | null>(null);
   const scrollContainerRef = useRef<HTMLDivElement>(null);
   const [showScrollIndicator, setShowScrollIndicator] = useState(true);
 
@@ -69,6 +70,7 @@ export default function ProviderGuard({ didSelectProvider, children }: ProviderG
     title: string;
     message: string;
     showProgress?: boolean;
+    showSuccess?: boolean;
     showRetry: boolean;
     closeLabel?: string;
     autoClose?: number;
@@ -107,9 +109,20 @@ export default function ProviderGuard({ didSelectProvider, children }: ProviderG
       setIsTetrateSetupInProgress(false);
 
       if (result.success) {
-        setTetrateSetupState(null);
-        setSwitchModelProvider('tetrate');
-        setShowSwitchModelModal(true);
+        setTetrateSetupState({
+          show: true,
+          title: 'Setup Complete',
+          message: 'Your account has been connected successfully.',
+          showSuccess: true,
+          showProgress: false,
+          showRetry: false,
+        });
+        tetrateSuccessTimerId.current = setTimeout(() => {
+          tetrateSuccessTimerId.current = null;
+          setTetrateSetupState(null);
+          setSwitchModelProvider('tetrate');
+          setShowSwitchModelModal(true);
+        }, 1500);
       } else {
         trackOnboardingSetupFailed('tetrate', result.message);
         setTetrateSetupState({
@@ -165,6 +178,16 @@ export default function ProviderGuard({ didSelectProvider, children }: ProviderG
   };
 
   const handleCancelTetrateSetup = async () => {
+    if (tetrateSuccessTimerId.current) {
+      // User dismissed the success modal early — still advance to model selection
+      clearTimeout(tetrateSuccessTimerId.current);
+      tetrateSuccessTimerId.current = null;
+      setTetrateSetupState(null);
+      setSwitchModelProvider('tetrate');
+      setShowSwitchModelModal(true);
+      return;
+    }
+
     if (!isTetrateSetupInProgress) {
       setTetrateSetupState(null);
       return;
@@ -538,6 +561,7 @@ export default function ProviderGuard({ didSelectProvider, children }: ProviderG
             title={tetrateSetupState.title}
             message={tetrateSetupState.message}
             showProgress={tetrateSetupState.showProgress}
+            showSuccess={tetrateSetupState.showSuccess}
             showRetry={tetrateSetupState.showRetry}
             onRetry={() => handleRetrySetup('tetrate')}
             onClose={() => closeSetupModal('tetrate')}

ui/desktop/src/components/SetupModal.tsx

@@ -5,6 +5,7 @@ interface SetupModalProps {
   title: string;
   message: string;
   showProgress?: boolean;
+  showSuccess?: boolean;
   showRetry?: boolean;
   onRetry?: () => void;
   autoClose?: number;
@@ -16,6 +17,7 @@ export function SetupModal({
   title,
   message,
   showProgress,
+  showSuccess,
   showRetry,
   onRetry,
   autoClose,
@@ -44,6 +46,15 @@ export function SetupModal({
           </div>
         )}
 
+        {showSuccess && (
+          <div className="flex justify-center mb-4">
+            <svg className="h-10 w-10 text-green-500" fill="none" viewBox="0 0 24 24" stroke="currentColor" strokeWidth={2}>
+              <circle cx="12" cy="12" r="10" className="stroke-green-500" fill="none" />
+              <path strokeLinecap="round" strokeLinejoin="round" d="M8 12l2.5 2.5L16 9" />
+            </svg>
+          </div>
+        )}
+
         {onClose && (
           <div className="mb-4">
             <Button onClick={onClose} className="w-full">

ui/desktop/src/main.ts

@@ -194,6 +194,19 @@ if (process.platform !== 'darwin') {
   }
 }
 
+function focusGooseWindow(): void {
+  const windows = BrowserWindow.getAllWindows();
+  if (windows.length > 0) {
+    const win = windows[0];
+    if (win.isMinimized()) win.restore();
+    win.show();
+    win.focus();
+  }
+  if (process.platform === 'darwin') {
+    app.dock?.bounce('informational');
+  }
+}
+
 let firstOpenWindow: BrowserWindow;
 let pendingDeepLink: string | null = null;
 let openUrlHandledLaunch = false;
@@ -206,6 +219,7 @@ async function handleProtocolUrl(url: string) {
       pendingDeepLink = null;
     })
   ) {
+    focusGooseWindow();
     return;
   }
 
@@ -286,6 +300,7 @@ app.on('open-url', async (_event, url) => {
         pendingDeepLink = null;
       })
     ) {
+      focusGooseWindow();
       return;
     }