1: <?php
  2: 
  3: /**
  4:  * This file is part of the Nette Framework (https://nette.org)
  5:  * Copyright (c) 2004 David Grudl (https://davidgrudl.com)
  6:  */
  7: 
  8: namespace Nette\Mail;
  9: 
 10: use Nette;
 11: use Nette\Utils\Strings;
 12: 
 13: 
 14: /**
 15:  * MIME message part.
 16:  *
 17:  * @author     David Grudl
 18:  *
 19:  * @property-read array $headers
 20:  * @property-write $contentType
 21:  * @property   string $encoding
 22:  * @property   mixed $body
 23:  */
 24: class MimePart extends Nette\Object
 25: {
 26:     /** encoding */
 27:     const ENCODING_BASE64 = 'base64',
 28:         ENCODING_7BIT = '7bit',
 29:         ENCODING_8BIT = '8bit',
 30:         ENCODING_QUOTED_PRINTABLE = 'quoted-printable';
 31: 
 32:     /** @internal */
 33:     const EOL = "\r\n";
 34:     const LINE_LENGTH = 76;
 35: 
 36:     /** @var array */
 37:     private $headers = array();
 38: 
 39:     /** @var array */
 40:     private $parts = array();
 41: 
 42:     /** @var string */
 43:     private $body;
 44: 
 45: 
 46:     /**
 47:      * Sets a header.
 48:      * @param  string
 49:      * @param  string|array  value or pair email => name
 50:      * @param  bool
 51:      * @return self
 52:      */
 53:     public function setHeader($name, $value, $append = FALSE)
 54:     {
 55:         if (!$name || preg_match('#[^a-z0-9-]#i', $name)) {
 56:             throw new Nette\InvalidArgumentException("Header name must be non-empty alphanumeric string, '$name' given.");
 57:         }
 58: 
 59:         if ($value == NULL) { // intentionally ==
 60:             if (!$append) {
 61:                 unset($this->headers[$name]);
 62:             }
 63: 
 64:         } elseif (is_array($value)) { // email
 65:             $tmp = & $this->headers[$name];
 66:             if (!$append || !is_array($tmp)) {
 67:                 $tmp = array();
 68:             }
 69: 
 70:             foreach ($value as $email => $recipient) {
 71:                 if ($recipient !== NULL && !Strings::checkEncoding($recipient)) {
 72:                     Nette\Utils\Validators::assert($recipient, 'unicode', "header '$name'");
 73:                 }
 74:                 if (preg_match('#[\r\n]#', $recipient)) {
 75:                     throw new Nette\InvalidArgumentException('Name must not contain line separator.');
 76:                 }
 77:                 Nette\Utils\Validators::assert($email, 'email', "header '$name'");
 78:                 $tmp[$email] = $recipient;
 79:             }
 80: 
 81:         } else {
 82:             $value = (string) $value;
 83:             if (!Strings::checkEncoding($value)) {
 84:                 throw new Nette\InvalidArgumentException('Header is not valid UTF-8 string.');
 85:             }
 86:             $this->headers[$name] = preg_replace('#[\r\n]+#', ' ', $value);
 87:         }
 88:         return $this;
 89:     }
 90: 
 91: 
 92:     /**
 93:      * Returns a header.
 94:      * @param  string
 95:      * @return mixed
 96:      */
 97:     public function getHeader($name)
 98:     {
 99:         return isset($this->headers[$name]) ? $this->headers[$name] : NULL;
100:     }
101: 
102: 
103:     /**
104:      * Removes a header.
105:      * @param  string
106:      * @return self
107:      */
108:     public function clearHeader($name)
109:     {
110:         unset($this->headers[$name]);
111:         return $this;
112:     }
113: 
114: 
115:     /**
116:      * Returns an encoded header.
117:      * @param  string
118:      * @param  string
119:      * @return string
120:      */
121:     public function getEncodedHeader($name)
122:     {
123:         $offset = strlen($name) + 2; // colon + space
124: 
125:         if (!isset($this->headers[$name])) {
126:             return NULL;
127: 
128:         } elseif (is_array($this->headers[$name])) {
129:             $s = '';
130:             foreach ($this->headers[$name] as $email => $name) {
131:                 if ($name != NULL) { // intentionally ==
132:                     $s .= self::encodeHeader($name, $offset, TRUE);
133:                     $email = " <$email>";
134:                 }
135:                 $s .= self::append($email . ',', $offset);
136:             }
137:             return ltrim(substr($s, 0, -1)); // last comma
138: 
139:         } elseif (preg_match('#^(\S+; (?:file)?name=)"(.*)"\z#', $this->headers[$name], $m)) { // Content-Disposition
140:             $offset += strlen($m[1]);
141:             return $m[1] . '"' . self::encodeHeader($m[2], $offset) . '"';
142: 
143:         } else {
144:             return ltrim(self::encodeHeader($this->headers[$name], $offset));
145:         }
146:     }
147: 
148: 
149:     /**
150:      * Returns all headers.
151:      * @return array
152:      */
153:     public function getHeaders()
154:     {
155:         return $this->headers;
156:     }
157: 
158: 
159:     /**
160:      * Sets Content-Type header.
161:      * @param  string
162:      * @param  string
163:      * @return self
164:      */
165:     public function setContentType($contentType, $charset = NULL)
166:     {
167:         $this->setHeader('Content-Type', $contentType . ($charset ? "; charset=$charset" : ''));
168:         return $this;
169:     }
170: 
171: 
172:     /**
173:      * Sets Content-Transfer-Encoding header.
174:      * @param  string
175:      * @return self
176:      */
177:     public function setEncoding($encoding)
178:     {
179:         $this->setHeader('Content-Transfer-Encoding', $encoding);
180:         return $this;
181:     }
182: 
183: 
184:     /**
185:      * Returns Content-Transfer-Encoding header.
186:      * @return string
187:      */
188:     public function getEncoding()
189:     {
190:         return $this->getHeader('Content-Transfer-Encoding');
191:     }
192: 
193: 
194:     /**
195:      * Adds or creates new multipart.
196:      * @return MimePart
197:      */
198:     public function addPart(MimePart $part = NULL)
199:     {
200:         return $this->parts[] = $part === NULL ? new self : $part;
201:     }
202: 
203: 
204:     /**
205:      * Sets textual body.
206:      * @return self
207:      */
208:     public function setBody($body)
209:     {
210:         if ($body instanceof Nette\Templating\ITemplate) {
211:             $body->mail = $this;
212:             $body = $body->__toString(TRUE);
213:         }
214:         $this->body = $body;
215:         return $this;
216:     }
217: 
218: 
219:     /**
220:      * Gets textual body.
221:      * @return mixed
222:      */
223:     public function getBody()
224:     {
225:         return $this->body;
226:     }
227: 
228: 
229:     /********************* building ****************d*g**/
230: 
231: 
232:     /**
233:      * Returns encoded message.
234:      * @return string
235:      */
236:     public function getEncodedMessage()
237:     {
238:         $output = '';
239:         $boundary = '--------' . Strings::random();
240: 
241:         foreach ($this->headers as $name => $value) {
242:             $output .= $name . ': ' . $this->getEncodedHeader($name);
243:             if ($this->parts && $name === 'Content-Type') {
244:                 $output .= ';' . self::EOL . "\tboundary=\"$boundary\"";
245:             }
246:             $output .= self::EOL;
247:         }
248:         $output .= self::EOL;
249: 
250:         $body = (string) $this->body;
251:         if ($body !== '') {
252:             switch ($this->getEncoding()) {
253:                 case self::ENCODING_QUOTED_PRINTABLE:
254:                     $output .= quoted_printable_encode($body);
255:                     break;
256: 
257:                 case self::ENCODING_BASE64:
258:                     $output .= rtrim(chunk_split(base64_encode($body), self::LINE_LENGTH, self::EOL));
259:                     break;
260: 
261:                 case self::ENCODING_7BIT:
262:                     $body = preg_replace('#[\x80-\xFF]+#', '', $body);
263:                     // break intentionally omitted
264: 
265:                 case self::ENCODING_8BIT:
266:                     $body = str_replace(array("\x00", "\r"), '', $body);
267:                     $body = str_replace("\n", self::EOL, $body);
268:                     $output .= $body;
269:                     break;
270: 
271:                 default:
272:                     throw new Nette\InvalidStateException('Unknown encoding.');
273:             }
274:         }
275: 
276:         if ($this->parts) {
277:             if (substr($output, -strlen(self::EOL)) !== self::EOL) {
278:                 $output .= self::EOL;
279:             }
280:             foreach ($this->parts as $part) {
281:                 $output .= '--' . $boundary . self::EOL . $part->getEncodedMessage() . self::EOL;
282:             }
283:             $output .= '--' . $boundary.'--';
284:         }
285: 
286:         return $output;
287:     }
288: 
289: 
290:     /********************* QuotedPrintable helpers ****************d*g**/
291: 
292: 
293:     /**
294:      * Converts a 8 bit header to a string.
295:      * @param  string
296:      * @param  int
297:      * @param  bool
298:      * @return string
299:      */
300:     private static function encodeHeader($s, & $offset = 0, $quotes = FALSE)
301:     {
302:         if (strspn($s, "!\"#$%&\'()*+,-./0123456789:;<>@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^`abcdefghijklmnopqrstuvwxyz{|}~=? _\r\n\t") === strlen($s)) {
303:             if ($quotes && preg_match('#[^ a-zA-Z0-9!\#$%&\'*+/?^_`{|}~-]#', $s)) { // RFC 2822 atext except =
304:                 return self::append('"' . addcslashes($s, '"\\') . '"', $offset);
305:             }
306:             return self::append($s, $offset);
307:         }
308: 
309:         $o = '';
310:         if ($offset >= 55) { // maximum for iconv_mime_encode
311:             $o = self::EOL . "\t";
312:             $offset = 1;
313:         }
314: 
315:         $s = iconv_mime_encode(str_repeat(' ', $old = $offset), $s, array(
316:             'scheme' => 'B', // Q is broken
317:             'input-charset' => 'UTF-8',
318:             'output-charset' => 'UTF-8',
319:         ));
320: 
321:         $offset = strlen($s) - strrpos($s, "\n");
322:         $s = str_replace("\n ", "\n\t", substr($s, $old + 2)); // adds ': '
323:         return $o . $s;
324:     }
325: 
326: 
327:     private static function append($s, & $offset = 0)
328:     {
329:         if ($offset + strlen($s) > self::LINE_LENGTH) {
330:             $offset = 1;
331:             $s = self::EOL . "\t" . $s;
332:         }
333:         $offset += strlen($s);
334:         return $s;
335:     }
336: 
337: }
338: