fix: TSTypeAnnotation printer when parent is a TSFunction type

benjamn · Jun 12, 2024 · ead67b6 · ead67b6
1 parent 887f5b7
commit ead67b6
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 2 deletions.
diff --git a/lib/printer.ts b/lib/printer.ts
@@ -2307,8 +2307,16 @@ function genericPrintNoParens(path: any, options: any, print: any) {
     case "TSNonNullExpression":
       return concat([path.call(print, "expression"), "!"]);
 
-    case "TSTypeAnnotation":
-      return concat([": ", path.call(print, "typeAnnotation")]);
+    case "TSTypeAnnotation": {
+      const isFunctionType = namedTypes.TSFunctionType.check(
+        path.getParentNode(0),
+      );
+
+      return concat([
+        isFunctionType ? "=> " : ": ",
+        path.call(print, "typeAnnotation"),
+      ]);
+    }
 
     case "TSIndexSignature":
       return concat([

diff --git a/test/printer.ts b/test/printer.ts
@@ -2600,4 +2600,32 @@ describe("printer", function () {
       ),
     );
   });
+
+  it("should reprint TSTypeAnnotation correctly", function () {
+    const code = "type Foo = () => Bar;";
+
+    const ast = parse(code, {
+      parser: tsParser,
+    });
+
+    recast.visit(ast, {
+      visitTSTypeReference(path) {
+        if (
+          path.node.typeName.type === "Identifier" &&
+          path.node.typeName.name === "Bar" &&
+          path.parentPath.node.type === "TSTypeAnnotation"
+        ) {
+          path.replace(
+            b.tsQualifiedName(b.identifier("Bar"), b.identifier("Baz")),
+          );
+        }
+
+        this.traverse(path);
+      },
+    });
+
+    const pretty = new Printer().print(ast).code;
+
+    assert.strictEqual(pretty, "type Foo = () => Bar.Baz;");
+  });
 });