fix(parser): panic when parsing JSDoc @Satisfies with export assignments (#1647)

Author: camc314

internal/ast/ast.go

@@ -779,6 +779,8 @@ func (n *Node) Initializer() *Node {
 		return n.AsForInOrOfStatement().Initializer
 	case KindJsxAttribute:
 		return n.AsJsxAttribute().Initializer
+	case KindCommonJSExport:
+		return n.AsCommonJSExport().Initializer
 	}
 	panic("Unhandled case in Node.Initializer")
 }
@@ -806,6 +808,8 @@ func (m *mutableNode) SetInitializer(initializer *Node) {
 		n.AsForInOrOfStatement().Initializer = initializer
 	case KindJsxAttribute:
 		n.AsJsxAttribute().Initializer = initializer
+	case KindCommonJSExport:
+		n.AsCommonJSExport().Initializer = initializer
 	default:
 		panic("Unhandled case in mutableNode.SetInitializer")
 	}

internal/parser/reparser.go

@@ -357,7 +357,7 @@ func (p *Parser) reparseHosted(tag *ast.Node, parent *ast.Node, jsDoc *ast.Node)
 				}
 			}
 		case ast.KindVariableDeclaration,
-			ast.KindCommonJSExport, ast.KindExportAssignment, ast.KindJSExportAssignment,
+			ast.KindCommonJSExport,
 			ast.KindPropertyDeclaration, ast.KindPropertyAssignment, ast.KindShorthandPropertyAssignment:
 			if parent.Initializer() != nil && tag.AsJSDocSatisfiesTag().TypeExpression != nil {
 				parent.AsMutable().SetInitializer(p.makeNewCast(
@@ -366,7 +366,8 @@ func (p *Parser) reparseHosted(tag *ast.Node, parent *ast.Node, jsDoc *ast.Node)
 					false /*isAssertion*/))
 				p.finishMutatedNode(parent)
 			}
-		case ast.KindReturnStatement, ast.KindParenthesizedExpression:
+		case ast.KindReturnStatement, ast.KindParenthesizedExpression,
+			ast.KindExportAssignment, ast.KindJSExportAssignment:
 			if parent.Expression() != nil && tag.AsJSDocSatisfiesTag().TypeExpression != nil {
 				parent.AsMutable().SetExpression(p.makeNewCast(
 					p.factory.DeepCloneReparse(tag.AsJSDocSatisfiesTag().TypeExpression.Type()),

testdata/baselines/reference/compiler/panicSatisfiesOnExportEqualsDeclaration.symbols

@@ -0,0 +1,11 @@
+//// [tests/cases/compiler/panicSatisfiesOnExportEqualsDeclaration.ts] ////
+
+=== panicSatisfiesOnExportEqualsDeclaration.js ===
+/**
+ * @satisfies {Record<string, never>}
+ */
+module.exports = {};
+>module.exports : Symbol(export=, Decl(panicSatisfiesOnExportEqualsDeclaration.js, 0, 0))
+>module : Symbol(module.exports)
+>exports : Symbol(export=, Decl(panicSatisfiesOnExportEqualsDeclaration.js, 0, 0))
+

testdata/baselines/reference/compiler/panicSatisfiesOnExportEqualsDeclaration.types

@@ -0,0 +1,13 @@
+//// [tests/cases/compiler/panicSatisfiesOnExportEqualsDeclaration.ts] ////
+
+=== panicSatisfiesOnExportEqualsDeclaration.js ===
+/**
+ * @satisfies {Record<string, never>}
+ */
+module.exports = {};
+>module.exports = {} : {}
+>module.exports : {}
+>module : { "export=": {}; }
+>exports : {}
+>{} : {}
+

testdata/tests/cases/compiler/panicSatisfiesOnExportEqualsDeclaration.ts

@@ -0,0 +1,11 @@
+// @allowJs: true
+// @checkJs: true
+// @noEmit: true
+// @module: commonjs
+// @lib: esnext,dom,dom.iterable
+// @Filename: panicSatisfiesOnExportEqualsDeclaration.js
+
+/**
+ * @satisfies {Record<string, never>}
+ */
+module.exports = {};