openmw-tes3mp/apps/openmw/mwmp/processors/object/ProcessorContainer.hpp

#ifndef OPENMW_PROCESSORCONTAINER_HPP
#define OPENMW_PROCESSORCONTAINER_HPP

#include "BaseObjectProcessor.hpp"

namespace mwmp
{
    class ProcessorContainer final: public BaseObjectProcessor
    {
    public:
        ProcessorContainer()
        {
            BPP_INIT(ID_CONTAINER)
        }

        virtual void Do(ObjectPacket &packet, ObjectList &objectList)
        {
            BaseObjectProcessor::Do(packet, objectList);

            ptrCellStore = Main::get().getCellController()->getCellStore(objectList.cell);

            if (!ptrCellStore) return;

            std::string debugMessage = "- action ";
            unsigned char action = objectList.action;
            unsigned char containerSubAction = objectList.containerSubAction;

            if (action == mwmp::BaseObjectList::SET)
                debugMessage += "SET";
            else if (action == mwmp::BaseObjectList::ADD)
                debugMessage += "ADD";
            else if (action == mwmp::BaseObjectList::REMOVE)
                debugMessage += "REMOVE";

            debugMessage += " and subaction ";

            if (containerSubAction == mwmp::BaseObjectList::NONE)
                debugMessage += "NONE";
            else if (containerSubAction == mwmp::BaseObjectList::DRAG)
                debugMessage += "DRAG";
            else if (containerSubAction == mwmp::BaseObjectList::DROP)
                debugMessage += "DROP";
            else if (containerSubAction == mwmp::BaseObjectList::TAKE_ALL)
                debugMessage += "TAKE_ALL";
            else if (containerSubAction == mwmp::BaseObjectList::REPLY_TO_REQUEST)
                debugMessage += "REPLY_TO_REQUEST";

            LOG_APPEND(TimedLog::LOG_VERBOSE, "%s", debugMessage.c_str());

            // If we've received a request for information, comply with it
            if (objectList.action == mwmp::BaseObjectList::REQUEST)
            {
                if (objectList.baseObjectCount == 0)
                {
                    LOG_APPEND(TimedLog::LOG_VERBOSE, "- Request had no objects attached, so we are sending all containers in the cell %s",
                        objectList.cell.getDescription().c_str());
                    objectList.reset();
                    objectList.cell = *ptrCellStore->getCell();
                    objectList.action = mwmp::BaseObjectList::SET;
                    objectList.containerSubAction = mwmp::BaseObjectList::REPLY_TO_REQUEST;
                    objectList.addAllContainers(ptrCellStore);
                    objectList.sendContainer();
                }
                else
                {
                    LOG_APPEND(TimedLog::LOG_VERBOSE, "- Request was for %i %s", objectList.baseObjectCount, objectList.baseObjectCount == 1 ? "object" : "objects");
                    std::vector<BaseObject> requestObjects = objectList.baseObjects;
                    objectList.reset();
                    objectList.cell = *ptrCellStore->getCell();
                    objectList.action = mwmp::BaseObjectList::SET;
                    objectList.containerSubAction = mwmp::BaseObjectList::REPLY_TO_REQUEST;
                    objectList.addRequestedContainers(ptrCellStore, requestObjects);

                    if (objectList.baseObjects.size() > 0)
                        objectList.sendContainer();
                }
            }
            // Otherwise, edit containers based on the information received
            else
            {
                LOG_APPEND(TimedLog::LOG_VERBOSE, "- Editing container contents to match those of packet", objectList.baseObjectCount);
                objectList.editContainers(ptrCellStore);
            }
        }

    };
}

#endif //OPENMW_PROCESSORCONTAINER_HPP