#!/usr/bin/env bash
set -eo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
TACHYON_LIB="$ROOT_DIR/tachyon/files/usr/lib"
VALIDATOR="$ROOT_DIR/tachyon/files/usr/lib/providers/adguard/validator.uc"

fail() {
  printf 'FAIL: %s\n' "$1" >&2
  exit 1
}

assert_json_field() {
  local json="$1"
  local field="$2"
  local expected="$3"

  JSON_VALUE="$json" node - "$field" "$expected" <<'NODE'
const field = process.argv[2];
const expected = process.argv[3];
const value = JSON.parse(process.env.JSON_VALUE);
const actual = String(value[field]);
if (actual !== expected) {
  console.error(`expected ${field}=${expected}, got ${actual}`);
  process.exit(1);
}
NODE
}

doh_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream 'https://dns.google/dns-query')"
assert_json_field "$doh_json" valid true

dot_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream 'tls://1.1.1.1')"
assert_json_field "$dot_json" valid true

plain_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream '8.8.8.8')"
assert_json_field "$plain_json" valid true

udp_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream 'udp://1.1.1.1:53')"
assert_json_field "$udp_json" valid true

bad_doh_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream 'http://not-tls.example')"
assert_json_field "$bad_doh_json" valid false

empty_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream '')"
assert_json_field "$empty_json" valid false

bad_port_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-upstream '1.1.1.1:99999')"
assert_json_field "$bad_port_json" valid false

if ucode -L "$TACHYON_LIB" -- "$VALIDATOR" dns-main-mode-valid 'dnsmasq' >/dev/null 2>&1; then
  :
else
  fail "dnsmasq should be a valid dns_main_mode"
fi

if ucode -L "$TACHYON_LIB" -- "$VALIDATOR" dns-main-mode-valid 'adguard' >/dev/null 2>&1; then
  :
else
  fail "adguard should be a valid dns_main_mode"
fi

if ucode -L "$TACHYON_LIB" -- "$VALIDATOR" dns-main-mode-valid 'bogus' >/dev/null 2>&1; then
  fail "bogus dns_main_mode should be rejected"
fi

if ucode -L "$TACHYON_LIB" -- "$VALIDATOR" valid-port '3000' >/dev/null 2>&1; then
  :
else
  fail "3000 should be a valid port"
fi

if ucode -L "$TACHYON_LIB" -- "$VALIDATOR" valid-port 'notaport' >/dev/null 2>&1; then
  fail "non-numeric port should be rejected"
fi

filter_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-filter-url 'https://adguardteam.github.io/AdGuardSDNSFilter/Filters/filter.txt')"
assert_json_field "$filter_json" valid true

bad_filter_json="$(ucode -L "$TACHYON_LIB" -- "$VALIDATOR" validate-filter-url 'ftp://filter')"
assert_json_field "$bad_filter_json" valid false

printf 'OK: adguard validator tests passed\n'