Allow providing own driver for REST

dc-Ragnarok · Mar 19, 2023 · 1709cc9 · 1709cc9
1 parent 8dce8a9
commit 1709cc9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 102 deletions.
diff --git a/src/CommandHandler.php b/src/CommandHandler.php
diff --git a/src/Discord.php b/src/Discord.php
@@ -4,6 +4,7 @@
 
 namespace Exan\Fenrir;
 
+use Discord\Http\DriverInterface;
 use Discord\Http\Drivers\Guzzle;
 use Discord\Http\Http;
 use Exan\Fenrir\Bitwise\Bitwise;
@@ -55,15 +56,18 @@ public function withGateway(
         return $this;
     }
 
-    public function withRest()
-    {
+    public function withRest(
+        DriverInterface $driver = null,
+    ) {
+        $driver ??= new Guzzle(
+            $this->loop
+        );
+
         $this->http = new Http(
             'Bot ' . $this->token,
             $this->loop,
             $this->logger,
-            new Guzzle(
-                $this->loop
-            )
+            $driver
         );
 
         $this->rest = new Rest($this->http, $this->mapper);