123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215 |
- <?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\Part;
- use Symfony\Component\Mime\Encoder\Base64ContentEncoder;
- use Symfony\Component\Mime\Encoder\ContentEncoderInterface;
- use Symfony\Component\Mime\Encoder\EightBitContentEncoder;
- use Symfony\Component\Mime\Encoder\QpContentEncoder;
- use Symfony\Component\Mime\Exception\InvalidArgumentException;
- use Symfony\Component\Mime\Header\Headers;
- /**
- * @author Fabien Potencier <fabien@symfony.com>
- */
- class TextPart extends AbstractPart
- {
- /** @internal */
- protected $_headers;
- private static $encoders = [];
- private $body;
- private $charset;
- private $subtype;
- /**
- * @var ?string
- */
- private $disposition;
- private $name;
- private $encoding;
- private $seekable;
- /**
- * @param resource|string $body
- */
- public function __construct($body, ?string $charset = 'utf-8', string $subtype = 'plain', string $encoding = null)
- {
- unset($this->_headers);
- parent::__construct();
- if (!\is_string($body) && !\is_resource($body)) {
- throw new \TypeError(sprintf('The body of "%s" must be a string or a resource (got "%s").', self::class, get_debug_type($body)));
- }
- $this->body = $body;
- $this->charset = $charset;
- $this->subtype = $subtype;
- $this->seekable = \is_resource($body) ? stream_get_meta_data($body)['seekable'] && 0 === fseek($body, 0, \SEEK_CUR) : null;
- if (null === $encoding) {
- $this->encoding = $this->chooseEncoding();
- } else {
- if ('quoted-printable' !== $encoding && 'base64' !== $encoding && '8bit' !== $encoding) {
- throw new InvalidArgumentException(sprintf('The encoding must be one of "quoted-printable", "base64", or "8bit" ("%s" given).', $encoding));
- }
- $this->encoding = $encoding;
- }
- }
- public function getMediaType(): string
- {
- return 'text';
- }
- public function getMediaSubtype(): string
- {
- return $this->subtype;
- }
- /**
- * @param string $disposition one of attachment, inline, or form-data
- *
- * @return $this
- */
- public function setDisposition(string $disposition)
- {
- $this->disposition = $disposition;
- return $this;
- }
- /**
- * Sets the name of the file (used by FormDataPart).
- *
- * @return $this
- */
- public function setName(string $name)
- {
- $this->name = $name;
- return $this;
- }
- public function getBody(): string
- {
- if (null === $this->seekable) {
- return $this->body;
- }
- if ($this->seekable) {
- rewind($this->body);
- }
- return stream_get_contents($this->body) ?: '';
- }
- public function bodyToString(): string
- {
- return $this->getEncoder()->encodeString($this->getBody(), $this->charset);
- }
- public function bodyToIterable(): iterable
- {
- if (null !== $this->seekable) {
- if ($this->seekable) {
- rewind($this->body);
- }
- yield from $this->getEncoder()->encodeByteStream($this->body);
- } else {
- yield $this->getEncoder()->encodeString($this->body);
- }
- }
- public function getPreparedHeaders(): Headers
- {
- $headers = parent::getPreparedHeaders();
- $headers->setHeaderBody('Parameterized', 'Content-Type', $this->getMediaType().'/'.$this->getMediaSubtype());
- if ($this->charset) {
- $headers->setHeaderParameter('Content-Type', 'charset', $this->charset);
- }
- if ($this->name && 'form-data' !== $this->disposition) {
- $headers->setHeaderParameter('Content-Type', 'name', $this->name);
- }
- $headers->setHeaderBody('Text', 'Content-Transfer-Encoding', $this->encoding);
- if (!$headers->has('Content-Disposition') && null !== $this->disposition) {
- $headers->setHeaderBody('Parameterized', 'Content-Disposition', $this->disposition);
- if ($this->name) {
- $headers->setHeaderParameter('Content-Disposition', 'name', $this->name);
- }
- }
- return $headers;
- }
- public function asDebugString(): string
- {
- $str = parent::asDebugString();
- if (null !== $this->charset) {
- $str .= ' charset: '.$this->charset;
- }
- if (null !== $this->disposition) {
- $str .= ' disposition: '.$this->disposition;
- }
- return $str;
- }
- private function getEncoder(): ContentEncoderInterface
- {
- if ('8bit' === $this->encoding) {
- return self::$encoders[$this->encoding] ?? (self::$encoders[$this->encoding] = new EightBitContentEncoder());
- }
- if ('quoted-printable' === $this->encoding) {
- return self::$encoders[$this->encoding] ?? (self::$encoders[$this->encoding] = new QpContentEncoder());
- }
- return self::$encoders[$this->encoding] ?? (self::$encoders[$this->encoding] = new Base64ContentEncoder());
- }
- private function chooseEncoding(): string
- {
- if (null === $this->charset) {
- return 'base64';
- }
- return 'quoted-printable';
- }
- /**
- * @return array
- */
- public function __sleep()
- {
- // convert resources to strings for serialization
- if (null !== $this->seekable) {
- $this->body = $this->getBody();
- }
- $this->_headers = $this->getHeaders();
- return ['_headers', 'body', 'charset', 'subtype', 'disposition', 'name', 'encoding'];
- }
- public function __wakeup()
- {
- $r = new \ReflectionProperty(AbstractPart::class, 'headers');
- $r->setAccessible(true);
- $r->setValue($this, $this->_headers);
- unset($this->_headers);
- }
- }
|