package au.id.zancanaro.javacheck.state.queue.commands;

import au.id.zancanaro.javacheck.Generator;
import au.id.zancanaro.javacheck.state.Command;
import au.id.zancanaro.javacheck.state.CommandValue;
import au.id.zancanaro.javacheck.state.queue.QueueState;

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

public class Offer<T> extends Command<QueueState<T>, T, Boolean> {
    public final Generator<T> generator;

    public Offer(Generator<T> argGen) {
        this.generator = argGen;
    }

    @Override
    public Generator<T> argsGenerator(QueueState<T> state) {
        return generator;
    }

    @Override
    public boolean preCondition(QueueState<T> state, T args) {
        return state != null;
    }

    @Override
    public Boolean runCommand(QueueState<T> state, T args) {
        return state.getConcreteQueue().offer(args);
    }

    @Override
    public QueueState<T> nextState(QueueState<T> state, T args, CommandValue<Boolean> result) {
        List<T> newState = new ArrayList<>(state.getAbstractQueue());
        newState.add(args);
        return state.withAbstractQueue(newState);
    }
}