wbgui

#!/usr/bin/env bash
# wbgui — interactive TUI menu for WBeam
# Navigation: arrow keys or j/k.  Enter = select.  q/Esc = quit.
set -euo pipefail

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

if [[ ! -x "$WBEAM" ]]; then
  echo "wbeam not found or not executable: $WBEAM" >&2
  exit 1
fi

MENU_ENTRIES=()
selectable=()
n=0
sel_pos=0

ANDROID_SERIALS=()
ANDROID_APIS=()
ANDROID_LABELS=()

collect_android_devices() {
  ANDROID_SERIALS=()
  ANDROID_APIS=()
  ANDROID_LABELS=()

  if ! command -v adb >/dev/null 2>&1; then
    return 1
  fi

  adb start-server >/dev/null 2>&1 || true

  # -l gives transport/product/model fields; tr -d '\r' strips CR from ADB on
  #  USB firmwares that return CRLF — without it state=="device\r" never matches.
  while IFS= read -r line; do
    # strip trailing CR
    line="${line%$'\r'}"
    [[ -z "$line" ]] && continue
    [[ "$line" == "List of devices attached" ]] && continue
    [[ "$line" == "* daemon"* ]] && continue

    # first field = serial, second = state; rest are key:value from -l
    serial="$(awk '{print $1}' <<<"$line")"
    state="$(awk '{print $2}' <<<"$line")"

    [[ -z "$serial" || "$state" != "device" ]] && continue

    # Use || true on every getprop call — old devices (API 17) may return
    # non-zero exit codes which would abort the loop under set -euo pipefail.
    api="$(adb -s "$serial" shell getprop ro.build.version.sdk 2>/dev/null | tr -d '\r[:space:]' || true)"
    release="$(adb -s "$serial" shell getprop ro.build.version.release 2>/dev/null | tr -d '\r[:space:]' || true)"
    model="$(adb -s "$serial" shell getprop ro.product.model 2>/dev/null | tr -d '\r\n' || true)"
    model="${model:-unknown-model}"
    api="${api:-?}"
    release="${release:-?}"

    ANDROID_SERIALS+=("$serial")
    ANDROID_APIS+=("$api")
    ANDROID_LABELS+=("$model  android $release  api $api  $serial")
  done < <(adb devices -l 2>/dev/null || true)

  [[ ${#ANDROID_SERIALS[@]} -gt 0 ]]
}

auto_pipeline_for_api() {
  local api="$1"
  # modern pipeline builds with minSdk=19, so API 18 must stay on legacy
  if [[ "$api" =~ ^[0-9]+$ ]] && (( api <= 18 )); then
    echo "legacy"
  else
    echo "modern"
  fi
}

auto_pipeline_for_all_devices() {
  local min_api=999
  local api
  for api in "${ANDROID_APIS[@]}"; do
    if [[ "$api" =~ ^[0-9]+$ ]] && (( api < min_api )); then
      min_api="$api"
    fi
  done
  auto_pipeline_for_api "$min_api"
}

pick_android_device() {
  local n i
  n=${#ANDROID_SERIALS[@]}

  while true; do
    tput clear >/dev/tty
    printf '\033[1;36m  WBeam: Android deploy target\033[0m\n' >/dev/tty
    printf '\033[90m  wybierz urzadzenie (numer), a=all, q=anuluj\033[0m\n\n' >/dev/tty

    for (( i=0; i<n; i++ )); do
      printf '  %d) %s\n' "$((i + 1))" "${ANDROID_LABELS[$i]}" >/dev/tty
    done
    printf '\n  a) all devices (adb default)\n  q) anuluj\n\n> ' >/dev/tty

    choice=""
    IFS= read -r choice </dev/tty
    case "$choice" in
      q|Q)
        return 1
        ;;
      a|A|"")
        SELECTED_ANDROID_SERIAL=""
        SELECTED_ANDROID_API=""
        return 0
        ;;
      *)
        if [[ "$choice" =~ ^[0-9]+$ ]] && (( choice >= 1 && choice <= n )); then
          idx=$((choice - 1))
          SELECTED_ANDROID_SERIAL="${ANDROID_SERIALS[$idx]}"
          SELECTED_ANDROID_API="${ANDROID_APIS[$idx]}"
          return 0
        fi
        ;;
    esac
  done
}

pick_android_pipeline() {
  local auto_mode="$1"

  while true; do
    tput clear >/dev/tty
    printf '\033[1;36m  WBeam: Android pipeline\033[0m\n' >/dev/tty
    printf '\033[90m  wybierz pipeline buildu\033[0m\n\n' >/dev/tty
    printf '  1) auto   (api-based, default: %s)\n' "$auto_mode" >/dev/tty
    printf '  2) modern (minSdk 19)\n' >/dev/tty
    printf '  3) legacy (minSdk 17)\n' >/dev/tty
    printf '  q) anuluj\n\n> ' >/dev/tty

    choice=""
    IFS= read -r choice </dev/tty
    case "$choice" in
      q|Q)
        return 1
        ;;
      1|"")
        SELECTED_ANDROID_PIPELINE="$auto_mode"
        return 0
        ;;
      2)
        SELECTED_ANDROID_PIPELINE="modern"
        return 0
        ;;
      3)
        SELECTED_ANDROID_PIPELINE="legacy"
        return 0
        ;;
    esac
  done
}

menu_rebuild_selectable() {
  n=${#MENU_ENTRIES[@]}
  selectable=()
  local i
  for (( i=0; i<n; i++ )); do
    [[ "${MENU_ENTRIES[$i]}" == ──* ]] || selectable+=("$i")
  done
}

menu_device_letter() {
  local idx="$1"
  if (( idx < 26 )); then
    printf "\\$(printf '%03o' $((65 + idx)))"
  else
    printf "%d" "$((idx + 1))"
  fi
}

build_menu_entries() {
  MENU_ENTRIES=(
    "──── Android ────────────────────────────────────────"
    "  Deploy  (build + install + launch):android:deploy"
  )

  if collect_android_devices; then
    local i letter label
    for (( i=0; i<${#ANDROID_SERIALS[@]}; i++ )); do
      letter="$(menu_device_letter "$i")"
      label="${ANDROID_LABELS[$i]}"
      MENU_ENTRIES+=("  Deploy Device ${letter}  (${label}):android:deploy-device|${ANDROID_SERIALS[$i]}|${ANDROID_APIS[$i]}")
    done
  fi

  MENU_ENTRIES+=(
    "  Build APK:android:build"
    "  Install APK:android:install"
    "  Launch App:android:launch"
    "  Build APK (release):android:build-release"
    "  Install APK (release):android:install-release"
    "  Deploy APK (release):android:deploy-release"
    "  Rebuild All  (host + legacy + modern):android:build-all"
    "──── Host ───────────────────────────────────────────────"
    "  Run Host  (release, foreground):host:run"
    "  Run Host  (debug + coloured logs):host:debug"
    "  Host Status:host:status"
    "──── Network ────────────────────────────────────────"
    "  Transport Up  (adb reverse / RNDIS auto-detect):ip:up"
    "  Transport Down  (remove adb reverse):ip:down"
    "  Transport Status:ip:status"
    "──────────────────────────────────────────────────────"
    "  Quit:__quit__:__quit__"
  )

  menu_rebuild_selectable
}

build_menu_entries

# ── terminal setup ────────────────────────────────────────────────────────────
if [[ ! -t 1 ]]; then
  echo "wbgui requires an interactive terminal" >&2
  exit 1
fi

saved_stty="$(stty -g </dev/tty 2>/dev/null || true)"
tty_restored=0

cleanup() {
  if [[ "$tty_restored" -eq 1 ]]; then
    return
  fi
  tty_restored=1
  # Restore terminal unconditionally
  tput rmcup  >/dev/tty 2>/dev/null || true
  tput cnorm  >/dev/tty 2>/dev/null || true
  if [[ -n "$saved_stty" ]]; then
    stty "$saved_stty" </dev/tty 2>/dev/null || stty sane </dev/tty 2>/dev/null || true
  else
    stty sane </dev/tty 2>/dev/null || true
  fi
}
trap cleanup EXIT INT TERM HUP

tput smcup  >/dev/tty
tput civis  >/dev/tty
stty -echo cbreak </dev/tty

# ── render ────────────────────────────────────────────────────────────────────
render() {
  tput clear >/dev/tty
  printf '\033[1;36m  WBeam\033[0m\n' >/dev/tty
  printf '\033[90m  \xe2\x86\x91/\xe2\x86\x93  j/k  navigate    Enter  select    q  quit\033[0m\n\n' >/dev/tty

  local cur="${selectable[$sel_pos]}"
  for (( i=0; i<n; i++ )); do
    local e="${MENU_ENTRIES[$i]}"
    if [[ "$e" == ──* ]]; then
      printf '\033[90m%s\033[0m\n' "$e" >/dev/tty
    elif [[ "$i" == "$cur" ]]; then
      local lbl="${e%%:*}"
      printf '\033[1;33m\xe2\x96\xb6%s\033[0m\n' "${lbl:1}" >/dev/tty
    else
      printf '\033[37m%s\033[0m\n' "${e%%:*}" >/dev/tty
    fi
  done
}

# ── main loop ─────────────────────────────────────────────────────────────────
selected_action=""

while true; do
  render

  # Read one key directly from /dev/tty (never a subshell).
  # For escape-based keys (arrows/home/end/etc), read the full sequence.
  key=""
  IFS= read -r -s -n1 key </dev/tty

  if [[ "$key" == $'\x1b' ]]; then
    # Slurp any immediately following bytes that belong to this escape sequence.
    # This avoids treating a slow arrow sequence as bare ESC.
    ch=""
    while IFS= read -r -s -n1 -t 0.05 ch </dev/tty; do
      key+="$ch"
    done
  fi

  case "$key" in
    # ── movement ──────────────────────────────────────────────────────────────
    $'\x1b[A'|$'\x1bOA'|$'\x1b['*A)           # Up arrow
      if (( sel_pos > 0 )); then
        sel_pos=$((sel_pos - 1))
      fi
      ;;
    $'\x1b[B'|$'\x1bOB'|$'\x1b['*B)           # Down arrow
      if (( sel_pos < ${#selectable[@]} - 1 )); then
        sel_pos=$((sel_pos + 1))
      fi
      ;;
    k|K)
      if (( sel_pos > 0 )); then
        sel_pos=$((sel_pos - 1))
      fi
      ;;
    j|J)
      if (( sel_pos < ${#selectable[@]} - 1 )); then
        sel_pos=$((sel_pos + 1))
      fi
      ;;
    # ── select ────────────────────────────────────────────────────────────────
    $'\n'|$'\r'|'')
      ci="${selectable[$sel_pos]}"
      ce="${MENU_ENTRIES[$ci]}"
      rest="${ce#*:}"        # strip label
      grp="${rest%%:*}"      # first field after label = group
      cmd_="${rest#*:}"      # everything after group = cmd
      selected_action="${grp}:${cmd_}"
      break
      ;;
    # ── quit ──────────────────────────────────────────────────────────────────
    $'\x03'|q|Q)
      selected_action="quit"
      break
      ;;
  esac
done

# cleanup runs via trap
trap - EXIT INT TERM HUP
cleanup

# ── dispatch ──────────────────────────────────────────────────────────────────
case "$selected_action" in
  quit|__quit__:__quit__)
    exit 0
    ;;
  *:*)
    grp="${selected_action%%:*}"
    cmd_="${selected_action#*:}"
    echo ""

    if [[ "$grp" == "android" ]]; then
      SELECTED_ANDROID_SERIAL=""
      SELECTED_ANDROID_API=""
      SELECTED_ANDROID_PIPELINE="modern"
      PRESET_ANDROID_SERIAL=""
      PRESET_ANDROID_API=""

      if [[ "$cmd_" == deploy-device\|* ]]; then
        IFS='|' read -r _ PRESET_ANDROID_SERIAL PRESET_ANDROID_API <<<"$cmd_"
        cmd_="deploy"
      fi

      if [[ -n "$PRESET_ANDROID_SERIAL" ]]; then
        SELECTED_ANDROID_SERIAL="$PRESET_ANDROID_SERIAL"
        SELECTED_ANDROID_API="$PRESET_ANDROID_API"
        SELECTED_ANDROID_PIPELINE="$(auto_pipeline_for_api "$SELECTED_ANDROID_API")"
      elif [[ "$cmd_" == "build" || "$cmd_" == "build-release" ]]; then
        if ! pick_android_pipeline "modern"; then
          exit 0
        fi
      elif [[ "$cmd_" == "build-all" ]]; then
        # no prompts — always builds both pipelines
        true
      else
        if collect_android_devices; then
          if ! pick_android_device; then
            exit 0
          fi
          if [[ -n "$SELECTED_ANDROID_API" ]]; then
            auto_mode="$(auto_pipeline_for_api "$SELECTED_ANDROID_API")"
          else
            auto_mode="$(auto_pipeline_for_all_devices)"
          fi
          if ! pick_android_pipeline "$auto_mode"; then
            exit 0
          fi
        else
          if ! pick_android_pipeline "modern"; then
            exit 0
          fi
        fi
      fi

      # ── build-all: build legacy + modern without any device/pipeline prompt ─
      if [[ "$cmd_" == "build-all" ]]; then
        echo "[wbgui] === Build All: host backend (Rust release) ==="
        "$WBEAM" host build
        rc_host=$?
        echo "[wbgui] === Build All: Android legacy (minSdk=17) ==="
        WBEAM_ANDROID_PIPELINE="legacy" "$WBEAM" android build
        rc_legacy=$?
        echo "[wbgui] === Build All: Android modern (minSdk=19) ==="
        WBEAM_ANDROID_PIPELINE="modern" "$WBEAM" android build
        rc_modern=$?
        if (( rc_host != 0 || rc_legacy != 0 || rc_modern != 0 )); then
          echo "[wbgui] build-all FAILED (host=$rc_host legacy=$rc_legacy modern=$rc_modern)" >&2
          exit 1
        fi
        echo "[wbgui] build-all OK"
        exit 0
      fi

      ALLOW_NO_TRANSPORT="0"
      API_ONLY_FALLBACK="0"
      if [[ "$cmd_" == "deploy" || "$cmd_" == "deploy-release" ]]; then
        ALLOW_NO_TRANSPORT="1"
        API_ONLY_FALLBACK="1"
      fi

      if [[ -n "$SELECTED_ANDROID_SERIAL" ]]; then
        WBEAM_ANDROID_SERIAL="$SELECTED_ANDROID_SERIAL" \
          WBEAM_ANDROID_PIPELINE="$SELECTED_ANDROID_PIPELINE" \
          WBEAM_ANDROID_ALLOW_NO_TRANSPORT="$ALLOW_NO_TRANSPORT" \
            WBEAM_ANDROID_API_ONLY_FALLBACK="$API_ONLY_FALLBACK" \
          "$WBEAM" "$grp" "$cmd_"
      else
        WBEAM_ANDROID_PIPELINE="$SELECTED_ANDROID_PIPELINE" \
          WBEAM_ANDROID_ALLOW_NO_TRANSPORT="$ALLOW_NO_TRANSPORT" \
            WBEAM_ANDROID_API_ONLY_FALLBACK="$API_ONLY_FALLBACK" \
          "$WBEAM" "$grp" "$cmd_"
      fi
      exit $?
    fi

    "$WBEAM" "$grp" "$cmd_"
    exit $?
    ;;
  *)
    exit 0
    ;;
esac