Mercury-IM/data/src/main/java/org/mercury_im/messenger/data/mapping/MessageMapping.java

package org.mercury_im.messenger.data.mapping;

import org.mercury_im.messenger.data.model.MessageModel;
import org.mercury_im.messenger.data.model.MessagePayloadContainerModel;
import org.mercury_im.messenger.entity.message.IMessage;
import org.mercury_im.messenger.entity.message.Message;
import org.mercury_im.messenger.entity.message.PayloadContainer;

import javax.inject.Inject;

public class MessageMapping extends AbstractMapping<Message, MessageModel> {

    private final MessagePayloadContainerMapping messagePayloadContainerMapping;

    @Inject
    public MessageMapping(MessagePayloadContainerMapping messagePayloadContainerMapping) {
        this.messagePayloadContainerMapping = messagePayloadContainerMapping;
    }

    @Override
    public Message newEntity(MessageModel model) {
        return new IMessage();
    }

    @Override
    public MessageModel newModel(Message entity) {
        return new MessageModel();
    }

    @Override
    public MessageModel mapToModel(Message entity, MessageModel model) {
        model.setSender(entity.getSender());
        model.setRecipient(entity.getRecipient());
        model.setTimestamp(entity.getTimestamp());

        model.getPayloads().clear();
        for (PayloadContainer payload : entity.getMessagePayloads()) {
            MessagePayloadContainerModel payloadModel = messagePayloadContainerMapping.toModel(payload, new MessagePayloadContainerModel());
            payloadModel.setMessage(model);
            model.getPayloads().add(payloadModel);
        }

        return model;
    }

    @Override
    public Message mapToEntity(MessageModel model, Message entity) {
        return null;
    }
}