package expression.exceptions;

import base.Functional;
import base.Named;
import base.Pair;
import expression.ToMiniString;
import expression.Variable;
import expression.common.ExpressionKind;
import expression.parser.ParserTestSet;

import java.util.ArrayList;
import java.util.List;
import java.util.function.LongBinaryOperator;

/**
 * @author Georgiy Korneev (kgeorgiy@kgeorgiy.info)
 */
public class ExceptionsTestSet<E extends ToMiniString, C> extends ParserTestSet<E, C> {
    private static final int D = 5;
    private static final List<Integer> OVERFLOW_VALUES = new ArrayList<>();
    private final char[] CHARS = "AZ+-*%()[]<>".toCharArray();

    static {
        Functional.addRange(OVERFLOW_VALUES, D, Integer.MIN_VALUE + D);
        Functional.addRange(OVERFLOW_VALUES, D, Integer.MIN_VALUE / 2);
        Functional.addRange(OVERFLOW_VALUES, D, (int) -Math.sqrt(Integer.MAX_VALUE));
        Functional.addRange(OVERFLOW_VALUES, D, 0);
        Functional.addRange(OVERFLOW_VALUES, D, (int) Math.sqrt(Integer.MAX_VALUE));
        Functional.addRange(OVERFLOW_VALUES, D, Integer.MAX_VALUE / 2);
        Functional.addRange(OVERFLOW_VALUES, D, Integer.MAX_VALUE - D);
    }

    private final List<Named<String>> parsingTest;

    public ExceptionsTestSet(final ExceptionsTester tester, final ParsedKind<E, C> kind) {
        super(tester, kind, false);
        parsingTest = tester.parsingTest;
    }

    private void testParsingErrors() {
        counter.testForEach(parsingTest, op -> {
            final List<String> names = Functional.map(kind.kind().variables().generate(counter.random(), 3), Pair::first);
            final String expr = mangle(op.value(), names);
            try {
                kind.parse(expr, names);
                counter.fail("Successfully parsed '%s'", op.value());
            } catch (final Exception e) {
                counter.format("%-30s %s%n", op.name(), e.getClass().getSimpleName() + ": " + e.getMessage());
            }
        });
    }

    private void testOverflow() {
        final List<Pair<String, E>> variables = kind.kind().variables().generate(counter.random(), 3);
        final List<String> names = Functional.map(variables, Pair::first);
        final Variable vx = (Variable) variables.get(0).second();
        final Variable vy = (Variable) variables.get(1).second();

        //noinspection Convert2MethodRef
        testOverflow(names, (a, b) -> a + b, "+", new CheckedAdd(vx, vy));
        testOverflow(names, (a, b) -> a - b, "-", new CheckedSubtract(vx, vy));
        testOverflow(names, (a, b) -> a * b, "*", new CheckedMultiply(vx, vy));
        testOverflow(names, (a, b) -> b == 0 ? Long.MAX_VALUE : a / b, "/", new CheckedDivide(vx, vy));
        testOverflow(names, (a, b) -> -b, "<- ignore first argument, unary -", new CheckedNegate(vy));
    }

    private void testOverflow(final List<String> names, final LongBinaryOperator f, final String op, final Object expression) {
        final ExpressionKind<E, C> kind = this.kind.kind();
        for (final int a : OVERFLOW_VALUES) {
            for (final int b : OVERFLOW_VALUES) {
                final long expected = f.applyAsLong(a, b);
                final boolean isInt = Integer.MIN_VALUE <= expected && expected <= Integer.MAX_VALUE;
                try {
                    final C actual = kind.evaluate(
                            kind.cast(expression),
                            names,
                            kind.fromInts(List.of(a, b, 0))
                    );
                    counter.checkTrue(
                            isInt && kind.fromInt((int) expected).equals(actual),
                            "%d %s %d == %d", a, op, b, actual
                    );
                } catch (final Exception e) {
                    if (isInt) {
                        counter.fail(e, "Unexpected error in %d %s %d", a, op, b);
                    }
                }
            }
        }
    }

    @Override
    protected void test() {
        counter.scope("Overflow tests", (Runnable) this::testOverflow);
        super.test();
        counter.scope("Parsing error tests", this::testParsingErrors);
    }


    @Override
    protected E parse(final String expression, final List<String> variables, final boolean reparse) {
        final String expr = expression.strip();
        if (expr.length() > 10) {
            for (final char ch : CHARS) {
                for (int i = 0; i < 10; i++) {
                    final int index = 1 + tester.random().nextInt(expr.length() - 2);
                    int pi = index - 1;
                    while (Character.isWhitespace(expr.charAt(pi))) {
                        pi--;
                    }
                    int ni = index;
                    while (Character.isWhitespace(expr.charAt(ni))) {
                        ni++;
                    }
                    final char pc = expr.charAt(pi);
                    final char nc = expr.charAt(ni);
                    if (
                            "-([{*∛√²³₂₃!‖⎵⎴⌊⌈=?".indexOf(nc) < 0 &&
                            (!Character.isLetterOrDigit(pc) || !Character.isLetterOrDigit(ch)) &&
                            nc != ch && pc != ch &&
                            !Character.isLetterOrDigit(nc) && nc != '$'
                    ) {
                        shouldFail(
                                variables,
                                "Parsing error expected for " + insert(expr, index, "<ERROR_INSERTED -->" + ch + "<-- ERROR_INSERTED>"),
                                insert(expr, index, String.valueOf(ch))
                        );
                        break;
                    }
                }
            }
            parens(variables, expr, '[', ']');
            parens(variables, expr, '{', '}');
        }

        return counter.testV(() -> counter.call("parse", () -> kind.parse(expr, variables)));
    }

    private static String insert(final String expr, final int index, final String value) {
        return expr.substring(0, index) + value + expr.substring(index);
    }

    private void parens(final List<String> variables, final String expr, final char open, final char close) {
        if (expr.indexOf(open) >= 0) {
            replaces(variables, expr, open, '(');
            replaces(variables, expr, close, ')');
            if (expr.indexOf('(') >= 0) {
                replaces(variables, expr, '(', open);
                replaces(variables, expr, ')', close);
            }
        }
    }

    private void replaces(final List<String> variables, final String expr, final char what, final char by) {
        final String input = expr.replace(what, by);
        shouldFail(variables, "Unmatched parentheses: " + input, input);
    }

    private void shouldFail(final List<String> variables, final String message, final String input) {
        counter.shouldFail(message, () -> kind.parse(input, variables));
    }
}