Restore console code after page reloads (#75)

* restore console code after page reload, button to reset console to initial state * fix
qdrant · Jul 7, 2023 · 944da28 · 944da28
1 parent cb951c9
commit 944da28
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/src/components/CodeEditorWindow/Menu/SpeedDialMenu.jsx b/src/components/CodeEditorWindow/Menu/SpeedDialMenu.jsx
@@ -1,13 +1,13 @@
 import React from "react";
 import PropTypes from "prop-types";
-import Box from "@mui/material/Box";
 import SpeedDial from "@mui/material/SpeedDial";
 import Bolt from "@mui/icons-material/Bolt";
 import SpeedDialAction from "@mui/material/SpeedDialAction";
 import SaveIcon from "@mui/icons-material/Save";
 import HistoryRounded from "@mui/icons-material/HistoryRounded";
+import RestartAlt from "@mui/icons-material/RestartAlt";
 
-function SpeedDialMenu({ openSavedCode, openHistory }) {
+function SpeedDialMenu({ openSavedCode, openHistory, resetConsole }) {
   const [open, setOpen] = React.useState(false);
   const handleOpen = () => setOpen(true);
   const handleClose = () => setOpen(false);
@@ -36,6 +36,13 @@ function SpeedDialMenu({ openSavedCode, openHistory }) {
         tooltipOpen
         onClick={openHistory}
       />
+      <SpeedDialAction
+        key={"Reset"}
+        icon={<RestartAlt/>}
+        tooltipTitle={"Reset"}
+        tooltipOpen
+        onClick={resetConsole}
+      />
     </SpeedDial>
   );
 }
@@ -44,6 +51,7 @@ function SpeedDialMenu({ openSavedCode, openHistory }) {
 SpeedDialMenu.propTypes = {
   openSavedCode: PropTypes.func,
   openHistory: PropTypes.func,
+  resetConsole: PropTypes.func,
 };
 
 export default SpeedDialMenu;
diff --git a/src/pages/Console.jsx b/src/pages/Console.jsx
@@ -5,7 +5,6 @@ import Box from "@mui/material/Box";
 import { Typography, Grid } from "@mui/material";
 import ErrorNotifier from "../components/ToastNotifications/ErrorNotifier";
 import { Panel, PanelGroup, PanelResizeHandle } from "react-resizable-panels";
-import Menu from "../components/CodeEditorWindow/Menu";
 import CodeEditorWindow from "../components/CodeEditorWindow";
 import ResultEditorWindow from "../components/ResultEditorWindow";
 import SpeedDialMenu from "../components/CodeEditorWindow/Menu/SpeedDialMenu";
@@ -43,7 +42,7 @@ const defaultResult = `{}`;
 
 function Console() {
   const theme = useTheme();
-  const [code, setCode] = useState(query);
+  const [code, setCode] = useState( localStorage.getItem('qwuiConsoleCode') ?? query);
   const [result, setResult] = useState(defaultResult);
   const [errorMessage, setErrorMessage] = useState(null);
 
@@ -53,6 +52,7 @@ function Console() {
   const onChangeCode = (action, data) => {
     switch (action) {
       case "code": {
+        localStorage.setItem("qwuiConsoleCode", data);
         setCode(data);
         break;
       }
@@ -114,7 +114,9 @@ function Console() {
           </Grid>
           <SpeedDialMenu
             openHistory={() => setOpenHistory(true)}
-            openSavedCode={() => setOpenSavedCode(true)}/>
+            openSavedCode={() => setOpenSavedCode(true)}
+            resetConsole={() => onChangeCode("code", query)}
+          />
           <History
             code={code}
             handleEditorChange={onChangeCode}