From 9cc7ec3e36ab449bec29c9aea709c1969b97ace1 Mon Sep 17 00:00:00 2001
From: Sergii Baitala <sbaitala@gmail.com>
Date: Mon, 29 Mar 2021 19:54:22 +0300
Subject: [PATCH] Support WiX files

---
 main.go      | 2 +-
 main_test.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index caf14d9..105a006 100644
--- a/main.go
+++ b/main.go
@@ -297,7 +297,7 @@ func licenseHeader(path string, tmpl *template.Template, data licenseData) ([]by
 		lic, err = executeTemplate(tmpl, data, "", "% ", "")
 	case ".hs", ".sql", ".sdl":
 		lic, err = executeTemplate(tmpl, data, "", "-- ", "")
-	case ".html", ".xml", ".vue":
+	case ".html", ".xml", ".vue", ".wxi", ".wxl", ".wxs":
 		lic, err = executeTemplate(tmpl, data, "<!--", " ", "-->")
 	case ".php":
 		lic, err = executeTemplate(tmpl, data, "", "// ", "")
diff --git a/main_test.go b/main_test.go
index 90b56ef..94998d5 100644
--- a/main_test.go
+++ b/main_test.go
@@ -332,7 +332,7 @@ func TestLicenseHeader(t *testing.T) {
 			"-- HYS\n\n",
 		},
 		{
-			[]string{"f.html", "f.xml", "f.vue"},
+			[]string{"f.html", "f.xml", "f.vue", "f.wxi", "f.wxl", "f.wxs"},
 			"<!--\n HYS\n-->\n\n",
 		},
 		{