diff options
author | Joas Schilling <coding@schilljs.com> | 2022-03-31 10:28:08 +0300 |
---|---|---|
committer | Joas Schilling <coding@schilljs.com> | 2022-03-31 16:05:34 +0300 |
commit | 5278e716c7e45c0751257cc0b14db1fbe344f40b (patch) | |
tree | 9f053333edd813f67d5aa7ce9b29abe44fd737b1 /lib | |
parent | 1fa1bb56cd566719df1f35343f3877cdf3670bee (diff) |
Add the entity and a service to handle attachments
Signed-off-by: Joas Schilling <coding@schilljs.com>
Diffstat (limited to 'lib')
-rw-r--r-- | lib/Model/Attachment.php | 92 | ||||
-rw-r--r-- | lib/Model/AttachmentMapper.php | 79 | ||||
-rw-r--r-- | lib/Service/AttachmentService.php | 62 |
3 files changed, 233 insertions, 0 deletions
diff --git a/lib/Model/Attachment.php b/lib/Model/Attachment.php new file mode 100644 index 000000000..a808fa367 --- /dev/null +++ b/lib/Model/Attachment.php @@ -0,0 +1,92 @@ +<?php + +declare(strict_types=1); +/** + * @copyright Copyright (c) 2022 Joas Schilling <coding@schilljs.com> + * + * @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\Talk\Model; + +use OCP\AppFramework\Db\Entity; + +/** + * @method void setRoomId(int $roomId) + * @method int getRoomId() + * @method void setMessageId(int $messageId) + * @method int getMessageId() + * @method void setMessageTime(int $messageTime) + * @method int getMessageTime() + * @method void setObjectType(string $objectType) + * @method string getObjectType() + * @method void setActorType(string $actorType) + * @method string getActorType() + * @method void setActorId(string $actorId) + * @method string getActorId() + */ +class Attachment extends Entity { + public const TYPE_AUDIO = 'audio'; + public const TYPE_DECK_CARD = 'deckcard'; + public const TYPE_FILE = 'file'; + public const TYPE_LOCATION = 'location'; + public const TYPE_MEDIA = 'media'; + public const TYPE_OTHER = 'other'; + public const TYPE_VOICE = 'voice'; + + /** @var int */ + protected $roomId; + + /** @var int */ + protected $messageId; + + /** @var int */ + protected $messageTime; + + /** @var string */ + protected $objectType; + + /** @var string */ + protected $actorType; + + /** @var string */ + protected $actorId; + + public function __construct() { + $this->addType('roomId', 'int'); + $this->addType('messageId', 'int'); + $this->addType('messageTime', 'int'); + $this->addType('objectType', 'string'); + $this->addType('actorType', 'string'); + $this->addType('actorId', 'string'); + } + + /** + * @return array + */ + public function asArray(): array { + return [ + 'id' => $this->getId(), + 'room_id' => $this->getRoomId(), + 'message_id' => $this->getMessageId(), + 'message_time' => $this->getMessageTime(), + 'object_type' => $this->getObjectType(), + 'actor_type' => $this->getActorType(), + 'actor_id' => $this->getActorId(), + ]; + } +} diff --git a/lib/Model/AttachmentMapper.php b/lib/Model/AttachmentMapper.php new file mode 100644 index 000000000..039d7c1c1 --- /dev/null +++ b/lib/Model/AttachmentMapper.php @@ -0,0 +1,79 @@ +<?php + +declare(strict_types=1); +/** + * @copyright Copyright (c) 2022 Joas Schilling <coding@schilljs.com> + * + * @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\Talk\Model; + +use OCP\AppFramework\Db\QBMapper; +use OCP\DB\QueryBuilder\IQueryBuilder; +use OCP\IDBConnection; + +/** + * @method Attachment mapRowToEntity(array $row) + * @method Attachment findEntity(IQueryBuilder $query) + * @method Attachment[] findEntities(IQueryBuilder $query) + */ +class AttachmentMapper extends QBMapper { + + /** + * @param IDBConnection $db + */ + public function __construct(IDBConnection $db) { + parent::__construct($db, 'talk_attachments', Attachment::class); + } + + public function createAttachmentFromRow(array $row): Attachment { + return $this->mapRowToEntity([ + 'id' => (int) $row['id'], + 'room_id' => (int) $row['room_id'], + 'message_id' => (int) $row['message_id'], + 'message_time' => (int) $row['message_time'], + 'object_type' => (string) $row['object_type'], + 'actor_type' => (string) $row['actor_type'], + 'actor_id' => (string) $row['actor_id'], + ]); + } + + /** + * @param int $roomId + * @param string $objectType + * @param int $offset + * @param int $limit + * @return Attachment[] + * @throws \OCP\DB\Exception + */ + public function getAttachmentsByType(int $roomId, string $objectType, int $offset, int $limit): array { + $query = $this->db->getQueryBuilder(); + $query->select('*') + ->from($this->getTableName()) + ->where($query->expr()->eq('room_id', $query->createNamedParameter($roomId, IQueryBuilder::PARAM_INT))) + ->andWhere($query->expr()->eq('object_type', $query->createNamedParameter($objectType))) + ->setMaxResults($limit) + ->orderBy('id', 'DESC'); + + if ($offset > 0) { + $query->andWhere($query->expr()->lt('message_id', $query->createNamedParameter($offset))); + } + + return $this->findEntities($query); + } +} diff --git a/lib/Service/AttachmentService.php b/lib/Service/AttachmentService.php new file mode 100644 index 000000000..f26a22967 --- /dev/null +++ b/lib/Service/AttachmentService.php @@ -0,0 +1,62 @@ +<?php + +namespace OCA\Talk\Service; + +use OCA\Talk\Model\Attachment; +use OCA\Talk\Model\AttachmentMapper; +use OCA\Talk\Room; +use OCP\Comments\IComment; + +class AttachmentService { + public AttachmentMapper $attachmentMapper; + + public function __construct(AttachmentMapper $attachmentMapper) { + $this->attachmentMapper = $attachmentMapper; + } + + public function createAttachmentEntry(Room $room, IComment $comment, string $messageType, array $parameters): void { + $attachment = new Attachment(); + $attachment->setRoomId($room->getId()); + $attachment->setActorType($comment->getActorType()); + $attachment->setActorId($comment->getActorId()); + $attachment->setMessageId((int) $comment->getId()); + $attachment->setMessageTime($comment->getCreationDateTime()->getTimestamp()); + + if ($messageType === 'object_shared') { + $objectType = $parameters['objectType'] ?? ''; + if ($objectType === 'geo-location') { + $attachment->setObjectType(Attachment::TYPE_LOCATION); + } elseif ($objectType === 'deck-card') { + $attachment->setObjectType(Attachment::TYPE_DECK_CARD); + } else { + $attachment->setObjectType(Attachment::TYPE_OTHER); + } + } else { + $messageType = $parameters['metaData']['messageType'] ?? ''; + $mimetype = $parameters['metaData']['mimeType'] ?? ''; + + if ($messageType === 'voice-message') { + $attachment->setObjectType(Attachment::TYPE_VOICE); + } elseif (str_starts_with($mimetype, 'audio/')) { + $attachment->setObjectType(Attachment::TYPE_AUDIO); + } elseif (str_starts_with($mimetype, 'image/') || str_starts_with($mimetype, 'video/')) { + $attachment->setObjectType(Attachment::TYPE_MEDIA); + } else { + $attachment->setObjectType(Attachment::TYPE_FILE); + } + } + + $this->attachmentMapper->insert($attachment); + } + + /** + * @param Room $room + * @param string $objectType + * @param int $offset + * @param int $limit + * @return Attachment[] + */ + public function getAttachmentsByType(Room $room, string $objectType, int $offset, int $limit): array { + return $this->attachmentMapper->getAttachmentsByType($room->getId(), $objectType, $offset, $limit); + } +} |