package io.github.vmzakharov.ecdataframe.dsl;

import io.github.vmzakharov.ecdataframe.dataset.AvroDataSet;
import io.github.vmzakharov.ecdataframe.dsl.value.Value;
import org.eclipse.collections.api.RichIterable;
import org.eclipse.collections.api.map.MutableMap;
import org.eclipse.collections.impl.factory.Maps;

public class SimpleEvalContext
extends EvalContextAbstract
{
    private final MutableMap<String, AvroDataSet> dataSets = Maps.mutable.of();

    @Override
    public Value getVariable(String newVariableName)
    {
        Value value = this.getContextVariables().get(newVariableName);

        if (value == null)
        {
            throw new RuntimeException("Uninitialized variable: " + newVariableName);
        }

        return value;
    }

    @Override
    public Value getVariableOrDefault(String newVariableName, Value defaultValue)
    {
        Value value = this.getContextVariables().get(newVariableName);

        return (value == null) ? defaultValue : value;
    }

    @Override
    public void addDataSet(AvroDataSet dataSet)
    {
        this.dataSets.put(dataSet.getName(), dataSet);
    }

    @Override
    public AvroDataSet getDataSet(String dataSetName)
    {
        return this.dataSets.get(dataSetName);
    }

    @Override
    public RichIterable<String> getVariableNames()
    {
        return this.getContextVariables().keysView();
    }
}