more accurate check with support for zero

tedious · May 24, 2015 · c40d2ce · c40d2ce
1 parent 397b748
commit c40d2ce
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/Fetch/Server.php b/src/Fetch/Server.php
@@ -323,13 +323,16 @@ protected function setImapStream()
      */
     public function numMessages( $mailbox='' )
     {
-        $oldMailbox = $this->getMailBox();
-        if( $mailbox !== '' ) {
+        $mboxExists = $this->hasMailbox( $mailbox );
+        if( $mboxExists ) {
+            $oldMailbox = $this->getMailBox();
             $this->setMailbox( $mailbox );
         }
-        $cnt = imap_num_msg($this->getImapStream());
-        $this->setMailbox( $oldMailbox );
-        return $cnt;
+        $cnt = imap_num_msg( $this->getImapStream() );
+        if( $mboxExists ) {
+            $this->setMailbox( $oldMailbox );
+        }
+        return ( !$mboxExists && $mailbox !== '' ) ? 0 : $cnt;
     }
 
     /**