From 7d49dacf3df6f8effc591b9d8868f7caaa31ff4f Mon Sep 17 00:00:00 2001
From: Jake Gehrke <gehrkej@oregonstate.edu>
Date: Tue, 9 Apr 2024 13:53:42 -0700
Subject: [PATCH] Restart Drill functionality working

---
 app/segments/drill/[id]/submission/index.js  |  8 +++++++-
 app/segments/drill/[id]/submission/result.js | 16 +++++++++++++++-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/app/segments/drill/[id]/submission/index.js b/app/segments/drill/[id]/submission/index.js
index ba9ad1c1..37b8136d 100644
--- a/app/segments/drill/[id]/submission/index.js
+++ b/app/segments/drill/[id]/submission/index.js
@@ -26,7 +26,13 @@ export default function Index() {
 
   const display = () => {
     if (toggleResult == true) {
-      return <Result submission={outputData} drill={drillInfo} />;
+      return (
+        <Result
+          submission={outputData}
+          drill={drillInfo}
+          setToggleResult={setToggleResult}
+        />
+      );
     } else {
       return (
         <Input
diff --git a/app/segments/drill/[id]/submission/result.js b/app/segments/drill/[id]/submission/result.js
index e74566b2..4057b2c0 100644
--- a/app/segments/drill/[id]/submission/result.js
+++ b/app/segments/drill/[id]/submission/result.js
@@ -1,6 +1,6 @@
 import { useLocalSearchParams, useNavigation } from "expo-router";
 import { StyleSheet } from "react-native";
-import { Appbar } from "react-native-paper";
+import { Appbar, Button } from "react-native-paper";
 import { SafeAreaView } from "react-native-safe-area-context";
 import ResultScreen from "../../../../../components/resultScreen";
 
@@ -8,6 +8,7 @@ function Result(props) {
   const submission = props.submission;
   const navigation = useNavigation();
   const drillId = useLocalSearchParams()["id"];
+  const assignedTime = useLocalSearchParams()["assignedTime"];
 
   return (
     <SafeAreaView style={{ flex: 1 }} edges={["right", "top", "left"]}>
@@ -20,6 +21,19 @@ function Result(props) {
         <Appbar.Content title={props.drill.drillType} />
       </Appbar.Header>
       <ResultScreen drillId={drillId} attemptData={submission} />
+      <Button
+        style={{
+          margin: 10,
+        }}
+        mode="contained"
+        buttonColor="#F24E1E"
+        textColor="white"
+        onPress={() => {
+          props.setToggleResult(false);
+        }}
+      >
+        Restart Drill
+      </Button>
     </SafeAreaView>
   );
 }