harukin/core
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor OAuth2Server a bit.

Browse Source
This commit is contained in:
Klaus Weidenbach 2018-02-17 01:49:01 +01:00
parent 547ef772ab
commit 4a5d1076ed
3 changed files with 46 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
Zotlabs/Identity/OAuth2Server.php
View File

@ -2,27 +2,24 @@
namespace Zotlabs\Identity;
class OAuth2Server {
class OAuth2Server extends \OAuth2\Server {
public $server;
public function __construct() {
$storage = new OAuth2Storage(\DBA::$dba->db);
public function __construct(OAuth2Storage $storage, $config = []) {
if(! is_array($config)) {
$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' => [
@ -31,13 +28,7 @@ class OAuth2Server {
]
]);
$this->server->addStorage($keyStorage,'public_key');
$this->addStorage($keyStorage, 'public_key');
}
public function get_server() {
return $this->server;
}
}

11
Zotlabs/Module/Authorize.php
View File

@ -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,13 +28,13 @@ 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)) {
if (! $s->validateAuthorizeRequest($request, $response)) {
$response->send();
killme();
}
@ -52,13 +52,12 @@ class Authorize extends \Zotlabs\Web\Controller {
// 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());
$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();

7
Zotlabs/Module/Token.php
View File

@ -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();
}