| #!/bin/bash | 
 | set -e | 
 |  | 
 | # This script reformats source files using various formatters and linters. | 
 | # | 
 | # Files are changed in-place, so make sure you don't have anything open in an | 
 | # editor, and you may want to commit before formatting in case of awryness. | 
 | # | 
 | # This must be run on a clean repository to succeed | 
 | # | 
 | function display_help() | 
 | { | 
 |     echo "usage: format-code.sh [-h | --help] [--no-diff] [--list-tools]" | 
 |     echo "                      [--disable <tool>] [--enable <tool>] [<path>]" | 
 |     echo | 
 |     echo "Format and lint a repository." | 
 |     echo | 
 |     echo "Arguments:" | 
 |     echo "    --list-tools      Display available linters and formatters" | 
 |     echo "    --no-diff         Don't show final diff output" | 
 |     echo "    --disable <tool>  Disable linter" | 
 |     echo "    --enable <tool>   Enable only specific linters" | 
 |     echo "    --allow-missing   Run even if linters are not all present" | 
 |     echo "    path              Path to git repository (default to pwd)" | 
 | } | 
 |  | 
 | LINTERS_ALL=( \ | 
 |         commit_gitlint \ | 
 |         commit_spelling \ | 
 |         beautysh \ | 
 |         beautysh_sh \ | 
 |         black \ | 
 |         clang_format \ | 
 |         eslint \ | 
 |         flake8 \ | 
 |         isort \ | 
 |         markdownlint \ | 
 |         prettier \ | 
 |         shellcheck \ | 
 |     ) | 
 | LINTERS_DISABLED=() | 
 | LINTERS_ENABLED=() | 
 | declare -A LINTERS_FAILED=() | 
 |  | 
 | eval set -- "$(getopt -o 'h' --long 'help,list-tools,no-diff,disable:,enable:,allow-missing' -n 'format-code.sh' -- "$@")" | 
 | while true; do | 
 |     case "$1" in | 
 |         '-h'|'--help') | 
 |             display_help && exit 0 | 
 |             ;; | 
 |  | 
 |         '--list-tools') | 
 |             echo "Available tools:" | 
 |             for t in "${LINTERS_ALL[@]}"; do | 
 |                 echo "    $t" | 
 |             done | 
 |             exit 0 | 
 |             ;; | 
 |  | 
 |         '--no-diff') | 
 |             OPTION_NO_DIFF=1 | 
 |             shift | 
 |             ;; | 
 |  | 
 |         '--disable') | 
 |             LINTERS_DISABLED+=("$2") | 
 |             shift && shift | 
 |             ;; | 
 |  | 
 |         '--enable') | 
 |             LINTERS_ENABLED+=("$2") | 
 |             shift && shift | 
 |             ;; | 
 |  | 
 |         '--allow-missing') | 
 |             ALLOW_MISSING=yes | 
 |             shift | 
 |             ;; | 
 |  | 
 |         '--') | 
 |             shift | 
 |             break | 
 |             ;; | 
 |  | 
 |         *) | 
 |             echo "unknown option: $1" | 
 |             display_help && exit 1 | 
 |             ;; | 
 |     esac | 
 | done | 
 |  | 
 | # Detect tty and set nicer colors. | 
 | if [ -t 1 ]; then | 
 |     BLUE="\e[34m" | 
 |     GREEN="\e[32m" | 
 |     NORMAL="\e[0m" | 
 |     RED="\e[31m" | 
 |     YELLOW="\e[33m" | 
 | else # non-tty, no escapes. | 
 |     BLUE="" | 
 |     GREEN="" | 
 |     NORMAL="" | 
 |     RED="" | 
 |     YELLOW="" | 
 | fi | 
 |  | 
 | # Allow called scripts to know which clang format we are using | 
 | export CLANG_FORMAT="clang-format" | 
 |  | 
 | # Path to default config files for linters. | 
 | CONFIG_PATH="$(git -C "$(dirname "${BASH_SOURCE[0]}")" rev-parse --show-toplevel)/config" | 
 |  | 
 | # Find repository root for `pwd` or $1. | 
 | if [ -z "$1" ]; then | 
 |     DIR="$(git rev-parse --show-toplevel || pwd)" | 
 | else | 
 |     DIR="$(git -C "$1" rev-parse --show-toplevel)" | 
 | fi | 
 | if [ ! -e "$DIR/.git" ]; then | 
 |     echo -e "${RED}Error:${NORMAL} Directory ($DIR) does not appear to be a git repository" | 
 |     exit 1 | 
 | fi | 
 |  | 
 | cd "${DIR}" | 
 | echo -e "    ${BLUE}Formatting code under${NORMAL} $DIR" | 
 |  | 
 | # Config hashes: | 
 | #   LINTER_REQUIRE - The requirements to run a linter, semi-colon separated. | 
 | #       1. Executable. | 
 | #       2. [optional] Configuration file. | 
 | #       3. [optional] Global fallback configuration file. | 
 | # | 
 | #   LINTER_IGNORE - An optional set of semi-colon separated ignore-files | 
 | #       specific to the linter. | 
 | # | 
 | #   LINTER_TYPES - The file types supported by the linter, semi-colon separated. | 
 | # | 
 | #   LINTER_CONFIG - The config (from LINTER_REQUIRE) chosen for the repository. | 
 | # | 
 | declare -A LINTER_REQUIRE=() | 
 | declare -A LINTER_IGNORE=() | 
 | declare -A LINTER_TYPES=() | 
 | declare -A LINTER_CONFIG=() | 
 |  | 
 | LINTER_REQUIRE+=([commit_spelling]="codespell") | 
 | LINTER_TYPES+=([commit_spelling]="commit") | 
 |  | 
 | commit_filename="$(mktemp)" | 
 | function clean_up_file() { | 
 |     rm "$commit_filename" | 
 | } | 
 | trap clean_up_file EXIT | 
 |  | 
 | function find_codespell_dict_file() { | 
 |     local python_codespell_dict | 
 |     # @formatter:off | 
 |     python_codespell_dict=$(python3 -c " | 
 | import os.path as op | 
 | import codespell_lib | 
 | codespell_dir = op.dirname(codespell_lib.__file__) | 
 | codespell_file = op.join(codespell_dir, 'data', 'dictionary.txt') | 
 | print(codespell_file if op.isfile(codespell_file) else '', end='') | 
 | " 2> /dev/null) | 
 |     # @formatter:on | 
 |  | 
 |     # Return the path if found, otherwise return an empty string | 
 |     echo "$python_codespell_dict" | 
 | } | 
 |  | 
 | function do_commit_spelling() { | 
 |     # Write the commit message to a temporary file | 
 |     git log --format='%B' -1 > "$commit_filename" | 
 |  | 
 |     # Some names or emails appear as false-positive misspellings, remove them | 
 |     sed -i "s/Signed-off-by.*//" "$commit_filename" | 
 |  | 
 |     # Get the path to the dictionary.txt file | 
 |     local codespell_dict | 
 |     codespell_dict=$(find_codespell_dict_file) | 
 |  | 
 |     # Check if the dictionary file was found | 
 |     if [[ -z "$codespell_dict" ]]; then | 
 |         echo "Error: Could not find dictionary.txt file" | 
 |         exit 1 | 
 |     fi | 
 |  | 
 |     # Run the codespell with codespell dictionary on the patchset | 
 |     echo -n "codespell-dictionary - misspelling count >> " | 
 |     codespell -D "$codespell_dict" -d --count "$commit_filename" | 
 |  | 
 |     # Run the codespell with builtin dictionary on the patchset | 
 |     echo -n "generic-dictionary - misspelling count >> " | 
 |     codespell --builtin clear,rare,en-GB_to_en-US -d --count "$commit_filename" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([commit_gitlint]="gitlint") | 
 | LINTER_TYPES+=([commit_gitlint]="commit") | 
 | function do_commit_gitlint() { | 
 |     gitlint --extra-path "${CONFIG_PATH}/gitlint/" \ | 
 |         --config "${CONFIG_PATH}/.gitlint" | 
 | } | 
 |  | 
 | # We need different function style for bash/zsh vs plain sh, so beautysh is | 
 | # split into two linters.  "function foo()" is not traditionally accepted | 
 | # POSIX-shell syntax, so shellcheck barfs on it. | 
 | LINTER_REQUIRE+=([beautysh]="beautysh") | 
 | LINTER_IGNORE+=([beautysh]=".beautysh-ignore") | 
 | LINTER_TYPES+=([beautysh]="bash;zsh") | 
 | function do_beautysh() { | 
 |     beautysh --force-function-style fnpar "$@" | 
 | } | 
 | LINTER_REQUIRE+=([beautysh_sh]="beautysh") | 
 | LINTER_IGNORE+=([beautysh_sh]=".beautysh-ignore") | 
 | LINTER_TYPES+=([beautysh_sh]="sh") | 
 | function do_beautysh_sh() { | 
 |     beautysh --force-function-style paronly "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([black]="black") | 
 | LINTER_TYPES+=([black]="python") | 
 | function do_black() { | 
 |     black -l 79 "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([eslint]="eslint;.eslintrc.json;${CONFIG_PATH}/eslint-global-config.json") | 
 | LINTER_IGNORE+=([eslint]=".eslintignore") | 
 | LINTER_TYPES+=([eslint]="json") | 
 | function do_eslint() { | 
 |     eslint --no-eslintrc -c "${LINTER_CONFIG[eslint]}" \ | 
 |         --ext .json --format=stylish \ | 
 |         --resolve-plugins-relative-to /usr/local/lib/node_modules \ | 
 |         --no-error-on-unmatched-pattern "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([flake8]="flake8") | 
 | LINTER_IGNORE+=([flake8]=".flake8-ignore") | 
 | LINTER_TYPES+=([flake8]="python") | 
 | function do_flake8() { | 
 |     flake8 --show-source --extend-ignore=E203,E501 "$@" | 
 |     # We disable E203 and E501 because 'black' is handling these and they | 
 |     # disagree on best practices. | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([isort]="isort") | 
 | LINTER_TYPES+=([isort]="python") | 
 | function do_isort() { | 
 |     isort --profile black "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([markdownlint]="markdownlint;.markdownlint.yaml;${CONFIG_PATH}/markdownlint.yaml") | 
 | LINTER_IGNORE+=([markdownlint]=".markdownlint-ignore") | 
 | LINTER_TYPES+=([markdownlint]="markdown") | 
 | function do_markdownlint() { | 
 |     markdownlint --config "${LINTER_CONFIG[markdownlint]}" \ | 
 |         --disable line-length -- "$@" || \ | 
 |         echo -e "    ${YELLOW}Failed markdownlint; temporarily ignoring." | 
 |     # We disable line-length because prettier should handle prose wrap for us. | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([prettier]="prettier;.prettierrc.yaml;${CONFIG_PATH}/prettierrc.yaml") | 
 | LINTER_IGNORE+=([prettier]=".prettierignore") | 
 | LINTER_TYPES+=([prettier]="json;markdown;yaml") | 
 | function do_prettier() { | 
 |     prettier --config "${LINTER_CONFIG[prettier]}" --write "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([shellcheck]="shellcheck") | 
 | LINTER_IGNORE+=([shellcheck]=".shellcheck-ignore") | 
 | LINTER_TYPES+=([shellcheck]="bash;sh") | 
 | function do_shellcheck() { | 
 |     shellcheck --color=never -x "$@" | 
 | } | 
 |  | 
 | LINTER_REQUIRE+=([clang_format]="clang-format;.clang-format") | 
 | LINTER_IGNORE+=([clang_format]=".clang-ignore;.clang-format-ignore") | 
 | LINTER_TYPES+=([clang_format]="c;cpp") | 
 | function do_clang_format() { | 
 |     "${CLANG_FORMAT}" -i "$@" | 
 | } | 
 |  | 
 | function get_file_type() | 
 | { | 
 |     case "$(basename "$1")" in | 
 |             # First to early detect template files. | 
 |         *.in | *.meson) echo "meson-template" && return ;; | 
 |         *.mako | *.mako.*) echo "mako" && return ;; | 
 |  | 
 |         *.ac) echo "autoconf" && return ;; | 
 |         *.[ch]) echo "c" && return ;; | 
 |         *.[ch]pp) echo "cpp" &&  return ;; | 
 |         *.json) echo "json" && return ;; | 
 |         *.md) echo "markdown" && return ;; | 
 |         *.py) echo "python" && return ;; | 
 |         *.tcl) echo "tcl" && return ;; | 
 |         *.yaml | *.yml) echo "yaml" && return ;; | 
 |  | 
 |             # Special files. | 
 |         .git/COMMIT_EDITMSG) echo "commit" && return ;; | 
 |         meson.build) echo "meson" && return ;; | 
 |     esac | 
 |  | 
 |     case "$(file "$1")" in | 
 |         *Bourne-Again\ shell*) echo "bash" && return ;; | 
 |         *C++\ source*) echo "cpp" && return ;; | 
 |         *C\ source*) echo "c" && return ;; | 
 |         *JSON\ data*) echo "json" && return ;; | 
 |         *POSIX\ shell*) echo "sh" && return ;; | 
 |         *Python\ script*) echo "python" && return ;; | 
 |         *python3\ script*) echo "python" && return ;; | 
 |         *zsh\ shell*) echo "zsh" && return ;; | 
 |     esac | 
 |  | 
 |     echo "unknown" | 
 | } | 
 |  | 
 | LINTERS_AVAILABLE=() | 
 | function check_linter() | 
 | { | 
 |     TITLE="$1" | 
 |     IFS=";" read -r -a ARGS <<< "$2" | 
 |  | 
 |     if [[ "${LINTERS_DISABLED[*]}" =~ $1 ]]; then | 
 |         return | 
 |     fi | 
 |  | 
 |     if [ 0 -ne "${#LINTERS_ENABLED[@]}" ]; then | 
 |         if ! [[ "${LINTERS_ENABLED[*]}" =~ $1 ]]; then | 
 |             return | 
 |         fi | 
 |     fi | 
 |  | 
 |     EXE="${ARGS[0]}" | 
 |     if [ ! -x "${EXE}" ]; then | 
 |         if ! which "${EXE}" > /dev/null 2>&1 ; then | 
 |             echo -e "    ${YELLOW}${TITLE}:${NORMAL} cannot find ${EXE}" | 
 |             if [ -z "$ALLOW_MISSING" ]; then | 
 |                 exit 1 | 
 |             fi | 
 |             return | 
 |         fi | 
 |     fi | 
 |  | 
 |     CONFIG="${ARGS[1]}" | 
 |     FALLBACK="${ARGS[2]}" | 
 |  | 
 |     if [ -n "${CONFIG}" ]; then | 
 |         if [ -e "${CONFIG}" ]; then | 
 |             LINTER_CONFIG+=( [${TITLE}]="${CONFIG}" ) | 
 |         elif [ -n "${FALLBACK}" ] && [ -e "${FALLBACK}" ]; then | 
 |             echo -e "    ${YELLOW}${TITLE}:${NORMAL} cannot find ${CONFIG}; using ${FALLBACK}" | 
 |             LINTER_CONFIG+=( [${TITLE}]="${FALLBACK}" ) | 
 |         else | 
 |             echo -e "    ${YELLOW}${TITLE}:${NORMAL} cannot find config ${CONFIG}" | 
 |             return | 
 |         fi | 
 |     fi | 
 |  | 
 |     LINTERS_AVAILABLE+=( "${TITLE}" ) | 
 | } | 
 |  | 
 | # Check for a global .linter-ignore file. | 
 | GLOBAL_IGNORE=("cat") | 
 | if [ -e ".linter-ignore" ]; then | 
 |     GLOBAL_IGNORE=("${CONFIG_PATH}/lib/ignore-filter" ".linter-ignore") | 
 | fi | 
 |  | 
 | # Find all the files in the git repository and organize by type. | 
 | declare -A FILES=() | 
 | FILES+=([commit]=".git") | 
 |  | 
 | while read -r file; do | 
 |     ftype="$(get_file_type "$file")" | 
 |     FILES+=([$ftype]="$(echo -ne "$file;${FILES[$ftype]:-}")") | 
 | done < <(git ls-files | xargs realpath --relative-base=. | "${GLOBAL_IGNORE[@]}") | 
 |  | 
 | # For each linter, check if there are an applicable files and if it can | 
 | # be enabled. | 
 | for op in "${LINTERS_ALL[@]}"; do | 
 |     for ftype in ${LINTER_TYPES[$op]//;/ }; do | 
 |         if [[ -v FILES["$ftype"] ]]; then | 
 |             check_linter "$op" "${LINTER_REQUIRE[${op}]}" | 
 |             break | 
 |         fi | 
 |     done | 
 | done | 
 |  | 
 | # Call each linter. | 
 | for op in "${LINTERS_AVAILABLE[@]}"; do | 
 |  | 
 |     # Determine the linter-specific ignore file(s). | 
 |     LOCAL_IGNORE=("${CONFIG_PATH}/lib/ignore-filter") | 
 |     if [[ -v LINTER_IGNORE["$op"] ]]; then | 
 |         for ignorefile in ${LINTER_IGNORE["$op"]//;/ } ; do | 
 |             if [ -e "$ignorefile" ]; then | 
 |                 LOCAL_IGNORE+=("$ignorefile") | 
 |             fi | 
 |         done | 
 |     fi | 
 |     if [ 1 -eq ${#LOCAL_IGNORE[@]} ]; then | 
 |         LOCAL_IGNORE=("cat") | 
 |     fi | 
 |  | 
 |     # Find all the files for this linter, filtering out the ignores. | 
 |     LINTER_FILES=() | 
 |     while read -r file ; do | 
 |         if [ -e "$file" ]; then | 
 |             LINTER_FILES+=("$file") | 
 |         fi | 
 |         done < <(for ftype in ${LINTER_TYPES[$op]//;/ }; do | 
 |             # shellcheck disable=SC2001 | 
 |             echo "${FILES["$ftype"]:-}" | sed "s/;/\\n/g" | 
 |     done | "${LOCAL_IGNORE[@]}") | 
 |  | 
 |     # Call the linter now with all the files. | 
 |     if [ 0 -ne ${#LINTER_FILES[@]} ]; then | 
 |         echo -e "    ${BLUE}Running $op${NORMAL}" | 
 |         if ! "do_$op" "${LINTER_FILES[@]}" ; then | 
 |             LINTERS_FAILED+=([$op]=1) | 
 |         fi | 
 |     else | 
 |         echo -e "    ${YELLOW}${op}:${NORMAL} all applicable files are on ignore-lists" | 
 |     fi | 
 | done | 
 |  | 
 | # Check for failing linters. | 
 | if [ 0 -ne ${#LINTERS_FAILED[@]} ]; then | 
 |     for op in "${!LINTERS_FAILED[@]}"; do | 
 |         echo -e "$op: ${RED}FAILED${NORMAL}" | 
 |     done | 
 |     exit 1 | 
 | fi | 
 |  | 
 | # Check for differences. | 
 | if [ -z "$OPTION_NO_DIFF" ]; then | 
 |     echo -e "    ${BLUE}Result differences...${NORMAL}" | 
 |     if ! git --no-pager diff --exit-code ; then | 
 |         echo -e "Format: ${RED}FAILED${NORMAL}" | 
 |         exit 1 | 
 |     else | 
 |         echo -e "Format: ${GREEN}PASSED${NORMAL}" | 
 |     fi | 
 | fi | 
 |  | 
 | # Sometimes your situation is terrible enough that you need the flexibility. | 
 | # For example, phosphor-mboxd. | 
 | for formatter in "format-code.sh" "format-code"; do | 
 |     if [[ -x "${formatter}" ]]; then | 
 |         echo -e "    ${BLUE}Calling secondary formatter:${NORMAL} ${formatter}" | 
 |         "./${formatter}" | 
 |         if [ -z "$OPTION_NO_DIFF" ]; then | 
 |             git --no-pager diff --exit-code | 
 |         fi | 
 |     fi | 
 | done |