315 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			315 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| 
 | |
| /*
 | |
|  * This file is part of the Symfony package.
 | |
|  *
 | |
|  * (c) Fabien Potencier <fabien@symfony.com>
 | |
|  *
 | |
|  * For the full copyright and license information, please view the LICENSE
 | |
|  * file that was distributed with this source code.
 | |
|  */
 | |
| 
 | |
| namespace Symfony\Component\Mime\Header;
 | |
| 
 | |
| use Symfony\Component\Mime\Address;
 | |
| use Symfony\Component\Mime\Exception\LogicException;
 | |
| 
 | |
| /**
 | |
|  * A collection of headers.
 | |
|  *
 | |
|  * @author Fabien Potencier <fabien@symfony.com>
 | |
|  */
 | |
| final class Headers
 | |
| {
 | |
|     private const UNIQUE_HEADERS = [
 | |
|         'date', 'from', 'sender', 'reply-to', 'to', 'cc', 'bcc',
 | |
|         'message-id', 'in-reply-to', 'references', 'subject',
 | |
|     ];
 | |
|     private const HEADER_CLASS_MAP = [
 | |
|         'date' => DateHeader::class,
 | |
|         'from' => MailboxListHeader::class,
 | |
|         'sender' => MailboxHeader::class,
 | |
|         'reply-to' => MailboxListHeader::class,
 | |
|         'to' => MailboxListHeader::class,
 | |
|         'cc' => MailboxListHeader::class,
 | |
|         'bcc' => MailboxListHeader::class,
 | |
|         'message-id' => IdentificationHeader::class,
 | |
|         'in-reply-to' => [UnstructuredHeader::class, IdentificationHeader::class], // `In-Reply-To` and `References` are less strict than RFC 2822 (3.6.4) to allow users entering the original email's ...
 | |
|         'references' => [UnstructuredHeader::class, IdentificationHeader::class], // ... `Message-ID`, even if that is no valid `msg-id`
 | |
|         'return-path' => PathHeader::class,
 | |
|     ];
 | |
| 
 | |
|     /**
 | |
|      * @var HeaderInterface[][]
 | |
|      */
 | |
|     private array $headers = [];
 | |
|     private int $lineLength = 76;
 | |
| 
 | |
|     public function __construct(HeaderInterface ...$headers)
 | |
|     {
 | |
|         foreach ($headers as $header) {
 | |
|             $this->add($header);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public function __clone()
 | |
|     {
 | |
|         foreach ($this->headers as $name => $collection) {
 | |
|             foreach ($collection as $i => $header) {
 | |
|                 $this->headers[$name][$i] = clone $header;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public function setMaxLineLength(int $lineLength)
 | |
|     {
 | |
|         $this->lineLength = $lineLength;
 | |
|         foreach ($this->all() as $header) {
 | |
|             $header->setMaxLineLength($lineLength);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public function getMaxLineLength(): int
 | |
|     {
 | |
|         return $this->lineLength;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @param array<Address|string> $addresses
 | |
|      *
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addMailboxListHeader(string $name, array $addresses): static
 | |
|     {
 | |
|         return $this->add(new MailboxListHeader($name, Address::createArray($addresses)));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addMailboxHeader(string $name, Address|string $address): static
 | |
|     {
 | |
|         return $this->add(new MailboxHeader($name, Address::create($address)));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addIdHeader(string $name, string|array $ids): static
 | |
|     {
 | |
|         return $this->add(new IdentificationHeader($name, $ids));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addPathHeader(string $name, Address|string $path): static
 | |
|     {
 | |
|         return $this->add(new PathHeader($name, $path instanceof Address ? $path : new Address($path)));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addDateHeader(string $name, \DateTimeInterface $dateTime): static
 | |
|     {
 | |
|         return $this->add(new DateHeader($name, $dateTime));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addTextHeader(string $name, string $value): static
 | |
|     {
 | |
|         return $this->add(new UnstructuredHeader($name, $value));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addParameterizedHeader(string $name, string $value, array $params = []): static
 | |
|     {
 | |
|         return $this->add(new ParameterizedHeader($name, $value, $params));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function addHeader(string $name, mixed $argument, array $more = []): static
 | |
|     {
 | |
|         $headerClass = self::HEADER_CLASS_MAP[strtolower($name)] ?? UnstructuredHeader::class;
 | |
|         if (\is_array($headerClass)) {
 | |
|             $headerClass = $headerClass[0];
 | |
|         }
 | |
|         $parts = explode('\\', $headerClass);
 | |
|         $method = 'add'.ucfirst(array_pop($parts));
 | |
|         if ('addUnstructuredHeader' === $method) {
 | |
|             $method = 'addTextHeader';
 | |
|         } elseif ('addIdentificationHeader' === $method) {
 | |
|             $method = 'addIdHeader';
 | |
|         }
 | |
| 
 | |
|         return $this->$method($name, $argument, $more);
 | |
|     }
 | |
| 
 | |
|     public function has(string $name): bool
 | |
|     {
 | |
|         return isset($this->headers[strtolower($name)]);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @return $this
 | |
|      */
 | |
|     public function add(HeaderInterface $header): static
 | |
|     {
 | |
|         self::checkHeaderClass($header);
 | |
| 
 | |
|         $header->setMaxLineLength($this->lineLength);
 | |
|         $name = strtolower($header->getName());
 | |
| 
 | |
|         if (\in_array($name, self::UNIQUE_HEADERS, true) && isset($this->headers[$name]) && \count($this->headers[$name]) > 0) {
 | |
|             throw new LogicException(sprintf('Impossible to set header "%s" as it\'s already defined and must be unique.', $header->getName()));
 | |
|         }
 | |
| 
 | |
|         $this->headers[$name][] = $header;
 | |
| 
 | |
|         return $this;
 | |
|     }
 | |
| 
 | |
|     public function get(string $name): ?HeaderInterface
 | |
|     {
 | |
|         $name = strtolower($name);
 | |
|         if (!isset($this->headers[$name])) {
 | |
|             return null;
 | |
|         }
 | |
| 
 | |
|         $values = array_values($this->headers[$name]);
 | |
| 
 | |
|         return array_shift($values);
 | |
|     }
 | |
| 
 | |
|     public function all(string $name = null): iterable
 | |
|     {
 | |
|         if (null === $name) {
 | |
|             foreach ($this->headers as $name => $collection) {
 | |
|                 foreach ($collection as $header) {
 | |
|                     yield $name => $header;
 | |
|                 }
 | |
|             }
 | |
|         } elseif (isset($this->headers[strtolower($name)])) {
 | |
|             foreach ($this->headers[strtolower($name)] as $header) {
 | |
|                 yield $header;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public function getNames(): array
 | |
|     {
 | |
|         return array_keys($this->headers);
 | |
|     }
 | |
| 
 | |
|     public function remove(string $name): void
 | |
|     {
 | |
|         unset($this->headers[strtolower($name)]);
 | |
|     }
 | |
| 
 | |
|     public static function isUniqueHeader(string $name): bool
 | |
|     {
 | |
|         return \in_array(strtolower($name), self::UNIQUE_HEADERS, true);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @throws LogicException if the header name and class are not compatible
 | |
|      */
 | |
|     public static function checkHeaderClass(HeaderInterface $header): void
 | |
|     {
 | |
|         $name = strtolower($header->getName());
 | |
|         $headerClasses = self::HEADER_CLASS_MAP[$name] ?? [];
 | |
|         if (!\is_array($headerClasses)) {
 | |
|             $headerClasses = [$headerClasses];
 | |
|         }
 | |
| 
 | |
|         if (!$headerClasses) {
 | |
|             return;
 | |
|         }
 | |
| 
 | |
|         foreach ($headerClasses as $c) {
 | |
|             if ($header instanceof $c) {
 | |
|                 return;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         throw new LogicException(sprintf('The "%s" header must be an instance of "%s" (got "%s").', $header->getName(), implode('" or "', $headerClasses), get_debug_type($header)));
 | |
|     }
 | |
| 
 | |
|     public function toString(): string
 | |
|     {
 | |
|         $string = '';
 | |
|         foreach ($this->toArray() as $str) {
 | |
|             $string .= $str."\r\n";
 | |
|         }
 | |
| 
 | |
|         return $string;
 | |
|     }
 | |
| 
 | |
|     public function toArray(): array
 | |
|     {
 | |
|         $arr = [];
 | |
|         foreach ($this->all() as $header) {
 | |
|             if ('' !== $header->getBodyAsString()) {
 | |
|                 $arr[] = $header->toString();
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         return $arr;
 | |
|     }
 | |
| 
 | |
|     public function getHeaderBody(string $name)
 | |
|     {
 | |
|         return $this->has($name) ? $this->get($name)->getBody() : null;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @internal
 | |
|      */
 | |
|     public function setHeaderBody(string $type, string $name, mixed $body): void
 | |
|     {
 | |
|         if ($this->has($name)) {
 | |
|             $this->get($name)->setBody($body);
 | |
|         } else {
 | |
|             $this->{'add'.$type.'Header'}($name, $body);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public function getHeaderParameter(string $name, string $parameter): ?string
 | |
|     {
 | |
|         if (!$this->has($name)) {
 | |
|             return null;
 | |
|         }
 | |
| 
 | |
|         $header = $this->get($name);
 | |
|         if (!$header instanceof ParameterizedHeader) {
 | |
|             throw new LogicException(sprintf('Unable to get parameter "%s" on header "%s" as the header is not of class "%s".', $parameter, $name, ParameterizedHeader::class));
 | |
|         }
 | |
| 
 | |
|         return $header->getParameter($parameter);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @internal
 | |
|      */
 | |
|     public function setHeaderParameter(string $name, string $parameter, ?string $value): void
 | |
|     {
 | |
|         if (!$this->has($name)) {
 | |
|             throw new LogicException(sprintf('Unable to set parameter "%s" on header "%s" as the header is not defined.', $parameter, $name));
 | |
|         }
 | |
| 
 | |
|         $header = $this->get($name);
 | |
|         if (!$header instanceof ParameterizedHeader) {
 | |
|             throw new LogicException(sprintf('Unable to set parameter "%s" on header "%s" as the header is not of class "%s".', $parameter, $name, ParameterizedHeader::class));
 | |
|         }
 | |
| 
 | |
|         $header->setParameter($parameter, $value);
 | |
|     }
 | |
| }
 | 
