Fix initial state bug.

agrawal-d · Jun 16, 2024 · ca322db · ca322db
1 parent 0b7e447
commit ca322db
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/webview/frontend/App.tsx b/src/webview/frontend/App.tsx
@@ -40,10 +40,12 @@ function Judge(props: {
     const [webviewState, setWebviewState] = useState<WebViewpersistenceState>(
         () => {
             const vscodeState = vscodeApi.getState();
-            console.log('Restored state:', vscodeState);
-            return {
+            let ret = {
                 dialogCloseDate: vscodeState?.dialogCloseDate || Date.now(),
             };
+            vscodeApi.setState(ret);
+            console.log('Restored to state:', ret);
+            return ret;
         },
     );