1 files changed, 131 insertions, 0 deletions
diff --git a/app/Model/UserSession.php b/app/Model/UserSession.php
new file mode 100644
index 00000000..6703a1bc
--- /dev/null
+++ b/app/Model/UserSession.php
@@ -0,0 +1,131 @@
+<?php
+
+namespace Model;
+
+use Core\Translator;
+
+/**
+ * User Session
+ *
+ * @package  model
+ * @author   Frederic Guillot
+ */
+class UserSession extends Base
+{
+    /**
+     * Update user session information
+     *
+     * @access public
+     * @param  array  $user  User data
+     */
+    public function refresh(array $user = array())
+    {
+        if (empty($user)) {
+            $user = $this->user->getById($this->userSession->getId());
+        }
+
+        if (isset($user['password'])) {
+            unset($user['password']);
+        }
+
+        if (isset($user['twofactor_secret'])) {
+            unset($user['twofactor_secret']);
+        }
+
+        $user['id'] = (int) $user['id'];
+        $user['default_project_id'] = (int) $user['default_project_id'];
+        $user['is_admin'] = (bool) $user['is_admin'];
+        $user['is_ldap_user'] = (bool) $user['is_ldap_user'];
+        $user['twofactor_activated'] = (bool) $user['twofactor_activated'];
+
+        $this->session['user'] = $user;
+    }
+
+    /**
+     * Return true if the user has validated the 2FA key
+     *
+     * @access public
+     * @return bool
+     */
+    public function check2FA()
+    {
+        return isset($this->session['2fa_validated']) && $this->session['2fa_validated'] === true;
+    }
+
+    /**
+     * Return true if the user has 2FA enabled
+     *
+     * @access public
+     * @return bool
+     */
+    public function has2FA()
+    {
+        return isset($this->session['user']['twofactor_activated']) && $this->session['user']['twofactor_activated'] === true;
+    }
+
+    /**
+     * Return true if the logged user is admin
+     *
+     * @access public
+     * @return bool
+     */
+    public function isAdmin()
+    {
+        return isset($this->session['user']['is_admin']) && $this->session['user']['is_admin'] === true;
+    }
+
+    /**
+     * Get the connected user id
+     *
+     * @access public
+     * @return integer
+     */
+    public function getId()
+    {
+        return isset($this->session['user']['id']) ? (int) $this->session['user']['id'] : 0;
+    }
+
+    /**
+     * Check is the user is connected
+     *
+     * @access public
+     * @return bool
+     */
+    public function isLogged()
+    {
+        return ! empty($this->session['user']);
+    }
+
+    /**
+     * Get the last seen project from the session
+     *
+     * @access public
+     * @return integer
+     */
+    public function getLastSeenProjectId()
+    {
+        return empty($this->session['last_show_project_id']) ? 0 : $this->session['last_show_project_id'];
+    }
+
+    /**
+     * Get the default project from the session
+     *
+     * @access public
+     * @return integer
+     */
+    public function getFavoriteProjectId()
+    {
+        return isset($this->session['user']['default_project_id']) ? $this->session['user']['default_project_id'] : 0;
+    }
+
+    /**
+     * Set the last seen project from the session
+     *
+     * @access public
+     * @param integer    $project_id    Project id
+     */
+    public function storeLastSeenProjectId($project_id)
+    {
+        $this->session['last_show_project_id'] = (int) $project_id;
+    }
+}