#!/usr/bin/env bash
# shellcheck shell=bash
#
# Render or execute the contributor-setup plan defined by the
# orchestrator's scripts/contributor_setup_steps.json. The welcome
# repo does not hardcode step lists.
#
# Modes:
#   default : print the plan.
#   --run   : exec `just contributor-setup` inside the orchestrator.
#
# Flags (forwarded to the orchestrator under --run):
#   --headless, --no-browser   headless mode
#   --yes                      non-interactive
#   --dry-run, --plan-only     plan-only execution
#   --skip-optional            skip optional steps
#   --only ID[,ID...]          run the listed step ids
#   -h, --help                 usage

set -euo pipefail

here="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd -P)"
# shellcheck disable=SC1091
. "$here/common.sh"

load_env

MODE="print"          # "print" or "run"
forward_args=()

usage() {
    cat <<'USAGE'
Usage: just next-steps [--run] [flags]
       just run-next-steps [flags]

Flags: --headless, --no-browser, --yes, --dry-run, --skip-optional,
       --only ID[,ID...]

Step list source: <orchestrator>/scripts/contributor_setup_steps.json.
USAGE
}

while [ $# -gt 0 ]; do
    case "$1" in
        --run)                          MODE=run; shift ;;
        --headless|--no-browser)        forward_args+=("--headless"); shift ;;
        --yes)                          forward_args+=("--yes"); shift ;;
        --dry-run|--plan-only)          forward_args+=("--dry-run"); shift ;;
        --skip-optional)                forward_args+=("--skip-optional"); shift ;;
        --only)                         shift; [ $# -gt 0 ] || die "--only needs an argument"; forward_args+=("--only" "$1"); shift ;;
        --only=*)                       forward_args+=("--only" "${1#--only=}"); shift ;;
        -h|--help)                      usage; exit 0 ;;
        --)                             shift; break ;;
        *)                              die "unexpected argument: $1 (try --help)" ;;
    esac
done

# Locate the orchestrator checkout.
workspace_root="${FORGE_WORKSPACE_ROOT:-.}"
repo_url="${FORGE_ORCHESTRATOR_REPO_URL:-}"
if [ -z "$repo_url" ]; then
    die "FORGE_ORCHESTRATOR_REPO_URL is unset; run 'just init-env' and edit .env"
fi
repo_name="$(basename "$repo_url" .git)"
orchestrator="$workspace_root/$repo_name"

if [ ! -d "$orchestrator/.git" ]; then
    err "orchestrator checkout not found at $orchestrator"
    note "required command: just clone-orchestrator"
    exit 1
fi

manifest="$orchestrator/scripts/contributor_setup_steps.json"
if [ ! -f "$manifest" ]; then
    warn "orchestrator checkout at $orchestrator does not ship"
    warn "  scripts/contributor_setup_steps.json (expected in newer versions)."
    note "falling back to the orchestrator README; open:"
    note "    $orchestrator/README.md"
    exit 1
fi

step "next steps (source: orchestrator manifest)"
note "source:       $manifest"
note "orchestrator: $orchestrator"
printf '\n' >&2

python3 - "$manifest" >&2 <<'PY'
import json, sys, textwrap
path = sys.argv[1]
with open(path, "r", encoding="utf-8") as f:
    data = json.load(f)
if data.get("schema_version") != 1:
    sys.stderr.write(
        f"[err]  unsupported contributor_setup_steps schema: "
        f"{data.get('schema_version')!r}\n"
    )
    sys.exit(2)
steps = data.get("steps", [])
total = len(steps)
for i, s in enumerate(steps, 1):
    sid   = s["id"]
    title = s.get("title", sid)
    desc  = s.get("desc", "")
    cmd   = " ".join(s["cmd"])
    opt = "" if "optional" in title.lower() else (" [optional]" if s.get("optional") else "")
    sys.stderr.write(f"  ({i}/{total}) {title}{opt}\n")
    sys.stderr.write(f"        id:  {sid}\n")
    if desc:
        for line in textwrap.wrap(desc, width=76,
                                  initial_indent="        ",
                                  subsequent_indent="        "):
            sys.stderr.write(line + "\n")
    sys.stderr.write(f"        run: {cmd}\n\n")
PY

if [ "$MODE" != "run" ]; then
    # ``${arr:+...}`` on an empty array triggers set -u; build explicitly.
    fwd_suffix=""
    if [ "${#forward_args[@]}" -gt 0 ]; then
        fwd_suffix=" ${forward_args[*]}"
    fi
    note "Command:"
    note "    just run-next-steps${fwd_suffix}"
    note "Orchestrator command:"
    note "    cd $orchestrator && just contributor-setup${fwd_suffix}"
    exit 0
fi

step "exec just contributor-setup (cwd=$orchestrator)"
if ! command -v just >/dev/null 2>&1; then
    die "just not on PATH; install via 'just doctor' fix hint"
fi
cd "$orchestrator"
exec just contributor-setup "${forward_args[@]}"