Merge pull request #980 from dawnbreak/dev
Refactor OAuth2Server a bit.
This commit is contained in:
		| @@ -2,42 +2,33 @@ | ||||
|  | ||||
| namespace Zotlabs\Identity; | ||||
|  | ||||
| class OAuth2Server { | ||||
| class OAuth2Server extends \OAuth2\Server { | ||||
|  | ||||
| 	public $server; | ||||
| 	public function __construct(OAuth2Storage $storage, $config = []) { | ||||
|  | ||||
| 	public function __construct() { | ||||
| 		if(! is_array($config)) { | ||||
| 			$config = [ | ||||
| 				'use_openid_connect' => true, | ||||
| 				'issuer' => \Zotlabs\Lib\System::get_site_name() | ||||
| 			]; | ||||
| 		} | ||||
|  | ||||
| 		$storage = new OAuth2Storage(\DBA::$dba->db); | ||||
|  | ||||
| 		$config = [ | ||||
| 			'use_openid_connect' => true, | ||||
| 			'issuer' => \Zotlabs\Lib\System::get_site_name() | ||||
| 		]; | ||||
|  | ||||
| 		// Pass a storage object or array of storage objects to the OAuth2 server class | ||||
| 		$this->server = new \OAuth2\Server($storage,$config); | ||||
| 		parent::__construct($storage, $config); | ||||
|  | ||||
| 		// Add the "Client Credentials" grant type (it is the simplest of the grant types) | ||||
| 		$this->server->addGrantType(new \OAuth2\GrantType\ClientCredentials($storage)); | ||||
| 		$this->addGrantType(new \OAuth2\GrantType\ClientCredentials($storage)); | ||||
|  | ||||
| 		// Add the "Authorization Code" grant type (this is where the oauth magic happens) | ||||
| 		$this->server->addGrantType(new \OAuth2\GrantType\AuthorizationCode($storage)); | ||||
| 		$this->addGrantType(new \OAuth2\GrantType\AuthorizationCode($storage)); | ||||
|  | ||||
| 		$keyStorage = new \OAuth2\Storage\Memory( [ | ||||
| 			'keys' => [ | ||||
| 				'public_key'  => get_config('system','pubkey'), | ||||
| 				'private_key' => get_config('system','prvkey') | ||||
| 				'public_key'  => get_config('system', 'pubkey'), | ||||
| 				'private_key' => get_config('system', 'prvkey') | ||||
| 			] | ||||
| 		]); | ||||
|  | ||||
| 		$this->server->addStorage($keyStorage,'public_key'); | ||||
|  | ||||
| 		$this->addStorage($keyStorage, 'public_key'); | ||||
| 	} | ||||
|  | ||||
| 	public function get_server() { | ||||
| 		return $this->server; | ||||
| 	}  | ||||
|  | ||||
|  | ||||
| } | ||||
| @@ -2,13 +2,13 @@ | ||||
|  | ||||
| namespace Zotlabs\Module; | ||||
|  | ||||
| use Zotlabs\Identity\OAuth2Storage; | ||||
|  | ||||
|  | ||||
| class Authorize extends \Zotlabs\Web\Controller { | ||||
|  | ||||
|  | ||||
| 	function init() { | ||||
|  | ||||
|  | ||||
| 		// workaround for HTTP-auth in CGI mode | ||||
| 		if (x($_SERVER, 'REDIRECT_REMOTE_USER')) { | ||||
| 			$userpass = base64_decode(substr($_SERVER["REDIRECT_REMOTE_USER"], 6)) ; | ||||
| @@ -28,41 +28,40 @@ class Authorize extends \Zotlabs\Web\Controller { | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		$s = new \Zotlabs\Identity\OAuth2Server(); | ||||
| 		$s = new \Zotlabs\Identity\OAuth2Server(new OAuth2Storage(\DBA::$dba->db)); | ||||
|  | ||||
| 		$request = \OAuth2\Request::createFromGlobals(); | ||||
| 		$response = new \OAuth2\Response(); | ||||
|  | ||||
| 		// validate the authorize request | ||||
| 		if (! $s->server->validateAuthorizeRequest($request, $response)) { | ||||
| 		    $response->send(); | ||||
|     		killme(); | ||||
| 		if (! $s->validateAuthorizeRequest($request, $response)) { | ||||
| 			$response->send(); | ||||
| 			killme(); | ||||
| 		} | ||||
|  | ||||
| 	// display an authorization form | ||||
| 	if (empty($_POST)) { | ||||
| 		// display an authorization form | ||||
| 		if (empty($_POST)) { | ||||
|  | ||||
| 	  return ' | ||||
| 			return ' | ||||
| <form method="post"> | ||||
|   <label>Do You Authorize TestClient?</label><br /> | ||||
|   <input type="submit" name="authorized" value="yes"> | ||||
|   <input type="submit" name="authorized" value="no"> | ||||
| </form>'; | ||||
| 	} | ||||
| 		} | ||||
|  | ||||
| 	// print the authorization code if the user has authorized your client | ||||
| 	$is_authorized = ($_POST['authorized'] === 'yes'); | ||||
| 	$s->server->handleAuthorizeRequest($request, $response, $is_authorized, local_channel()); | ||||
| 	if ($is_authorized) { | ||||
| 		// this is only here so that you get to see your code in the cURL request. Otherwise,  | ||||
| 		// we'd redirect back to the client | ||||
| 		$code = substr($response->getHttpHeader('Location'), strpos($response->getHttpHeader('Location'), 'code=')+5, 40); | ||||
| 		echo("SUCCESS! Authorization Code: $code"); | ||||
| 		// print the authorization code if the user has authorized your client | ||||
| 		$is_authorized = ($_POST['authorized'] === 'yes'); | ||||
| 		$s->handleAuthorizeRequest($request, $response, $is_authorized, local_channel()); | ||||
| 		if ($is_authorized) { | ||||
| 			// this is only here so that you get to see your code in the cURL request. Otherwise, | ||||
| 			// we'd redirect back to the client | ||||
| 			$code = substr($response->getHttpHeader('Location'), strpos($response->getHttpHeader('Location'), 'code=')+5, 40); | ||||
| 			echo("SUCCESS! Authorization Code: $code"); | ||||
| 		} | ||||
|  | ||||
| 	} | ||||
|  | ||||
| 	$response->send(); | ||||
| 	killme(); | ||||
| 		$response->send(); | ||||
| 		killme(); | ||||
| 	} | ||||
|  | ||||
| } | ||||
| @@ -2,6 +2,8 @@ | ||||
|  | ||||
| namespace Zotlabs\Module; | ||||
|  | ||||
| use Zotlabs\Identity\OAuth2Storage; | ||||
|  | ||||
|  | ||||
| class Token extends \Zotlabs\Web\Controller { | ||||
|  | ||||
| @@ -26,9 +28,8 @@ class Token extends \Zotlabs\Web\Controller { | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
|  | ||||
| 		$s = new \Zotlabs\Identity\OAuth2Server(); | ||||
| 		$s->server->handleTokenRequest(\OAuth2\Request::createFromGlobals())->send(); | ||||
| 		$s = new \Zotlabs\Identity\OAuth2Server(new OAuth2Storage(\DBA::$dba->db)); | ||||
| 		$s->handleTokenRequest(\OAuth2\Request::createFromGlobals())->send(); | ||||
|  | ||||
| 		killme(); | ||||
| 	} | ||||
|   | ||||
		Reference in New Issue
	
	Block a user