Update parser for Number token compatibility

Author: stevebauman

src/Connection/ImapParser.php

@@ -13,6 +13,7 @@
 use DirectoryTree\ImapEngine\Connection\Tokens\Crlf;
 use DirectoryTree\ImapEngine\Connection\Tokens\ListClose;
 use DirectoryTree\ImapEngine\Connection\Tokens\ListOpen;
+use DirectoryTree\ImapEngine\Connection\Tokens\Number;
 use DirectoryTree\ImapEngine\Connection\Tokens\ResponseCodeClose;
 use DirectoryTree\ImapEngine\Connection\Tokens\ResponseCodeOpen;
 use DirectoryTree\ImapEngine\Connection\Tokens\Token;
@@ -54,8 +55,8 @@ public function next(): Data|Token|Response|null
             return $this->parseList();
         }
 
-        // If the token is an ATOM, check its value for special markers.
-        if ($this->currentToken instanceof Atom) {
+        // If the token is an Atom or Number, check its value for special markers.
+        if ($this->currentToken instanceof Atom || $this->currentToken instanceof Number) {
             // '*' marks an untagged response.
             if ($this->currentToken->value === '*') {
                 return $this->parseUntaggedResponse();

src/Connection/Responses/TaggedResponse.php

@@ -3,13 +3,15 @@
 namespace DirectoryTree\ImapEngine\Connection\Responses;
 
 use DirectoryTree\ImapEngine\Connection\Tokens\Atom;
+use DirectoryTree\ImapEngine\Connection\Tokens\Number;
+use DirectoryTree\ImapEngine\Connection\Tokens\Token;
 
 class TaggedResponse extends Response
 {
     /**
      * Get the response tag.
      */
-    public function tag(): Atom
+    public function tag(): Atom|Number
     {
         return $this->tokens[0];
     }
@@ -25,7 +27,7 @@ public function status(): Atom
     /**
      * Get the response data tokens.
      *
-     * @return Atom[]
+     * @return Token[]
      */
     public function data(): array
     {

src/Connection/Responses/UntaggedResponse.php

@@ -3,13 +3,14 @@
 namespace DirectoryTree\ImapEngine\Connection\Responses;
 
 use DirectoryTree\ImapEngine\Connection\Tokens\Atom;
+use DirectoryTree\ImapEngine\Connection\Tokens\Number;
 
 class UntaggedResponse extends Response
 {
     /**
      * Get the response type token.
      */
-    public function type(): Atom
+    public function type(): Atom|Number
     {
         return $this->tokens[1];
     }