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

package org.mercury_im.messenger.data.mapping;

import org.mercury_im.messenger.data.model.AccountModel;
import org.mercury_im.messenger.entity.Account;
import org.mercury_im.messenger.entity.PasswordAuthentication;

import javax.inject.Inject;

public class AccountMapping implements Mapping<Account, AccountModel> {

    @Inject
    public AccountMapping() {

    }

    @Override
    public AccountModel entityToModel(Account entity, AccountModel model) {
        if (entity == null) {
            return null;
        }
        if (model == null) {
            model = new AccountModel();
        }
        model.setPassword(entity.getAuthentication().getPassword());
        model.setAddress(entity.getAddress());
        model.setEnabled(entity.isEnabled());

        return model;
    }

    @Override
    public Account modelToEntity(AccountModel model, Account entity) {
        if (model == null) {
            return null;
        }

        entity.setId(model.getId());
        entity.setAddress(model.getAddress());
        entity.setEnabled(model.isEnabled());
        entity.setAuthentication(new PasswordAuthentication(model.getPassword()));

        return entity;
    }
}