make that work

Author: romank-msft

Diff for: .github/scripts/add_unsafe_reviewers/add-unsafe-reviewers.py

@@ -1,3 +1,6 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
 import click
 from git import Repo
 from github import Github

Diff for: .github/scripts/refresh_mirror/refresh-mirror.py

@@ -1,3 +1,6 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
 import click
 import time
 import sys

Diff for: openhcl/gen_init_ramfs.py

@@ -1,5 +1,4 @@
 #!/usr/bin/python3
-
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 

Diff for: openhcl/update-rootfs.py

@@ -1,5 +1,4 @@
 #!/usr/bin/python3
-
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 

Diff for: support/clap_dyn_complete/src/templates/complete.ps1

@@ -1,3 +1,6 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
 Register-ArgumentCompleter -Native -CommandName '__COMMAND_NAME__' -ScriptBlock {
     param($wordToComplete, $commandAst, $cursorPosition)
 

Diff for: vmm_tests/vmm_tests/test_data/tpm.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-#
+
 # The test script that reads AK CERT from TPM and validate its content
 
 # TPM NV READ command for Index 0x01c101d0 (AK CERT) that takes size (u32) and offset (u32) arguments through appending

Diff for: xtask/src/tasks/fmt/house_rules/copyright.rs

@@ -17,24 +17,34 @@ pub fn check_copyright(path: &Path, fix: bool) -> anyhow::Result<()> {
         .and_then(|e| e.to_str())
         .unwrap_or_default();
 
-    if !matches!(ext, "rs" | "c" | "proto" | "toml" | "ts" | "js" | "py" | "ps1" | "config") {
+    if !matches!(
+        ext,
+        "rs" | "c" | "proto" | "toml" | "ts" | "js" | "py" | "ps1" | "config"
+    ) {
         return Ok(());
     }
 
     let f = BufReader::new(File::open(path)?);
     let mut lines = f.lines();
-    let first_line = lines.next().unwrap_or(Ok(String::new()))?;
-    let second_line = lines.next().unwrap_or(Ok(String::new()))?;
-    let third_line = lines.next().unwrap_or(Ok(String::new()))?;
+    let first_content_line = {
+        let line = lines.next().unwrap_or(Ok(String::new()))?;
+        if line.starts_with("#!") {
+            lines.next().unwrap_or(Ok(String::new()))?
+        } else {
+            line
+        }
+    };
+    let second_content_line = lines.next().unwrap_or(Ok(String::new()))?;
+    let third_content_line = lines.next().unwrap_or(Ok(String::new()))?;
 
     // Preserve any files which are copyright, but not by Microsoft.
-    if first_line.contains("Copyright") && !first_line.contains("Microsoft") {
+    if first_content_line.contains("Copyright") && !first_content_line.contains("Microsoft") {
         return Ok(());
     }
 
-    let mut missing_banner =
-        !first_line.contains(HEADER_MIT_FIRST) || !second_line.contains(HEADER_MIT_SECOND);
-    let mut missing_blank_line = !third_line.is_empty();
+    let mut missing_banner = !first_content_line.contains(HEADER_MIT_FIRST)
+        || !second_content_line.contains(HEADER_MIT_SECOND);
+    let mut missing_blank_line = !third_content_line.is_empty();
 
     // TEMP: until we have more robust infrastructure for distinct
     // microsoft-internal checks, include this "escape hatch" for preserving
@@ -44,8 +54,9 @@ pub fn check_copyright(path: &Path, fix: bool) -> anyhow::Result<()> {
     let is_msft_internal = std::env::var("XTASK_FMT_COPYRIGHT_ALLOW_MISSING_MIT").is_ok();
     if is_msft_internal {
         // support both new and existing copyright banner styles
-        missing_banner = !(first_line.contains("Copyright") && first_line.contains("Microsoft"));
-        missing_blank_line = !second_line.is_empty();
+        missing_banner =
+            !(first_content_line.contains("Copyright") && first_content_line.contains("Microsoft"));
+        missing_blank_line = !second_content_line.is_empty();
     }
 
     if fix {