Add support in babel-types babel-generator

babel · Apr 7, 2021 · bc141f2 · bc141f2
1 parent 69a1f2b
commit bc141f2
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 0 deletions.
diff --git a/packages/babel-generator/src/generators/typescript.ts b/packages/babel-generator/src/generators/typescript.ts
@@ -127,6 +127,11 @@ export function tsPrintPropertyOrMethodName(this: Printer, node) {
 }
 
 export function TSMethodSignature(this: Printer, node: t.TSMethodSignature) {
+  const { kind } = node;
+  if (kind === "set" || kind === "get") {
+    this.word(kind);
+    this.space();
+  }
   this.tsPrintPropertyOrMethodName(node);
   this.tsPrintSignatureDeclarationBase(node);
   this.token(";");

diff --git a/packages/babel-generator/test/fixtures/typescript/interface-method-kind/input.js b/packages/babel-generator/test/fixtures/typescript/interface-method-kind/input.js
@@ -0,0 +1,4 @@
+interface Foo {
+  get foo();
+  set bar(v);
+}
diff --git a/packages/babel-generator/test/fixtures/typescript/interface-method-kind/output.js b/packages/babel-generator/test/fixtures/typescript/interface-method-kind/output.js
@@ -0,0 +1,4 @@
+interface Foo {
+  get foo();
+  set bar(v);
+}
diff --git a/packages/babel-types/src/ast-types/generated/index.ts b/packages/babel-types/src/ast-types/generated/index.ts
@@ -1701,6 +1701,7 @@ export interface TSMethodSignature extends BaseNode {
   parameters: Array<Identifier | RestElement>;
   typeAnnotation?: TSTypeAnnotation | null;
   computed?: boolean | null;
+  kind: "method" | "get" | "set";
   optional?: boolean | null;
 }
 

diff --git a/packages/babel-types/src/definitions/typescript.ts b/packages/babel-types/src/definitions/typescript.ts
@@ -118,6 +118,9 @@ defineType("TSMethodSignature", {
   fields: {
     ...signatureDeclarationCommon,
     ...namedTypeElementCommon,
+    kind: {
+      validate: assertOneOf("method", "get", "set"),
+    },
   },
 });