#!/bin/bash

# ----------------------------------------------------------------------
# COLORS
# ----------------------------------------------------------------------

GRAY="\e[0;30m"
RED="\e[0;31m"
GREEN="\e[0;32m"
YELLOW="\e[0;33m"
BLUE="\e[0;34m"
PURPLE="\e[0;35m"
CYAN="\e[0;36m"
WHITE="\e[0;37m"

B_GRAY="\e[1;30m"
B_RED="\e[1;31m"
B_GREEN="\e[1;32m"
B_YELLOW="\e[1;33m"
B_BLUE="\e[1;34m"
B_PURPLE="\e[1;35m"
B_CYAN="\e[1;36m"
B_WHITE="\e[1;37m"

RESET="\e[0m"

# ----------------------------------------------------------------------
# CONFIGURATION
# ----------------------------------------------------------------------

NAME="computor"
BINARY="./$NAME"

if [[ ! -x "$BINARY" ]]; then
    echo -e >&2 $B_RED"Error:$RESET binary $B_PURPLE'$BINARY'$RESET not found or not executable."
    echo >&2 "Did you compile your C program?"
    exit 1
fi

# flags
SKIP_REDUCED=false
SKIP_SOLUTION=false
RUN_ONLY=false
FLAGS="" # default empty

# Parse command-line arguments
while [[ "$#" -gt 0 ]]; do
    case $1 in
        --skip-reduced) SKIP_REDUCED=true ;;
        --skip-solution) SKIP_SOLUTION=true ;;
        --run-only) RUN_ONLY=true ;;
        -b) FLAGS="$FLAGS -b" ;; # append -b to FLAGS
        -d) FLAGS="$FLAGS -d" ;; # append -d to FLAGS
        *) echo "Unknown parameter passed: $1"; exit 1 ;;
    esac
    shift
done

# ----------------------------------------------------------------------
# TEST LOGIC
# ----------------------------------------------------------------------

run_test() {
    local test_name="$1"
    local input="$2"
    local expected="$3"
    local check_error="${4:-no_error}"  # Default: false (compare output)

    # If RUN_ONLY is true, skip all test logic and just run the binary
    if [[ "$RUN_ONLY" == "true" ]]; then
        printf $B_PURPLE"Running: '%s'$RESET\n" "$test_name"
        printf "Input: '%s'\n" "$input"
        printf "Output:\n"
        printf "%s" "$input" | "$BINARY" $FLAGS 2>&1
        printf "\n"
        return
    fi

    printf $B_PURPLE"Test : '%s'$RESET - " "$test_name"

    # Run the program and capture the output
    out=$(printf "%s" "$input" | "$BINARY" $FLAGS 2>&1)
    exit_code=$?

    # If SKIP_REDUCED is true, remove the "Reduced form" line from both output and expected
    if [[ "$SKIP_REDUCED" == "true" ]]; then
        out=$(echo "$out" | grep -v "Reduced form:")
        expected=$(echo "$expected" | grep -v "Reduced form:")
    fi

    # If SKIP_SOLUTION is true, remove all lines after the 2nd line following "Polynomial degree:"
    if [[ "$SKIP_SOLUTION" == "true" ]]; then
        # Get the line number of "Polynomial degree:"
        line_num=$(echo "$out" | grep -n "Polynomial degree:" | head -1 | cut -d: -f1)
        if [[ -n "$line_num" ]]; then
            # Keep lines up to 1 lines after "Polynomial degree:"
            out=$(echo "$out" | head -n $((line_num + 1)))
            expected=$(echo "$expected" | head -n $((line_num + 1)))
        fi
    fi

    # If check_error is true, verify if the output contains an error message or non-zero exit code
    if [[ "$check_error" == "error" ]]; then
        if [[ $exit_code -ne 0 || "$out" == *"Error"* ]]; then
            printf $B_GREEN"PASS$RESET (Input: '%s')\n" "$input"
        else
            printf $B_RED"FAIL$RESET\n"
            printf "  "$B_WHITE"Input:$RESET '%s'\n" "$input"
            printf "  "$B_WHITE"Got:$RESET\n"
            printf '%s\n' "$out" | sed 's/^/    | /'
            printf "  "$B_WHITE"Expected:$RESET an error (non-zero exit code or 'Error' in output: $exit_code)\n"
            echo
        fi
    else
        # Compare with expected output
        if [[ "$out" == "$expected" ]]; then
            printf $B_GREEN"PASS$RESET (Input: '%s')\n" "$input"
        else
            printf $B_RED"FAIL$RESET\n"
            printf "  "$B_WHITE"Input:$RESET '%s'\n" "$input"
            printf "  "$B_WHITE"Got:$RESET\n"
            printf '%s\n' "$out" | sed 's/^/    | /'
            printf "  "$B_WHITE"Expected:$RESET\n"
            printf '%s\n' "$expected" | sed 's/^/    | /'
            echo
        fi
    fi
}

# ----------------------------------------------------------------------
# TESTS CASES
# ----------------------------------------------------------------------

run_test \
"1. degree 2" \
"3 * x^2 + 5 * x^1 - 2 * x^0 = 5 * x^1" "\
Reduced form: 2 * x^0 + 0 * x^1 + 3 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly negative, the two complex solutions are:
-√(2/3)
√(2/3)"

run_test \
"2. degree 2" \
"3.4 * x^2 + 1 * x^1 - 2.0 * x^0 = 5 * x^1" "\
Reduced form: 2 * x^0 - 4 * x^1 + 3.4 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly negative, the two complex solutions are:
4/6.8 + 3.34664i/6.8
4/6.8 - 3.34664i/6.8"

run_test \
"3. degree 2" \
"3 * x^2 + 2 * x^2 = 5 * x^1" "\
Reduced form: 0 * x^0 - 5 * x^1 + 5 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly positive, the two solutions are:
13
-12"

run_test \
"4. flag -e" \
'-e "3 * x^2 + 2 * x - 7 * x^4 = 1 * x^4"' \
"" \
error

run_test \
"5. degree 2" \
"3*x^2 + 2x = 0" "\
Reduced form: 0 * x^0 + 2 * x^1 + 3 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly positive, the two solutions are:
2.66667
-3.33333"

run_test \
"6. degree 2" \
"3.4 * x^2 + 1 * x^1 - 2.0 * x^0 = 5.123 * x^1" "\
Reduced form: 2 * x^0 - 4.123 * x^1 + 3.4 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly negative, the two complex solutions are:
4.123/6.8 + 3.19388i/6.8
4.123/6.8 - 3.19388i/6.8"

run_test \
"7. float exponent" \
"3.4 * x^2 + 1 * x^1 - 2.0 * x^0 = 5 * x^1.2" \
"" \
error

run_test \
"8. degree 4" \
"3x^2 + 2x -7x^4 = x^4" "\
Reduced form: 0 * x^0 + 2 * x^1 + 3 * x^2 + 6 * x^4 = 0
Polynomial degree: 4
The polynomial degree is strictly greater than 2, I can't solve."

run_test \
"9. degree 2" \
"3x² + 2x -7x¹ = x" "\
Reduced form: 0 * x^0 - 6 * x^1 + 3 * x^2 = 0
Polynomial degree: 2
Discriminant is strictly positive, the two solutions are:
0
2"

# run_test \
# "10. degree 2" \
# "3 * x^2 + 2 * x - 7 * x^1 = x" "\
# Reduced form: 0 * x^0 - 6 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 2
# 0"

# run_test \
# "11. degree 2" \
# "-3x² + 2x -7 = x" "\
# Reduced form: -7 * x^0 + 1 * x^1 - 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly negative, the two complex solutions are:
# 0.166667 - 1.51841*i
# 0.166667 + 1.51841*i"

# run_test \
# "12. degree 2" \
# "+3x² + 2x -7 = x" "\
# Reduced form: -7 * x^0 + 1 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 1.36992
# -1.70326"

# run_test \
# "13. degree 2" \
# "3x² + 0x -7 = x" "\
# Reduced form: -7 * x^0 - 1 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 1.70326
# -1.36992"

# run_test \
# "14. degree 2" \
# "3x² + 0x -0 = x" "\
# Reduced form: 0 * x^0 - 1 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 0.333333
# 0"

# run_test \
# "15. degree 2" \
# "3x² + 2x -0 = x" "\
# Reduced form: 0 * x^0 + 1 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 0
# -0.333333"

# run_test \
# "16. degree 1" \
# "3x + 2x -0 = x" "\
# Reduced form: 0 * x^0 + 4 * x^1 = 0
# Polynomial degree: 1
# The solution is:
# 0"

# run_test \
# "17. degree 2" \
# "3x² + x -0 = x" "\
# Reduced form: 0 * x^0 + 0 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Radicant is equal to zero, the solution is:
# 0"

# run_test \
# "18. degree 2" \
# "0x² + x -0 = x" "\
# Reduced form: 0 * x^0 = 0
# Any real number is a solution."

# run_test \
# "19. degree 5" \
# "2x⁵ + x -0 = -7x^5" "\
# Reduced form: 0 * x^0 + 1 * x^1 + 0 * x^2 + 0 * x^3 + 0 * x^4 + 9 * x^5 = 0
# Polynomial degree: 5
# The polynomial degree is strictly greater than 2, I can't solve."

# run_test \
# "20. degree 1" \
# "2x + x -0 = -7x" "\
# Reduced form: 0 * x^0 + 10 * x^1 = 0
# Polynomial degree: 1
# The solution is:
# 0"

# run_test \
# "21. degree 1" \
# "2x + x -3 = -7x" "\
# Reduced form: -3 * x^0 + 10 * x^1 = 0
# Polynomial degree: 1
# The solution is:
# 0.3"

# run_test \
# "22. degree 1" \
# "-2x + x -3 = -7x" "\
# Reduced form: -3 * x^0 + 6 * x^1 = 0
# Polynomial degree: 1
# The solution is:
# 0.5"

# run_test \
# "23. degree 2 without [=]" \
# "3x^2" \
# "" \
# error

# run_test \
# "24. degree 2" \
# "3x^2 = 0" "\
# Reduced form: 0 * x^0 + 0 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Radicant is equal to zero, the solution is:
# 0"

# run_test \
# "25. degree 2" \
# "3x^2 + 2 = 0" "\
# Reduced form: 2 * x^0 + 0 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly negative, the two complex solutions are:
# i√(2/3)
# -i√(2/3)"

# run_test \
# "26. degree 2 pure" \
# "3 * x^2 + 5 * x^1 - 2 * x^0 = 5 * x" "\
# Reduced form: -2 * x^0 + 0 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly positive, the two solutions are:
# √(2/3)
# -√(2/3)"

# run_test \
# "27. degree 2 pure" \
# "9 * x^2 + 5 * x^1 - 2 * x^0 = 5 * x" "\
# Reduced form: -2 * x^0 + 0 * x^1 + 9 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly positive, the two solutions are:
# √(2)/3
# -√(2)/3"

# run_test \
# "28. degree 2 pure" \
# "3 * x^2 + 5 * x^1 - 4 * x^0 = 5 * x" "\
# Reduced form: -4 * x^0 + 0 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly positive, the two solutions are:
# 2/√(3)
# -2/√(3)"

# run_test \
# "29. degree 2 pure" \
# "16 * x^2 + 5 * x^1 - 4 * x^0 = 5 * x" "\
# Reduced form: -4 * x^0 + 0 * x^1 + 16 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly positive, the two solutions are:
# 1/2
# -1/2"

# run_test \
# "30. degree 2 pure" \
# "4 * x^2 + 5 * x^1 - 16 * x^0 = 5 * x" "\
# Reduced form: -16 * x^0 + 0 * x^1 + 4 * x^2 = 0
# Polynomial degree: 2
# Radicant is strictly positive, the two solutions are:
# 2
# -2"

# run_test \
# "31. degree 2" \
# "3x² + 2x -7 = x" "\
# Reduced form: -7 * x^0 + 1 * x^1 + 3 * x^2 = 0
# Polynomial degree: 2
# Discriminant is strictly positive, the two solutions are:
# 1.36992
# -1.70326"