chore: update method isMakeComment

Bin-Huang · Aug 16, 2022 · 64e6182 · 64e6182
1 parent 0913328
commit 64e6182
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/parser.go b/parser.go
@@ -168,7 +168,12 @@ func ParseCodeFile(filename string) ([]StructInfo, []ImportInfo, error) {
 // isMakeComment ...
 func isMakeComment(s string) bool {
 	s = strings.TrimSpace(s)
-	return strings.HasPrefix(s, "//go:generate") && strings.Contains(s, "newc")
+	if !strings.HasPrefix(s, "//go:generate") {
+		return false
+	}
+	s = strings.TrimLeft(s, "//go:generate")
+	s = strings.TrimSpace(s)
+	return strings.HasPrefix(s, "newc ")
 }
 
 // isInitModeEnable check if this struct enable the init mode