lint.sh

#!/bin/bash
if [ "$1" = "" ]; then
  ACTION="fix"
else
  ACTION="$1"
fi

if [ "$ACTION" = "install" ]; then
  set -ex
  pip install ruff==0.15.14 pyrefly==1.0.0
  exit 0
fi

if ! (which ruff >/dev/null && which pyrefly >/dev/null); then
  echo "ruff/pyrefly not installed. Run ./lint.sh install"
  exit 1
fi

VALID_ACTION="false"
ERRORS=""

function run {
  echo "+" $@ 1>&2
  $@
  if [ $? -ne 0 ]; then
    ERRORS="$ERRORS"$'\n'"ERROR running: $@"
  fi
  VALID_ACTION="true"
}

PYREFLY="scripts/pyrefly_check.sh"

if [ "$ACTION" = "fix" ]; then
  run ruff format
  run ruff check --fix
  run pre-commit run codespell --all-files
  run $PYREFLY
fi

if [ "$ACTION" = "unsafe" ]; then
  run ruff format
  run ruff check --fix --unsafe-fixes
  run pre-commit run codespell --all-files
  run $PYREFLY
fi

if [ "$ACTION" = "check" ]; then
  run ruff format --check --diff
  run ruff check --no-fix
  run pre-commit run codespell --all-files
  run $PYREFLY
fi

if [ "$ERRORS" != "" ]; then
  echo "$ERRORS" 1>&2
  exit 1
fi

if [ "$VALID_ACTION" = "false" ]; then
  echo "Invalid argument: $ACTION" 1>&2
  echo "Usage: ./lint.sh [fix|check|install|unsafe]" 1>&2
  exit 1
fi

exit 0