Merge pull request #245 from uclaacm/view-only-fix-dl

View-Only: Fixes Redux State Corruption

src/components/ViewOnly.js

@@ -31,11 +31,18 @@ class ViewOnly extends React.Component {
       code: "",
       loaded: false,
       notfound: false,
+      originalCode: "",
     };
+    this.savePrevProgram = (this.props.uid !== "");
   }
 
   componentDidMount = async () => {
+    if (this.savePrevProgram){
+      await this.codeSaverHelper();
+    }
+
     const { ok, sketch } = await fetch.getSketch(this.props.programid);
+
     if (!ok) {
       this.setState({ notfound: true });
       return;
@@ -62,6 +69,27 @@ class ViewOnly extends React.Component {
     }
   }
 
+  componentWillUnmount = () => {
+    if (this.savePrevProgram){
+      this.props.setProgramCode(this.props.mostRecentProgram, this.state.originalCode);
+    }
+  };
+
+  codeSaverHelper = async () => {
+    const { ok: okOriginal, sketch: original } = await fetch.getSketch(
+      this.props.mostRecentProgram,
+    );
+
+    if (!okOriginal) {
+      this.setState({ notfound: true });
+      return;
+    }
+
+    this.setState({
+      originalCode: original.code,
+    });
+  }
+
   onThemeChange = () => {
     let newTheme = this.props.theme === "dark" ? "light" : "dark";
     cookies.setThemeCookie(newTheme);