add ReferToken Plugin

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 All notable changes to `ghtk-sdk` will be documented in this file.
 
+## 0.0.2 - 2021-06-29
+
+- initial release
+
 ## 0.0.1 - 2021-06-24
 
 - initial release

README.md

@@ -292,6 +292,13 @@ $client = new \Vanthao03596\GhtkSdk\Client(
 
 $client->authenticate('APITokenSample-ca441e70288cB0515F310742');
 
+// Or use ReferToken plugin
+
+$client = Vanthao03596\GhtkSdk\Client::createWithHttpClient(new Http\Adapter\Guzzle7\Client, true); // true is production mode
+
+$client->authenticate('APITokenSample-ca441e70288cB0515F310742');
+$client->setReferToken('B2CToken-hlsheiwquhrksadlfkjahsdfjaaljh')
+
 ....
 ```
 

src/Client.php

@@ -16,6 +16,7 @@
 use Vanthao03596\GhtkSdk\HttpClient\Builder;
 use Vanthao03596\GhtkSdk\HttpClient\Plugin\Authentication;
 use Vanthao03596\GhtkSdk\HttpClient\Plugin\History;
+use Vanthao03596\GhtkSdk\HttpClient\Plugin\ReferToken;
 
 /**
  * PHP GHTK client.
@@ -129,7 +130,7 @@ public function api($name): AbstractApi
     /**
      * Authenticate a user for all next requests.
      *
-     * @param string      $token GitHub private token
+     * @param string      $token GHTK private token
      *
      * @throws InvalidArgumentException If no token
      *
@@ -214,4 +215,23 @@ protected function getHttpClientBuilder(): Builder
     {
         return $this->httpClientBuilder;
     }
+
+    /**
+     * Set Refer Token for all next requests.
+     *
+     * @param string      $token GHTK private token
+     *
+     * @throws InvalidArgumentException If no token
+     *
+     * @return void
+     */
+    public function setReferToken(string $token): void
+    {
+        if (empty($token)) {
+            throw new InvalidArgumentException('You need token!');
+        }
+
+        $this->getHttpClientBuilder()->removePlugin(ReferToken::class);
+        $this->getHttpClientBuilder()->addPlugin(new ReferToken($token));
+    }
 }

src/HttpClient/Plugin/ReferToken.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Vanthao03596\GhtkSdk\HttpClient\Plugin;
+
+use Http\Client\Common\Plugin;
+use Http\Promise\Promise;
+use Psr\Http\Message\RequestInterface;
+
+final class ReferToken implements Plugin
+{
+    /**
+     * The authorization token.
+     *
+     * @var string
+     */
+    private $token;
+
+    /**
+     * Create a new authentication plugin instance.
+     *
+     * @param string      $token
+     * @return void
+     */
+    public function __construct(string $token)
+    {
+        $this->token = $token;
+    }
+
+    /**
+     * @return Promise
+     */
+    public function handleRequest(RequestInterface $request, callable $next, callable $first): Promise
+    {
+        $request = $request->withHeader('X-Refer-Token', $this->token);
+
+        return $next($request);
+    }
+}