#!/usr/bin/env bash
# Тесты Bash-функций proxy-client: вместо настоящего CLI — поддельный
# исполняемый файл в PROXY_CLIENT_BIN.
# Строчные proxy-переменные задаёт proxy_on, shellcheck этого не видит.
# shellcheck disable=SC2154
set -euo pipefail

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
FUNCTIONS_FILE="$REPO_ROOT/bash/.bash_functions"
CUSTOM_FILE="$REPO_ROOT/bash/.bash_custom"
SHELL_MODULE="$REPO_ROOT/proxy-client/.config/proxy-client/shell.bash"
TEST_ROOT="$(mktemp -d /tmp/proxy-launchers-test.XXXXXX)"
trap 'rm -rf "$TEST_ROOT"' EXIT

PROXY_KEYS=(HTTP_PROXY HTTPS_PROXY ALL_PROXY NO_PROXY
            http_proxy https_proxy all_proxy no_proxy)

fail() {
  echo "FAIL: $*" >&2
  exit 1
}

ok() {
  echo "ok - $*"
}

assert_eq() {
  [ "$1" = "$2" ] || fail "$3 (ожидалось '$1', получено '$2')"
}

# Поддельный CLI: на "env" печатает $FAKE_ENV_OUTPUT, на остальное — свои
# аргументы по одному в строке; завершается с кодом $FAKE_RC.
PROXY_CLIENT_BIN="$TEST_ROOT/proxy-client"
cat > "$PROXY_CLIENT_BIN" <<'EOF'
#!/usr/bin/env bash
if [ -n "${FAKE_CALL_LOG:-}" ]; then
  printf '<%s>\n' "$@" >> "$FAKE_CALL_LOG"
fi
if [ "$1" = env ]; then
  printf '%s\n' "$FAKE_ENV_OUTPUT"
elif [ "$#" -eq 3 ] && [ "$1" = --profile ] && [ "$2" = work ] && [ "$3" = env ]; then
  printf '%s\n' "$FAKE_PROFILE_ENV_OUTPUT"
else
  printf '<%s>\n' "$@"
fi
exit "${FAKE_RC:-0}"
EOF
chmod +x "$PROXY_CLIENT_BIN"
export PROXY_CLIENT_BIN

VALID_ENV='HTTP_PROXY=http://proxy.test:1234
HTTPS_PROXY=http://proxy.test:1234
ALL_PROXY=socks5://proxy.test:1234
NO_PROXY=proxy.test,localhost,.example.test
http_proxy=http://proxy.test:1234
https_proxy=http://proxy.test:1234
all_proxy=socks5://proxy.test:1234
no_proxy=proxy.test,localhost,.example.test'

# shellcheck source=/dev/null
source "$SHELL_MODULE"

# proxy_on применяет все восемь переменных.
(
  unset "${PROXY_KEYS[@]}"
  FAKE_ENV_OUTPUT="$VALID_ENV" proxy_on >/dev/null
  assert_eq http://proxy.test:1234 "$HTTP_PROXY" HTTP_PROXY
  assert_eq http://proxy.test:1234 "$https_proxy" https_proxy
  assert_eq socks5://proxy.test:1234 "$ALL_PROXY" ALL_PROXY
  assert_eq proxy.test,localhost,.example.test "$no_proxy" no_proxy
  for key in "${PROXY_KEYS[@]}"; do
    [ -n "$(printenv "$key")" ] || fail "$key не экспортирован"
  done
)
ok "proxy_on применяет восемь переменных"

# Выбор профиля меняет окружение только текущей оболочки.
(
  unset "${PROXY_KEYS[@]}"
  FAKE_ENV_OUTPUT="$VALID_ENV" \
    FAKE_PROFILE_ENV_OUTPUT='HTTP_PROXY=http://work.test:4321
HTTPS_PROXY=http://work.test:4321
ALL_PROXY=socks5://work.test:4321
NO_PROXY=work.test,localhost
http_proxy=http://work.test:4321
https_proxy=http://work.test:4321
all_proxy=socks5://work.test:4321
no_proxy=work.test,localhost' \
    proxy_on --profile work >/dev/null
  assert_eq http://work.test:4321 "$HTTP_PROXY" "proxy_on не выбрал профиль work"
  assert_eq socks5://work.test:4321 "$all_proxy" "SOCKS выбранного профиля"
  assert_eq work.test,localhost "$NO_PROXY" "исключения выбранного профиля"
)
ok "proxy_on выбирает именованный профиль"

# Неверный синтаксис proxy_on отклоняется до изменения окружения.
assert_invalid_proxy_on() (
  for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
  local rc=0
  FAKE_ENV_OUTPUT="$VALID_ENV" proxy_on "$@" >/dev/null 2>&1 || rc=$?
  assert_eq 2 "$rc" "proxy_on принял неверные аргументы: $*"
  for key in "${PROXY_KEYS[@]}"; do
    assert_eq old "${!key}" "$key изменился после неверных аргументов"
  done
)
assert_invalid_proxy_on env
assert_invalid_proxy_on --profile
assert_invalid_proxy_on --profile ""
assert_invalid_proxy_on --profile work extra
assert_invalid_proxy_on --unknown work
ok "proxy_on отклоняет неверные аргументы без изменения окружения"

# Ошибка CLI или непонятный вывод не меняют окружение ни на одну переменную.
for case_spec in "1|$VALID_ENV" "0|$VALID_ENV
EXTRA_PROXY=bad" "0|BROKEN-LINE"; do
  (
    for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
    if FAKE_RC="${case_spec%%|*}" FAKE_ENV_OUTPUT="${case_spec#*|}" \
        proxy_on >/dev/null 2>&1; then
      fail "proxy_on принял ошибочный ответ: ${case_spec#*|}"
    fi
    for key in "${PROXY_KEYS[@]}"; do
      assert_eq old "${!key}" "$key изменился после ошибки"
    done
  )
done
ok "неуспешный env не меняет окружение"

# Даже пригодный вывод не применяется, если проверка выбранного прокси не прошла.
(
  for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
  unset no_proxy
  if FAKE_RC=1 FAKE_PROFILE_ENV_OUTPUT="$VALID_ENV" proxy_on --profile work >/dev/null 2>&1; then
    fail "proxy_on принял отказ проверки выбранного профиля"
  fi
  for key in "${PROXY_KEYS[@]}"; do
    if [[ "$key" == no_proxy ]]; then
      [ -z "${!key+set}" ] || fail "$key появился после ошибки"
    else
      assert_eq old "${!key}" "$key изменился после ошибки выбранного профиля"
    fi
  done
)
ok "отказ проверки выбранного профиля сохраняет окружение"

# proxy_off очищает оба регистра.
(
  for key in "${PROXY_KEYS[@]}"; do export "$key=x"; done
  FAKE_CALL_LOG="$TEST_ROOT/off-calls" proxy_off >/dev/null
  [ ! -e "$TEST_ROOT/off-calls" ] || fail "proxy_off вызвал CLI"
  for key in "${PROXY_KEYS[@]}"; do
    [ -z "${!key+set}" ] || fail "$key не очищен"
  done
)
ok "proxy_off очищает восемь переменных без вызова CLI"

# Обёртки передают CLI команду, аргументы и код возврата без искажений.
(
  export FAKE_RC=42
  set +e
  output="$(pclaude "a b" -- --flag)"
  rc=$?
  set -e
  assert_eq 42 "$rc" "pclaude не сохранил код возврата"
  assert_eq $'<run>\n<-->\n<claude>\n<a b>\n<-->\n<--flag>' "$output" \
    "pclaude исказил аргументы"

  set +e
  output="$(pcodex --search)"
  rc=$?
  set -e
  assert_eq 42 "$rc" "pcodex не сохранил код возврата"
  assert_eq $'<run>\n<-->\n<codex>\n<--search>' "$output" \
    "pcodex исказил аргументы"
)
ok "pclaude и pcodex передают аргументы и код выхода"

# Ведущий --proxy-profile относится к прокси, обычный --profile — к программе.
(
  for wrapper in pclaude pcodex; do
    rc=0
    output="$(FAKE_RC=42 "$wrapper" --proxy-profile work --profile personal "a b" "")" || rc=$?
    assert_eq 42 "$rc" "$wrapper не сохранил код возврата с профилем"
    assert_eq $'<--profile>\n<work>\n<run>\n<-->\n<'"${wrapper#p}"$'>\n<--profile>\n<personal>\n<a b>\n<>' "$output" \
      "$wrapper исказил выбор профиля или аргументы ребенка"
  done
)
ok "обёртки выбирают профиль прокси и сохраняют аргументы ребенка"

# Отсутствующее имя профиля не должно запускать дочернюю команду.
(
  for wrapper in pclaude pcodex; do
    for empty_name in missing empty; do
      args=(--proxy-profile)
      if [[ "$empty_name" == empty ]]; then args+=(""); fi
      rc=0
      output="$("$wrapper" "${args[@]}" 2>/dev/null)" || rc=$?
      assert_eq 2 "$rc" "$wrapper принял --proxy-profile без имени"
      assert_eq "" "$output" "$wrapper вызвал CLI без имени профиля"
    done
  done
)
ok "обёртки отклоняют --proxy-profile без имени"

# Только ведущий --proxy-profile выбирает профиль прокси.
(
  for wrapper in pclaude pcodex; do
    output="$("$wrapper" --profile personal --proxy-profile child "a b" "")"
    assert_eq $'<run>\n<-->\n<'"${wrapper#p}"$'>\n<--profile>\n<personal>\n<--proxy-profile>\n<child>\n<a b>\n<>' "$output" \
      "$wrapper забрал аргументы ребенка"
  done
)
ok "обычный --profile и неведущий --proxy-profile остаются ребенку"

# Запуск с профилем не меняет окружение вызывающей оболочки.
(
  for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
  unset no_proxy
  for wrapper in pclaude pcodex; do
    "$wrapper" --proxy-profile work >/dev/null
    for key in "${PROXY_KEYS[@]}"; do
      if [[ "$key" == no_proxy ]]; then
        [ -z "${!key+set}" ] || fail "$wrapper экспортировал $key"
      else
        assert_eq old "${!key}" "$wrapper изменил $key"
      fi
    done
  done
)
ok "обёртки сохраняют окружение вызывающей оболочки"

# .bash_custom подключает и общие функции, и proxy-модуль.
(
  home="$TEST_ROOT/home"
  mkdir -p "$home/.config/proxy-client"
  ln -s "$CUSTOM_FILE" "$home/.bash_custom"
  ln -s "$FUNCTIONS_FILE" "$home/.bash_functions"
  ln -s "$SHELL_MODULE" "$home/.config/proxy-client/shell.bash"
  output="$(
    HOME="$home" bash --noprofile --norc -ic '
      source "$HOME/.bash_custom"
      declare -F proxy_on update_hproxy >/dev/null && printf ready
    ' 2>/dev/null
  )"
  assert_eq ready "$output" ".bash_custom не подключил модули"
)
ok ".bash_custom подключает общие функции и proxy-модуль"

if command -v shellcheck >/dev/null 2>&1; then
  shellcheck -s bash "$SHELL_MODULE" "$0" \
    || fail "shellcheck нашёл проблемы"
  ok "shellcheck чист"
else
  echo "skip - shellcheck не установлен"
fi

echo "PASS: тесты Bash-функций proxy-client"