1 files changed, 55 insertions, 0 deletions
diff --git a/OAuth/src/Entity/UserEntity.php b/OAuth/src/Entity/UserEntity.php
new file mode 100644
index 00000000..73c803b0
--- /dev/null
+++ b/OAuth/src/Entity/UserEntity.php
@@ -0,0 +1,55 @@
+<?php
+
+namespace MediaWiki\Extensions\OAuth\Entity;
+
+use League\OAuth2\Server\Entities\UserEntityInterface;
+use MediaWiki\Extensions\OAuth\Backend\Utils;
+use MWException;
+use User;
+
+class UserEntity implements UserEntityInterface {
+
+	/**
+	 * @var int
+	 */
+	private $identifier = 0;
+
+	/**
+	 * @param User $user
+	 * @return UserEntity|null
+	 */
+	public static function newFromMWUser( User $user ) {
+		try {
+			$userId = Utils::getCentralIdFromLocalUser( $user );
+			if ( !$userId ) {
+				return null;
+			}
+			return new static( $userId );
+		} catch ( MWException $ex ) {
+			return null;
+		}
+	}
+
+	/**
+	 * @param string $identifier
+	 */
+	public function __construct( $identifier ) {
+		$this->identifier = $identifier;
+	}
+
+	/**
+	 * Return the user's identifier.
+	 *
+	 * @return mixed
+	 */
+	public function getIdentifier() {
+		return $this->identifier;
+	}
+
+	/**
+	 * @return bool|User
+	 */
+	public function getMWUser() {
+		return Utils::getLocalUserFromCentralId( $this->identifier );
+	}
+}