#1859 - First pass at an implementation of the full reset method. Will now start testing...

src/primaite/simulator/file_system/folder.py

@@ -51,6 +51,44 @@ class Folder(FileSystemItemABC):
 
         self.sys_log.info(f"Created file /{self.name} (id: {self.uuid})")
 
+    def set_original_state(self):
+        """Sets the original state."""
+        for file in self.files.values():
+            file.set_original_state()
+        super().set_original_state()
+        vals_to_include = {
+            "scan_duration",
+            "scan_countdown",
+            "red_scan_duration",
+            "red_scan_countdown",
+            "restore_duration",
+            "restore_countdown",
+        }
+        self._original_state.update(self.model_dump(include=vals_to_include))
+        self._original_state["original_file_uuids"] = list(self.files.keys())
+
+    def reset_component_for_episode(self, episode: int):
+        """Reset the original state of the SimComponent."""
+        # Move any 'original' file that have been deleted back to files
+        original_file_uuids = self._original_state.pop("original_file_uuids")
+        for uuid in original_file_uuids:
+            if uuid in self.deleted_files:
+                self.files[uuid] = self.deleted_files.pop(uuid)
+
+        # Clear any other deleted files that aren't original (have been created by agent)
+        self.deleted_files.clear()
+
+        # Now clear all non-original files created by agent
+        current_file_uuids = list(self.files.keys())
+        for uuid in current_file_uuids:
+            if uuid not in original_file_uuids:
+                self.files.pop(uuid)
+
+        # Now reset all remaining files
+        for file in self.files.values():
+            file.reset_component_for_episode(episode)
+        super().reset_component_for_episode(episode)
+
     def _init_request_manager(self) -> RequestManager:
         rm = super()._init_request_manager()
         rm.add_request(