Add support for //go:build in minify function

koki-develop · Mar 6, 2024 · acd826a · acd826a
1 parent 31d4de0
commit acd826a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/mingo/mingo.go b/internal/mingo/mingo.go
@@ -34,8 +34,11 @@ func (m *mingo) Minify(filename string, src []byte) (string, error) {
 
 	for _, cg := range file.Comments {
 		for _, c := range cg.List {
-			if strings.HasPrefix(c.Text, "//go:generate ") {
-				fmt.Fprintln(sb, c.Text)
+			dirs := []string{"//go:build ", "//go:generate "}
+			for _, prefix := range dirs {
+				if strings.HasPrefix(c.Text, prefix) {
+					fmt.Fprintln(sb, c.Text)
+				}
 			}
 		}
 	}