feat: rewrite response transformers to new parser

hey-api · Oct 24, 2024 · 7a41471 · 7a41471
1 parent 2834ccf
commit 7a41471
Show file tree

Hide file tree

Showing 56 changed files with 17,229 additions and 13,725 deletions.
diff --git a/packages/openapi-ts/src/compiler/classes.ts b/packages/openapi-ts/src/compiler/classes.ts
@@ -3,6 +3,7 @@ import ts from 'typescript';
 import { createCallExpression } from './module';
 import {
   type AccessLevel,
+  createBlock,
   createTypeNode,
   type FunctionParameter,
   type FunctionTypeParameter,
@@ -39,7 +40,7 @@ export const createConstructorDeclaration = ({
   const node = ts.factory.createConstructorDeclaration(
     toAccessLevelModifiers(accessLevel),
     toParameterDeclarations(parameters),
-    ts.factory.createBlock(statements, multiLine),
+    createBlock({ multiLine, statements }),
   );
 
   addLeadingComments({
@@ -100,7 +101,7 @@ export const createMethodDeclaration = ({
     types ? toTypeParameters(types) : undefined,
     toParameterDeclarations(parameters),
     returnType ? createTypeNode(returnType) : undefined,
-    ts.factory.createBlock(statements, multiLine),
+    createBlock({ multiLine, statements }),
   );
 
   addLeadingComments({

diff --git a/packages/openapi-ts/src/compiler/index.ts b/packages/openapi-ts/src/compiler/index.ts
@@ -16,8 +16,10 @@ export const compiler = {
   anonymousFunction: types.createAnonymousFunction,
   arrayLiteralExpression: types.createArrayLiteralExpression,
   arrowFunction: types.createArrowFunction,
+  assignment: types.createAssignment,
   awaitExpression: types.createAwaitExpression,
   binaryExpression: transform.createBinaryExpression,
+  block: types.createBlock,
   callExpression: module.createCallExpression,
   classDeclaration: classes.createClassDeclaration,
   conditionalExpression: types.createConditionalExpression,
@@ -27,6 +29,8 @@ export const compiler = {
   exportAllDeclaration: module.createExportAllDeclaration,
   exportNamedDeclaration: module.createNamedExportDeclarations,
   expressionToStatement: convert.expressionToStatement,
+  forOfStatement: types.createForOfStatement,
+  functionTypeNode: types.createFunctionTypeNode,
   identifier: utils.createIdentifier,
   ifStatement: transform.createIfStatement,
   indexedAccessTypeNode: types.createIndexedAccessTypeNode,
@@ -36,13 +40,16 @@ export const compiler = {
   methodDeclaration: classes.createMethodDeclaration,
   namedImportDeclarations: module.createNamedImportDeclarations,
   namespaceDeclaration: types.createNamespaceDeclaration,
+  newExpression: types.createNewExpression,
   nodeToString: utils.tsNodeToString,
   null: types.createNull,
   objectExpression: types.createObjectType,
   ots: utils.ots,
+  parameterDeclaration: types.createParameterDeclaration,
   propertyAccessExpression: types.createPropertyAccessExpression,
   propertyAccessExpressions: transform.createPropertyAccessExpressions,
   returnFunctionCall: _return.createReturnFunctionCall,
+  returnStatement: _return.createReturnStatement,
   returnVariable: _return.createReturnVariable,
   safeAccessExpression: transform.createSafeAccessExpression,
   stringLiteral: types.createStringLiteral,

diff --git a/packages/openapi-ts/src/compiler/return.ts b/packages/openapi-ts/src/compiler/return.ts
@@ -4,7 +4,7 @@ import { createCallExpression } from './module';
 import { createTypeReferenceNode } from './types';
 import { createIdentifier, isType } from './utils';
 
-const createReturnStatement = ({
+export const createReturnStatement = ({
   expression,
 }: {
   expression?: ts.Expression;

diff --git a/packages/openapi-ts/src/compiler/transform.ts b/packages/openapi-ts/src/compiler/transform.ts
@@ -5,6 +5,8 @@ import { expressionToStatement } from './convert';
 import { createCallExpression } from './module';
 import {
   createArrowFunction,
+  createBlock,
+  createNewExpression,
   createPropertyAccessChain,
   createPropertyAccessExpression,
 } from './types';
@@ -99,19 +101,20 @@ export const createDateTransformMutation = ({
   const safeAccessExpression = createSafeAccessExpression(path);
   const accessExpression = createAccessExpression(path);
 
-  const thenStatement = ts.factory.createBlock([
-    expressionToStatement({
-      expression: ts.factory.createBinaryExpression(
-        accessExpression,
-        ts.SyntaxKind.EqualsToken,
-        ts.factory.createNewExpression(
-          createIdentifier({ text: 'Date' }),
-          undefined,
-          [accessExpression],
+  const thenStatement = createBlock({
+    statements: [
+      expressionToStatement({
+        expression: ts.factory.createBinaryExpression(
+          accessExpression,
+          ts.SyntaxKind.EqualsToken,
+          createNewExpression({
+            argumentsArray: [accessExpression],
+            expression: createIdentifier({ text: 'Date' }),
+          }),
         ),
-      ),
-    }),
-  ]);
+      }),
+    ],
+  });
 
   const statement = createIfStatement({
     expression: safeAccessExpression,
@@ -131,17 +134,16 @@ export const createFunctionTransformMutation = ({
   const safeAccessExpression = createSafeAccessExpression(path);
   const accessExpression = createAccessExpression(path);
 
-  const thenStatement = ts.factory.createBlock(
-    [
+  const thenStatement = createBlock({
+    statements: [
       expressionToStatement({
         expression: createCallExpression({
           functionName: transformerName,
           parameters: [accessExpression],
         }),
       }),
     ],
-    true,
-  );
+  });
 
   const statement = [
     createIfStatement({
@@ -171,8 +173,8 @@ export const createArrayTransformMutation = ({
       }),
       parameters: [safeAccessExpression],
     }),
-    thenStatement: ts.factory.createBlock(
-      [
+    thenStatement: createBlock({
+      statements: [
         expressionToStatement({
           expression: ts.factory.createCallChain(
             createPropertyAccessExpression({
@@ -185,8 +187,7 @@ export const createArrayTransformMutation = ({
           ),
         }),
       ],
-      true,
-    ),
+    }),
   });
 
   return statement;
@@ -198,9 +199,10 @@ export const createDateTransformerExpression = ({
   parameterName: string;
 }) => {
   const expression = createIdentifier({ text: 'Date' });
-  const newExpression = ts.factory.createNewExpression(expression, undefined, [
-    createIdentifier({ text: parameterName }),
-  ]);
+  const newExpression = createNewExpression({
+    argumentsArray: [createIdentifier({ text: parameterName })],
+    expression,
+  });
   return newExpression;
 };
 
@@ -222,8 +224,8 @@ export const createArrayMapTransform = ({
       }),
       parameters: [safeAccessExpression],
     }),
-    thenStatement: ts.factory.createBlock(
-      [
+    thenStatement: createBlock({
+      statements: [
         expressionToStatement({
           expression: ts.factory.createBinaryExpression(
             accessExpression,
@@ -249,8 +251,7 @@ export const createArrayMapTransform = ({
           ),
         }),
       ],
-      true,
-    ),
+    }),
   });
 
   return statement;

diff --git a/packages/openapi-ts/src/compiler/typedef.ts b/packages/openapi-ts/src/compiler/typedef.ts
@@ -3,6 +3,7 @@ import ts from 'typescript';
 import { validTypescriptIdentifierRegExp } from '../utils/regexp';
 import {
   createKeywordTypeNode,
+  createParameterDeclaration,
   createStringLiteral,
   createTypeNode,
   createTypeReferenceNode,
@@ -109,16 +110,10 @@ export const createTypeInterfaceNode = ({
     const indexSignature = ts.factory.createIndexSignature(
       modifiers,
       [
-        ts.factory.createParameterDeclaration(
-          undefined,
-          undefined,
-          createIdentifier({ text: indexProperty.name }),
-          undefined,
-          createKeywordTypeNode({
-            keyword: 'string',
-          }),
-          undefined,
-        ),
+        createParameterDeclaration({
+          name: createIdentifier({ text: indexProperty.name }),
+          type: createKeywordTypeNode({ keyword: 'string' }),
+        }),
       ],
       createTypeNode(indexProperty.type),
     );