1 files changed, 131 insertions, 0 deletions
diff --git a/lib/Service/SubscriptionService.php b/lib/Service/SubscriptionService.php
new file mode 100644
index 00000000..f378e476
--- /dev/null
+++ b/lib/Service/SubscriptionService.php
@@ -0,0 +1,131 @@
+<?php
+/**
+ * @copyright Copyright (c) 2017 Vinzenz Rosenkranz <vinzenz.rosenkranz@gmail.com>
+ *
+ * @author René Gieling <github@dartcafe.de>
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU Affero General Public License as
+ *  published by the Free Software Foundation, either version 3 of the
+ *  License, or (at your option) any later version.
+ *
+ *  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
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+namespace OCA\Polls\Service;
+
+use Exception;
+use OCA\Polls\Exceptions\NotAuthorizedException;
+use OCP\AppFramework\Db\MultipleObjectsReturnedException;
+use OCP\AppFramework\Db\DoesNotExistException;
+use OCP\ILogger;
+
+use OCA\Polls\Db\Subscription;
+use OCA\Polls\Db\SubscriptionMapper;
+use OCA\Polls\Model\Acl;
+
+class SubscriptionService  {
+
+	private $acl;
+	private $subscriptionMapper;
+	private $logger;
+
+	/**
+	 * SubscriptionController constructor.
+	 * @param SubscriptionMapper $subscriptionMapper
+	 * @param ILogger $logger
+	 * @param Acl $acl
+	 */
+
+	public function __construct(
+		SubscriptionMapper $subscriptionMapper,
+		ILogger $logger,
+		Acl $acl
+	) {
+		$this->subscriptionMapper = $subscriptionMapper;
+		$this->acl = $acl;
+		$this->logger = $logger;
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @param integer $pollId
+	 * @return array
+	 */
+	public function get($pollId) {
+		if (!$this->acl->setPollId($pollId)->getAllowView()) {
+			throw new NotAuthorizedException;
+		}
+		try {
+			return $this->subscriptionMapper->findByUserAndPoll($pollId, $this->acl->getUserId());
+		} catch (MultipleObjectsReturnedException $e) {
+			// subscription should be unique. If duplicates are found resubscribe
+			// duplicates are removed in $this->set()
+			return $this->set($pollId, true);
+		}
+
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @param integer $pollId
+	 * @return array
+	 */
+	public function set($pollId, $subscribed) {
+		if (!$this->acl->setPollId($pollId)->getAllowView()) {
+			throw new NotAuthorizedException;
+		}
+		try {
+			$subscription = $this->subscriptionMapper->findByUserAndPoll($pollId, $this->acl->getUserId());
+			if (!$subscribed) {
+				$this->subscriptionMapper->delete($subscription);
+				return ['status' => 'Unsubscribed from poll ' . $pollId];
+			} else {
+				// subscription already exists, just return the existing subscription
+				return ['status' => 'Subscribed to poll ' . $pollId];
+			}
+
+		} catch (DoesNotExistException $e){
+
+			if ($subscribed) {
+				$subscription = new Subscription();
+				$subscription->setPollId($pollId);
+				$subscription->setUserId($this->acl->getUserId());
+
+				$this->subscriptionMapper->insert($subscription);
+				return ['status' => 'Subscribed to poll ' . $pollId];
+			} else {
+				// subscription is not found, just approve the unsubscription
+				return ['status' => 'Unsubscribed from poll ' . $pollId];
+			}
+
+		} catch (MultipleObjectsReturnedException $e) {
+			// Duplicates should not exist but if found, fix it
+			// unsubscribe from all and resubscribe, if requested
+			$this->logger->debug('Multiple subscription (dulpicates) found');
+			$this->subscriptionMapper->unsubscribe($pollId, $this->acl->getUserId());
+			$this->logger->debug('Unsubscribed all for user ' . $this->acl->getUserId() . 'in poll' . $pollId);
+			if ($subscribed) {
+				$subscription = new Subscription();
+				$subscription->setPollId($pollId);
+				$subscription->setUserId($this->acl->getUserId());
+				$this->subscriptionMapper->insert($subscription);
+				$this->logger->debug('Added new subscription');
+				return $subscription;
+			} else {
+				return ['status' => 'Unsubscribed from poll ' . $pollId];
+			}
+
+		}
+
+	}
+}