dev

#!/bin/bash
set -euf

# print all commands if ci
if [[ -n ${CI:-} ]]; then
  set -x
fi

DOCKER_IMAGE_NAME="${DOCKER_IMAGE_NAME:-dns-test:latest}"

case "${1:-}" in
  "fmt")
    find . -not -path '*/\.*' -type f -name '*.py' -exec \
      yapf -i --style .style.yapf -p -vv {} '+' || true
    ;;
  "run")
    exec python3 -O ./dns-test.py ${@:2}
    ;;
  "docker-build")
    exec docker build -t "${DOCKER_IMAGE_NAME}" --cache-from "${DOCKER_IMAGE_NAME}" .
    ;;
  "docker-push")
    exec docker push "${DOCKER_IMAGE_NAME}"
    ;;
  "docker-run")
    exec docker run --rm -t \
      --user "$(id -u):$(id -g)" \
      -v "$(pwd):/workdir:rw" \
      "${DOCKER_IMAGE_NAME}" ${@:2}
    ;;
  "download-umbrella")
    cd domains
    wget -O- 'http://s3-us-west-1.amazonaws.com/umbrella-static/top-1m.csv.zip' | gunzip >umbrella.csv
    ;;
  *)
    echo "usage: ${0} <fmt|docker-build|docker-run|docker-push|download-umbrella>"
    ;;
esac