src/Entity/Vulnerability.php line 17

Open in your IDE?
  1. <?php
  3. namespace App\Entity;
  5. use App\Repository\VulnerabilityRepository;
  6. use App\Traits\DateTimeTrait;
  7. use Doctrine\ORM\Mapping as ORM;
  8. use Overblog\GraphQLBundle\Annotation\Field;
  9. use Doctrine\Common\Collections\ArrayCollection;
  10. use Doctrine\Common\Collections\Collection;
  11. use Doctrine\Common\Collections\ReadableCollection;
  12. use App\Attribute\Searchable;
  13. use DateTimeImmutable;
  15. #[ORM\Entity(repositoryClass: VulnerabilityRepository::class)]
  16. #[ORM\HasLifecycleCallbacks]
  17. class Vulnerability extends BasicEntity
  18. {
  19.     use DateTimeTrait;
  21.     #[ORM\Id]
  22.     #[ORM\GeneratedValue]
  23.     #[ORM\Column(type: 'integer')]
  24.     #[Field(name: "id", type: "int")]
  25.     private $id;
  27.     #[ORM\Column(type: 'string', length: 255)]
  28.     #[Field(name: "title", type: "string")]
  29.     #[Searchable(name: 'title')]
  30.     private $title;
  32.     #[ORM\Column(type: 'string', length: 255, nullable: true)]
  33.     #[Field(name: "componenent", type: "string")]
  34.     private $componenent;
  36.     #[ORM\Column(type: 'text', nullable: true)]
  37.     #[Field(name: "metadata", type: "string")]
  38.     #[Searchable(name: 'metadata')]
  39.     private $metadata;
  41.     #[ORM\Column(type: 'text', nullable: true)]
  42.     #[Field(name: "description", type: "string")]
  43.     private $description;
  45.     #[ORM\ManyToOne(targetEntity: VulnerabilityLevel::class)]
  46.     #[ORM\JoinColumn(nullable: false)]
  47.     #[Field(name: "level", type: "string")]
  48.     private $level;
  50.     #[ORM\ManyToOne(targetEntity: Domain::class, inversedBy: 'vulnerabilities')]
  51.     #[ORM\JoinColumn(nullable: false)]
  52.     #[Field(name: "domain")]
  53.     private $domain;
  55.     #[ORM\Column(type: 'string', length: 255)]
  56.     #[Field(name: "uid", type: "string")]
  57.     private $uid;
  59.     #[ORM\OneToMany(mappedBy: "vulnurability", targetEntity: Recommendation::class, cascade: ["remove"])]
  60.     private $recommendations;
  62.     #[ORM\Column(type: "boolean", options: ['default' => true])]
  63.     #[Field(name: "is_active", type: "boolean")]
  64.     private $isActive;
  66.     #[ORM\Column(type: 'datetime_immutable', nullable: true)]
  67.     #[Field(name: "last_seen", type: "string")]
  68.     private $lastSeen;
  70.     #[ORM\ManyToOne(targetEntity: Status::class, inversedBy: 'exposedIps')]
  71.     #[ORM\JoinColumn(nullable: true, options: ['default' => 1])]
  72.     #[Field(name: "status")]
  73.     private $status;
  75.     public function __construct()
  76.     {
  77.         $this->recommendations = new ArrayCollection();
  78.     }
  81.     public function getId(): ?int
  82.     {
  83.         return $this->id;
  84.     }
  86.     public function getTitle(): ?string
  87.     {
  88.         return $this->title;
  89.     }
  91.     public function setTitle(string $title): self
  92.     {
  93.         $this->title = $title;
  95.         return $this;
  96.     }
  98.     public function getComponenent(): ?string
  99.     {
  100.         return $this->componenent;
  101.     }
  103.     public function setComponenent(?string $componenent): self
  104.     {
  105.         $this->componenent = $componenent;
  107.         return $this;
  108.     }
  110.     public function getMetadata(): ?string
  111.     {
  112.         return $this->metadata;
  113.     }
  115.     public function setMetadata(?string $metadata): self
  116.     {
  117.         $this->metadata = $metadata;
  119.         return $this;
  120.     }
  122.     public function getDescription(): ?string
  123.     {
  124.         return $this->description;
  125.     }
  127.     public function setDescription(?string $description): self
  128.     {
  129.         $this->description = $description;
  131.         return $this;
  132.     }
  134.     public function getLevel(): ?VulnerabilityLevel
  135.     {
  136.         return $this->level;
  137.     }
  139.     public function setLevel(?VulnerabilityLevel $level): self
  140.     {
  141.         $this->level = $level;
  143.         return $this;
  144.     }
  146.     public function getDomain(): ?Domain
  147.     {
  148.         return $this->domain;
  149.     }
  151.     public function setDomain(?Domain $domain): self
  152.     {
  153.         $this->domain = $domain;
  155.         return $this;
  156.     }
  158.     public function getUid(): ?string
  159.     {
  160.         return $this->uid;
  161.     }
  163.     public function setUid(string $uid): self
  164.     {
  165.         $this->uid = $uid;
  167.         return $this;
  168.     }
  170.     /**
  171.      * @return Collection|Recommendations[]
  172.      */
  173.     public function getRecommendations(): Collection
  174.     {
  175.         return $this->recommendations;
  176.     }
  178.     public function getIsActive(): ?bool
  179.     {
  180.         return $this->isActive;
  181.     }
  183.     public function setIsActive(?bool $isActive): self
  184.     {
  185.         $this->isActive = $isActive;
  187.         return $this;
  188.     }
  190.     public function getLastSeen(): ?DateTimeImmutable
  191.     {
  192.         return $this->lastSeen;
  193.     }
  195.     public function setLastSeen(?DateTimeImmutable $lastSeen): self
  196.     {
  197.         $this->lastSeen = $lastSeen;
  199.         return $this;
  200.     }
  202.     public function getStatus(): ?Status
  203.     {
  204.         return $this->status;
  205.     }
  207.     public function setStatus(?Status $status): self
  208.     {
  209.         $this->status = $status;
  211.         return $this;
  212.     }
  214.     public function getSubDomain(): ?ExposedSubDomain
  215.     {
  216.         $metadata = $this->getMetadata();
  217.         $subDomain = $this->getDomain()->getExposedSubDomains()
  218.             ->filter(function(ExposedSubDomain $sub) use ($metadata){
  219.                return str_contains(
  220.                    $metadata,
  221.                    $sub->getSubDomain().".".str_replace('@', '', $this->getDomain()->getUrl())
  222.                );
  223.             })->current();
  224.         
  225.         // Check if current() returned false and return null if so
  226.         return $subDomain !== false ? $subDomain : null;
  227.     }
  228. }