#!/usr/bin/env bash
#
# Tests for ASDL.
#
# Usage:
#   asdl/TEST.sh <function name>

set -o nounset
set -o pipefail
set -o errexit

REPO_ROOT=$(cd "$(dirname $0)/.."; pwd)

source build/dev-shell.sh  # python3 in $PATH
source devtools/common.sh  # banner
source test/common.sh      # run-one-test

unit() {
  ### Run unit tests

  for variant in asan asan+gcalways ubsan; do
    run-one-test 'asdl/gen_cpp_test' '' $variant
    run-one-test 'asdl/gc_test' '' $variant
  done
}

#
# Python codegen
#

readonly PY_PATH='.:vendor/'  # note: could consolidate with other scripts

asdl-check() {
  # Unlike Python code, we use --strict mode
  python3 -m mypy --py2 --strict --follow-imports=silent "$@"
}

# NOTE: We're testing ASDL code generation with --strict because we might want
# Oils to pass under --strict someday.
typed-demo-asdl() {
  # We want to exclude ONLY pylib.collections_, but somehow --exclude
  # '.*collections_\.py' does not do it.  So --follow-imports=silent.  Tried
  # --verbose too
  asdl-check \
    _devbuild/gen/typed_demo_asdl.py asdl/examples/typed_demo.py

  PYTHONPATH=$PY_PATH asdl/examples/typed_demo.py "$@"
}

check-arith() {
  # NOTE: There are still some Any types here!  We don't want them for
  # translation.

  asdl-check \
    asdl/examples/typed_arith_parse.py \
    asdl/examples/typed_arith_parse_test.py \
    asdl/examples/tdop.py
}

typed-arith-asdl() {
  check-arith

  PYTHONPATH=$PY_PATH asdl/examples/typed_arith_parse_test.py

  banner 'parse'
  PYTHONPATH=$PY_PATH asdl/examples/typed_arith_parse.py parse '40+2'
  echo

  banner 'eval'
  PYTHONPATH=$PY_PATH asdl/examples/typed_arith_parse.py eval '40+2+5'
  echo
}

check-types() {
  build/py.sh py-asdl-examples

  asdl-check _devbuild/gen/shared_variant_asdl.py

  banner 'typed-arith-asdl'
  typed-arith-asdl

  banner 'typed-demo-asdl'
  typed-demo-asdl

  asdl-check asdl/target_lang_test.py
}

pretty-demo() {  
  local cpp=${1:-}

  OSH=bin/osh
  YSH=bin/ysh

  if test -n "$cpp"; then
    ninja _bin/cxx-asan/{osh,ysh}
    export OSH=_bin/cxx-asan/osh
    export YSH=_bin/cxx-asan/ysh
  fi

  # osh -n
  test/parse-errors.sh test-syntax-abbrev
  echo

  #return

  # Show Dict[BigInt, str]
  for i in 12 50 80; do
    $OSH -c 'declare -a a=(a b); a[$1]=zzz; pp asdl_ (a)' dummy $i
    echo
  done

  # Tabular
  for i in 12 50 100; do
    $YSH -c 'var i = $1; var x = []; for i in (1 ..= i) { call x->append(i) }; pp (x)' \
      dummy $i
    echo
  done


  # Show Dict[str, value_t]
  $YSH -c 'var d = {x:42}; setvar d.k = d; pp asdl_ (d)'
  echo

  # hnode::External
  $YSH -c 'pp asdl_ (len)'
  echo
}

"$@"
