#!/usr/bin/env bash
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"

pass_count=0

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

ok() {
  pass_count=$((pass_count + 1))
  echo "ok - $*"
}

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

valid_env() {
  cat <<'EOF'
HTTP_PROXY=http://proxy.test:1234
HTTPS_PROXY=http://proxy.test:1234
ALL_PROXY=socks5://proxy.test:1234
NO_PROXY=proxy.test,localhost,127.0.0.1,::1,[::1],.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,127.0.0.1,::1,[::1],.example.test
EOF
}

assert_five_proxy_commands() {
  local command_name
  for command_name in proxy_on proxy_off pclaude pcodex myip; do
    declare -F "$command_name" >/dev/null || fail "нет функции $command_name"
  done
}

# Самостоятельный модуль предоставляет весь публичный Bash-интерфейс.
(
  source "$SHELL_MODULE"
  assert_five_proxy_commands
) && ok "самостоятельное подключение предоставляет пять proxy-команд"

# Обычная .bash_custom независимо подключает общие функции и proxy-модуль.
(
  fixture="$(mktemp -d /tmp/proxy-bash-custom-test.XXXXXX)"
  mkdir -p "$fixture/.config/proxy-client"
  ln -s "$CUSTOM_FILE" "$fixture/.bash_custom"
  ln -s "$FUNCTIONS_FILE" "$fixture/.bash_functions"
  ln -s "$SHELL_MODULE" "$fixture/.config/proxy-client/shell.bash"
  output="$(
    HOME="$fixture" bash --noprofile --norc -ic '
      source "$HOME/.bash_custom"
      for name in proxy_on proxy_off pclaude pcodex myip update_hproxy; do
        declare -F "$name" >/dev/null || exit 1
      done
      printf ready
    ' 2>/dev/null
  )"
  assert_eq ready "$output" ".bash_custom не подключил оба независимых модуля"
) && ok "обычное подключение предоставляет proxy-команды и update_hproxy"

# proxy_on применяет восемь известных переменных и не запрашивает IP.
(
  source "$SHELL_MODULE"
  _proxy_cli() {
    [ "$1" = env ] || return 99
    valid_env
  }
  curl() { fail "proxy_on не должен вызывать curl"; }
  unset HTTP_PROXY HTTPS_PROXY ALL_PROXY NO_PROXY \
        http_proxy https_proxy all_proxy no_proxy || true
  proxy_on >/dev/null
  assert_eq "http://proxy.test:1234" "$HTTP_PROXY" "HTTP_PROXY"
  assert_eq "$HTTP_PROXY" "$http_proxy" "http_proxy"
  assert_eq "http://proxy.test:1234" "$HTTPS_PROXY" "HTTPS_PROXY"
  assert_eq "$HTTPS_PROXY" "$https_proxy" "https_proxy"
  assert_eq "socks5://proxy.test:1234" "$ALL_PROXY" "ALL_PROXY"
  assert_eq "$ALL_PROXY" "$all_proxy" "all_proxy"
  assert_eq "proxy.test,localhost,127.0.0.1,::1,[::1],.example.test" "$NO_PROXY" "NO_PROXY"
  assert_eq "$NO_PROXY" "$no_proxy" "no_proxy"
) && ok "proxy_on применяет известные ключи без запроса IP"

# Неизвестная строка в машинном выводе оставляет прежнее окружение целиком.
unknown_output="$(valid_env; printf 'EXTRA_PROXY=bad\n')"
for bad_output in \
  "$unknown_output" \
  "BROKEN-LINE"; do
  (
    source "$SHELL_MODULE"
    export HTTP_PROXY=old-http HTTPS_PROXY=old-https ALL_PROXY=old-all NO_PROXY=old-no
    export http_proxy=old-http-lower https_proxy=old-https-lower
    export all_proxy=old-all-lower no_proxy=old-no-lower
    _proxy_cli() { printf '%s\n' "$bad_output"; }
    if proxy_on >/dev/null 2>&1; then
      fail "proxy_on принял неверный вывод"
    fi
    assert_eq old-http "$HTTP_PROXY" "HTTP_PROXY изменился при ошибке"
    assert_eq old-https "$HTTPS_PROXY" "HTTPS_PROXY изменился при ошибке"
    assert_eq old-all "$ALL_PROXY" "ALL_PROXY изменился при ошибке"
    assert_eq old-no "$NO_PROXY" "NO_PROXY изменился при ошибке"
    assert_eq old-http-lower "$http_proxy" "http_proxy изменился при ошибке"
    assert_eq old-https-lower "$https_proxy" "https_proxy изменился при ошибке"
    assert_eq old-all-lower "$all_proxy" "all_proxy изменился при ошибке"
    assert_eq old-no-lower "$no_proxy" "no_proxy изменился при ошибке"
  ) || fail "атомарность нарушена для неверного вывода"
done
ok "неизвестная env-строка отклоняется без частичного изменения окружения"

# Даже полный stdout не применяется, если CLI завершился с ошибкой.
(
  source "$SHELL_MODULE"
  export HTTP_PROXY=old HTTPS_PROXY=old ALL_PROXY=old NO_PROXY=old
  export http_proxy=old https_proxy=old all_proxy=old no_proxy=old
  _proxy_cli() {
    valid_env
    return 1
  }
  if proxy_on >/dev/null 2>&1; then
    fail "proxy_on принял вывод неуспешного CLI"
  fi
  for key in HTTP_PROXY HTTPS_PROXY ALL_PROXY NO_PROXY \
             http_proxy https_proxy all_proxy no_proxy; do
    assert_eq old "${!key}" "$key изменился после ошибки CLI"
  done
) && ok "stdout неуспешного env не изменяет окружение"

# Обёртки передают CLI команду, аргументы и код возврата без искажений.
(
  source "$SHELL_MODULE"
  _proxy_cli() {
    printf '<%s>\n' "$@"
    return 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 делегируют run и сохраняют аргументы/код"

# proxy_off очищает оба регистра и не смешивает действие с диагностикой.
(
  source "$SHELL_MODULE"
  export HTTP_PROXY=x HTTPS_PROXY=x ALL_PROXY=x NO_PROXY=x
  export http_proxy=x https_proxy=x all_proxy=x no_proxy=x
  curl() { fail "proxy_off не должен вызывать curl"; }
  proxy_off >/dev/null
  for key in HTTP_PROXY HTTPS_PROXY ALL_PROXY NO_PROXY \
             http_proxy https_proxy all_proxy no_proxy; do
    [ -z "${!key+x}" ] || fail "$key не очищен"
  done
) && ok "proxy_off очищает восемь переменных без запроса IP"

# Общий файл больше не владеет proxy-интерфейсом, но сохраняет update_hproxy.
(
  source "$FUNCTIONS_FILE"
  declare -F update_hproxy >/dev/null || fail "update_hproxy потерян"
  for name in proxy_on proxy_off pclaude pcodex myip _proxy_run _proxy_cli; do
    if declare -F "$name" >/dev/null; then
      fail "$name всё ещё определён в bash/.bash_functions"
    fi
  done
) && ok "владение proxy-функциями перенесено, update_hproxy сохранён"

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

echo "PASS: $pass_count проверок пройдено"