Merge pull request #3610 from hey-api/fix/transformers-plugin-arg

fix: expose plugin in transformers

Author: mrlubos

.changeset/two-cameras-check.md

@@ -0,0 +1,5 @@
+---
+"@hey-api/openapi-ts": patch
+---
+
+**plugin(@hey-api/transformers)**: expose plugin in context

packages/openapi-ts/src/plugins/@hey-api/transformers/expressions.ts

@@ -2,15 +2,13 @@ import type { IR } from '@hey-api/shared';
 import type ts from 'typescript';
 
 import { $ } from '../../../ts-dsl';
-import type { UserConfig } from './types';
+import type { HeyApiTransformersPlugin, UserConfig } from './types';
 
-export type ExpressionTransformer = ({
-  config,
-  dataExpression,
-  schema,
-}: {
+export type ExpressionTransformer = (ctx: {
+  /** @deprecated Use `plugin` instead and access the config via `plugin.config` */
   config: Omit<UserConfig, 'name'>;
-  dataExpression?: ts.Expression | ReturnType<typeof $.expr | typeof $.attr> | string;
+  dataExpression?: ts.Expression | ReturnType<typeof $.attr | typeof $.expr> | string;
+  plugin: HeyApiTransformersPlugin['Instance'];
   schema: IR.SchemaObject;
 }) => Array<ReturnType<typeof $.fromValue>> | undefined;
 

packages/openapi-ts/src/plugins/@hey-api/transformers/plugin.ts

@@ -14,15 +14,17 @@ const buildingSymbols = new Set<number>();
 
 type Expr = ReturnType<typeof $.fromValue | typeof $.return | typeof $.if>;
 
-const isNodeReturnStatement = (node: Expr) => node['~dsl'] === 'ReturnTsDsl';
+function isNodeReturnStatement(node: Expr) {
+  return node['~dsl'] === 'ReturnTsDsl';
+}
 
-const schemaResponseTransformerNodes = ({
+function schemaResponseTransformerNodes({
   plugin,
   schema,
 }: {
   plugin: HeyApiTransformersPlugin['Instance'];
   schema: IR.SchemaObject;
-}): Array<ts.Expression | ts.Statement | Expr> => {
+}): Array<ts.Expression | ts.Statement | Expr> {
   const nodes = processSchemaType({
     dataExpression: $(dataVariableName),
     plugin,
@@ -36,17 +38,17 @@ const schemaResponseTransformerNodes = ({
     }
   }
   return nodes;
-};
+}
 
-const processSchemaType = ({
+function processSchemaType({
   dataExpression,
   plugin,
   schema,
 }: {
   dataExpression?: ts.Expression | string | ReturnType<typeof $.attr | typeof $.expr>;
   plugin: HeyApiTransformersPlugin['Instance'];
   schema: IR.SchemaObject;
-}): Array<Expr> => {
+}): Array<Expr> {
   if (schema.$ref) {
     const query: SymbolMeta = {
       category: 'transform',
@@ -265,6 +267,7 @@ const processSchemaType = ({
     const t = transformer({
       config: plugin.config,
       dataExpression,
+      plugin,
       schema,
     });
     if (t) {
@@ -273,7 +276,7 @@ const processSchemaType = ({
   }
 
   return [];
-};
+}
 
 // handles only response transformers for now
 export const handler: HeyApiTransformersPlugin['Handler'] = ({ plugin }) => {