<?php

namespace Mautic\CampaignBundle\Executioner\Result;

use Doctrine\Common\Collections\ArrayCollection;
use Mautic\LeadBundle\Entity\Lead;

class EvaluatedContacts
{
    private \Doctrine\Common\Collections\ArrayCollection $passed;

    private \Doctrine\Common\Collections\ArrayCollection $failed;

    public function __construct(ArrayCollection $passed = null, ArrayCollection $failed = null)
    {
        $this->passed = $passed ?? new ArrayCollection();
        $this->failed = $failed ?? new ArrayCollection();
    }

    public function pass(Lead $contact): void
    {
        $this->passed->set($contact->getId(), $contact);
    }

    public function fail(Lead $contact): void
    {
        $this->failed->set($contact->getId(), $contact);
    }

    /**
     * @return ArrayCollection|Lead[]
     */
    public function getPassed()
    {
        return $this->passed;
    }

    /**
     * @return ArrayCollection|Lead[]
     */
    public function getFailed()
    {
        return $this->failed;
    }
}
