fix(fmt): stop adding extra lines between Go comments and templates (#…

…351)

Co-authored-by: Adrian Hesketh <adrianhesketh@hushmail.com>

.version

@@ -1 +1 @@
-0.2.503
+0.2.507

generator/generator.go

@@ -292,6 +292,15 @@ func (g *generator) writeGoExpression(n parser.TemplateFileGoExpression) (err er
 		return err
 	}
 	g.sourceMap.Add(n.Expression, r)
+	v := n.Expression.Value
+	lineSlice := strings.Split(v, "\n")
+	lastLine := lineSlice[len(lineSlice)-1]
+	if strings.HasPrefix(lastLine, "//") {
+		if _, err = g.w.WriteIndent(0, "\n"); err != nil {
+			return err
+		}
+		return err
+	}
 	if _, err = g.w.WriteIndent(0, "\n\n"); err != nil {
 		return err
 	}

generator/generator_test.go

@@ -14,22 +14,36 @@ func TestGeneratorSourceMap(t *testing.T) {
 		w:         NewRangeWriter(w),
 		sourceMap: parser.NewSourceMap(),
 	}
-	exp := parser.TemplateFileGoExpression{
+	invalidExp := parser.TemplateFileGoExpression{
 		Expression: parser.Expression{
 			Value: "line1\nline2",
 		},
 	}
-	err := g.writeGoExpression(exp)
-	if err != nil {
+	if err := g.writeGoExpression(invalidExp); err != nil {
 		t.Fatalf("failed to write Go expression: %v", err)
 	}
-	expected := parser.NewPosition(0, 0, 0)
 
+	expected := parser.NewPosition(0, 0, 0)
 	actual, ok := g.sourceMap.TargetPositionFromSource(0, 0)
 	if !ok {
 		t.Errorf("failed to get matching target")
 	}
 	if diff := cmp.Diff(expected, actual); diff != "" {
 		t.Errorf("unexpected target:\n%v", diff)
 	}
+
+	withCommentExp := parser.TemplateFileGoExpression{
+		Expression: parser.Expression{
+			Value: `package main
+
+// A comment.
+templ h1() {
+	<h1></h1>
+}
+			`,
+		},
+	}
+	if err := g.writeGoExpression(withCommentExp); err != nil {
+		t.Fatalf("failed to write Go expression: %v", err)
+	}
 }

generator/test-css-usage/template.templ

@@ -1,7 +1,6 @@
 package testcssusage
 
 // Constant class.
-
 templ StyleTagsAreSupported() {
 	<style>
 	.test {
@@ -25,15 +24,13 @@ templ CSSComponentsAreSupported() {
 
 // Both CSS components and constants are supported.
 // Only string names are really required. There is no need to use templ.Class or templ.SafeClass.
-
 templ CSSComponentsAndConstantsAreSupported() {
 	<div class={ cssComponentGreen(), "classA", templ.Class("&&&classB"), templ.SafeClass("classC"), "d e" } type="button">Both CSS components and constants are supported</div>
 	// The following is also valid, but not required - you can put the class names in directly.
 	<div class={ templ.Classes(cssComponentGreen(), "classA", templ.Class("&&&classB"), templ.SafeClass("classC")), "d e" } type="button">Both CSS components and constants are supported</div>
 }
 
 // Maps can be used to determine if a class should be added or not.
-
 templ MapsCanBeUsedToConditionallySetClasses() {
 	<div class={ map[string]bool{"a": true, "b": false, "c": true} }>Maps can be used to determine if a class should be added or not.</div>
 }
@@ -53,19 +50,16 @@ templ KVCanBeUsedToConditionallySetClasses() {
 }
 
 // Pseudo attributes can be used without any special syntax.
-
 templ PsuedoAttributesAndComplexClassNamesAreSupported() {
 	<div class={ "bg-violet-500", "hover:bg-red-600", "hover:bg-sky-700", "text-[#50d71e]", "w-[calc(100%-4rem)" }>Psuedo attributes and complex class names are supported.</div>
 }
 
 // Class names are HTML escaped.
-
 templ ClassNamesAreHTMLEscaped() {
 	<div class={ "a\" onClick=\"alert('hello')\"" }>Class names are HTML escaped.</div>
 }
 
 // Combine all tests.
-
 templ TestComponent() {
 	@StyleTagsAreSupported()
 	@CSSComponentsAreSupported()

parser/v2/types.go

@@ -125,27 +125,37 @@ func (tf TemplateFile) Write(w io.Writer) error {
 	if err := tf.Package.Write(w, indent); err != nil {
 		return err
 	}
-	if _, err := w.Write([]byte("\n\n")); err != nil {
+	if _, err := io.WriteString(w, "\n\n"); err != nil {
 		return err
 	}
-	count := len(tf.Nodes)
-	for i := 0; i < count; i++ {
-		var err error
-		if err = tf.Nodes[i].Write(w, indent); err != nil {
+	for i := 0; i < len(tf.Nodes); i++ {
+		if err := tf.Nodes[i].Write(w, indent); err != nil {
 			return err
 		}
-		if i == count-1 {
-			_, err = w.Write([]byte("\n"))
-		} else {
-			_, err = w.Write([]byte("\n\n"))
-		}
-		if err != nil {
+		if _, err := io.WriteString(w, getNodeWhitespace(tf.Nodes, i)); err != nil {
 			return err
 		}
 	}
 	return nil
 }
 
+func getNodeWhitespace(nodes []TemplateFileNode, i int) string {
+	if i == len(nodes)-1 {
+		return "\n"
+	}
+	if _, nextIsTemplate := nodes[i+1].(HTMLTemplate); nextIsTemplate {
+		if e, isGo := nodes[i].(TemplateFileGoExpression); isGo && endsWithComment(e.Expression.Value) {
+			return "\n"
+		}
+	}
+	return "\n\n"
+}
+
+func endsWithComment(s string) bool {
+	lineSlice := strings.Split(s, "\n")
+	return strings.HasPrefix(lineSlice[len(lineSlice)-1], "//")
+}
+
 // TemplateFileNode can be a Template, CSS, Script or Go.
 type TemplateFileNode interface {
 	IsTemplateFileNode() bool

parser/v2/types_test.go

@@ -776,6 +776,25 @@ templ test() {
 		Leave this alone.
 	*/
 }
+`,
+		},
+		{
+			name: "godoc comments are preserved",
+			input: ` // first line removed to make indentation clear
+package main
+
+// test the comment handling.
+templ test() {
+	Test
+}
+`,
+			expected: ` // first line removed to make indentation clear
+package main
+
+// test the comment handling.
+templ test() {
+	Test
+}
 `,
 		},
 	}