Add custom user tags

Signed-off-by: Daniel Kesselberg <mail@danielkesselberg.de>

1 files changed, 140 insertions, 0 deletions

diff --git a/lib/Controller/TagsController.php b/lib/Controller/TagsController.php
new file mode 100644
index 000000000..feccf38ed
--- /dev/null
+++ b/lib/Controller/TagsController.php
@@ -0,0 +1,140 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @author Daniel Kesselberg <mail@danielkesselberg.de>
+ *
+ * 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\AppInfo\Application;
+use OCA\Mail\Db\Tag;
+use OCA\Mail\Db\TagMapper;
+use OCA\Mail\Exception\ClientException;
+use OCP\AppFramework\Controller;
+use OCP\AppFramework\Db\DoesNotExistException;
+use OCP\AppFramework\Http\JSONResponse;
+use OCP\IRequest;
+
+class TagsController extends Controller {
+
+	/** @var string */
+	private $currentUserId;
+
+	/** @var TagMapper */
+	private $tagMapper;
+
+	/**
+	 * TagsController constructor.
+	 *
+	 * @param IRequest $request
+	 * @param $UserId
+	 * @param TagMapper $tagMapper
+	 */
+	public function __construct(IRequest $request,
+								$UserId,
+								TagMapper $tagMapper
+	) {
+		parent::__construct(Application::APP_ID, $request);
+		$this->currentUserId = $UserId;
+		$this->tagMapper = $tagMapper;
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @TrapError
+	 *
+	 * @param string $displayName
+	 * @param string $color
+	 *
+	 * @return JSONResponse
+	 * @throws ClientException
+	 */
+	public function create(string $displayName, string $color): JSONResponse {
+		$this->validateDisplayName($displayName);
+		$this->validateColor($color);
+
+		$imapLabel = str_replace(' ', '_', $displayName);
+		/** @var string|false $imapLabel */
+		$imapLabel = mb_convert_encoding($imapLabel, 'UTF7-IMAP', 'UTF-8');
+		if ($imapLabel === false) {
+			throw new ClientException('Error converting display name to UTF7-IMAP ', 0);
+		}
+		$imapLabel = mb_strcut($imapLabel, 0, 64);
+
+		try {
+			return new JSONResponse($this->tagMapper->getTagByImapLabel($imapLabel, $this->currentUserId));
+		} catch (DoesNotExistException $e) {
+			// it's valid that a tag does not exist.
+		}
+
+		$tag = new Tag();
+		$tag->setUserId($this->currentUserId);
+		$tag->setDisplayName($displayName);
+		$tag->setImapLabel($imapLabel);
+		$tag->setColor($color);
+		$tag->setIsDefaultTag(false);
+
+		$tag = $this->tagMapper->insert($tag);
+		return new JSONResponse($tag);
+	}
+
+	/**
+	 * @NoAdminRequired
+	 * @TrapError
+	 *
+	 * @param int $id
+	 * @param string $displayName
+	 * @param string $color
+	 *
+	 * @return JSONResponse
+	 * @throws ClientException
+	 * @throws DoesNotExistException
+	 */
+	public function update(int $id, string $displayName, string $color): JSONResponse {
+		$this->validateDisplayName($displayName);
+		$this->validateColor($color);
+
+		$tag = $this->tagMapper->getTagForUser($id, $this->currentUserId);
+
+		$tag->setDisplayName($displayName);
+		$tag->setColor($color);
+
+		$tag = $this->tagMapper->update($tag);
+		return new JSONResponse($tag);
+	}
+
+	/**
+	 * @throws ClientException
+	 */
+	private function validateDisplayName(string $displayName): void {
+		if (mb_strlen($displayName) > 128) {
+			throw new ClientException('The maximum length for displayName is 128');
+		}
+	}
+
+	/**
+	 * @throws ClientException
+	 */
+	private function validateColor(string $color): void {
+		if (mb_strlen($color) > 9) {
+			throw new ClientException('The maximum length for color is 9');
+		}
+	}
+}