Store the output of the script in the database

Author: vmcj

webapp/src/Controller/API/JudgehostController.php

@@ -8,6 +8,8 @@
 use App\Entity\DebugPackage;
 use App\Entity\Executable;
 use App\Entity\ExecutableFile;
+use App\Entity\GenericTask;
+use App\Entity\GenericTaskOutput;
 use App\Entity\InternalError;
 use App\Entity\Judgehost;
 use App\Entity\JudgeTask;
@@ -553,6 +555,63 @@ public function addDebugInfo(
         $this->em->flush();
     }
 
+    /**
+     * Add generic task output.
+     */
+    #[IsGranted('ROLE_JUDGEHOST')]
+    #[Rest\Post('/add-generic-task/{hostname}/{judgeTaskId<\d+>}')]
+    #[OA\Response(response: 200, description: 'When the task output has been added')]
+    public function addGenericTaskOutput(
+        Request $request,
+        #[OA\PathParameter(description: 'The hostname of the judgehost that wants to add the task output')]
+        string $hostname,
+        #[OA\PathParameter(description: 'The ID of the judgetask to add', schema: new OA\Schema(type: 'integer'))]
+        int $judgeTaskId
+    ): void {
+        $judgeTask = $this->em->getRepository(JudgeTask::class)->find($judgeTaskId);
+        if ($judgeTask === null) {
+            throw new BadRequestHttpException(
+                'Inconsistent data, no judgetask known with judgetaskid = ' . $judgeTaskId . '.');
+        }
+
+        $required = ['generic_task'];
+        foreach ($required as $argument) {
+            if (!$request->request->has($argument)) {
+                throw new BadRequestHttpException(
+                    sprintf("Argument '%s' is mandatory", $argument));
+            }
+        }
+
+        $judgehost = $this->em->getRepository(Judgehost::class)->findOneBy(['hostname' => $hostname]);
+        if (!$judgehost) {
+            throw new BadRequestHttpException("Who are you and why are you sending us any data?");
+        }
+
+        $genericTask = $judgeTask->getGenericTasks();
+        if (count($genericTask) === 0) {
+            $genericTask = new GenericTask();
+            $genericTask
+                ->setJudgeTask($judgeTask)
+                ->setStarttime($judgeTask->getStarttime())
+                ->setEndtime(Utils::now());
+            $genericTaskOutput = new GenericTaskOutput();
+            $genericTaskOutput->setGenericTask($genericTask);
+            $genericTask->setOutput($genericTaskOutput);
+            $this->em->persist($genericTask);
+            $this->em->persist($genericTaskOutput);
+        } elseif (count($genercTasks) !== 1) {
+            throw new BadRequestHttpException("There should be only one generic task for this judgetask.");
+        } else {
+            $genericTask = $genericTask->first();
+        }
+
+        $genericTask->setEndtime(Utils::now());
+        $outputTask = base64_decode($request->request->get('generic_task'));
+        $genericTaskOutput = $genericTask->getOutput();
+        $genericTaskOutput->setOutputTask($outputTask);
+        $this->em->flush();
+    }
+
     /**
      * Add one JudgingRun. When relevant, finalize the judging.
      * @throws DBALException
@@ -1238,7 +1297,7 @@ public function getFilesAction(
         return match ($type) {
             'source' => $this->getSourceFiles($id),
             'testcase' => $this->getTestcaseFiles($id),
-            'compare', 'compile', 'debug', 'run' => $this->getExecutableFiles($id),
+            'compare', 'compile', 'debug', 'run', 'generic_task' => $this->getExecutableFiles($id),
             default => throw new BadRequestHttpException('Unknown type requested.'),
         };
     }

webapp/src/Entity/GenericTask.php

@@ -155,7 +155,7 @@ public function setOutput(GenericTaskOutput $output): GenericTask
     {
         $this->output->clear();
         $this->output->add($output);
-        $output->setRun($this);
+        $output->setGenericTask($this);
 
         return $this;
     }

webapp/src/Entity/JudgeTask.php

@@ -161,6 +161,13 @@ public function getSubmitid(): ?int
     #[Serializer\Exclude]
     private Collection $judging_runs;
 
+    /**
+     * @var Collection<int, GenericTask>
+     */
+    #[ORM\OneToMany(mappedBy: 'judgetask', targetEntity: GenericTask::class)]
+    #[Serializer\Exclude]
+    private Collection $generic_tasks;
+
     #[ORM\ManyToOne(inversedBy: 'judgeTasks')]
     #[ORM\JoinColumn(name: 'versionid', referencedColumnName: 'versionid', onDelete: 'SET NULL')]
     #[Serializer\Exclude]