Handled sentinel policy scenario

Author: sonamtenzin6

addlicense/main.go

@@ -414,8 +414,31 @@ var head = []string{
 	"/** @jest-environment",    // Jest Environment string https://jestjs.io/docs/configuration#testenvironment-string
 }
 
+var headPatterns = []string{
+	`# This policy requires.*\s*(#.*\s*)*`,
+}
+
+// matches regex patterns to extract headings to skip
+func matchPattern(b []byte) []byte {
+
+	for _, v := range headPatterns {
+		re := regexp.MustCompile(v)
+		match := re.Find(b)
+		if len(match) > 0 {
+			return match
+		}
+	}
+	return []byte{}
+}
+
 func hashBang(b []byte) []byte {
 	var line []byte
+
+	line = matchPattern(b)
+	if len(line) > 0 {
+		return line
+	}
+
 	for _, c := range b {
 		line = append(line, c)
 		if c == '\n' {
@@ -428,6 +451,7 @@ func hashBang(b []byte) []byte {
 			return line
 		}
 	}
+
 	return nil
 }
 

addlicense/main_test.go

@@ -245,6 +245,17 @@ func TestAddLicense(t *testing.T) {
 		{"# escape: `\ncontent", "# escape: `\n// HYS\n\ncontent", true},
 		{"# syntax: docker/dockerfile:1.3\ncontent", "# syntax: docker/dockerfile:1.3\n// HYS\n\ncontent", true},
 		{"/** @jest-environment jsdom */\ncontent", "/** @jest-environment jsdom */\n// HYS\n\ncontent", true},
+		{
+			"# This policy requires immediate action.\n# This is a follow-up comment.\n# Another line of policy.\nSome text that should not match.",
+			"# This policy requires immediate action.\n# This is a follow-up comment.\n# Another line of policy.\n// HYS\n\nSome text that should not match.",
+			true,
+		},
+		{
+			`# This policy requires that the max_password_age attribute of the aws_iam_account_password_policy
+			# resource is according to CIS standards.`,
+			"# This policy requires that the max_password_age attribute of the aws_iam_account_password_policy\n\t\t\t# resource is according to CIS standards.\n// HYS\n\n",
+			true,
+		},
 
 		// ensure files with existing license or generated files are
 		// skipped. No need to test all permutations of these, since