#!/usr/bin/env bash

set -euo pipefail

SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
REPO_ROOT=$(cd "${SCRIPT_DIR}/.." && pwd)

usage() {
    cat <<'EOF'
Usage: ./scripts/formatthecode.sh [JAVA_FILE ...]

Formats Java sources with Artistic Style using the project's historical style.

Without arguments, formats all Java files under:
  - src/main/java
  - src/test/java

With arguments, formats only the provided files.

Options:
  -h, --help        Show this help text
  --check-deps      Verify required formatter dependencies are installed
EOF
}

check_deps() {
    if ! command -v astyle >/dev/null 2>&1; then
        echo "error: required formatter 'astyle' was not found in PATH" >&2
        echo "Install Artistic Style to use ./scripts/formatthecode.sh" >&2
        return 1
    fi
}

collect_default_targets() {
    find "${REPO_ROOT}/src/main/java" "${REPO_ROOT}/src/test/java" \
        -type f -name '*.java' -print0
}

main() {
    if [[ $# -gt 0 ]]; then
        case "$1" in
            -h|--help)
                usage
                return 0
                ;;
            --check-deps)
                check_deps
                return 0
                ;;
        esac
    fi

    check_deps

    if [[ $# -gt 0 ]]; then
        local target
        for target in "$@"; do
            if [[ ! -f "${target}" ]]; then
                echo "error: file not found: ${target}" >&2
                return 1
            fi
        done

        astyle --style=java --mode=java -n "$@"
        return 0
    fi

    mapfile -d '' targets < <(collect_default_targets)

    if [[ ${#targets[@]} -eq 0 ]]; then
        echo "No Java files found to format." >&2
        return 0
    fi

    astyle --style=java --mode=java -n "${targets[@]}"
}

main "$@"