hu.blackbelt.epsilon.runtime.execution.model.excel.ExcelModelContext Maven / Gradle / Ivy

Go to download
package hu.blackbelt.epsilon.runtime.execution.model.excel;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import hu.blackbelt.epsilon.runtime.execution.api.Log;
import hu.blackbelt.epsilon.runtime.execution.api.ModelContext;
import lombok.Builder;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NonNull;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.epsilon.eol.exceptions.models.EolModelLoadingException;
import org.eclipse.epsilon.eol.models.IModel;
import org.eclipse.epsilon.eol.models.ModelReference;
import org.eclipse.epsilon.eol.models.ModelRepository;

import java.util.List;
import java.util.Map;

@Data
@Builder(builderMethodName = "excelModelContextBuilder")
@EqualsAndHashCode
public class ExcelModelContext implements ModelContext {


    public static final String EXCEL_CONFIGURATION = "excelConfiguration";
    public static final String EXCEL = "excel";
    String spreadSheetPassword;


    @NonNull
    String name;

    @Builder.Default
    List aliases = ImmutableList.of();

    @Builder.Default
    boolean readOnLoad = true;

    @Builder.Default
    boolean storeOnDisposal = false;

    @Builder.Default
    boolean cached = true;

    @NonNull
    String excel;

    String excelConfiguration;

    @Override
    public String toString() {
        return "ExcelModelContext{" +
                "artifacts='" + getArtifacts() + '\'' +
                ", name='" + name + '\'' +
                ", aliases=" + aliases +
                ", uriConverterMap='" + getUriConverterMap() + '\'' +
                ", readOnLoad=" + readOnLoad +
                ", storeOnDisposal=" + storeOnDisposal +
                ", cached=" + cached +
                ", spreadSheetPassword='" + spreadSheetPassword + '\'' +
                '}';
    }

    @Override
    public void addAliases(ModelRepository repository, ModelReference ref) {
        ref.setName(this.getName());
        if (this.getAliases() != null) {
            for (String alias : this.getAliases()) {
                ref.getAliases().add(alias);
            }
        }
        repository.addModel(ref);
    }

    @Override
    public Map getArtifacts() {
        return ImmutableMap.of(EXCEL, excel, EXCEL_CONFIGURATION,  excelConfiguration);
    }

    @Override
    public Map getUriConverterMap() {
        return ImmutableMap.of();
    }

    @Override
    public IModel load(Log log, ResourceSet resourceSet, ModelRepository repository, Map uri, Map uriMap) throws EolModelLoadingException {
        return ExcelModelUtil.loadExcel(log, repository, this, uri.get(EXCEL), uri.get(EXCEL_CONFIGURATION));
    }

}