FormAttributeEntityDataMapper.java

package com.ushahidi.platform.mobile.app.data.entity.mapper;

import com.ushahidi.platform.mobile.app.data.entity.FormAttributeEntity;
import com.ushahidi.platform.mobile.app.domain.entity.FormAttribute;

import android.support.annotation.Nullable;

import java.util.ArrayList;
import java.util.List;

import javax.inject.Inject;

/**
 * @author Ushahidi Team <team@ushahidi.com>
 */
public class FormAttributeEntityDataMapper {

    /**
     * Default constructor
     */
    @Inject
    public FormAttributeEntityDataMapper() {
        // Do nothing
    }

    /**
     * Maps {@link FormAttributeEntity} to {@link FormAttribute}
     *
     * @param formAttributeEntity The form entity to be mapped
     * @return The mapped form entity
     */
    @Nullable
    public FormAttribute map(FormAttributeEntity formAttributeEntity) {
        FormAttribute formAttribute = null;
        if (formAttributeEntity != null) {
            formAttribute = new FormAttribute();
            formAttribute._id = formAttributeEntity._id;
            formAttribute.setCardinality(formAttributeEntity.getCardinality());
            formAttribute.setPriority(formAttributeEntity.getPriority());
            formAttribute.setKey(formAttributeEntity.getKey());
            formAttribute.setDeploymentId(formAttributeEntity.getDeploymentId());
            formAttribute.setOptions(formAttributeEntity.getOptions());
            formAttribute.setFormId(formAttributeEntity.getFormId());
            formAttribute.setRequired(formAttributeEntity.getRequired());
            formAttribute
                    .setInput(
                            FormAttribute.Input.valueOf(formAttributeEntity.getInput().name()));
            formAttribute
                    .setType(FormAttribute.Type.valueOf(formAttributeEntity.getType().name()));
            formAttribute.setLabel(formAttributeEntity.getLabel());
            formAttribute.setFormStageId(formAttributeEntity.getFormStageId());
        }
        return formAttribute;
    }

    /**
     * Maps {@link FormAttribute} to {@link FormAttributeEntity}
     *
     * @param formAttribute The form to be mapped
     * @return The mapped form
     */
    @Nullable
    public FormAttributeEntity map(FormAttribute formAttribute) {
        FormAttributeEntity formAttributeEntity = null;
        if (formAttribute != null) {
            formAttributeEntity = new FormAttributeEntity();
            formAttributeEntity._id = formAttribute._id;
            formAttributeEntity.setCardinality(formAttribute.getCardinality());
            formAttributeEntity.setPriority(formAttribute.getPriority());
            formAttributeEntity.setKey(formAttribute.getKey());
            formAttributeEntity.setDeploymentId(formAttribute.getDeploymentId());
            formAttributeEntity.setOptions(formAttribute.getOptions());
            formAttributeEntity.setFormId(formAttribute.getFormId());
            formAttributeEntity.setRequired(formAttribute.getRequired());
            formAttributeEntity
                    .setInput(
                            FormAttributeEntity.Input.valueOf(formAttribute.getInput().name()));
            formAttributeEntity
                    .setType(FormAttributeEntity.Type.valueOf(formAttribute.getType().name()));
            formAttributeEntity.setLabel(formAttribute.getLabel());
            formAttributeEntity.setFormStageId(formAttribute.getFormStageId());
            formAttribute.setCardinality(formAttribute.getCardinality());
        }
        return formAttributeEntity;
    }

    /**
     * Maps a list of {@link FormAttributeEntity} to a list of {@link FormAttribute}
     *
     * @param formAttributeEntityList The form entity list
     * @return The mapped form entity list
     */
    @Nullable
    public List<FormAttribute> map(List<FormAttributeEntity> formAttributeEntityList) {
        List<FormAttribute> formAttributeList = null;
        if (formAttributeEntityList != null) {
            formAttributeList = new ArrayList<>();
            for (FormAttributeEntity formAttributeEntity : formAttributeEntityList) {
                FormAttribute formAttribute = map(formAttributeEntity);
                formAttributeList.add(formAttribute);
            }
        }
        return formAttributeList;
    }
}