<?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 OCP\AppFramework\Db\DoesNotExistException;
use OCA\Polls\Exceptions\NotAuthorizedException;

use OCA\Polls\Db\VoteMapper;
use OCA\Polls\Db\Vote;
use OCA\Polls\Db\OptionMapper;
use OCA\Polls\Model\Acl;

class VoteService {

	/** @var VoteMapper */
	private $voteMapper;

	/** @var Vote */
	private $vote;

	/** @var OptionMapper */
	private $optionMapper;

	/** @var AnonymizeService */
	private $anonymizer;

	/** @var LogService */
	private $logService;

	/** @var Acl */
	private $acl;

	/**
	 * VoteController constructor.
	 * @param VoteMapper $voteMapper
	 * @param Vote $vote
	 * @param OptionMapper $optionMapper
	 * @param AnonymizeService $anonymizer
	 * @param LogService $logService
	 * @param Acl $acl
	 */
	public function __construct(
		VoteMapper $voteMapper,
		Vote $vote,
		OptionMapper $optionMapper,
		AnonymizeService $anonymizer,
		LogService $logService,
		Acl $acl
	) {
		$this->voteMapper = $voteMapper;
		$this->vote = $vote;
		$this->optionMapper = $optionMapper;
		$this->anonymizer = $anonymizer;
		$this->logService = $logService;
		$this->acl = $acl;
	}

	/**
	 * Read all votes of a poll based on the poll id and return list as array
	 * @NoAdminRequired
	 * @param int $pollId
	 * @param string $token
	 * @return array
	 * @throws NotAuthorizedException
	 */
	public function list($pollId = 0, $token = '') {
		if (!$this->acl->set($pollId, $token)->getAllowView()) {
			throw new NotAuthorizedException;
		}

		if (!$this->acl->getAllowSeeResults()) {
			return $this->voteMapper->findByPollAndUser($this->acl->getpollId(), $this->acl->getUserId());
		} elseif (!$this->acl->getAllowSeeUsernames()) {
			$this->anonymizer->set($this->acl->getpollId(), $this->acl->getUserId());
			return $this->anonymizer->getVotes();
		} else {
			return $this->voteMapper->findByPoll($this->acl->getpollId());
		}
	}

	/**
	 * Set vote
	 * @NoAdminRequired
	 * @param int $optionId
	 * @param string $setTo
	 * @param string $token
	 * @return Vote
	 * @throws NotAuthorizedException
	 */
	public function set($optionId, $setTo, $token = '') {
		$option = $this->optionMapper->find($optionId);

		if (!$this->acl->set($option->getPollId(), $token)->getAllowVote()) {
			throw new NotAuthorizedException;
		}

		if (intval($option->getPollId()) !== $this->acl->getPollId()) {
			throw new NotAuthorizedException;
		}

		try {
			$this->vote = $this->voteMapper->findSingleVote($this->acl->getPollId(), $option->getPollOptionText(), $this->acl->getUserId());
			$this->vote->setVoteAnswer($setTo);
			$this->voteMapper->update($this->vote);
		} catch (DoesNotExistException $e) {
			// Vote does not exist, insert as new Vote
			$this->vote = new Vote();

			$this->vote->setPollId($this->acl->getPollId());
			$this->vote->setUserId($this->acl->getUserId());
			$this->vote->setVoteOptionText($option->getPollOptionText());
			$this->vote->setVoteOptionId($option->getId());
			$this->vote->setVoteAnswer($setTo);
			$this->voteMapper->insert($this->vote);
		} finally {
			$this->logService->setLog($this->acl->getPollId(), 'setVote', $this->vote->getUserId());
			return $this->vote;
		}
	}

	/**
	 * Remove user from poll
	 * @NoAdminRequired
	 * @param int $voteId
	 * @param string $userId
	 * @param int $pollId
	 * @return boolean
	 * @throws NotAuthorizedException
	 */
	public function delete($pollId, $userId) {
		if (!$this->acl->set($pollId)->getAllowEdit()) {
			throw new NotAuthorizedException;
		}

		$this->voteMapper->deleteByPollAndUser($pollId, $userId);
		return $userId;
	}
}