add license header check v2 (#38)

fix #37

Add v2 for new branch strategy. Allow to check main branch

test run:
https://github.com/YanxuanLiu/spark-rapids/actions/runs/15484134743/job/43595238006?pr=31

---------

Signed-off-by: Yanxuan Liu <[email protected]>

Author: YanxuanLiu

.github/workflows/license-header-check.yml

@@ -36,6 +36,7 @@ jobs:
       - name: license-header-check
         uses: NVIDIA/spark-rapids-common/license-header-check@main
         with:
+          check_main: true
           included_file_patterns: |
             *.yml,
             *Dockerfile*,

license-header-check/action.yml

@@ -23,6 +23,11 @@ inputs:
     description: "excluded file pattern"
     required: false
     type: string
+  check_main:
+    description: "check main branch or not"
+    required: false
+    type: boolean
+    default: false
 
 runs:
   using: "composite"
@@ -36,8 +41,8 @@ runs:
         CURRENT_YEAR=$(date +%Y)
         LICENSE_PATTERN="Copyright \(c\) .*?${CURRENT_YEAR},? NVIDIA CORPORATION."
 
-        # Skip for merge to main PR
-        if [[ "${{ github.event.pull_request.base.ref }}" == "main" ]]; then
+        # Skip check for main branch when check_main == false
+        if [[ "${{ inputs.check_main }}" == "false" && "${{ github.event.pull_request.base.ref }}" == "main" ]]; then
           echo "Skip this check for Merge to main PR."
           exit 0
         fi