#!/bin/bash

set -euo pipefail

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

ROOT_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"

CONFIG_PATH="${ROOT_DIR}/cluster-context/madagascar.json"

CLUSTER_NAME="madagascar"

COMMAND="install"

REMOTE_USER="root"

DRY_RUN=0

PROJECT_NAME=""

PROJECT_DIR=""

DEPLOY_MODE=""

NODE_FILTERS=()

TARGETS=()

usage() {

    cat <<EOF

Usage: $0 <project> [command] [options]

Commands:

  install      Deploy/install project on selected nodes (default)

  uninstall    Remove project from selected nodes

  status       Query project status on selected nodes (deploy.sh projects only)

  start        Start services on selected nodes (deploy.sh projects only)

  restart      Restart services on selected nodes (deploy.sh projects only)

  stop         Stop services on selected nodes (deploy.sh projects only)

Options:

  --cluster <name>    Cluster name in cluster-context/madagascar.json (default: madagascar)

  --node <name|ip>    Restrict to one node. Can be repeated.

  --user <user>       Remote SSH user for setup.sh projects (default: root)

  --dry-run           Show resolved targets and commands without executing

  -h, --help          Show this help

Examples:

  $0 pve-guests-state

  $0 pve-guests-state install --node ebony

  $0 autoNAS install

  $0 autoSMART status --node 192.168.2.92

EOF

}

require_config() {

    if [[ ! -f "${CONFIG_PATH}" ]]; then

        echo "ERROR: missing cluster config: ${CONFIG_PATH}" >&2

        exit 1

    fi

}

require_project() {

    PROJECT_DIR="${ROOT_DIR}/projects/${PROJECT_NAME}"

    if [[ ! -d "${PROJECT_DIR}" ]]; then

        echo "ERROR: unknown project: ${PROJECT_NAME}" >&2

        exit 1

    fi

    if [[ -x "${PROJECT_DIR}/setup.sh" ]]; then

        DEPLOY_MODE="setup"

        return

    fi

    if [[ -x "${PROJECT_DIR}/deploy.sh" ]]; then

        DEPLOY_MODE="deploy"

        return

    fi

    echo "ERROR: project ${PROJECT_NAME} has neither setup.sh nor deploy.sh" >&2

    exit 1

}

load_targets() {

    local entry=""

    TARGETS=()

    while IFS= read -r entry; do

        [[ -n "${entry}" ]] && TARGETS+=("${entry}")

    done < <(

        jq -r --arg cluster "${CLUSTER_NAME}" '

            .clusters[$cluster].nodes

            | to_entries[]

            | .key + "\t" + ((.value.ip // .value.wan.vmbr443.address // empty) | split("/")[0])

        ' "${CONFIG_PATH}"

    )

    if [[ ${#TARGETS[@]} -eq 0 ]]; then

        echo "ERROR: no targets found for cluster ${CLUSTER_NAME}" >&2

        exit 1

    fi

}

match_filter() {

    local filter="$1"

    local node_name="$2"

    local node_ip="$3"

    [[ "${filter}" == "${node_name}" || "${filter}" == "${node_ip}" ]]

}

filter_targets() {

    local filtered=()

    local entry=""

    local filter=""

    local node_name=""

    local node_ip=""

    if [[ ${#NODE_FILTERS[@]} -eq 0 ]]; then

        return

    fi

    for entry in "${TARGETS[@]}"; do

        node_name="${entry%%$'\t'*}"

        node_ip="${entry#*$'\t'}"

        for filter in "${NODE_FILTERS[@]}"; do

            if match_filter "${filter}" "${node_name}" "${node_ip}"; then

                filtered+=("${entry}")

                break

            fi

        done

    done

    TARGETS=("${filtered[@]}")

    if [[ ${#TARGETS[@]} -eq 0 ]]; then

        echo "ERROR: no targets matched the provided --node filters" >&2

        exit 1

    fi

}

run_setup_project() {

    local node_name="$1"

    local node_ip="$2"

    local cmd=()

    case "${COMMAND}" in

        install)

            cmd=(bash "${PROJECT_DIR}/setup.sh" --user "${REMOTE_USER}" "${node_ip}")

            ;;

        uninstall)

            cmd=(bash "${PROJECT_DIR}/setup.sh" --user "${REMOTE_USER}" --uninstall "${node_ip}")

            ;;

        *)

            echo "ERROR: command ${COMMAND} is not supported for setup.sh-only projects" >&2

            exit 1

            ;;

    esac

    echo "==> ${PROJECT_NAME}: ${COMMAND} on ${node_name} (${node_ip})"

    if [[ "${DRY_RUN}" -eq 1 ]]; then

        printf 'DRY-RUN:'

        printf ' %q' "${cmd[@]}"

        echo

        return

    fi

    (cd "${PROJECT_DIR}" && "${cmd[@]}")

}

run_deploy_project() {

    local node_name="$1"

    local node_ip="$2"

    local cmd=(bash "${PROJECT_DIR}/deploy.sh" "${COMMAND}" "${node_ip}")

    echo "==> ${PROJECT_NAME}: ${COMMAND} on ${node_name} (${node_ip})"

    if [[ "${DRY_RUN}" -eq 1 ]]; then

        printf 'DRY-RUN:'

        printf ' %q' "${cmd[@]}"

        echo

        return

    fi

    (cd "${PROJECT_DIR}" && "${cmd[@]}")

}

parse_args() {

    if [[ $# -lt 1 ]]; then

        usage

        exit 1

    fi

    PROJECT_NAME="$1"

    shift

    if [[ $# -gt 0 ]]; then

        case "$1" in

            install|uninstall|status|start|restart|stop)

                COMMAND="$1"

                shift

                ;;

        esac

    fi

    while [[ $# -gt 0 ]]; do

        case "$1" in

            --cluster)

                CLUSTER_NAME="$2"

                shift 2

                ;;

            --node)

                NODE_FILTERS+=("$2")

                shift 2

                ;;

            --user)

                REMOTE_USER="$2"

                shift 2

                ;;

            --dry-run)

                DRY_RUN=1

                shift

                ;;

            -h|--help)

                usage

                exit 0

                ;;

            *)

                echo "ERROR: unknown option: $1" >&2

                usage

                exit 1

                ;;

        esac

    done

}

main() {

    parse_args "$@"

    require_config

    require_project

    load_targets

    filter_targets

    echo "Project: ${PROJECT_NAME}"

    echo "Mode: ${DEPLOY_MODE}"

    echo "Command: ${COMMAND}"

    echo "Cluster: ${CLUSTER_NAME}"

    echo "Targets:"

    printf '  %s\n' "${TARGETS[@]}"

    echo

    local entry=""

    local node_name=""

    local node_ip=""

    for entry in "${TARGETS[@]}"; do

        node_name="${entry%%$'\t'*}"

        node_ip="${entry#*$'\t'}"

        if [[ "${DEPLOY_MODE}" == "setup" ]]; then

            run_setup_project "${node_name}" "${node_ip}"

        else

            run_deploy_project "${node_name}" "${node_ip}"

        fi

        echo

    done

}

main "$@"