reset active defences when changing phase (#168)

backend/src/router.js

@@ -12,6 +12,7 @@ const {
   setOpenAiApiKey,
   setGptModel,
 } = require("./openai");
+const { getInitialDefences } = require("./defence");
 const { initQAModel } = require("./langchain");
 const { retrievalQAPrePrompt } = require("./promptTemplates");
 const router = express.Router();
@@ -89,6 +90,13 @@ router.get("/defence/status", (req, res, next) => {
   res.send(req.session.defences);
 });
 
+// reset the active defences
+router.post("/defence/reset", (req, res, next) => {
+  req.session.defences = getInitialDefences();
+  console.debug("Defences reset");
+  res.send("Defences reset");
+});
+
 // Get sent emails
 router.get("/email/get", (req, res, next) => {
   res.send(req.session.sentEmails);

frontend/src/App.tsx

@@ -14,6 +14,7 @@ import { DefenceInfo } from "./models/defence";
 import { getCompletedPhases } from "./service/phaseService";
 import { clearEmails } from "./service/emailService";
 import { clearChat } from "./service/chatService";
+import { resetActiveDefences } from "./service/defenceService";
 import { PHASES } from "./Phases";
 import { ATTACKS_ALL, ATTACKS_PHASE_1 } from "./Attacks";
 import { DEFENCE_DETAILS_ALL, DEFENCE_DETAILS_PHASE } from "./Defences";
@@ -87,6 +88,8 @@ function App() {
     setMessages([]);
     setCurrentPhase(newPhase);
 
+    resetActiveDefences();
+
     // add the preamble to the chat
     const preambleMessage = PHASES[newPhase].preamble;
     addPhasePreambleMessage(preambleMessage.toLowerCase());

frontend/src/service/defenceService.ts

@@ -71,4 +71,15 @@ function validateDefence(id: string, configName: string, config: string) {
   }
 }
 
-export { getDefences, activateDefence, deactivateDefence, configureDefence };
+async function resetActiveDefences() {
+  const response = await sendRequest(PATH + "reset", "POST");
+  return response.status === 200;
+}
+
+export {
+  getDefences,
+  activateDefence,
+  deactivateDefence,
+  configureDefence,
+  resetActiveDefences,
+};