src/Entity/Organisation.php line 19

Open in your IDE?
  1. <?php
  3. namespace App\Entity;
  5. use App\Repository\OrganisationRepository;
  6. use DateTimeImmutable;
  7. use Doctrine\Common\Collections\ArrayCollection;
  8. use Doctrine\Common\Collections\Collection;
  9. use Doctrine\ORM\Mapping as ORM;
  10. use Overblog\GraphQLBundle\Annotation\Field;
  11. use Symfony\Component\Validator\Constraints as Assert;
  12. use Symfony\Bridge\Doctrine\Validator\Constraints\UniqueEntity;
  14. #[ORM\Entity(repositoryClass: OrganisationRepository::class)]
  15. #[UniqueEntity(
  16.     fields: "uuid",
  17.     message: "This uuid is already used."
  18. )]
  19. class Organisation extends BasicEntity
  20. {
  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: 50)]
  28.     #[Assert\NotBlank]
  29.     #[Field(name: "uuid", type: "string")]
  30.     private $uuid;
  32.     #[ORM\Column(type: "string", length: 255)]
  33.     #[Assert\NotBlank]
  34.     #[Field(name: "title", type: "string")]
  35.     private $title;
  37.     #[ORM\Column(type: "string", length: 255)]
  38.     #[Field(name: "website", type: "string")]
  39.     private $website;
  41.     #[ORM\Column(type: "string", length: 50)]
  42.     #[Field(name: "country", type: "string")]
  43.     private $country;
  45.     #[ORM\Column(type: "string", length: 255)]
  46.     #[Field(name: "address", type: "string")]
  47.     private $address;
  49.     #[ORM\Column(type: "string", length: 50)]
  50.     #[Field(name: "num_employees", type: "string")]
  51.     private $numEmployees;
  53.     #[ORM\ManyToOne(targetEntity: Industry::class, cascade: ["persist"])]
  54.     #[ORM\JoinColumn(nullable: false)]
  55.     #[Field(name: "industry")]
  56.     private $industry;
  58.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: BankIdentification::class, cascade: ["persist"])]
  59.     #[Field(name: "bank_identifications")]
  60.     private $bankIdentifications;
  62.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: LeakedIp::class, cascade: ["persist"])]
  63.     #[Field(name: "leaked_ips")]
  64.     private $leakedIps;
  66.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: Cryptocurrency::class, cascade: ["persist"])]
  67.     #[Field(name: "cyptocurrencys")]
  68.     private $cryptocurrencys;
  70.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: Domain::class, cascade: ["persist"])]
  71.     #[Field(name: "domains")]
  72.     private $domains;
  74.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: GitDepot::class, cascade: ["persist"])]
  75.     #[Field(name: "git_depots")]
  76.     private $gitDepots;
  78.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: ExposedIp::class)]
  79.     #[Field(name: "exposed_ips")]
  80.     private $exposedIps;
  82.     #[ORM\Column(type: "boolean")]
  83.     #[Field(name: "is_added", type: "boolean")]
  84.     private $isAdded;
  86.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: Subscription::class, cascade: ["persist"], fetch: 'EAGER')]
  87.     private $subscriptions;
  89.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: LeakedPassword::class)]
  90.     private $leakedPasswords;
  92.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: UserToNotify::class, cascade: ["persist"], orphanRemoval: true)]
  93.     #[Field(name: "users_tonotify")]
  94.     private $usersToNotify;
  96.     #[ORM\OneToMany(mappedBy: "organisation", targetEntity: CronLog::class)]
  97.     private $cronLogs;
  99.     #[ORM\Column(type: "boolean", nullable: true, options: ["default" => false])]
  100.     #[Field(name: "is_demo", type: "boolean")]
  101.     private $isDemo;
  103.     #[ORM\Column(type: "string", length: 255, nullable: true)]
  104.     #[Field(name: "avatar", type: "string")]
  105.     private $avatar;
  107.     #[ORM\ManyToMany(targetEntity: User::class, mappedBy: "organisations", cascade: ["persist"])]
  108.     #[Field(name: "users")]
  109.     private $users;
  111.     #[ORM\Column(options: ['default' => 0])]
  112.     #[Field(name: "quota", type: "int")]
  113.     private ?int $quota = 0;
  114.     private ?int $consumedQuota = 0;
  116.     #[ORM\OneToMany(mappedBy: 'organisation', targetEntity: InternalSecurityAnswer::class, orphanRemoval: true)]
  117.     private Collection $internalSecurityAnswers;
  119.     public function __construct()
  120.     {
  121.         $this->domains = new ArrayCollection();
  122.         $this->gitDepots = new ArrayCollection();
  123.         $this->exposedSubDomains = new ArrayCollection();
  124.         $this->exposedIps = new ArrayCollection();
  125.         $this->subscriptions = new ArrayCollection();
  126.         $this->leakedPasswords = new ArrayCollection();
  127.         $this->usersToNotify = new ArrayCollection();
  128.         $this->cronLogs = new ArrayCollection();
  129.         $this->users = new ArrayCollection();
  130.         $this->internalSecurityAnswers = new ArrayCollection();
  131.     }
  133.     public function getId(): ?int
  134.     {
  135.         return $this->id;
  136.     }
  138.     public function getUuid(): ?string
  139.     {
  140.         return $this->uuid;
  141.     }
  143.     public function setUuid(string $uuid): self
  144.     {
  145.         $this->uuid = $uuid;
  147.         return $this;
  148.     }
  150.     public function getTitle(): ?string
  151.     {
  152.         return $this->title;
  153.     }
  155.     public function setTitle(string $title): self
  156.     {
  157.         $this->title = $title;
  159.         return $this;
  160.     }
  162.     public function getWebsite(): ?string
  163.     {
  164.         return $this->website;
  165.     }
  167.     public function setWebsite(string $website): self
  168.     {
  169.         $this->website = $website;
  171.         return $this;
  172.     }
  174.     public function getCountry(): ?string
  175.     {
  176.         return $this->country;
  177.     }
  179.     public function setCountry(string $country): self
  180.     {
  181.         $this->country = $country;
  183.         return $this;
  184.     }
  186.     public function getAddress(): ?string
  187.     {
  188.         return $this->address;
  189.     }
  191.     public function setAddress(string $address): self
  192.     {
  193.         $this->address = $address;
  195.         return $this;
  196.     }
  198.     public function getNumEmployees(): ?string
  199.     {
  200.         return $this->numEmployees;
  201.     }
  203.     public function setNumEmployees(string $numEmployees): self
  204.     {
  205.         $this->numEmployees = $numEmployees;
  207.         return $this;
  208.     }
  210.     public function getIndustry(): ?Industry
  211.     {
  212.         return $this->industry;
  213.     }
  215.     public function setIndustry(?Industry $industry): self
  216.     {
  217.         $this->industry = $industry;
  219.         return $this;
  220.     }
  222.     /**
  223.      * @return Collection|Domain[]
  224.      */
  225.     public function getDomains(): Collection
  226.     {
  227.         return $this->domains;
  228.     }
  230.     public function addDomain(Domain $domain): self
  231.     {
  232.         if (!$this->domains->contains($domain)) {
  233.             $this->domains[] = $domain;
  234.             $domain->setOrganisation($this);
  235.         }
  237.         return $this;
  238.     }
  240.     public function removeDomain(Domain $domain): self
  241.     {
  242.         if ($this->domains->removeElement($domain)) {
  243.             // set the owning side to null (unless already changed)
  244.             if ($domain->getOrganisation() === $this) {
  245.                 $domain->setOrganisation(null);
  246.             }
  247.         }
  249.         return $this;
  250.     }
  252.     /**
  253.      * @return Collection|GitDepot[]
  254.      */
  255.     public function getGitDepots(): Collection
  256.     {
  257.         return $this->gitDepots;
  258.     }
  260.     public function addGitDepot(GitDepot $gitDepot): self
  261.     {
  262.         if (!$this->gitDepots->contains($gitDepot)) {
  263.             $this->gitDepots[] = $gitDepot;
  264.             $gitDepot->setOrganisation($this);
  265.         }
  267.         return $this;
  268.     }
  270.     public function removeGitDepot(GitDepot $gitDepot): self
  271.     {
  272.         if ($this->gitDepots->removeElement($gitDepot)) {
  273.             // set the owning side to null (unless already changed)
  274.             if ($gitDepot->getOrganisation() === $this) {
  275.                 $gitDepot->setOrganisation(null);
  276.             }
  277.         }
  279.         return $this;
  280.     }
  282.     /**
  283.      * @return Collection|ExposedIp[]
  284.      */
  285.     public function getExposedIps(): Collection
  286.     {
  287.         return $this->exposedIps;
  288.     }
  290.     public function addExposedIp(ExposedIp $exposedIp): self
  291.     {
  292.         if (!$this->exposedIps->contains($exposedIp)) {
  293.             $this->exposedIps[] = $exposedIp;
  294.             $exposedIp->setOrganisation($this);
  295.         }
  297.         return $this;
  298.     }
  300.     public function removeExposedIp(ExposedIp $exposedIp): self
  301.     {
  302.         if ($this->exposedIps->removeElement($exposedIp)) {
  303.             // set the owning side to null (unless already changed)
  304.             if ($exposedIp->getOrganisation() === $this) {
  305.                 $exposedIp->setOrganisation(null);
  306.             }
  307.         }
  309.         return $this;
  310.     }
  312.     public function getIsAdded(): ?bool
  313.     {
  314.         return $this->isAdded;
  315.     }
  317.     public function setIsAdded(bool $isAdded): self
  318.     {
  319.         $this->isAdded = $isAdded;
  321.         return $this;
  322.     }
  324.     /**
  325.      * @return Collection|Subscription[]
  326.      */
  327.     public function getSubscriptions(): Collection
  328.     {
  329.         return $this->subscriptions;
  330.     }
  332.     public function getSubscription(): ?Subscription
  333.     {
  334.         $actif = $this->subscriptions
  335.             ->filter(function (Subscription $s) {
  336.                 return $s->getIsActif();
  337.             });
  338.         return $actif->isEmpty() ? null : $actif->first();
  339.     }
  341.     public function addSubscription(Subscription $subscription): self
  342.     {
  343.         if (!$this->subscriptions->contains($subscription)) {
  344.             $this->subscriptions[] = $subscription;
  345.             $subscription->setOrganisation($this);
  346.         }
  348.         return $this;
  349.     }
  351.     public function removeSubscription(Subscription $subscription): self
  352.     {
  353.         if ($this->subscriptions->removeElement($subscription)) {
  354.             // set the owning side to null (unless already changed)
  355.             if ($subscription->getOrganisation() === $this) {
  356.                 $subscription->setOrganisation(null);
  357.             }
  358.         }
  360.         return $this;
  361.     }
  363.     public function __toString()
  364.     {
  365.         return $this->getTitle() ?? '';
  366.     }
  368.     /**
  369.      * @return Collection
  370.      */
  371.     public function getLeakedPasswords(): Collection
  372.     {
  373.         return $this->leakedPasswords;
  374.     }
  376.     public function addLeakedPassword(LeakedPassword $leakedPassword): self
  377.     {
  378.         if (!$this->leakedPasswords->contains($leakedPassword)) {
  379.             $this->leakedPasswords[] = $leakedPassword;
  380.             $leakedPassword->setOrganisation($this);
  381.         }
  383.         return $this;
  384.     }
  386.     public function removeLeakedPassword(LeakedPassword $leakedPassword): self
  387.     {
  388.         if ($this->leakedPasswords->removeElement($leakedPassword)) {
  389.             // set the owning side to null (unless already changed)
  390.             if ($leakedPassword->getOrganisation() === $this) {
  391.                 $leakedPassword->setOrganisation(null);
  392.             }
  393.         }
  395.         return $this;
  396.     }
  398.     /**
  399.      * @return Collection|UserToNotify[]
  400.      */
  401.     public function getUsersToNotify(): Collection
  402.     {
  403.         return $this->usersToNotify;
  404.     }
  406.     public function addUsersToNotify(UserToNotify $usersToNotify): self
  407.     {
  408.         if (!$this->usersToNotify->contains($usersToNotify)) {
  409.             $this->usersToNotify[] = $usersToNotify;
  410.             $usersToNotify->setOrganisation($this);
  411.         }
  413.         return $this;
  414.     }
  416.     public function removeUsersToNotify(UserToNotify $usersToNotify): self
  417.     {
  418.         if ($this->usersToNotify->removeElement($usersToNotify)) {
  419.             // set the owning side to null (unless already changed)
  420.             if ($usersToNotify->getOrganisation() === $this) {
  421.                 $usersToNotify->setOrganisation(null);
  422.             }
  423.         }
  425.         return $this;
  426.     }
  428.     /**
  429.      * @return Collection|CronLog[]
  430.      */
  431.     public function getCronLogs(): Collection
  432.     {
  433.         return $this->cronLogs;
  434.     }
  436.     public function addCronLog(CronLog $cronLog): self
  437.     {
  438.         if (!$this->cronLogs->contains($cronLog)) {
  439.             $this->cronLogs[] = $cronLog;
  440.             $cronLog->setOrganisation($this);
  441.         }
  443.         return $this;
  444.     }
  446.     public function removeCronLog(CronLog $cronLog): self
  447.     {
  448.         if ($this->cronLogs->removeElement($cronLog)) {
  449.             // set the owning side to null (unless already changed)
  450.             if ($cronLog->getOrganisation() === $this) {
  451.                 $cronLog->setOrganisation(null);
  452.             }
  453.         }
  455.         return $this;
  456.     }
  458.     public function getIsDemo(): ?bool
  459.     {
  460.         return $this->isDemo;
  461.     }
  463.     public function setIsDemo(?bool $isDemo): self
  464.     {
  465.         $this->isDemo = $isDemo;
  467.         return $this;
  468.     }
  470.     // public function getAvatar(): ?string
  471.     // {
  472.     //     return $this->avatar;
  473.     // }
  475.     public function getAvatar(): ?string
  476.     {
  477.         $path = __DIR__.'/../../public/uploads/avatar/'.$this->avatar; // Path to the image file
  478.         if (!empty($this->avatar) && file_exists($path)) {
  479.             $type = pathinfo($path, PATHINFO_EXTENSION); // Get the file extension
  480.             $data = file_get_contents($path); // Read the file's contents
  481.             $base64 = 'data:image/' . $type . ';base64,' . base64_encode($data); // Convert to base64
  482.             return $base64; // Return the base64 encoded string
  483.         }
  485.         return null; // Return null if the file doesn't exist
  486.     }
  489.     public function setAvatar(?string $avatar): self
  490.     {
  491.         $this->avatar = $avatar;
  493.         return $this;
  494.     }
  496.     /**
  497.      * @return Collection<int, User>
  498.      */
  499.     public function getUsers(): Collection
  500.     {
  501.         return $this->users;
  502.     }
  504.     public function addUser(User $user): self
  505.     {
  506.         if (!$this->users->contains($user)) {
  507.             $this->users[] = $user;
  508.             $user->addOrganisation($this);
  509.         }
  511.         return $this;
  512.     }
  514.     public function removeUser(User $user): self
  515.     {
  516.         if ($this->users->removeElement($user)) {
  517.             $user->removeOrganisation($this);
  518.         }
  520.         return $this;
  521.     }
  523.     public function getQuota(): ?int
  524.     {
  525.         return $this->quota;
  526.     }
  528.     public function setQuota(int $quota): static
  529.     {
  530.         $this->quota = $quota;
  532.         return $this;
  533.     }
  535.     public function setConsumedQuota(?int $consumedQuota): static
  536.     {
  537.         $this->consumedQuota = $consumedQuota;
  539.         return $this;
  540.     }
  542.     public function getConsumedQuota(): ?int
  543.     {
  544.         return $this->consumedQuota;
  545.     }
  547.     /**
  548.      * @return Collection<int, InternalSecurityAnswer>
  549.      */
  550.     public function getInternalSecurityAnswers(): Collection
  551.     {
  552.         return $this->internalSecurityAnswers;
  553.     }
  555.     public function addInternalSecurityAnswer(InternalSecurityAnswer $internalSecurityAnswer): static
  556.     {
  557.         if (!$this->internalSecurityAnswers->contains($internalSecurityAnswer)) {
  558.             $this->internalSecurityAnswers->add($internalSecurityAnswer);
  559.             $internalSecurityAnswer->setOrganisation($this);
  560.         }
  562.         return $this;
  563.     }
  565.     public function removeInternalSecurityAnswer(InternalSecurityAnswer $internalSecurityAnswer): static
  566.     {
  567.         if ($this->internalSecurityAnswers->removeElement($internalSecurityAnswer)) {
  568.             // set the owning side to null (unless already changed)
  569.             if ($internalSecurityAnswer->getOrganisation() === $this) {
  570.                 $internalSecurityAnswer->setOrganisation(null);
  571.             }
  572.         }
  574.         return $this;
  575.     }
  577. }