use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\ORM\Mapping as ORM;
-use Symfony\Bridge\Doctrine\Validator\Constraints\UniqueEntity;
-use Symfony\Component\Security\Core\User\UserInterface;
+use Scheb\TwoFactorBundle\Model\Email\TwoFactorInterface;
+use Scheb\TwoFactorBundle\Model\TrustedComputerInterface;
+use FOS\UserBundle\Model\User as BaseUser;
use JMS\Serializer\Annotation\ExclusionPolicy;
use JMS\Serializer\Annotation\Expose;
-use FOS\UserBundle\Model\User as BaseUser;
+use Symfony\Bridge\Doctrine\Validator\Constraints\UniqueEntity;
+use Symfony\Component\Security\Core\User\UserInterface;
+use Wallabag\ApiBundle\Entity\Client;
use Wallabag\CoreBundle\Entity\Config;
use Wallabag\CoreBundle\Entity\Entry;
-use Wallabag\CoreBundle\Entity\Tag;
/**
* User.
*
* @ORM\Entity(repositoryClass="Wallabag\UserBundle\Repository\UserRepository")
- * @ORM\Table
+ * @ORM\Table(name="`user`")
* @ORM\HasLifecycleCallbacks()
* @ExclusionPolicy("all")
*
* @UniqueEntity("email")
* @UniqueEntity("username")
*/
-class User extends BaseUser
+class User extends BaseUser implements TwoFactorInterface, TrustedComputerInterface
{
/**
* @var int
protected $entries;
/**
- * @ORM\OneToOne(targetEntity="Wallabag\CoreBundle\Entity\Config", mappedBy="user")
+ * @ORM\OneToOne(targetEntity="Wallabag\CoreBundle\Entity\Config", mappedBy="user", cascade={"remove"})
*/
protected $config;
/**
- * @ORM\OneToMany(targetEntity="Wallabag\CoreBundle\Entity\Tag", mappedBy="user", cascade={"remove"})
+ * @ORM\Column(type="integer", nullable=true)
+ */
+ private $authCode;
+
+ /**
+ * @var bool Enabled yes/no
+ * @ORM\Column(type="boolean")
*/
- protected $tags;
+ private $twoFactorAuthentication = false;
+
+ /**
+ * @ORM\Column(type="json_array", nullable=true)
+ */
+ private $trusted;
+
+ /**
+ * @ORM\OneToMany(targetEntity="Wallabag\ApiBundle\Entity\Client", mappedBy="user", cascade={"remove"})
+ */
+ protected $clients;
public function __construct()
{
parent::__construct();
$this->entries = new ArrayCollection();
- $this->tags = new ArrayCollection();
- $this->roles = array('ROLE_USER');
+ $this->roles = ['ROLE_USER'];
}
/**
return $this->entries;
}
+ public function isEqualTo(UserInterface $user)
+ {
+ return $this->username === $user->getUsername();
+ }
+
/**
- * @param Entry $entry
+ * Set config.
+ *
+ * @param Config $config
*
* @return User
*/
- public function addTag(Tag $tag)
+ public function setConfig(Config $config = null)
{
- $this->tags[] = $tag;
+ $this->config = $config;
return $this;
}
/**
- * @return ArrayCollection<Tag>
+ * Get config.
+ *
+ * @return Config
*/
- public function getTags()
+ public function getConfig()
{
- return $this->tags;
+ return $this->config;
}
- public function isEqualTo(UserInterface $user)
+ /**
+ * @return bool
+ */
+ public function isTwoFactorAuthentication()
{
- return $this->username === $user->getUsername();
+ return $this->twoFactorAuthentication;
}
/**
- * Set config.
- *
- * @param Config $config
+ * @param bool $twoFactorAuthentication
+ */
+ public function setTwoFactorAuthentication($twoFactorAuthentication)
+ {
+ $this->twoFactorAuthentication = $twoFactorAuthentication;
+ }
+
+ public function isEmailAuthEnabled()
+ {
+ return $this->twoFactorAuthentication;
+ }
+
+ public function getEmailAuthCode()
+ {
+ return $this->authCode;
+ }
+
+ public function setEmailAuthCode($authCode)
+ {
+ $this->authCode = $authCode;
+ }
+
+ public function addTrustedComputer($token, \DateTime $validUntil)
+ {
+ $this->trusted[$token] = $validUntil->format('r');
+ }
+
+ public function isTrustedComputer($token)
+ {
+ if (isset($this->trusted[$token])) {
+ $now = new \DateTime();
+ $validUntil = new \DateTime($this->trusted[$token]);
+
+ return $now < $validUntil;
+ }
+
+ return false;
+ }
+
+ /**
+ * @param Client $client
*
* @return User
*/
- public function setConfig(Config $config = null)
+ public function addClient(Client $client)
{
- $this->config = $config;
+ $this->clients[] = $client;
return $this;
}
/**
- * Get config.
- *
- * @return Config
+ * @return ArrayCollection<Entry>
*/
- public function getConfig()
+ public function getClients()
{
- return $this->config;
+ return $this->clients;
}
}