From 562d4ee80042a3d77a1506c5e3f3bbc196a6bf30 Mon Sep 17 00:00:00 2001
From: TheMas3212 <mas3212@gmail.com>
Date: Tue, 11 Jan 2022 17:50:41 +1100
Subject: [PATCH] Add ErrorBoundary component to catch rendering error and
 redirect to recovery page also add softreset button on recovery page

---
 src/ui/ErrorBoundary.tsx      | 29 ++++++++++++++
 src/ui/GameRoot.tsx           | 73 ++++++++++++++++++++---------------
 src/ui/React/RecoveryRoot.tsx |  4 +-
 3 files changed, 74 insertions(+), 32 deletions(-)
 create mode 100644 src/ui/ErrorBoundary.tsx

diff --git a/src/ui/ErrorBoundary.tsx b/src/ui/ErrorBoundary.tsx
new file mode 100644
index 000000000..ab555124e
--- /dev/null
+++ b/src/ui/ErrorBoundary.tsx
@@ -0,0 +1,29 @@
+import React, { ErrorInfo } from "react";
+import { RecoveryRoot } from "./React/RecoveryRoot";
+import { IRouter } from "./Router";
+
+interface IProps {
+  router: IRouter;
+  softReset: () => void;
+}
+
+
+export class ErrorBoundary extends React.Component<IProps> {
+  state: { hasError: boolean }
+  constructor(props: IProps) {
+    super(props);
+    this.state = { hasError: false };
+  }
+  componentDidCatch(error: Error, errorInfo: ErrorInfo): void {
+    console.error(error, errorInfo);
+  }
+  render(): React.ReactNode {
+    if (this.state.hasError) {
+      return <RecoveryRoot router={this.props.router} softReset={this.props.softReset} />;
+    }
+    return this.props.children;
+  }
+  static getDerivedStateFromError(): { hasError: true} {
+    return { hasError: true };
+  }
+}
\ No newline at end of file
diff --git a/src/ui/GameRoot.tsx b/src/ui/GameRoot.tsx
index dd5633e3c..177d6fcf5 100644
--- a/src/ui/GameRoot.tsx
+++ b/src/ui/GameRoot.tsx
@@ -77,6 +77,7 @@ import { enterBitNode } from "../RedPill";
 import { Context } from "./Context";
 import { RecoveryMode, RecoveryRoot } from "./React/RecoveryRoot";
 import { AchievementsRoot } from "../Achievements/AchievementsRoot";
+import { ErrorBoundary } from "./ErrorBoundary";
 
 const htmlLocation = location;
 
@@ -234,6 +235,11 @@ export function GameRoot({ player, engine, terminal }: IProps): React.ReactEleme
     throw new Error("Trying to go to a page without the proper setup");
 
   const [cinematicText, setCinematicText] = useState("");
+  const [errorBoundaryKey, setErrorBoundaryKey] = useState<number>(0);
+
+  function resetErrorBoundary(): void {
+    setErrorBoundaryKey(errorBoundaryKey+1);
+  }
 
   function rerender(): void {
     setRerender((old) => old + 1);
@@ -382,12 +388,19 @@ export function GameRoot({ player, engine, terminal }: IProps): React.ReactEleme
     if (page !== Page.Terminal) window.scrollTo(0, 0);
   });
 
+  function softReset(): void {
+    dialogBoxCreate("Soft Reset!");
+    prestigeAugmentation();
+    resetErrorBoundary();
+    Router.toTerminal();
+  }
+
   let mainPage = <Typography>Cannot load</Typography>;
   let withSidebar = true;
   let withPopups = true;
   switch (page) {
     case Page.Recovery: {
-      mainPage = <RecoveryRoot router={Router} />;
+      mainPage = <RecoveryRoot router={Router} softReset={softReset} />;
       withSidebar = false;
       withPopups = false;
       break;
@@ -540,11 +553,7 @@ export function GameRoot({ player, engine, terminal }: IProps): React.ReactEleme
             saveObject.exportGame();
           }}
           forceKill={killAllScripts}
-          softReset={() => {
-            dialogBoxCreate("Soft Reset!");
-            prestigeAugmentation();
-            Router.toTerminal();
-          }}
+          softReset={softReset}
         />
       );
       break;
@@ -574,37 +583,39 @@ export function GameRoot({ player, engine, terminal }: IProps): React.ReactEleme
   return (
     <Context.Player.Provider value={player}>
       <Context.Router.Provider value={Router}>
-        <SnackbarProvider>
-          <Overview mode={ITutorial.isRunning ? "tutorial" : "overview"}>
-            {!ITutorial.isRunning ? (
-              <CharacterOverview
+        <ErrorBoundary key={errorBoundaryKey} router={Router} softReset={softReset}>
+          <SnackbarProvider>
+           <Overview mode={ITutorial.isRunning ? "tutorial" : "overview"}>
+              {!ITutorial.isRunning ? (
+                <CharacterOverview
                 save={() => saveObject.saveGame()}
                 killScripts={killAllScripts}
                 router={Router}
                 allowBackButton={withSidebar} />
+              ) : (
+                <InteractiveTutorialRoot />
+              )}
+            </Overview>
+            {withSidebar ? (
+              <Box display="flex" flexDirection="row" width="100%">
+                <SidebarRoot player={player} router={Router} page={page} />
+                <Box className={classes.root}>{mainPage}</Box>
+              </Box>
             ) : (
-              <InteractiveTutorialRoot />
-            )}
-          </Overview>
-          {withSidebar ? (
-            <Box display="flex" flexDirection="row" width="100%">
-              <SidebarRoot player={player} router={Router} page={page} />
               <Box className={classes.root}>{mainPage}</Box>
-            </Box>
-          ) : (
-            <Box className={classes.root}>{mainPage}</Box>
-          )}
-          <Unclickable />
-          {withPopups && (
-            <>
-              <LogBoxManager />
-              <AlertManager />
-              <PromptManager />
-              <InvitationModal />
-              <Snackbar />
-            </>
-          )}
-        </SnackbarProvider>
+            )}
+            <Unclickable />
+            {withPopups && (
+              <>
+                <LogBoxManager />
+                <AlertManager />
+                <PromptManager />
+                <InvitationModal />
+                <Snackbar />
+              </>
+            )}
+          </SnackbarProvider>
+        </ErrorBoundary>
       </Context.Router.Provider>
     </Context.Player.Provider>
   );
diff --git a/src/ui/React/RecoveryRoot.tsx b/src/ui/React/RecoveryRoot.tsx
index 267e265e3..5c0e2e42c 100644
--- a/src/ui/React/RecoveryRoot.tsx
+++ b/src/ui/React/RecoveryRoot.tsx
@@ -15,9 +15,10 @@ export function ActivateRecoveryMode(): void {
 
 interface IProps {
   router: IRouter;
+  softReset: () => void;
 }
 
-export function RecoveryRoot({ router }: IProps): React.ReactElement {
+export function RecoveryRoot({ router, softReset }: IProps): React.ReactElement {
   function recover(): void {
     RecoveryMode = false;
     router.toTerminal();
@@ -48,6 +49,7 @@ export function RecoveryRoot({ router }: IProps): React.ReactElement {
       <br />
       <Typography>You can disable recovery mode now. But chances are the game will not work correctly.</Typography>
       <Button onClick={recover}>DISABLE RECOVERY MODE</Button>
+      <Button onClick={softReset}>PERFORM SOFT RESET</Button>
     </>
   );
 }