Added CookieJar support to ArtaxHttpOptions.

Porter-connectors · Apr 16, 2018 · 04a2174 · 04a2174
1 parent 339ea83
commit 04a2174
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/src/ArtaxHttpOptions.php b/src/ArtaxHttpOptions.php
@@ -4,13 +4,26 @@
 namespace ScriptFUSION\Porter\Net\Http;
 
 use Amp\Artax\Client;
+use Amp\Artax\Cookie\CookieJar;
 use ScriptFUSION\Porter\Options\EncapsulatedOptions;
 
 /**
  * Encapsulates Artax HTTP client options.
  */
 final class ArtaxHttpOptions extends EncapsulatedOptions
 {
+    private $cookieJar;
+
+    public function __construct(CookieJar $cookieJar)
+    {
+        $this->cookieJar = $cookieJar;
+    }
+
+    public function __clone()
+    {
+        $this->cookieJar = clone $this->cookieJar;
+    }
+
     public function setAutoEncoding(bool $autoEncoding): self
     {
         return $this->set(Client::OP_AUTO_ENCODING, $autoEncoding);
@@ -51,6 +64,11 @@ public function setMaxBodyBytes(int $maxBodyBytes): self
         return $this->set(Client::OP_MAX_BODY_BYTES, $maxBodyBytes);
     }
 
+    public function getCookieJar(): CookieJar
+    {
+        return $this->cookieJar;
+    }
+
     public function extractArtaxOptions(): array
     {
         return $this->copy();

diff --git a/src/AsyncHttpConnector.php b/src/AsyncHttpConnector.php
@@ -3,6 +3,7 @@
 
 namespace ScriptFUSION\Porter\Net\Http;
 
+use Amp\Artax\Cookie\ArrayCookieJar;
 use Amp\Artax\DefaultClient;
 use Amp\Artax\Response;
 use Amp\Promise;
@@ -14,9 +15,11 @@ class AsyncHttpConnector implements AsyncConnector, ConnectorOptions
 {
     private $options;
 
+    private $cookieJar;
+
     public function __construct(ArtaxHttpOptions $options = null)
     {
-        $this->options = $options ?: new ArtaxHttpOptions;
+        $this->options = $options ?: new ArtaxHttpOptions($this->cookieJar = new ArrayCookieJar);
     }
 
     public function __clone()
@@ -27,7 +30,7 @@ public function __clone()
     public function fetchAsync(ConnectionContext $context, string $source): Promise
     {
         return \Amp\call(function () use ($context, $source) {
-            $client = new DefaultClient;
+            $client = new DefaultClient($this->cookieJar);
             $client->setOptions($this->getOptions()->extractArtaxOptions());
 
             return $context->retry(static function () use ($client, $source) {