package carpet.script.argument;

import carpet.script.CarpetContext;
import carpet.script.Context;
import carpet.script.LazyValue;
import carpet.script.bundled.Module;
import carpet.script.exception.InternalExpressionException;
import carpet.script.value.FunctionValue;
import carpet.script.value.Value;

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

public class FunctionArgument extends Argument
{
    public FunctionValue function;
    public List<LazyValue> args;

    private FunctionArgument(FunctionValue function, int offset, List<LazyValue> args)
    {
        super(offset);
        this.function = function;
        this.args = args;
    }
    public static FunctionArgument findIn(Context c, Module module, List<LazyValue> params, int offset)
    {
        return findIn(c, module, params, offset, false);
    }
    public static FunctionArgument findIn(Context c, Module module, List<LazyValue> params, int offset, boolean strict)
    {
        Value functionValue = params.get(offset).evalValue(c);
        if (!(functionValue instanceof FunctionValue))
        {
            String name = functionValue.getString();
            functionValue = c.host.getAssertFunction(module, name);
        }
        FunctionValue fun = (FunctionValue)functionValue;
        int argsize = fun.getArguments().size();
        int extraargs = params.size() - argsize - offset - 1;
        if (extraargs < 0 || (extraargs > 0 && strict))
            throw new InternalExpressionException("Function "+fun.getPrettyString()+" requires "+fun.getArguments().size()+" arguments");
        List<LazyValue> lvargs = new ArrayList<>();
        for (int i=0; i< argsize; i++)
        {
            lvargs.add(params.get(offset+1+i));
        }
        return new FunctionArgument(fun, offset+1+argsize, lvargs);
    }

    public List<Value> resolveArgs(Context context, Integer type)
    {
        List<Value> res = new ArrayList<>(args.size());
        for (LazyValue arg : args) res.add(arg.evalValue(context, type));
        return res;
    }
}