Add support for go:embed in stringifyVarDecl function

koki-develop · Mar 6, 2024 · 15c91ff · 15c91ff
1 parent acd826a
commit 15c91ff
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/internal/mingo/decl.go b/internal/mingo/decl.go
@@ -110,6 +110,15 @@ func (m *mingo) stringifyConstDecl(decl *ast.GenDecl) string {
 
 func (m *mingo) stringifyVarDecl(decl *ast.GenDecl) string {
 	sb := new(strings.Builder)
+
+	if decl.Doc != nil {
+		for _, cmt := range decl.Doc.List {
+			if strings.HasPrefix(cmt.Text, "//go:embed ") {
+				fmt.Fprintf(sb, "\n%s\n", cmt.Text)
+			}
+		}
+	}
+
 	sb.WriteString("var")
 
 	if len(decl.Specs) > 1 {
@@ -128,6 +137,14 @@ func (m *mingo) stringifyVarDecl(decl *ast.GenDecl) string {
 			if j > 0 {
 				sb.WriteString(",")
 			}
+
+			if spec.Doc != nil {
+				for _, cmt := range spec.Doc.List {
+					if strings.HasPrefix(cmt.Text, "//go:embed ") {
+						fmt.Fprintf(sb, "\n%s\n", cmt.Text)
+					}
+				}
+			}
 			sb.WriteString(name.Name)
 		}