From 6ea7d8f6cf985b4811d7f27d3b92dad9db3da350 Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Sat, 3 Feb 2024 16:45:34 +0700
Subject: [PATCH] fix: migration loading indicator (#1913)

---
 web/containers/Providers/index.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/web/containers/Providers/index.tsx b/web/containers/Providers/index.tsx
index 895c22177..c8a20bca7 100644
--- a/web/containers/Providers/index.tsx
+++ b/web/containers/Providers/index.tsx
@@ -21,6 +21,8 @@ import {
 
 import Umami from '@/utils/umami'
 
+import Loader from '../Loader'
+
 import KeyListener from './KeyListener'
 
 import { extensionManager } from '@/extension'
@@ -30,6 +32,7 @@ const Providers = (props: PropsWithChildren) => {
 
   const [setupCore, setSetupCore] = useState(false)
   const [activated, setActivated] = useState(false)
+  const [settingUp, setSettingUp] = useState(false)
 
   async function setupExtensions() {
     // Register all active extensions
@@ -37,11 +40,13 @@ const Providers = (props: PropsWithChildren) => {
 
     setTimeout(async () => {
       if (!isCoreExtensionInstalled()) {
-        setupBaseExtensions()
+        setSettingUp(true)
+        await setupBaseExtensions()
         return
       }
 
       extensionManager.load()
+      setSettingUp(false)
       setActivated(true)
     }, 500)
   }
@@ -71,6 +76,7 @@ const Providers = (props: PropsWithChildren) => {
     <JotaiWrapper>
       <ThemeWrapper>
         <Umami />
+        {settingUp && <Loader description="Preparing Update..." />}
         {setupCore && activated && (
           <KeyListener>
             <FeatureToggleWrapper>