allow setFlags() to set multiple flags as an array

This commit is contained in:
Geoffray 2014-11-01 09:11:38 +01:00
parent 4027b05fc6
commit 56983f0365

View File

@ -677,27 +677,36 @@ class Message
} }
/** /**
* This function is used to enable or disable a flag on the imap message. * This function is used to enable or disable one or more flags on the imap message.
* *
* @param string $flag Flagged, Answered, Deleted, Seen, Draft * @param string|array $flag Flagged, Answered, Deleted, Seen, Draft
* @param bool $enable * @param bool $enable
* @throws \InvalidArgumentException * @throws \InvalidArgumentException
* @return bool * @return bool
*/ */
public function setFlag($flag, $enable = true) public function setFlag($flag, $enable = true)
{ {
if (!in_array($flag, self::$flagTypes) || $flag == self::FLAG_RECENT) $flags = (is_array($flag)) ? $flag : array($flag);
throw new \InvalidArgumentException('Unable to set invalid flag "' . $flag . '"');
$imapifiedFlag = '\\' . ucfirst($flag); foreach ($flags as $i => $flag) {
$flag = ltrim(strtolower($flag), '\\');
if (!in_array($flag, self::$flagTypes) || $flag == self::FLAG_RECENT)
throw new \InvalidArgumentException('Unable to set invalid flag "' . $flag . '"');
if ($enable) {
$this->status[$flag] = true;
} else {
unset($this->status[$flag]);
}
$flags[$i] = $flag;
}
$imapifiedFlag = '\\'.implode(' \\', array_map('ucfirst', $flags));
if ($enable === true) { if ($enable === true) {
$this->status[$flag] = true;
return imap_setflag_full($this->imapStream, $this->uid, $imapifiedFlag, ST_UID); return imap_setflag_full($this->imapStream, $this->uid, $imapifiedFlag, ST_UID);
} else { } else {
unset($this->status[$flag]);
return imap_clearflag_full($this->imapStream, $this->uid, $imapifiedFlag, ST_UID); return imap_clearflag_full($this->imapStream, $this->uid, $imapifiedFlag, ST_UID);
} }
} }