chore: Update check_new_py_files.sh to run in multiple VCS environments

Co-authored-by: Xuan Yang <xygoogle@google.com>
PiperOrigin-RevId: 941262556

Author: xuanyang15

scripts/check_new_py_files.sh

@@ -16,12 +16,24 @@
 
 exit_code=0
 
-# Get list of newly added files using diff-filter=A
-# Using process substitution to avoid subshell and handle spaces in filenames
+get_added_files() {
+  if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
+    git diff --cached --name-only --diff-filter=A
+  elif jj root >/dev/null 2>&1; then
+    jj diff --summary 2>/dev/null | awk '/^A / {print $2}'
+  elif hg root >/dev/null 2>&1; then
+    hg status --added --no-status 2>/dev/null
+  elif g4 info >/dev/null 2>&1; then
+    g4 opened 2>/dev/null | awk '/ - add / {print $1}' | sed 's/#.*//'
+  elif p4 info >/dev/null 2>&1; then
+    p4 opened 2>/dev/null | awk '/ - add / {print $1}' | sed 's/#.*//'
+  fi
+}
+
 while read -r file; do
     # Check if file is not empty (happens if no new files)
     if [[ -n "$file" ]]; then
-        if [[ "$file" == src/google/adk/*.py ]]; then
+        if [[ "$file" == */google/adk/*.py ]] || [[ "$file" == google/adk/*.py ]]; then
             filename=$(basename "$file")
             if [[ ! "$filename" == _* ]]; then
                 echo "Error: New Python file '$file' must have a '_' prefix."
@@ -32,6 +44,6 @@ while read -r file; do
             fi
         fi
     fi
-done < <(git diff --cached --name-only --diff-filter=A)
+done < <(get_added_files)
 
 exit $exit_code