Making ParseUser and ParseRole subclassable

Author: Caleb Fidecaro

src/Parse/ParseClient.php

@@ -67,8 +67,14 @@ final class ParseClient
    */
   public static function initialize($app_id, $rest_key, $master_key, $enableCurlExceptions = true)
   {
-      ParseUser::registerSubclass();
-      ParseRole::registerSubclass();
+      if (! ParseObject::hasRegisteredSubclass('_User')) {
+          ParseUser::registerSubclass();
+      }
+
+      if (! ParseObject::hasRegisteredSubclass('_Role')) {
+          ParseRole::registerSubclass();
+      }
+
       ParseInstallation::registerSubclass();
       ParseSession::registerSubclass();
       self::$applicationId = $app_id;

src/Parse/ParseObject.php

@@ -1196,6 +1196,35 @@ public static function _unregisterSubclass()
       unset(self::$registeredSubclasses[$subclass]);
   }
 
+  /**
+   * Check whether there is a subclass registered for a given parse class
+   *
+   * @param $parseClassName
+   *
+   * @return bool
+   */
+  public static function hasRegisteredSubclass($parseClassName)
+  {
+      return array_key_exists($parseClassName, self::$registeredSubclasses);
+  }
+
+  /**
+   * Get the registered subclass for a Parse class, or a generic ParseObject
+   * if no subclass is registered.
+   *
+   * @param $parseClassName
+   *
+   * @return ParseObject
+   */
+  public static function getRegisteredSubclass($parseClassName)
+  {
+      if (self::hasRegisteredSubclass($parseClassName)) {
+          return self::$registeredSubclasses[$parseClassName];
+      }
+
+      return new static($parseClassName);
+  }
+
   /**
    * Creates a ParseQuery for the subclass of ParseObject.
    * Cannot be called on the base class ParseObject.

src/Parse/ParseUser.php

@@ -14,7 +14,7 @@ class ParseUser extends ParseObject
   /**
    * @var ParseUser The currently logged-in user.
    */
-  private static $currentUser = null;
+  protected static $currentUser = null;
 
   /**
    * @var string The sessionToken for an authenticated user.
@@ -90,7 +90,7 @@ public function isAuthenticated()
   /**
    * Signs up the current user, or throw if invalid.
    * This will create a new ParseUser on the server, and also persist the
-   * session so that you can access the user using ParseUser::getCurrentUser();.
+   * session so that you can access the user using static::getCurrentUser();.
    */
   public function signUp()
   {
@@ -133,7 +133,7 @@ public static function logIn($username, $password)
       }
       $data = ["username" => $username, "password" => $password];
       $result = ParseClient::_request("GET", "/1/login", "", $data);
-      $user = new ParseUser();
+      $user = new static();
       $user->_mergeAfterFetch($result);
       $user->handleSaveResult(true);
       ParseClient::getStorage()->set("user", $user);
@@ -152,7 +152,7 @@ public static function logIn($username, $password)
   public static function become($sessionToken)
   {
       $result = ParseClient::_request('GET', '/1/users/me', $sessionToken);
-      $user = new ParseUser();
+      $user = new static();
       $user->_mergeAfterFetch($result);
       $user->handleSaveResult(true);
       ParseClient::getStorage()->set("user", $user);
@@ -169,7 +169,7 @@ public static function become($sessionToken)
    */
   public static function logOut()
   {
-      $user = ParseUser::getCurrentUser();
+      $user = static::getCurrentUser();
       if ($user) {
           try {
               ParseClient::_request('POST', '/1/logout', $user->getSessionToken());
@@ -188,7 +188,7 @@ public static function logOut()
    *
    * @return null
    */
-  private function handleSaveResult($makeCurrent = false)
+  public function handleSaveResult($makeCurrent = false)
   {
       if (isset($this->serverData['password'])) {
           unset($this->serverData['password']);
@@ -223,7 +223,7 @@ public static function getCurrentUser()
           return $userData;
       }
       if (isset($userData["id"]) && isset($userData["_sessionToken"])) {
-          $user = ParseUser::create("_User", $userData["id"]);
+          $user = static::create("_User", $userData["id"]);
           unset($userData["id"]);
           $user->_sessionToken = $userData["_sessionToken"];
           unset($userData["_sessionToken"]);
@@ -247,7 +247,7 @@ public static function getCurrentUser()
   protected static function saveCurrentUser()
   {
       $storage = ParseClient::getStorage();
-      $storage->set('user', ParseUser::getCurrentUser());
+      $storage->set('user', static::getCurrentUser());
   }
 
   /**
@@ -267,8 +267,8 @@ public function getSessionToken()
    */
   public function isCurrent()
   {
-      if (ParseUser::getCurrentUser() && $this->getObjectId()) {
-          if ($this->getObjectId() == ParseUser::getCurrentUser()->getObjectId()) {
+      if (static::getCurrentUser() && $this->getObjectId()) {
+          if ($this->getObjectId() == static::getCurrentUser()->getObjectId()) {
               return true;
           }
       }