Maintain whitespace before tag close

.changeset/seven-moons-design.md

@@ -0,0 +1,5 @@
+---
+'@astrojs/compiler': patch
+---
+
+[TSX] maintain trailing whitespace before an element is closed, fixing TypeScript completion in some cases

internal/printer/print-to-tsx.go

@@ -375,15 +375,13 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 			p.print(`:`)
 			p.addSourceMapping(loc.Loc{Start: eqStart + 1})
 			p.print(`"` + encodeDoubleQuote(a.Val) + `"`)
-			endLoc = eqStart + 1 + len(a.Val) + 2
 		case astro.EmptyAttribute:
 			p.print(a.Key)
 			p.print(`"`)
 			p.addNilSourceMapping()
 			p.print(`:`)
 			p.addSourceMapping(a.KeyLoc)
 			p.print(`true`)
-			endLoc = a.KeyLoc.Start + len(a.Key)
 		case astro.ExpressionAttribute:
 			p.print(a.Key)
 			p.print(`"`)
@@ -394,7 +392,6 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 			p.printTextWithSourcemap(a.Val, loc.Loc{Start: eqStart + 2})
 			p.addSourceMapping(loc.Loc{Start: eqStart + 2 + len(a.Val)})
 			p.print(`)`)
-			endLoc = eqStart + len(a.Val) + 2
 		case astro.SpreadAttribute:
 			// noop
 		case astro.ShorthandAttribute:
@@ -404,14 +401,12 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 			}
 			p.addSourceMapping(a.KeyLoc)
 			p.print(a.Key)
-			endLoc = a.KeyLoc.Start + len(a.Key)
 		case astro.TemplateLiteralAttribute:
 			p.addSourceMapping(a.KeyLoc)
 			p.print(a.Key)
 			p.print(`":`)
 			p.addSourceMapping(a.ValLoc)
 			p.print(fmt.Sprintf("`%s`", a.Val))
-			endLoc = a.ValLoc.Start + len(a.Val) + 2
 		}
 		if i == len(invalidTSXAttributes)-1 {
 			p.addNilSourceMapping()
@@ -425,6 +420,7 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 		endLoc = 0
 	}
 	isSelfClosing := false
+	hasLeadingSpace := false
 	tmpLoc := endLoc
 	if len(p.sourcetext) > tmpLoc {
 		for i := 0; i < len(p.sourcetext[tmpLoc:]); i++ {
@@ -436,6 +432,9 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 				p.addSourceMapping(loc.Loc{Start: endLoc})
 				endLoc++
 				break
+			} else if unicode.IsSpace(rune(c)) {
+				hasLeadingSpace = true
+				endLoc++
 			} else {
 				endLoc++
 			}
@@ -444,13 +443,16 @@ declare const Astro: Readonly<import('astro').AstroGlobal<%s>>`, props.Ident)
 		endLoc++
 	}
 
+	if hasLeadingSpace {
+		p.addSourceMapping(loc.Loc{Start: endLoc - 1})
+		p.print(" ")
+	}
+
 	if voidElements[n.Data] && n.FirstChild == nil {
 		p.print("/>")
 		return
 	}
 	if isSelfClosing && n.FirstChild == nil {
-		p.addSourceMapping(loc.Loc{Start: endLoc - 1})
-		p.print(" ")
 		p.addSourceMapping(loc.Loc{Start: endLoc})
 		p.print("/>")
 		return

packages/compiler/test/tsx/basic.ts

@@ -214,4 +214,24 @@ export default function __AstroComponent_(_props: Record<string, any>): any {}\n
   assert.snapshot(code, output, `expected code to match snapshot`);
 });
 
+test('preserves spaces in tag', async () => {
+  const input = '<Button ></Button>';
+  const output = `<Fragment>
+<Button ></Button>
+</Fragment>
+export default function __AstroComponent_(_props: Record<string, any>): any {}\n`;
+  const { code } = await convertToTSX(input, { sourcemap: 'external' });
+  assert.snapshot(code, output, `expected code to match snapshot`);
+});
+
+test('preserves spaces after attributes in tag', async () => {
+  const input = '<Button a="b" ></Button>';
+  const output = `<Fragment>
+<Button a="b" ></Button>
+</Fragment>
+export default function __AstroComponent_(_props: Record<string, any>): any {}\n`;
+  const { code } = await convertToTSX(input, { sourcemap: 'external' });
+  assert.snapshot(code, output, `expected code to match snapshot`);
+});
+
 test.run();