SpecialOAuth2Client.php

<?php
/**
 * SpecialOAuth2Client.php
 * Based on TwitterLogin by David Raison, which is based on the guideline published by Dave Challis at http://blogs.ecs.soton.ac.uk/webteam/2010/04/13/254/
 * @license: LGPL (GNU Lesser General Public License) http://www.gnu.org/licenses/lgpl.html
 *
 * @file SpecialOAuth2Client.php
 * @ingroup OAuth2Client
 *
 * @author Joost de Keijzer
 *
 * Uses the OAuth2 library https://github.com/vznet/oauth_2.0_client_php
 *
 */

if ( !defined( 'MEDIAWIKI' ) ) {
	die( 'This is a MediaWiki extension, and must be run from within MediaWiki.' );
}

class SpecialOAuth2Client extends SpecialPage {

	private $_clientId;
	private $_clientSecret;
	private $_clientCallbackUrl;

	private $_serviceAuthorizeEndpointUrl;
	private $_serviceAccessTokenEndpointUrl;
	private $_serviceApiEndpointUrl;

	private $_euTable = 'external_user';

	private $_oAuth2Service;

	/**
	 * Required settings in global $wgOAuth2Client
	 *
	 * $wgOAuth2Client['client']['id']
	 * $wgOAuth2Client['client']['secret']
	 * //$wgOAuth2Client['client']['callback_url'] // extension should know
	 *
	 * $wgOAuth2Client['configuration']['authorize_endpoint']
	 * $wgOAuth2Client['configuration']['access_token_endpoint']
	 * $wgOAuth2Client['configuration']['http_bearer_token']
	 * $wgOAuth2Client['configuration']['query_parameter_token']
	 * $wgOAuth2Client['configuration']['api_endpoint']
	 */
	public function __construct() {
		if( !self::OAuthEnabled() ) return;

		parent::__construct('OAuth2Client'); // ???: wat doet dit?
		global $wgOAuth2Client, $wgScriptPath;
		global $wgServer, $wgArticlePath;

		$client = new OAuth2\Client(
			$wgOAuth2Client['client']['id'],
			$wgOAuth2Client['client']['secret'],
			$wgServer . str_replace( '$1', 'Special:OAuth2Client/callback', $wgArticlePath )
			//SpecialPage::getTitleFor( 'OAuth2Client', 'callback' )->getFullURL() // setting variant does not work on specialpages
		);

		// configuration of service
		$configuration = new OAuth2\Service\Configuration(
			$wgOAuth2Client['configuration']['authorize_endpoint'],
			$wgOAuth2Client['configuration']['access_token_endpoint'],
			$wgOAuth2Client['configuration']['http_bearer_token'],
			$wgOAuth2Client['configuration']['query_parameter_token']
		);
		//$configuration->setAuthorizationMethod( OAuth2\Service\Configuration::AUTHORIZATION_METHOD_ALTERNATIVE);

		// storage class for access token, just implement OAuth2\DataStore interface for
		// your own implementation
		$dataStore = new OAuth2\DataStore\Session();

		$scope = null;

		$this->_oAuth2Service = new OAuth2\Service($client, $configuration, $dataStore, $scope);
	}

	// default method being called by a specialpage
	public function execute( $parameter ){
		$this->setHeaders();
		switch($parameter){
			case 'redirect':
				$this->_redirect();
			break;
			case 'callback':
				$this->_handleCallback();
			break;
			case 'logout':
				$this->_logout();
			break;
			default:
				$this->_default();
			break;
		}
		
	}

	private function _redirect() {
		if( !self::OAuthEnabled() ) return false;

		global $wgRequest;
		$_SESSION['returnto'] = $wgRequest->getVal( 'returnto' );
		$this->_oAuth2Service->authorize();
	}

	private function _handleCallback(){
		if( !self::OAuthEnabled() ) return false;

		global $wgOAuth2Client, $wgOut;
		if( $this->_oAuth2Service->getAccessToken() ) {
			$requestApiResponse = $this->_oAuth2Service->callApiEndpoint($wgOAuth2Client['configuration']['api_endpoint']);
		} else {
			throw new MWException('Invalid callback');
		}

		$user = $this->_userHandling( $requestApiResponse );
		$user->setOption("rememberpassword", 1);
		$user->setCookies();

		if( $user->getRegistration() > wfTimestamp( TS_MW ) - 1 ) {
			// new user!
			$wgOut->redirect(SpecialPage::getTitleFor('Preferences')->getLinkUrl());
		} else {
			$title = null;
			if( isset( $_SESSION['returnto'] ) ) {
				$title = Title::newFromText( $_SESSION['returnto'] );
				unset( $_SESSION['returnto'] );
			}

			if( !$title instanceof Title ) {
				$title = Title::newMainPage();
			}
			$wgOut->redirect( $title->getFullURL() );
		}
		return true;
	}

	private function _logout() {
		global $wgOAuth2Client, $wgOut, $wgUser;
		if( $wgUser->isLoggedIn() ) $wgUser->logout();

		$sevice_name = ( isset( $wgOAuth2Client['configuration']['sevice_name'] ) && 0 < strlen( $wgOAuth2Client['configuration']['sevice_name'] ) ? $wgOAuth2Client['configuration']['sevice_name'] : 'OAuth2' );

		$wgOut->setPagetitle( wfMsg( 'oauth2client-logout-header', $sevice_name) );
		$wgOut->addWikiMsg( 'oauth2client-logged-out' );
		$wgOut->addWikiMsg( 'oauth2client-login-with-oauth2-again', $this->getTitle( 'redirect' )->getPrefixedURL(), $sevice_name );
	}

	private function _default(){
		global $wgOAuth2Client, $wgOut, $wgUser, $wgScriptPath, $wgExtensionAssetsPath;
		$sevice_name = ( isset( $wgOAuth2Client['configuration']['sevice_name'] ) && 0 < strlen( $wgOAuth2Client['configuration']['sevice_name'] ) ? $wgOAuth2Client['configuration']['sevice_name'] : 'OAuth2' );

		$wgOut->setPagetitle( wfMsg( 'oauth2client-login-header', $sevice_name) );
		if ( !$wgUser->isLoggedIn() ) {
			$wgOut->addWikiMsg( 'oauth2client-you-can-login-to-this-wiki-with-oauth2', $sevice_name );
			$wgOut->addWikiMsg( 'oauth2client-login-with-oauth2', $this->getTitle( 'redirect' )->getPrefixedURL(), $sevice_name );

		} else {
			$wgOut->addWikiMsg( 'oauth2client-youre-already-loggedin' );
		}
		return true;
	}

	protected function _userHandling( $response ) {
		global $wgOAuth2Client, $wgAuth;

		// TODO: make id, name etc. parameters configurable
		$oAuth2Id = $response['id'];
		$oAuth2Name = $response['first_name'] .( strlen($response['last_name']) > 0 ? ' ' . $response['last_name'] : '');

		// not required
		$oAuth2Email = ( isset( $response['email'] ) ? $response['email'] : '' );


		$externalId = 'OAuth2Client.' . $wgOAuth2Client['client']['id'] . '.' . $oAuth2Id;

		$dbr = wfGetDB( DB_SLAVE );
		$row = $dbr->selectRow(
			'external_user',
			'*',
			array( 'eu_external_id' => $externalId )
		);
		if( $row ) {
			// existing OAuth2 user
			return User::newFromId( $row->eu_local_id );
		}
		// create user based on $oAuth2Name
		$counter = 1;
		$success = false; 
		while( !$success && $counter <= 1000 ) {
			$checkName = $oAuth2Name . ( $counter > 1 ? ' ' . $counter : '' );
			$user = User::newFromName( $checkName, 'creatable' );
			$counter ++;
			$success = (false !== $user && $user->getId() == 0);
		}
		if( false === $user || $user->getId() != 0 ) {
			throw new MWException('Unable to create new user account, please contact the Wiki administrator');
		}
		$user->setRealName($oAuth2Name);
		if( strlen($oAuth2Email) > 0 ) {
			$user->setEmail($oAuth2Email);
			$user->setEmailAuthenticationTimestamp(time()); // ???: should we auto-authenticate e-mail?
		}
		if ( $wgAuth->allowPasswordChange() ) {
			$user->setPassword(User::randomPassword());
		}
		$user->addToDatabase();

		// link local user to remote OAuth2
		$dbw = wfGetDB( DB_MASTER );
		$dbw->replace( 'external_user',
			array( 'eu_local_id', 'eu_external_id' ),
			array( 'eu_local_id' => $user->getId(),
				   'eu_external_id' => $externalId ),
			__METHOD__ );

		return $user;
	}

	static function OAuthEnabled() {
		global $wgOAuth2Client;

		return isset(
			$wgOAuth2Client['client']['id'],
			$wgOAuth2Client['client']['secret'],
			$wgOAuth2Client['configuration']['authorize_endpoint'],
			$wgOAuth2Client['configuration']['access_token_endpoint'],
			$wgOAuth2Client['configuration']['http_bearer_token'],
			$wgOAuth2Client['configuration']['query_parameter_token']
		);
	}
}