Merge branch '5.5'

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

<?php

namespace wcf\system\worker;

use wcf\data\conversation\Conversation;
use wcf\data\conversation\ConversationAction;
use wcf\data\conversation\ConversationEditor;
use wcf\data\conversation\ConversationList;
use wcf\system\WCF;

/**
 * Worker implementation for updating conversations.
 *
 * @author  Marcel Werk
 * @copyright   2001-2019 WoltLab GmbH
 * @license GNU Lesser General Public License <http://opensource.org/licenses/lgpl-license.php>
 *
 * @method  ConversationList    getObjectList()
 */
class ConversationRebuildDataWorker extends AbstractRebuildDataWorker
{
    /**
     * @inheritDoc
     */
    protected $limit = 100;

    /**
     * @inheritDoc
     */
    public function countObjects()
    {
        if ($this->count === null) {
            $this->count = 0;
            $sql = "SELECT  MAX(conversationID) AS conversationID
                    FROM    wcf1_conversation";
            $statement = WCF::getDB()->prepare($sql);
            $statement->execute();
            $row = $statement->fetchArray();
            if ($row !== false) {
                $this->count = $row['conversationID'];
            }
        }
    }

    /**
     * @inheritDoc
     */
    protected function initObjectList()
    {
        $this->objectList = new ConversationList();
        $this->objectList->sqlOrderBy = 'conversation.conversationID';
    }

    /**
     * @inheritDoc
     */
    public function execute()
    {
        $this->objectList->getConditionBuilder()->add(
            'conversation.conversationID BETWEEN ? AND ?',
            [$this->limit * $this->loopCount + 1, $this->limit * $this->loopCount + $this->limit]
        );

        parent::execute();

        // prepare statements
        $sql = "SELECT      messageID, time, userID, username
                FROM        wcf1_conversation_message
                WHERE       conversationID = ?
                ORDER BY    time";
        $firstMessageStatement = WCF::getDB()->prepare($sql, 1);
        $sql = "SELECT      time, userID, username
                FROM        wcf1_conversation_message
                WHERE       conversationID = ?
                ORDER BY    time DESC";
        $lastMessageStatement = WCF::getDB()->prepare($sql, 1);
        $sql = "SELECT  COUNT(*) AS messages,
                        SUM(attachments) AS attachments
                FROM    wcf1_conversation_message
                WHERE   conversationID = ?";
        $statsStatement = WCF::getDB()->prepare($sql);
        $sql = "SELECT  COUNT(*) AS participants
                FROM    wcf1_conversation_to_user conversation_to_user
                WHERE   conversation_to_user.conversationID = ?
                    AND conversation_to_user.hideConversation <> ?
                    AND conversation_to_user.participantID <> ?
                    AND conversation_to_user.isInvisible = ?";
        $participantCounterStatement = WCF::getDB()->prepare($sql);
        $sql = "SELECT      conversation_to_user.participantID AS userID, conversation_to_user.hideConversation, user_table.username
                FROM        wcf1_conversation_to_user conversation_to_user
                LEFT JOIN   wcf1_user user_table
                ON          user_table.userID = conversation_to_user.participantID
                WHERE       conversation_to_user.conversationID = ?
                        AND conversation_to_user.participantID <> ?
                        AND conversation_to_user.isInvisible = ?
                ORDER BY    user_table.username";
        $participantStatement = WCF::getDB()->prepare($sql, 5);

        $sql = "SELECT  COUNT(*) AS participants
                FROM    wcf1_conversation_to_user
                WHERE   conversationID = ?
                    AND hideConversation <> ?
                    AND participantID IS NOT NULL";
        $existingParticipantStatement = WCF::getDB()->prepare($sql);

        $obsoleteConversations = [];
        $updateData = [];
        /** @var Conversation $conversation */
        foreach ($this->objectList as $conversation) {
            // get stats
            $statsStatement->execute([$conversation->conversationID]);
            $row = $statsStatement->fetchSingleRow();

            // update data
            $data = [
                'attachments' => $row['attachments'] ?: 0,
                'firstMessageID' => $conversation->firstMessageID,
                'lastPostTime' => $conversation->lastPostTime,
                'lastPosterID' => $conversation->lastPosterID,
                'lastPoster' => $conversation->lastPoster,
                'replies' => $row['messages'] ? $row['messages'] - 1 : 0,
                'userID' => $conversation->userID,
                'username' => $conversation->username,
            ];

            // check for obsolete conversations
            $obsolete = $row['messages'] == 0;
            if (!$obsolete) {
                if ($conversation->isDraft) {
                    if (!$conversation->userID) {
                        $obsolete = true;
                    }
                } else {
                    $existingParticipantStatement->execute([$conversation->conversationID, Conversation::STATE_LEFT]);
                    $row = $existingParticipantStatement->fetchSingleRow();
                    if (!$row['participants']) {
                        $obsolete = true;
                    }
                }
            }

            if ($obsolete) {
                $obsoleteConversations[] = new ConversationEditor($conversation);
                continue;
            }

            // get first post
            $firstMessageStatement->execute([$conversation->conversationID]);
            if (($row = $firstMessageStatement->fetchSingleRow()) !== false) {
                $data['firstMessageID'] = $row['messageID'];
                $data['lastPostTime'] = $data['time'] = $row['time'];
                $data['userID'] = $row['userID'];
                $data['username'] = $row['username'];
            }

            // get last post
            $lastMessageStatement->execute([$conversation->conversationID]);
            if (($row = $lastMessageStatement->fetchSingleRow()) !== false) {
                $data['lastPostTime'] = $row['time'];
                $data['lastPosterID'] = $row['userID'];
                $data['lastPoster'] = $row['username'];
            }

            // get number of participants
            $participantCounterStatement->execute([
                $conversation->conversationID,
                Conversation::STATE_LEFT,
                $conversation->userID,
                0,
            ]);
            $data['participants'] = $participantCounterStatement->fetchSingleColumn();

            // get participant summary
            $participantStatement->execute([$conversation->conversationID, $conversation->userID, 0]);
            $users = [];
            while ($row = $participantStatement->fetchArray()) {
                $users[] = $row;
            }
            $data['participantSummary'] = \serialize($users);

            $updateData[$conversation->conversationID] = $data;
        }

        $sql = "UPDATE  wcf1_conversation
                SET     firstMessageID = ?,
                        lastPostTime = ?,
                        lastPosterID = ?,
                        lastPoster = ?,
                        userID = ?,
                        username = ?,
                        replies = ?,
                        attachments = ?,
                        participants = ?,
                        participantSummary = ?
                WHERE   conversationID = ?";
        $statement = WCF::getDB()->prepare($sql);

        WCF::getDB()->beginTransaction();
        foreach ($updateData as $conversationID => $data) {
            $statement->execute([
                $data['firstMessageID'],
                $data['lastPostTime'],
                $data['lastPosterID'],
                $data['lastPoster'],
                $data['userID'],
                $data['username'],
                $data['replies'],
                $data['attachments'],
                $data['participants'],
                $data['participantSummary'],
                $conversationID,
            ]);
        }
        WCF::getDB()->commitTransaction();

        // delete obsolete conversations
        if (!empty($obsoleteConversations)) {
            $action = new ConversationAction($obsoleteConversations, 'delete');
            $action->executeAction();
        }
    }
}