1 files changed, 21 insertions, 0 deletions
diff --git a/lib/Db/FileOperation.php b/lib/Db/FileOperation.php
index 09bcedc..18d400b 100644
--- a/lib/Db/FileOperation.php
+++ b/lib/Db/FileOperation.php
@@ -22,6 +22,7 @@
 namespace OCA\RansomwareDetection\Db;
 
 use OCP\AppFramework\Db\Entity;
+use OCA\RansomwareDetection\Db\RecoveredFileOperation;
 
 class FileOperation extends Entity
 {
@@ -86,4 +87,24 @@ class FileOperation extends Entity
         $this->addType('fileExtensionClass', 'integer');
         $this->addType('fileClass', 'integer');
     }
+
+    public function toRecoveredFileOperation() {
+        $recoveredFileOperation = new RecoveredFileOperation();
+        $recoveredFileOperation->setUserId($this->getUserId());
+        $recoveredFileOperation->setPath($this->getPath());
+        $recoveredFileOperation->setOriginalName($this->getOriginalName());
+        $recoveredFileOperation->setNewName($this->getNewName());
+        $recoveredFileOperation->setType($this->getType());
+        $recoveredFileOperation->setMimeType($this->getMimeType());
+        $recoveredFileOperation->setSize($this->getSize());
+        $recoveredFileOperation->setTimestamp($this->getTimestamp());
+        $recoveredFileOperation->setCorrupted($this->getCorrupted());
+        $recoveredFileOperation->setCommand($this->getCommand());
+        $recoveredFileOperation->setSequence($this->getSequence());
+        $recoveredFileOperation->setEntropy($this->getEntropy());
+        $recoveredFileOperation->setStandardDeviation($this->getStandardDeviation());
+        $recoveredFileOperation->setFileClass($this->getFileClass());
+        $recoveredFileOperation->setFileExtensionClass($this->getFileExtensionClass());
+        return $recoveredFileOperation;
+    }
 }