Use PSR-4 loading for normal files

Signed-off-by: Joas Schilling <coding@schilljs.com>

1 files changed, 502 insertions, 0 deletions

diff --git a/lib/Controller/MessagesController.php b/lib/Controller/MessagesController.php
new file mode 100755
index 000000000..72951665f
--- /dev/null
+++ b/lib/Controller/MessagesController.php
@@ -0,0 +1,502 @@
+<?php
+
+/**
+ * @author Alexander Weidinger <alexwegoo@gmail.com>
+ * @author Christoph Wurst <christoph@winzerhof-wurst.at>
+ * @author Christoph Wurst <wurst.christoph@gmail.com>
+ * @author Jakob Sack <jakob@owncloud.org>
+ * @author Jan-Christoph Borchardt <hey@jancborchardt.net>
+ * @author Lukas Reschke <lukas@owncloud.com>
+ * @author Thomas Imbreckx <zinks@iozero.be>
+ * @author Thomas Müller <thomas.mueller@tmit.eu>
+ *
+ * Mail
+ *
+ * This code is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License, version 3,
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>
+ *
+ */
+
+namespace OCA\Mail\Controller;
+
+use OCA\Mail\Http\AttachmentDownloadResponse;
+use OCA\Mail\Http\HtmlResponse;
+use OCA\Mail\Service\AccountService;
+use OCA\Mail\Service\ContactsIntegration;
+use OCA\Mail\Service\IAccount;
+use OCA\Mail\Service\IMailBox;
+use OCA\Mail\Service\Logger;
+use OCA\Mail\Service\UnifiedAccount;
+use OCP\AppFramework\Controller;
+use OCP\AppFramework\Db\DoesNotExistException;
+use OCP\AppFramework\Http;
+use OCP\AppFramework\Http\ContentSecurityPolicy;
+use OCP\AppFramework\Http\JSONResponse;
+use OCP\AppFramework\Http\TemplateResponse;
+use OCP\Files\Folder;
+use OCP\Files\IMimeTypeDetector;
+use OCP\IL10N;
+use OCP\IRequest;
+use OCP\IURLGenerator;
+use OCP\Util;
+
+class MessagesController extends Controller {
+
+
+	/** @var AccountService */
+	private $accountService;
+
+	/** @var string */
+	private $currentUserId;
+
+	/** @var ContactsIntegration */
+	private $contactsIntegration;
+
+	/** @var Logger */
+	private $logger;
+
+	/** @var Folder */
+	private $userFolder;
+
+	/** @var IMimeTypeDetector */
+	private $mimeTypeDetector;
+
+	/** @var IL10N */
+	private $l10n;
+
+	/** @var IURLGenerator */
+	private $urlGenerator;
+
+	/** @var IAccount[] */
+	private $accounts = [];
+
+	/**
+	 * @param string $appName
+	 * @param IRequest $request
+	 * @param AccountService $accountService
+	 * @param string $UserId
+	 * @param $userFolder
+	 * @param ContactsIntegration $contactsIntegration
+	 * @param Logger $logger
+	 * @param IL10N $l10n
+	 * @param IMimeTypeDetector $mimeTypeDetector
+	 * @param IURLGenerator $urlGenerator
+	 */
+	public function __construct($appName,
+								IRequest $request,
+								AccountService $accountService,
+								$UserId,
+								$userFolder,
+								ContactsIntegration $contactsIntegration,
+								Logger $logger,
+								IL10N $l10n,
+								IMimeTypeDetector $mimeTypeDetector,
+								IURLGenerator $urlGenerator) {
+		parent::__construct($appName, $request);
+		$this->accountService = $accountService;
+		$this->currentUserId = $UserId;
+		$this->userFolder = $userFolder;
+		$this->contactsIntegration = $contactsIntegration;
+		$this->logger = $logger;
+		$this->l10n = $l10n;
+		$this->mimeTypeDetector = $mimeTypeDetector;
+		$this->urlGenerator = $urlGenerator;
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @NoCSRFRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param int $from
+	 * @param int $to
+	 * @param string $filter
+	 * @param array $ids
+	 * @return JSONResponse
+	 */
+	public function index($accountId, $folderId, $from=0, $to=20, $filter=null, $ids=null) {
+		if (!is_null($ids)) {
+			$ids = explode(',', $ids);
+
+			return $this->loadMultiple($accountId, $folderId, $ids);
+		}
+		$mailBox = $this->getFolder($accountId, $folderId);
+
+		$this->logger->debug("loading messages $from to $to of folder <$folderId>");
+
+		$json = $mailBox->getMessages($from, $to-$from+1, $filter);
+
+		$ci = $this->contactsIntegration;
+		$json = array_map(function($j) use ($ci, $mailBox) {
+			if ($mailBox->getSpecialRole() === 'trash') {
+				$j['delete'] = (string)$this->l10n->t('Delete permanently');
+			}
+
+			if ($mailBox->getSpecialRole() === 'sent') {
+				$j['fromEmail'] = $j['toEmail'];
+				$j['from'] = $j['to'];
+				if((count($j['toList']) > 1) || (count($j['ccList']) > 0)) {
+					$j['from'] .= ' ' . $this->l10n->t('& others');
+				}
+			}
+
+			$j['senderImage'] = $ci->getPhoto($j['fromEmail']);
+			return $j;
+		}, $json);
+
+		return new JSONResponse($json);
+	}
+
+	/**
+	 * @param integer $accountId
+	 * @param string $folderId
+	 */
+	private function loadMessage($accountId, $folderId, $id) {
+		$account = $this->getAccount($accountId);
+		$mailBox = $account->getMailbox(base64_decode($folderId));
+		$m = $mailBox->getMessage($id);
+
+		$json = $this->enhanceMessage($accountId, $folderId, $id, $m, $account, $mailBox);
+
+		// Unified inbox hack
+		$messageId = $id;
+		if ($accountId === UnifiedAccount::ID) {
+			// Add accountId, folderId for unified inbox replies
+			list($accountId, $messageId) = json_decode(base64_decode($id));
+			$account = $this->getAccount($accountId);
+			$inbox = $account->getInbox();
+			$folderId = base64_encode($inbox->getFolderId());
+		}
+		$json['messageId'] = $messageId;
+		$json['accountId'] = $accountId;
+		$json['folderId'] = $folderId;
+		// End unified inbox hack
+
+		return $json;
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @NoCSRFRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param mixed $id
+	 * @return JSONResponse
+	 */
+	public function show($accountId, $folderId, $id) {
+		try {
+			$json = $this->loadMessage($accountId, $folderId, $id);
+		} catch (DoesNotExistException $ex) {
+			return new JSONResponse([], 404);
+		}
+		return new JSONResponse($json);
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @NoCSRFRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param string $messageId
+	 * @return HtmlResponse|TemplateResponse
+	 */
+	public function getHtmlBody($accountId, $folderId, $messageId) {
+		try {
+			$mailBox = $this->getFolder($accountId, $folderId);
+
+			$m = $mailBox->getMessage($messageId, true);
+			$html = $m->getHtmlBody($accountId, $folderId, $messageId, function($cid) use ($m){
+				$match = array_filter($m->attachments, function($a) use($cid){
+					return $a['cid'] === $cid;
+				});
+				$match = array_shift($match);
+				if (is_null($match)) {
+					return null;
+				}
+				return $match['id'];
+			});
+
+			$htmlResponse = new HtmlResponse($html);
+
+			// Harden the default security policy
+			$policy = new ContentSecurityPolicy();
+			$policy->allowEvalScript(false);
+			$policy->disallowScriptDomain('\'self\'');
+			$policy->disallowConnectDomain('\'self\'');
+			$policy->disallowFontDomain('\'self\'');
+			$policy->disallowMediaDomain('\'self\'');
+			$htmlResponse->setContentSecurityPolicy($policy);
+
+			// Enable caching
+			$htmlResponse->cacheFor(60 * 60);
+			$htmlResponse->addHeader('Pragma', 'cache');
+
+			return $htmlResponse;
+		} catch(\Exception $ex) {
+			return new TemplateResponse($this->appName, 'error', ['message' => $ex->getMessage()], 'none');
+		}
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @NoCSRFRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param string $messageId
+	 * @param string $attachmentId
+	 * @return AttachmentDownloadResponse
+	 */
+	public function downloadAttachment($accountId, $folderId, $messageId, $attachmentId) {
+		$mailBox = $this->getFolder($accountId, $folderId);
+
+		$attachment = $mailBox->getAttachment($messageId, $attachmentId);
+
+		return new AttachmentDownloadResponse(
+			$attachment->getContents(),
+			$attachment->getName(),
+			$attachment->getType());
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @NoCSRFRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param string $messageId
+	 * @param int $attachmentId
+	 * @param string $targetPath
+	 * @return JSONResponse
+	 */
+	public function saveAttachment($accountId, $folderId, $messageId, $attachmentId, $targetPath) {
+		$mailBox = $this->getFolder($accountId, $folderId);
+
+		$attachmentIds = [];
+		if($attachmentId === 0) {
+			// Save all attachments
+			$m = $mailBox->getMessage($messageId);
+			$attachmentIds = array_map(function($a){
+				return $a['id'];
+			}, $m->attachments);
+		} else {
+			$attachmentIds = [$attachmentId];
+		}
+
+		foreach($attachmentIds as $attachmentId) {
+			$attachment = $mailBox->getAttachment($messageId, $attachmentId);
+
+			$fileName = $attachment->getName();
+			$fileParts = pathinfo($fileName);
+			$fileName = $fileParts['filename'];
+			$fileExtension = $fileParts['extension'];
+			$fullPath = "$targetPath/$fileName.$fileExtension";
+			$counter = 2;
+			while($this->userFolder->nodeExists($fullPath)) {
+				$fullPath = "$targetPath/$fileName ($counter).$fileExtension";
+				$counter++;
+			}
+
+			$newFile = $this->userFolder->newFile($fullPath);
+			$newFile->putContent($attachment->getContents());
+		}
+
+		return new JSONResponse();
+	}
+
+	/**
+	 * @NoAdminRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param string $messageId
+	 * @param array $flags
+	 * @return JSONResponse
+	 */
+	public function setFlags($accountId, $folderId, $messageId, $flags) {
+		$mailBox = $this->getFolder($accountId, $folderId);
+
+		foreach($flags as $flag => $value) {
+			$value = filter_var($value, FILTER_VALIDATE_BOOLEAN);
+			if ($flag === 'unseen') {
+				$flag = 'seen';
+				$value = !$value;
+			}
+			$mailBox->setMessageFlag($messageId, '\\'.$flag, $value);
+		}
+
+		return new JSONResponse();
+	}
+
+	/**
+	 * @NoAdminRequired
+	 *
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @param string $id
+	 * @return JSONResponse
+	 */
+	public function destroy($accountId, $folderId, $id) {
+		$this->logger->debug("deleting message <$id> of folder <$folderId>, account <$accountId>");
+		try {
+			$account = $this->getAccount($accountId);
+			$account->deleteMessage(base64_decode($folderId), $id);
+			return new JSONResponse();
+
+		} catch (DoesNotExistException $e) {
+			$this->logger->error("could not delete message <$id> of folder <$folderId>, "
+				. "account <$accountId> because it does not exist");
+			return new JSONResponse([], Http::STATUS_NOT_FOUND);
+		}
+	}
+
+	/**
+	 * @param int $accountId
+	 * @return IAccount
+	 */
+	private function getAccount($accountId) {
+		if (!array_key_exists($accountId, $this->accounts)) {
+			$this->accounts[$accountId] = $this->accountService->find($this->currentUserId, $accountId);
+		}
+		return $this->accounts[$accountId];
+	}
+
+	/**
+	 * @param int $accountId
+	 * @param string $folderId
+	 * @return IMailBox
+	 */
+	private function getFolder($accountId, $folderId) {
+		$account = $this->getAccount($accountId);
+		return $account->getMailbox(base64_decode($folderId));
+	}
+
+	/**
+	 * @param string $messageId
+	 * @param $accountId
+	 * @param $folderId
+	 * @return callable
+	 */
+	private function enrichDownloadUrl($accountId, $folderId, $messageId, $attachment) {
+		$downloadUrl = $this->urlGenerator->linkToRoute('mail.messages.downloadAttachment', [
+			'accountId' => $accountId,
+			'folderId' => $folderId,
+			'messageId' => $messageId,
+			'attachmentId' => $attachment['id'],
+		]);
+		$downloadUrl = $this->urlGenerator->getAbsoluteURL($downloadUrl);
+		$attachment['downloadUrl'] = $downloadUrl;
+		$attachment['mimeUrl'] = $this->mimeTypeDetector->mimeTypeIcon($attachment['mime']);
+
+		if ($this->attachmentIsImage($attachment)) {
+			$attachment['isImage'] = true;
+		} else if ($this->attachmentIsCalendarEvent($attachment)) {
+			$attachment['isCalendarEvent'] = true;
+		}
+		return $attachment;
+	}
+
+	/**
+	 * @param $attachment
+	 *
+	 * Determines if the content of this attachment is an image
+	 *
+	 * @return boolean
+	 */
+	private function attachmentIsImage($attachment) {
+		return in_array(
+			$attachment['mime'], [
+			'image/jpeg',
+			'image/png',
+			'image/gif'
+		]);
+	}
+
+	/**
+	 * @param type $attachment
+	 * @return boolean
+	 */
+	private function attachmentIsCalendarEvent($attachment) {
+		return $attachment['mime'] === 'text/calendar';
+	}
+
+	/**
+	 * @param string $accountId
+	 * @param string $folderId
+	 * @param string $messageId
+	 * @return string
+	 */
+	private function buildHtmlBodyUrl($accountId, $folderId, $messageId) {
+		$htmlBodyUrl = $this->urlGenerator->linkToRoute('mail.messages.getHtmlBody', [
+			'accountId' => $accountId,
+			'folderId' => $folderId,
+			'messageId' => $messageId,
+		]);
+		return $this->urlGenerator->getAbsoluteURL($htmlBodyUrl);
+	}
+
+	/**
+	 * @param integer $accountId
+	 * @param string $folderId
+	 */
+	private function loadMultiple($accountId, $folderId, $ids) {
+		$messages = array_map(function($id) use ($accountId, $folderId){
+			try {
+				return $this->loadMessage($accountId, $folderId, $id);
+			} catch (DoesNotExistException $ex) {
+				return null;
+			}
+		}, $ids);
+
+		return $messages;
+	}
+
+	/**
+	 * @param integer $accountId
+	 * @param string $folderId
+	 * @param $id
+	 * @param $m
+	 * @param IAccount $account
+	 * @param IMailBox $mailBox
+	 * @return mixed
+	 */
+	private function enhanceMessage($accountId, $folderId, $id, $m, IAccount $account, $mailBox) {
+		$json = $m->getFullMessage($account->getEmail(), $mailBox->getSpecialRole());
+		$json['senderImage'] = $this->contactsIntegration->getPhoto($m->getFromEmail());
+		if (isset($json['hasHtmlBody'])) {
+			$json['htmlBodyUrl'] = $this->buildHtmlBodyUrl($accountId, $folderId, $id);
+		}
+
+		if (isset($json['attachments'])) {
+			$json['attachments'] = array_map(function ($a) use ($accountId, $folderId, $id) {
+				return $this->enrichDownloadUrl($accountId, $folderId, $id, $a);
+			}, $json['attachments']);
+
+			// show images first
+			usort($json['attachments'], function($a, $b) {
+				if (isset($a['isImage']) && !isset($b['isImage'])) {
+					return -1;
+				} elseif (!isset($a['isImage']) && isset($b['isImage'])) {
+					return 1;
+				} else {
+					Util::naturalSortCompare($a['fileName'], $b['fileName']);
+				}
+			});
+			return $json;
+		}
+		return $json;
+	}
+
+}