2 files changed, 59 insertions, 0 deletions
diff --git a/app/Model/Project.php b/app/Model/Project.php
index 158de295..80e17437 100644
--- a/app/Model/Project.php
+++ b/app/Model/Project.php
@@ -84,6 +84,10 @@ class Project extends Base
      */
     public function getByToken($token)
     {
+        if (empty($token)) {
+            return false;
+        }
+
         return $this->db->table(self::TABLE)->eq('token', $token)->eq('is_public', 1)->findOne();
     }
 
diff --git a/app/Model/User.php b/app/Model/User.php
index be2b034b..845efac0 100644
--- a/app/Model/User.php
+++ b/app/Model/User.php
@@ -5,6 +5,7 @@ namespace Model;
 use SimpleValidator\Validator;
 use SimpleValidator\Validators;
 use Core\Session;
+use Core\Security;
 
 /**
  * User model
@@ -114,6 +115,10 @@ class User extends Base
      */
     public function getByGoogleId($google_id)
     {
+        if (empty($google_id)) {
+            return false;
+        }
+
         return $this->db->table(self::TABLE)->eq('google_id', $google_id)->findOne();
     }
 
@@ -126,6 +131,10 @@ class User extends Base
      */
     public function getByGitHubId($github_id)
     {
+        if (empty($github_id)) {
+            return false;
+        }
+
         return $this->db->table(self::TABLE)->eq('github_id', $github_id)->findOne();
     }
 
@@ -158,6 +167,22 @@ class User extends Base
     }
 
     /**
+     * Fetch user by using the token
+     *
+     * @access public
+     * @param  string   $token    Token
+     * @return array
+     */
+    public function getByToken($token)
+    {
+        if (empty($token)) {
+            return false;
+        }
+
+        return $this->db->table(self::TABLE)->eq('token', $token)->findOne();
+    }
+
+    /**
      * Get all users
      *
      * @access public
@@ -301,6 +326,36 @@ class User extends Base
     }
 
     /**
+     * Enable public access for a user
+     *
+     * @access public
+     * @param  integer   $user_id   User id
+     * @return bool
+     */
+    public function enablePublicAccess($user_id)
+    {
+        return $this->db
+                    ->table(self::TABLE)
+                    ->eq('id', $user_id)
+                    ->save(array('token' => Security::generateToken()));
+    }
+
+    /**
+     * Disable public access for a user
+     *
+     * @access public
+     * @param  integer   $user_id    User id
+     * @return bool
+     */
+    public function disablePublicAccess($user_id)
+    {
+        return $this->db
+                    ->table(self::TABLE)
+                    ->eq('id', $user_id)
+                    ->save(array('token' => ''));
+    }
+
+    /**
      * Common validation rules
      *
      * @access private