1 files changed, 100 insertions, 5 deletions

diff --git a/apps/files_sharing/lib/Notification/Notifier.php b/apps/files_sharing/lib/Notification/Notifier.php
index a9028ec9cf5..6ae009895d3 100644
--- a/apps/files_sharing/lib/Notification/Notifier.php
+++ b/apps/files_sharing/lib/Notification/Notifier.php
@@ -2,8 +2,10 @@
 declare(strict_types=1);
 /**
  * @copyright Copyright (c) 2019, Roeland Jago Douma <roeland@famdouma.nl>
+ * @copyright Copyright (c) 2019, Joas Schilling <coding@schilljs.com>
  *
  * @author Roeland Jago Douma <roeland@famdouma.nl>
+ * @author Joas Schilling <coding@schilljs.com>
  *
  * @license GNU AGPL version 3 or any later version
  *
@@ -25,43 +27,70 @@ declare(strict_types=1);
 namespace OCA\Files_Sharing\Notification;
 
 use OCP\Files\IRootFolder;
+use OCP\IL10N;
+use OCP\IURLGenerator;
 use OCP\L10N\IFactory;
 use OCP\Notification\AlreadyProcessedException;
 use OCP\Notification\INotification;
 use OCP\Notification\INotifier;
 use OCP\Share\Exceptions\ShareNotFound;
 use OCP\Share\IManager;
+use OCP\Share\IShare;
 
 class Notifier implements INotifier {
 
 	/** @var IFactory */
 	protected $l10nFactory;
-
 	/** @var IManager */
 	private $shareManager;
-
 	/** @var IRootFolder */
 	private $rootFolder;
+	/** @var IURLGenerator */
+	protected $url;
+
 
 	public function __construct(IFactory $l10nFactory,
 								IManager $shareManager,
-								IRootFolder $rootFolder) {
+								IRootFolder $rootFolder,
+								IURLGenerator $url) {
 		$this->l10nFactory = $l10nFactory;
 		$this->shareManager = $shareManager;
 		$this->rootFolder = $rootFolder;
+		$this->url = $url;
 	}
 
+	/**
+	 * Identifier of the notifier, only use [a-z0-9_]
+	 *
+	 * @return string
+	 * @since 17.0.0
+	 */
 	public function getID(): string {
 		return 'files_sharing';
 	}
 
+	/**
+	 * Human readable name describing the notifier
+	 *
+	 * @return string
+	 * @since 17.0.0
+	 */
 	public function getName(): string {
-		return $this->l10nFactory->get('files_sharing')->t('Files sharing');
+		return $this->l10nFactory->get('files_sharing')->t('File sharing');
 	}
 
+	/**
+	 * @param INotification $notification
+	 * @param string $languageCode The code of the language that should be used to prepare the notification
+	 * @return INotification
+	 * @throws \InvalidArgumentException When the notification was not prepared by a notifier
+	 * @throws AlreadyProcessedException When the notification is not needed anymore and should be deleted
+	 * @since 9.0.0
+	 */
 	public function prepare(INotification $notification, string $languageCode): INotification {
 		if ($notification->getApp() !== 'files_sharing' ||
-			$notification->getSubject() !== 'expiresTomorrow') {
+			($notification->getSubject() !== 'expiresTomorrow' &&
+				$notification->getObjectType() !== 'share')) {
 			throw new \InvalidArgumentException('Unhandled app or subject');
 		}
 
@@ -74,6 +103,15 @@ class Notifier implements INotifier {
 			throw new AlreadyProcessedException();
 		}
 
+		if ($notification->getSubject() === 'expiresTomorrow') {
+			$notification = $this->parseShareExpiration($share, $notification, $l);
+		} else {
+			$notification = $this->parseShareInvitation($share, $notification, $l);
+		}
+		return $notification;
+	}
+
+	protected function parseShareExpiration(IShare $share, INotification $notification, IL10N $l): INotification {
 		$node = $share->getNode();
 		$userFolder = $this->rootFolder->getUserFolder($notification->getUser());
 		$path = $userFolder->getRelativePath($node->getPath());
@@ -95,4 +133,61 @@ class Notifier implements INotifier {
 
 		return $notification;
 	}
+
+	protected function parseShareInvitation(IShare $share, INotification $notification, IL10N $l): INotification {
+		if ($share->getShareType() === IShare::TYPE_USER) {
+			if ($share->getSharedWith() !== $notification->getUser()) {
+				throw new AlreadyProcessedException();
+			}
+
+			if ($share->getStatus() !== IShare::STATUS_PENDING) {
+				throw new AlreadyProcessedException();
+			}
+		}
+
+		switch ($notification->getSubject()) {
+			case 'incoming_user_share':
+				$subject = $l->t('You received {share} as a share from {user}');
+				$subjectParameters = [
+					'share' => [
+						'type' => 'highlight',
+						'id' => $notification->getObjectId(),
+						'name' => $share->getNode()->getName(),
+					],
+					'user' =>  [
+						'type' => 'user',
+						'id' => $share->getShareOwner(),
+						'name' => $share->getShareOwner(),
+					],
+				];
+
+				$placeholders = $replacements = [];
+				foreach ($subjectParameters as $placeholder => $parameter) {
+					$placeholders[] = '{' . $placeholder . '}';
+					$replacements[] = $parameter['name'];
+				}
+
+				$notification->setParsedSubject(str_replace($placeholders, $replacements, $subject))
+					->setRichSubject($subject, $subjectParameters)
+					->setIcon($this->url->getAbsoluteURL($this->url->imagePath('core', 'actions/share.svg')));
+
+				$acceptAction = $notification->createAction();
+				$acceptAction->setParsedLabel($l->t('Accept'))
+					->setLink($this->url->linkToOCSRouteAbsolute('files_sharing.ShareAPI.acceptShare', ['id' => $share->getId()]), 'POST')
+					->setPrimary(true);
+				$notification->addParsedAction($acceptAction);
+
+				$rejectAction = $notification->createAction();
+				$rejectAction->setParsedLabel($l->t('Reject'))
+					->setLink($this->url->linkToOCSRouteAbsolute('files_sharing.ShareAPI.deleteShare', ['id' => $share->getId()]), 'DELETE')
+					->setPrimary(false);
+				$notification->addParsedAction($rejectAction);
+
+				return $notification;
+				break;
+
+			default:
+				throw new \InvalidArgumentException('Invalid subject');
+		}
+	}
 }