fix: persist webview state.

Author: cangzhang

src/codingServer.ts

@@ -72,7 +72,7 @@ export class CodingServer {
         }
 
         Logger.error(`Error reading sessions: ${e}`);
-        await keychain.deleteToken(TokenType.AccessToken);
+        // await keychain.deleteToken(TokenType.AccessToken);
       }
     }
 

webviews/App.tsx

@@ -4,6 +4,7 @@ import styled from 'styled-components';
 import { view } from '@risingstack/react-easy-state';
 import appStore from './store/appStore';
 import { actions } from './store/constants';
+import persistDataHook from './hooks/persistDataHook';
 
 const LoadingWrapper = styled.div`
   font-size: 16px;
@@ -31,12 +32,13 @@ function App() {
           break;
         }
         default:
-          console.log(type, value);
           break;
       }
     });
   }, [updateCurrentMR]);
 
+  persistDataHook();
+
   if (!currentMR.iid) {
     return <LoadingWrapper>Please select an merge request first.</LoadingWrapper>;
   }

webviews/hooks/persistDataHook.tsx

@@ -0,0 +1,19 @@
+import { useEffect, useRef } from 'react';
+import { persistData, removeDataPersist } from '../store/appStore';
+
+const persistDataHook = () => {
+  const effect = useRef(() => {
+  });
+
+  useEffect(() => {
+    effect.current = persistData();
+
+    return () => {
+      removeDataPersist(effect.current);
+    };
+  }, []);
+
+  return effect.current;
+}
+
+export default persistDataHook;

webviews/store/appStore.ts

@@ -1,11 +1,23 @@
-import { store } from '@risingstack/react-easy-state';
+import { autoEffect, clearEffect, store } from '@risingstack/react-easy-state';
 import { IMRWebViewDetail } from '../../src/typings/commonTypes';
 
+export const vscode = acquireVsCodeApi();
+
 const appStore = store({
-  currentMR: {} as IMRWebViewDetail,
+  currentMR: (vscode.getState()?.currentMR || {}) as IMRWebViewDetail,
   updateCurrentMR(data: IMRWebViewDetail) {
     appStore.currentMR = data;
   },
 });
 
+export const persistData = () =>
+  autoEffect(() => {
+    const p = vscode.getState();
+    vscode.setState({
+      ...p,
+      currentMR: appStore.currentMR,
+    });
+  });
+export const removeDataPersist = (e: () => void) => clearEffect(e);
+
 export default appStore;