ogcore/src/Controller/OgAgent/Webhook/StatusController.php

<?php

namespace App\Controller\OgAgent\Webhook;

use App\Controller\OgRepository\Image\CreateAuxFilesAction;
use App\Entity\Client;
use App\Entity\Image;
use App\Entity\ImageImageRepository;
use App\Entity\OperativeSystem;
use App\Entity\Partition;
use App\Entity\Software;
use App\Entity\SoftwareProfile;
use App\Entity\Trace;
use App\Model\ClientStatus;
use App\Model\CommandTypes;
use App\Model\ImageStatus;
use App\Model\SoftwareTypes;
use App\Model\TraceStatus;
use App\Service\CreatePartitionService;
use App\Service\Trace\CreateService;
use Doctrine\ORM\EntityManagerInterface;
use Psr\Log\LoggerInterface;
use Symfony\Bundle\FrameworkBundle\Controller\AbstractController;
use Symfony\Component\HttpFoundation\JsonResponse;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpKernel\Attribute\AsController;
use Symfony\Component\Routing\Annotation\Route;
use Symfony\Contracts\HttpClient\Exception\ClientExceptionInterface;
use Symfony\Contracts\HttpClient\Exception\RedirectionExceptionInterface;
use Symfony\Contracts\HttpClient\Exception\ServerExceptionInterface;
use Symfony\Contracts\HttpClient\Exception\TransportExceptionInterface;
use Symfony\Contracts\HttpClient\HttpClientInterface;

#[AsController]
class StatusController extends AbstractController
{
    const string CREATE_IMAGE = 'RESPUESTA_CrearImagen';
    const string RESTORE_IMAGE = 'RESPUESTA_RestaurarImagen';
    const string CONFIGURE_IMAGE = 'RESPUESTA_Configurar';


    public function __construct(
        protected readonly EntityManagerInterface $entityManager,
        public readonly CreateAuxFilesAction $createAuxFilesAction,
        protected readonly CreatePartitionService $createPartitionService,
        protected readonly LoggerInterface $logger,
        protected readonly CreateService $createService,
    )
    {
    }

    /**
     * @throws TransportExceptionInterface
     * @throws ServerExceptionInterface
     * @throws RedirectionExceptionInterface
     * @throws ClientExceptionInterface
     */
    #[Route('/opengnsys/rest/clients/status/webhook', methods: ['POST'])]
    public function index(Request $request): JsonResponse
    {
        $data = $request->toArray();

        $this->logger->info('Webhook data received', $data);

        // Esta parte del codigo nos indica si el cliente se encuentra activo
        if (isset($data['iph']) && isset($data['timestamp'])) {
            $client = $this->entityManager->getRepository(Client::class)->findOneBy(['ip' => $data['iph']]);
            if (!$client) {
                $this->logger->error('Client not found', $data);
                return new JsonResponse(['message' => 'Client not found'], Response::HTTP_NOT_FOUND);
            }

            $client->setUpdatedAt(new \DateTime());
            $this->entityManager->persist($client);
            $this->entityManager->flush();
        }

        if (isset($data['progress'])){
            $trace = $this->entityManager->getRepository(Trace::class)->findOneBy(['jobId' => $data['job_id']]);
            if ($trace){
                $trace->setProgress($data['progress'] * 100);
                $this->entityManager->persist($trace);
                $this->entityManager->flush();
            }
        }

        if (isset($data['nfn']) && $data['nfn'] === self::CREATE_IMAGE) {
            $trace = $this->entityManager->getRepository(Trace::class)->findOneBy(['jobId' => $data['job_id']]);
            /** @var ImageImageRepository $imageImageRepository */
            $imageImageRepository = $this->entityManager->getRepository(ImageImageRepository::class)->findOneBy(['uuid' => $data['idi']]);

            if (!$imageImageRepository) {
                $this->logger->error('Image not found', $data);
                return new JsonResponse(['message' => 'Image not found'], Response::HTTP_NOT_FOUND);
            }

            if (!$trace) {
                $this->logger->error('Trace not found', $data);
                return new JsonResponse(['message' => 'Trace not found'], Response::HTTP_NOT_FOUND);
            }

            if ($data['res'] === 1) {
                $trace->setStatus(TraceStatus::SUCCESS);
                $trace->setFinishedAt(new \DateTime());
                $imageImageRepository->setStatus(ImageStatus::AUX_FILES_PENDING);
                $imageImageRepository->setCreated(true);
                $this->entityManager->persist($imageImageRepository);

                $this->logger->info('Start partition creation. ', ['image' => (string) $imageImageRepository->getUuid()]);
                if (isset($data['cfg'])) {
                    $this->createPartitionService->__invoke($data, $imageImageRepository->getImage()->getClient());
                }
                $this->logger->info('Starting software profile creation. ', ['image' => (string) $imageImageRepository->getUuid()]);
                $this->createSoftwareProfile($data['inv_sft'], $imageImageRepository);
                $this->logger->info('Start aux files ogrepo API ', ['image' => (string) $imageImageRepository->getUuid()]);

                try {
                    $this->createAuxFilesAction->__invoke($imageImageRepository);
                } catch (\Exception $e) {
                    $this->logger->error('Error creating aux files', ['image' => (string) $imageImageRepository->getUuid(), 'error' => $e->getMessage()]);
                }

                $this->logger->info('End aux files ogrepo API ', ['image' => (string) $imageImageRepository->getUuid()]);
            } else {
                $trace->setStatus(TraceStatus::FAILED);
                $trace->setFinishedAt(new \DateTime());
                $trace->setOutput($data['der']);
                $imageImageRepository->setCreated(false);
                $imageImageRepository->setStatus(ImageStatus::FAILED);
                $this->logger->error('Image updated failed', $data);
            }

            $this->entityManager->persist($imageImageRepository);
            $client = $trace->getClient();
            $client->setStatus(ClientStatus::OG_LIVE);
            $this->entityManager->persist($client);
            $this->entityManager->persist($trace);
            $this->entityManager->flush();
            $this->logger->info('Image updated. Success.', ['image' => (string) $imageImageRepository->getUuid()]);
        }

        if (isset($data['nfn']) && ($data['nfn'] === self::RESTORE_IMAGE || $data['nfn'] === self::CONFIGURE_IMAGE)) {
            $trace = $this->entityManager->getRepository(Trace::class)->findOneBy(['jobId' => $data['job_id']]);
            $client = $trace->getClient();

            if ($data['res'] === 1) {
                $trace->setStatus(TraceStatus::SUCCESS);
                $trace->setFinishedAt(new \DateTime());
                if (isset($data['cfg'])) {
                    $this->createPartitionService->__invoke($data,$client);
                }
            } else {
                $trace->setStatus(TraceStatus::FAILED);
                $trace->setFinishedAt(new \DateTime());
                $trace->setOutput($data['der']);
            }

            $client->setStatus(ClientStatus::OG_LIVE);
            $this->entityManager->persist($client);
            $this->entityManager->persist($trace);
            $this->entityManager->flush();
        }

        return new JsonResponse(data: 'Webhook finished', status: Response::HTTP_OK);
    }

    public function createSoftwareProfile(string $base64Data, ImageImageRepository $imageImageRepository): void
    {
        $decodedData = base64_decode($base64Data);
        $this->logger->info('Software profile decoded', ['data' => '']);
        $softwareList = array_map('trim', explode("\n", $decodedData));
        $softwareList = array_filter($softwareList);
        $existingSoftware = $this->entityManager->getRepository(Software::class)->findBy(['name' => $softwareList]);
        $existingNames = array_map(fn($software) => $software->getName(), $existingSoftware);
        $newSoftwareNames = array_diff($softwareList, $existingNames);

        foreach ($newSoftwareNames as $software) {
            $softwareEntity = new Software();
            $softwareEntity->setName($software);
            $softwareEntity->setType(SoftwareTypes::APPLICATION);
            $this->entityManager->persist($softwareEntity);
        }

        $image = $imageImageRepository->getImage();

        $softwareProfile = new SoftwareProfile();
        $softwareProfile->setDescription('Perfil software: ' . $image->getClient()->getName());
        $softwareProfile->setOrganizationalUnit($image->getClient()->getOrganizationalUnit());

        foreach ($existingSoftware as $softwareEntity) {
            $softwareEntity->addSoftwareProfile($softwareProfile);
        }

        $image->setSoftwareProfile($softwareProfile);

        $this->entityManager->persist($image);
        $this->entityManager->persist($imageImageRepository);
        $this->entityManager->persist($softwareProfile);
        $this->entityManager->flush();
    }
}