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 implements Mapping<Message, MessageModel> {

    private final MessagePayloadMapping messagePayloadMapping;

    @Inject
    public MessageMapping(MessagePayloadMapping messagePayloadMapping) {
        this.messagePayloadMapping = messagePayloadMapping;
    }

    @Override
    public MessageModel entityToModel(Message entity, MessageModel model) {
        if (entity == null) {
            return null;
        }
        if (model == null) {
            model = new MessageModel();
        }
        model.setSender(entity.getSender());
        model.setRecipient(entity.getRecipient());
        model.setTimestamp(entity.getTimestamp());

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

        return model;
    }

    @Override
    public Message modelToEntity(MessageModel model, Message entity) {
        if (model == null) {
            return null;
        }
        if (entity == null) {
            entity = new IMessage();
        }
        return null;
    }
}