From a12b92c928ae64efa99b9bc824667daa7dd444b1 Mon Sep 17 00:00:00 2001
From: Matt Dale <9760375+matthewdale@users.noreply.github.com>
Date: Mon, 9 Jun 2025 21:15:31 -0700
Subject: [PATCH] GODRIVER-3543 Check license headers in pre-commit.

---
 .pre-commit-config.yaml |  6 ++++++
 etc/check_license.sh    | 23 ++++++++++++++++++-----
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 571868de35..27f3bd3147 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -65,3 +65,9 @@ repos:
     require_serial: true
     pass_filenames: false
     entry: etc/golangci-lint.sh
+
+  - id: check-licenses
+    name: check-licenses
+    language: system
+    types: [go]
+    entry: etc/check_license.sh
diff --git a/etc/check_license.sh b/etc/check_license.sh
index 5d13fa68e3..5ea046daa5 100755
--- a/etc/check_license.sh
+++ b/etc/check_license.sh
@@ -8,7 +8,7 @@ copyright=$"// Copyright (C) MongoDB, Inc. $year-present.
 // a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 "
 
-add_copyright() {
+check_or_add_copyright() {
     file=$1
 
     # Check if first 24 bytes match first 24 bytes of copyright notice.
@@ -54,10 +54,23 @@ do
     esac
 done
 
-# Find all .go files and try to write a license notice.
-GO_FILES=$(find . -type f -name "*.go" -print)
+# Shift script arguments so $1 contains only optional filename args.
+# E.g. check_license.sh -a ./mongo/cursor.go
+shift "$((OPTIND - 1))"
+FILES=$1
 
-for file in $GO_FILES
+# If no filenames were passed, find all .go files and try to write a license
+# notice.
+if [ -z "$FILES" ]; then
+    FILES=$(find . -type f -name "*.go" -print)
+fi
+
+for file in $FILES
 do
-    add_copyright "$file"
+    # Skip any files that aren't .go files.
+    if [[ ! "$file" =~ .go$ ]]; then
+        continue
+    fi
+
+    check_or_add_copyright "$file"
 done