From b441885d3009f4f2bff3e68386d7a350863ce5ca Mon Sep 17 00:00:00 2001
From: Taiki Endo <te316e89@gmail.com>
Date: Sat, 16 Sep 2023 17:56:46 +0900
Subject: [PATCH] tidy: Update tools/tidy.sh and add SPDX-License-Identifier

---
 tools/.tidy-check-license-headers |  2 +-
 tools/codegen/src/main.rs         |  2 ++
 tools/tidy.sh                     | 17 ++++++++++-------
 3 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/tools/.tidy-check-license-headers b/tools/.tidy-check-license-headers
index 3c5d9a0c..46c7417c 100644
--- a/tools/.tidy-check-license-headers
+++ b/tools/.tidy-check-license-headers
@@ -1 +1 @@
-git ls-files '*.sh' # TODO: check more files
+git ls-files
diff --git a/tools/codegen/src/main.rs b/tools/codegen/src/main.rs
index 17366d99..0d1cc1f5 100644
--- a/tools/codegen/src/main.rs
+++ b/tools/codegen/src/main.rs
@@ -1,3 +1,5 @@
+// SPDX-License-Identifier: Apache-2.0 OR MIT
+
 use anyhow::{bail, Context as _, Result};
 use fs_err as fs;
 use serde::{Deserialize, Serialize};
diff --git a/tools/tidy.sh b/tools/tidy.sh
index 4442e1dc..3e33ddca 100755
--- a/tools/tidy.sh
+++ b/tools/tidy.sh
@@ -250,19 +250,22 @@ if [[ -f tools/.tidy-check-license-headers ]]; then
     info "checking license headers (experimental)"
     failed_files=''
     for p in $(eval $(<tools/.tidy-check-license-headers)); do
-        # TODO: More file types?
         case "$(basename "${p}")" in
-            *.sh) prefix=("# ") ;;
-            *.rs | *.c | *.h | *.cpp | *.hpp | *.s | *.S) prefix=("// " "/* ") ;;
+            *.stderr | *.expanded.rs) continue ;; # generated files
+            *.sh | *.py | *.rb | *Dockerfile) prefix=("# ") ;;
+            *.rs | *.c | *.h | *.cpp | *.hpp | *.s | *.S | *.js) prefix=("// " "/* ") ;;
             *.ld | *.x) prefix=("/* ") ;;
-            *) error "unrecognized file type: ${p}" ;;
+            # TODO: More file types?
+            *) continue ;;
         esac
         # TODO: The exact line number is not actually important; it is important
         # that it be part of the top-level comments of the file.
         line="1"
-        case "${p}" in
-            *.sh) line="2" ;; # shebang
-        esac
+        if IFS= LC_ALL=C read -rn3 -d '' shebang <"${p}" && [[ "${shebang}" == '#!/' ]]; then
+            line="2"
+        elif [[ "${p}" == *"Dockerfile" ]] && IFS= LC_ALL=C read -rn9 -d '' syntax <"${p}" && [[ "${syntax}" == '# syntax=' ]]; then
+            line="2"
+        fi
         header_found=''
         for pre in "${prefix[@]}"; do
             if [[ "$(grep -E -n "${pre}SPDX-License-Identifier: " "${p}")" == "${line}:${pre}SPDX-License-Identifier: "* ]]; then