fix: allow providing custom query key to TanStack Query

packages/openapi-ts/src/plugins/@tanstack/query-core/infiniteQueryOptions.ts

@@ -32,6 +32,17 @@
 }) => {
   const file = context.file({ id: plugin.name })!;
 
+  if (plugin.runtimeConfigPath) {
+    file.import({
+      module: file.relativePathToFile({
+        context,
+        id: plugin.runtimeConfigPath,
+      }),
+      name: createInfiniteParamsFn,
+    });
+    return;
+  }
+
   const fn = compiler.constVariable({
     expression: compiler.arrowFunction({
       multiLine: true,

packages/openapi-ts/src/plugins/@tanstack/query-core/queryKey.ts

@@ -30,6 +30,17 @@
   });
 
   if (identifierCreateQueryKey.name) {
+    if (plugin.runtimeConfigPath) {
+      file.import({
+        module: file.relativePathToFile({
+          context,
+          id: plugin.runtimeConfigPath,
+        }),
+        name: identifierCreateQueryKey.name,
+      });
+      return;
+    }
+
     const returnType = compiler.indexedAccessTypeNode({
       indexType: compiler.literalTypeNode({
         literal: compiler.ots.number(0),
@@ -250,6 +261,18 @@
 }) => {
   const file = context.file({ id: plugin.name })!;
 
+  if (plugin.runtimeConfigPath) {
+    file.import({
+      asType: true,
+      module: file.relativePathToFile({
+        context,
+        id: plugin.runtimeConfigPath,
+      }),
+      name: queryKeyName,
+    });
+    return;
+  }
+
   const properties: Property[] = [
     {
       name: '_id',

packages/openapi-ts/src/plugins/@tanstack/query-core/types.d.ts

@@ -44,5 +44,9 @@ export namespace TanStackQuery {
      * @default false
      */
     exportFromIndex?: boolean;
+    /**
+     * TODO
+     */
+    runtimeConfigPath?: string;
   };
 }

packages/openapi-ts/test/hey-api.ts

@@ -0,0 +1,70 @@
+import { client as _heyApiClient } from './generated/sample/client.gen';
+import type { Options } from './generated/sample/sdk.gen';
+
+export type QueryKey<TOptions extends Options> = [
+  Pick<TOptions, 'baseUrl' | 'body' | 'headers' | 'path' | 'query'> & {
+    _id: string;
+    _infinite?: boolean;
+  },
+];
+
+export const createQueryKey = <TOptions extends Options>(
+  id: string,
+  options?: TOptions,
+  infinite?: boolean,
+): [QueryKey<TOptions>[0]] => {
+  const params: QueryKey<TOptions>[0] = {
+    _id: id,
+    baseUrl: (options?.client ?? _heyApiClient).getConfig().baseUrl,
+  } as QueryKey<TOptions>[0];
+  if (infinite) {
+    params._infinite = infinite;
+  }
+  if (options?.body) {
+    params.body = options.body;
+  }
+  if (options?.headers) {
+    params.headers = options.headers;
+  }
+  if (options?.path) {
+    params.path = options.path;
+  }
+  if (options?.query) {
+    params.query = options.query;
+  }
+  return [params];
+};
+
+export const createInfiniteParams = <
+  K extends Pick<QueryKey<Options>[0], 'body' | 'headers' | 'path' | 'query'>,
+>(
+  queryKey: QueryKey<Options>,
+  page: K,
+) => {
+  const params = queryKey[0];
+  if (page.body) {
+    params.body = {
+      ...(queryKey[0].body as any),
+      ...(page.body as any),
+    };
+  }
+  if (page.headers) {
+    params.headers = {
+      ...queryKey[0].headers,
+      ...page.headers,
+    };
+  }
+  if (page.path) {
+    params.path = {
+      ...(queryKey[0].path as any),
+      ...(page.path as any),
+    };
+  }
+  if (page.query) {
+    params.query = {
+      ...(queryKey[0].query as any),
+      ...(page.query as any),
+    };
+  }
+  return params as unknown as typeof page;
+};

packages/openapi-ts/test/openapi-ts.config.ts

@@ -78,6 +78,7 @@ export default defineConfig({
     // @ts-ignore
     {
       name: '@tanstack/react-query',
+      runtimeConfigPath: './hey-api.ts',
     },
     // @ts-ignore
     {