| 1 | #!/usr/bin/env bash
|
| 2 | #
|
| 3 | # Run yapf formatter; it's installed in ~/wedge/ by build/deps.sh
|
| 4 | #
|
| 5 | # Usage:
|
| 6 | # test/format.sh <function name>
|
| 7 |
|
| 8 | REPO_ROOT=$(cd "$(dirname $0)/.."; pwd)
|
| 9 |
|
| 10 | source build/dev-shell.sh # python3 in $PATH
|
| 11 | source devtools/run-task.sh
|
| 12 |
|
| 13 | # Hack to prevent interference. TODO: Make a separate wedge for yapf.
|
| 14 | unset PYTHONPATH
|
| 15 |
|
| 16 | . build/common.sh # $CLANG_DIR
|
| 17 |
|
| 18 | set -o nounset
|
| 19 | set -o pipefail
|
| 20 | set -o errexit
|
| 21 | shopt -s strict:all 2>/dev/null || true # dogfood for OSH
|
| 22 |
|
| 23 | #
|
| 24 | # Python
|
| 25 | #
|
| 26 |
|
| 27 | readonly YAPF_VENV='_tmp/yapf-venv'
|
| 28 |
|
| 29 | install-yapf() {
|
| 30 | local venv=$YAPF_VENV
|
| 31 |
|
| 32 | rm -r -f -v $venv
|
| 33 |
|
| 34 | python3 -m venv $venv
|
| 35 |
|
| 36 | . $venv/bin/activate
|
| 37 |
|
| 38 | # 0.40.1 is the 2023-06-20 release
|
| 39 | #
|
| 40 | # Pin the version so formatting is stable!
|
| 41 |
|
| 42 | python3 -m pip install 'yapf == 0.40.1'
|
| 43 |
|
| 44 | yapf-version
|
| 45 | }
|
| 46 |
|
| 47 | yapf-version() {
|
| 48 | . $YAPF_VENV/bin/activate
|
| 49 | python3 -m yapf --version
|
| 50 | }
|
| 51 |
|
| 52 | # For now, run yapf on specific files. TODO: could query git for the files
|
| 53 | # that are are different from master branch, and run it on those.
|
| 54 | yapf-files() {
|
| 55 | . $YAPF_VENV/bin/activate
|
| 56 | python3 -m yapf -i "$@"
|
| 57 | }
|
| 58 |
|
| 59 | yapf-known() {
|
| 60 | ### yapf some files that have been normalized
|
| 61 |
|
| 62 | time yapf-files \
|
| 63 | {asdl,benchmarks,builtin,core,data_lang,doctools,frontend,lazylex,mycpp,mycpp/examples,osh,spec/*,yaks,ysh}/*.py \
|
| 64 | */NINJA_subgraph.py
|
| 65 | }
|
| 66 |
|
| 67 | yapf-changed() {
|
| 68 | branch="${1:-master}"
|
| 69 |
|
| 70 | #git diff --name-only .."$branch" '*.py'
|
| 71 |
|
| 72 | git diff --name-only .."$branch" '*.py' \
|
| 73 | | xargs --no-run-if-empty -- $0 yapf-files
|
| 74 | }
|
| 75 |
|
| 76 | #
|
| 77 | # Doc strings - one off
|
| 78 | #
|
| 79 |
|
| 80 | install-docformatter() {
|
| 81 | python3 -m pip install docformatter
|
| 82 | }
|
| 83 |
|
| 84 | docstrings() {
|
| 85 | ### Format docstrings - NOT done automatically, because it can mangle them
|
| 86 |
|
| 87 | # Requires manual fix-up
|
| 88 |
|
| 89 | #time test/lint.sh py2-files-to-lint \
|
| 90 | # | xargs --verbose -- python3 -m docformatter --in-place
|
| 91 |
|
| 92 | python3 -m docformatter --in-place lazylex/*.py
|
| 93 | }
|
| 94 |
|
| 95 | #
|
| 96 | # C++
|
| 97 | #
|
| 98 |
|
| 99 | clang-format() {
|
| 100 | # See //.clang-format for the style config.
|
| 101 | $CLANG_DIR/bin/clang-format --style=file "$@"
|
| 102 | }
|
| 103 |
|
| 104 | readonly -a CPP_FILES=(
|
| 105 | {asdl,core}/*.cc
|
| 106 | benchmarks/*.c
|
| 107 | cpp/*.{c,cc,h}
|
| 108 | data_lang/*.{c,cc,h}
|
| 109 | mycpp/*.{cc,h}
|
| 110 | mycpp/demo/*.{cc,h}
|
| 111 | demo/*.c
|
| 112 | doctools/*.{h,cc}
|
| 113 | yaks/*.h
|
| 114 |
|
| 115 | # Could add pyext, but they have sort of a Python style
|
| 116 | # pyext/fanos.c
|
| 117 | )
|
| 118 |
|
| 119 | cpp-files() {
|
| 120 | shopt -s nullglob
|
| 121 | for file in "${CPP_FILES[@]}"; do
|
| 122 |
|
| 123 | echo $file
|
| 124 | done
|
| 125 | }
|
| 126 |
|
| 127 | all-cpp() {
|
| 128 | # see build/common.sh
|
| 129 | if test -n "$CLANG_IS_MISSING"; then
|
| 130 | log ''
|
| 131 | log " *** $0: Did not find $CLANG_DIR_1"
|
| 132 | log " *** Run deps/from-binary.sh to get it"
|
| 133 | log ''
|
| 134 | return 1
|
| 135 | fi
|
| 136 |
|
| 137 | cpp-files | egrep -v 'greatest.h' | xargs -- $0 clang-format -i
|
| 138 | git diff
|
| 139 | }
|
| 140 |
|
| 141 | test-asdl-format() {
|
| 142 | ### Test how clang-format would like our generated code
|
| 143 |
|
| 144 | local file=${1:-_gen/asdl/hnode.asdl.h}
|
| 145 |
|
| 146 | local tmp=_tmp/hnode
|
| 147 | clang-format $file > $tmp
|
| 148 | diff -u $file $tmp
|
| 149 | }
|
| 150 |
|
| 151 | run-task "$@"
|