Implemented message embedded object system

[GitHub/WoltLab/com.woltlab.wcf.conversation.git] / files / lib / system / attachment / ConversationMessageAttachmentObjectType.class.php

diff --git a/files/lib/system/attachment/ConversationMessageAttachmentObjectType.class.php b/files/lib/system/attachment/ConversationMessageAttachmentObjectType.class.php
index 83887c7cfddc7242f5346a594e0585d5940f1a6e..daf51fb4ccfe246a43ebf23e17976c965711ac4c 100644 (file)
--- a/files/lib/system/attachment/ConversationMessageAttachmentObjectType.class.php
+++ b/files/lib/system/attachment/ConversationMessageAttachmentObjectType.class.php
@@ -1,6 +1,7 @@
 <?php
 namespace wcf\system\attachment;
 use wcf\data\conversation\message\ConversationMessage;
+use wcf\data\conversation\message\ConversationMessageList;
 use wcf\data\conversation\Conversation;
 use wcf\system\WCF;
 use wcf\util\ArrayUtil;
@@ -73,4 +74,63 @@ class ConversationMessageAttachmentObjectType extends AbstractAttachmentObjectTy
                
                return false;
        }
+       
+       /**
+        * @see \wcf\system\attachment\IAttachmentObjectType::cacheObjects()
+        */
+       public function cacheObjects(array $objectIDs) {
+               $messageList = new ConversationMessageList();
+               $messageList->setObjectIDs($objectIDs);
+               $messageList->readObjects();
+               $conversationIDs = array();
+               foreach ($messageList as $message) {
+                       $conversationIDs[] = $message->conversationID;
+               }
+               if (!empty($conversationIDs)) {
+                       $conversations = Conversation::getUserConversations($conversationIDs, WCF::getUser()->userID);
+                       foreach ($messageList as $message) {
+                               if (isset($conversations[$message->conversationID])) $message->setConversation($conversations[$message->conversationID]);
+                       }
+               }
+               $this->cachedObjects = $messageList->getObjects();
+       }
+       
+       /**
+        * @see \wcf\system\attachment\IAttachmentObjectType::setPermissions()
+        */
+       public function setPermissions(array $attachments) {
+               $postIDs = array();
+               foreach ($attachments as $attachment) {
+                       // set default permissions
+                       $attachment->setPermissions(array(
+                               'canDownload' => false,
+                               'canViewPreview' => false
+                       ));
+                               
+                       if ($this->getObject($attachment->objectID) === null) {
+                               $messageIDs[] = $attachment->objectID;
+                       }
+               }
+       
+               if (!empty($messageIDs)) {
+                       $this->cacheObjects($messageIDs);
+               }
+       
+               foreach ($attachments as $attachment) {
+                       if (($message = $this->getObject($attachment->objectID)) !== null) {
+                               if (!$message->getConversation()->canRead()) continue;
+                               
+                               $attachment->setPermissions(array(
+                                       'canDownload' => true,
+                                       'canViewPreview' => true
+                               ));
+                       }
+                       else if ($attachment->tmpHash != '' && $attachment->userID == WCF::getUser()->userID) {
+                               $attachment->setPermissions(array(
+                                       'canDownload' => true,
+                                       'canViewPreview' => true
+                               ));
+                       }
+               }
+       }
 }