Merge pull request #2 from pjio/main

from pijo/php-imap2: Handle messages fetched by UID
arbor-education · Feb 12, 2023 · aad3c01 · aad3c01
2 parents 89922e8 + 25adff6
commit aad3c01
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/Functions.php b/src/Functions.php
@@ -195,4 +195,22 @@ public static function isRetrofitResource($imap)
     {
         return (class_exists('\IMAP\Connection') && $imap instanceof \IMAP\Connection) || (is_resource($imap) && get_resource_type($imap) == 'imap');
     }
+
+    public static function keyBy(string $name, array $list): array
+    {
+        $keyBy = [];
+        foreach ($list as $item) {
+            if (!isset($item->$name)) {
+                trigger_error('keyBy: key "' . $name . '" not found!', E_USER_WARNING);
+                continue;
+            }
+            if (isset($keyBy[$item->$name])) {
+                trigger_error('keyBy: duplicate key "' . $name . '" = "' . $item->$name . '"', E_USER_WARNING);
+                continue;
+            }
+            $keyBy[$item->$name] = $item;
+        }
+
+        return $keyBy;
+    }
 }
diff --git a/src/Message.php b/src/Message.php
@@ -165,6 +165,10 @@ public static function body($imap, $messageNum, $flags = 0)
 
         $messages = $client->fetch($imap->getMailboxName(), $messageNum, $isUid, ['BODY[TEXT]']);
 
+        if ($isUid && is_array($messages)) {
+            $messages = Functions::keyBy('uid', $messages);
+        }
+
         return $messages[$messageNum]->bodypart['TEXT'];
     }
 
@@ -186,6 +190,10 @@ public static function fetchBody($imap, $messageNum, $section, $flags = 0)
             return false;
         }
 
+        if ($isUid && is_array($messages)) {
+            $messages = Functions::keyBy('uid', $messages);
+        }
+
         if ($section) {
             return $messages[$messageNum]->bodypart[$section];
         }
@@ -217,6 +225,10 @@ public static function fetchMime($imap, $messageNum, $section, $flags = 0)
             return "";
         }
 
+        if ($isUid && is_array($messages)) {
+            $messages = Functions::keyBy('uid', $messages);
+        }
+
         if ($section && isset($messages[$messageNum]->bodypart[$sectionKey])) {
             return $messages[$messageNum]->bodypart[$sectionKey];
         }