@kubb/plugin-react-query 3.0.0-alpha.9 → 3.0.0-beta.10

package/src/components/Query.tsx

@@ -1,623 +1,176 @@
-import { PackageManager } from '@kubb/core'
-import transformers from '@kubb/core/transformers'
-import { FunctionParams, URLPath } from '@kubb/core/utils'
-import { File, Function, useApp } from '@kubb/react'
-import { pluginTsName } from '@kubb/plugin-ts'
-import { pluginZodName } from '@kubb/plugin-zod'
-import { useOperation, useOperationManager } from '@kubb/plugin-oas/hooks'
-import { getASTParams, getComments } from '@kubb/plugin-oas/utils'
+import { File, Function, FunctionParams } from '@kubb/react'
 
-import { getImportNames, reactQueryDepRegex } from '../utils.ts'
-import { QueryImports } from './QueryImports.tsx'
+import { type Operation, isOptional } from '@kubb/oas'
+import type { OperationSchemas } from '@kubb/plugin-oas'
+import { getComments, getPathParams } from '@kubb/plugin-oas/utils'
+import type { ReactNode } from 'react'
+import type { PluginReactQuery } from '../types.ts'
 import { QueryKey } from './QueryKey.tsx'
 import { QueryOptions } from './QueryOptions.tsx'
-import { SchemaType } from './SchemaType.tsx'
-
-import { isRequired } from '@kubb/oas'
-import type { ReactNode } from 'react'
-import type { QueryOptions as QueryOptionsPluginOptions, Query as QueryPluginOptions } from '../types.ts'
-import type { FileMeta, Infinite, PluginReactQuery, Suspense } from '../types.ts'
 
-type TemplateProps = {
+type Props = {
   /**
    * Name of the function
    */
   name: string
-  /**
-   * Parameters/options/props that need to be used
-   */
-  params: string
-  /**
-   * Generics that needs to be added for TypeScript
-   */
-  generics?: string
-  /**
-   * ReturnType(see async for adding Promise type)
-   */
-  returnType?: string
-  /**
-   * Options for JSdocs
-   */
-  JSDoc?: {
-    comments: string[]
-  }
-  hook: {
-    name: string
-    generics?: string
-    queryKey: string
-    queryOptions: string
-  }
-  infinite: Infinite | false
-}
-
-function Template({ name, generics, returnType, params, JSDoc, hook, infinite }: TemplateProps): ReactNode {
-  const isV5 = new PackageManager().isValidSync(reactQueryDepRegex, '>=5')
-  const resolvedReturnType = `${returnType} & { queryKey: TQueryKey }`
-
-  if (isV5) {
-    return (
-      <File.Source name={name} isExportable isIndexable>
-        <Function name={name} export generics={generics} returnType={resolvedReturnType} params={params} JSDoc={JSDoc}>
-          {`
-         const { query: queryOptions, client: clientOptions = {} } = options ?? {}
-         const queryKey = queryOptions?.queryKey ?? ${hook.queryKey}
-
-         const query = ${hook.name}({
-          ...${hook.queryOptions} as unknown as ${infinite ? 'InfiniteQueryObserverOptions' : 'QueryObserverOptions'},
-          queryKey,
-          ...queryOptions as unknown as ${infinite ? 'Omit<InfiniteQueryObserverOptions, "queryKey">' : 'Omit<QueryObserverOptions, "queryKey">'}
-        }) as ${resolvedReturnType}
-
-        query.queryKey = queryKey as TQueryKey
-
-        return query
-
-         `}
-        </Function>
-      </File.Source>
-    )
-  }
-
-  return (
-    <File.Source name={name} isExportable isIndexable>
-      <Function name={name} export generics={generics} returnType={resolvedReturnType} params={params} JSDoc={JSDoc}>
-        {`
-       const { query: queryOptions, client: clientOptions = {} } = options ?? {}
-       const queryKey = queryOptions?.queryKey ?? ${hook.queryKey}
-
-       const query = ${hook.name}<${hook.generics}>({
-        ...${hook.queryOptions},
-        queryKey,
-        ...queryOptions
-      }) as ${resolvedReturnType}
-
-      query.queryKey = queryKey as TQueryKey
-
-      return query
-
-       `}
-      </Function>
-    </File.Source>
-  )
+  queryOptionsName: string
+  queryKeyName: string
+  queryKeyTypeName: string
+  typeSchemas: OperationSchemas
+  operation: Operation
+  paramsType: PluginReactQuery['resolvedOptions']['paramsType']
+  pathParamsType: PluginReactQuery['resolvedOptions']['pathParamsType']
+  dataReturnType: PluginReactQuery['resolvedOptions']['client']['dataReturnType']
 }
 
-type FrameworkProps = TemplateProps & {
-  context: {
-    factory: {
-      name: string
-    }
-    queryKey: string
-  }
+type GetParamsProps = {
+  paramsType: PluginReactQuery['resolvedOptions']['paramsType']
+  pathParamsType: PluginReactQuery['resolvedOptions']['pathParamsType']
+  dataReturnType: PluginReactQuery['resolvedOptions']['client']['dataReturnType']
+  typeSchemas: OperationSchemas
 }
 
-const defaultTemplates = {
-  get react() {
-    return function (props: FrameworkProps): ReactNode {
-      return <Template {...props} />
-    }
-  },
-  get solid() {
-    return function (props: FrameworkProps): ReactNode {
-      return <Template {...props} />
-    }
-  },
-  get svelte() {
-    return function (props: FrameworkProps): ReactNode {
-      return <Template {...props} />
-    }
-  },
-  get vue() {
-    return function ({ context, hook, ...rest }: FrameworkProps): ReactNode {
-      const { factory, queryKey } = context
-
-      const {
-        pluginManager,
-        plugin: {
-          key: pluginKey,
-          options: { pathParamsType },
-        },
-      } = useApp<PluginReactQuery>()
-      const operation = useOperation()
-      const { getSchemas } = useOperationManager()
-
-      const importNames = getImportNames()
-
-      const queryOptions = pluginManager.resolveName({
-        name: `${factory.name}QueryOptions`,
-        pluginKey,
-      })
-
-      const hookName = rest.infinite ? importNames.queryInfinite.vue.hookName : importNames.query.vue.hookName
-      const resultType = rest.infinite ? importNames.queryInfinite.vue.resultType : importNames.query.vue.resultType
-      const optionsType = rest.infinite ? importNames.queryInfinite.vue.optionsType : importNames.query.vue.optionsType
-
-      const schemas = getSchemas(operation, { pluginKey: [pluginTsName], type: 'type' })
-      const isV5 = new PackageManager().isValidSync(reactQueryDepRegex, '>=5')
-      const params = new FunctionParams()
-      const queryParams = new FunctionParams()
-      const queryKeyParams = new FunctionParams()
-      const client = {
-        withQueryParams: !!schemas.queryParams?.name,
-        withData: !!schemas.request?.name,
-        withPathParams: !!schemas.pathParams?.name,
-        withHeaders: !!schemas.headerParams?.name,
-      }
-
-      const resultGenerics = ['TData', `${factory.name}['error']`]
-
-      // only needed for the options to override the useQuery options/params
-      const queryOptionsOverrideGenerics = [`${factory.name}['response']`, `${factory.name}['error']`, 'TData', 'TQueryKey']
-      const queryOptionsGenerics = ['TData', 'TQueryData']
-
-      params.add([
-        ...(pathParamsType === 'object'
-          ? [
-              getASTParams(schemas.pathParams, {
-                typed: true,
-                override: (item) => ({
-                  ...item,
-                  name: item.name ? `ref${transformers.pascalCase(item.name)}` : undefined,
-                }),
-              }),
-            ]
-          : getASTParams(schemas.pathParams, {
-              typed: true,
-              override: (item) => ({
-                ...item,
-                name: item.name ? `ref${transformers.pascalCase(item.name)}` : undefined,
-              }),
-            })),
-        {
-          name: 'refParams',
-          type: `MaybeRef<${schemas.queryParams?.name}>`,
-          enabled: client.withQueryParams,
-          required: isRequired(schemas.queryParams?.schema),
-        },
-        {
-          name: 'refHeaders',
-          type: `MaybeRef<${schemas.headerParams?.name}>`,
-          enabled: client.withHeaders,
-          required: isRequired(schemas.headerParams?.schema),
-        },
-        {
-          name: 'refData',
-          type: `MaybeRef<${schemas.request?.name}>`,
-          enabled: client.withData,
-          required: isRequired(schemas.request?.schema),
+function getParams({ paramsType, pathParamsType, dataReturnType, typeSchemas }: GetParamsProps) {
+  const TData = dataReturnType === 'data' ? typeSchemas.response.name : `ResponseConfig<${typeSchemas.response.name}>`
+
+  if (paramsType === 'object') {
+    return FunctionParams.factory({
+      data: {
+        mode: 'object',
+        children: {
+          ...getPathParams(typeSchemas.pathParams, { typed: true }),
+          data: typeSchemas.request?.name
+            ? {
+                type: typeSchemas.request?.name,
+                optional: isOptional(typeSchemas.request?.schema),
+              }
+            : undefined,
+          params: typeSchemas.queryParams?.name
+            ? {
+                type: typeSchemas.queryParams?.name,
+                optional: isOptional(typeSchemas.queryParams?.schema),
+              }
+            : undefined,
+          headers: typeSchemas.headerParams?.name
+            ? {
+                type: typeSchemas.headerParams?.name,
+                optional: isOptional(typeSchemas.headerParams?.schema),
+              }
+            : undefined,
         },
-        {
-          name: 'options',
-          type: `{
-              query?: Partial<${optionsType}<${queryOptionsOverrideGenerics.join(', ')}>>,
-              client?: ${factory.name}['client']['parameters']
-          }`,
-          default: '{}',
-        },
-      ])
-
-      queryParams.add([
-        ...getASTParams(schemas.pathParams, {
-          typed: false,
-          override: (item) => ({
-            ...item,
-            name: item.name ? `ref${transformers.pascalCase(item.name)}` : undefined,
-          }),
-        }),
-        {
-          name: 'refParams',
-          enabled: client.withQueryParams,
-          required: isRequired(schemas.queryParams?.schema),
-        },
-        {
-          name: 'refHeaders',
-          enabled: client.withHeaders,
-          required: isRequired(schemas.headerParams?.schema),
-        },
-        {
-          name: 'clientOptions',
-          required: false,
-        },
-      ])
-
-      queryKeyParams.add([
-        ...(pathParamsType === 'object'
-          ? [
-              getASTParams(schemas.pathParams, {
-                override: (item) => ({
-                  ...item,
-                  name: item.name ? `ref${transformers.pascalCase(item.name)}` : undefined,
-                }),
-              }),
-            ]
-          : getASTParams(schemas.pathParams, {
-              override: (item) => ({
-                ...item,
-                name: item.name ? `ref${transformers.pascalCase(item.name)}` : undefined,
-              }),
-            })),
-        {
-          name: 'refParams',
-          enabled: client.withQueryParams,
-          required: isRequired(schemas.queryParams?.schema),
-        },
-        {
-          name: 'refData',
-          enabled: client.withData,
-          required: isRequired(schemas.request?.schema),
-        },
-      ])
-
-      return (
-        <Template
-          {...rest}
-          params={params.toString()}
-          returnType={`${resultType}<${resultGenerics.join(', ')}>`}
-          hook={{
-            ...hook,
-            name: hookName,
-            queryOptions: isV5
-              ? `${queryOptions}(${queryParams.toString()})`
-              : `${queryOptions}<${queryOptionsGenerics.join(', ')}>(${queryParams.toString()})`,
-            queryKey: `${queryKey}(${queryKeyParams.toString()})`,
-          }}
-        />
-      )
-    }
-  },
-} as const
-
-type Props = {
-  factory: {
-    name: string
+      },
+      options: {
+        type: `
+{
+  query?: Partial<QueryObserverOptions<${[TData, typeSchemas.errors?.map((item) => item.name).join(' | ') || 'Error', 'TData', 'TQueryData', 'TQueryKey'].join(', ')}>>,
+  client?: ${typeSchemas.request?.name ? `Partial<RequestConfig<${typeSchemas.request?.name}>>` : 'Partial<RequestConfig>'}
+}
+`,
+        default: '{}',
+      },
+    })
   }
-  resultType: string
-  hookName: string
-  optionsType: string
-  infinite: Infinite | false
-  query: QueryPluginOptions | false
-  queryOptions: QueryOptionsPluginOptions | false
-  suspense: Suspense | false
-  /**
-   * This will make it possible to override the default behaviour.
-   */
-  Template?: React.ComponentType<FrameworkProps>
-  /**
-   * This will make it possible to override the default behaviour.
-   */
-  QueryKeyTemplate?: React.ComponentType<React.ComponentProps<typeof QueryKey.templates.react>>
-  /**
-   * This will make it possible to override the default behaviour.
-   */
-  QueryOptionsTemplate?: React.ComponentType<React.ComponentProps<typeof QueryOptions.templates.react>>
+
+  return FunctionParams.factory({
+    pathParams: typeSchemas.pathParams?.name
+      ? {
+          mode: pathParamsType === 'object' ? 'object' : 'inlineSpread',
+          children: getPathParams(typeSchemas.pathParams, { typed: true }),
+          type: typeSchemas.pathParams?.name,
+          optional: isOptional(typeSchemas.pathParams?.schema),
+        }
+      : undefined,
+    data: typeSchemas.request?.name
+      ? {
+          type: typeSchemas.request?.name,
+          optional: isOptional(typeSchemas.request?.schema),
+        }
+      : undefined,
+    params: typeSchemas.queryParams?.name
+      ? {
+          type: typeSchemas.queryParams?.name,
+          optional: isOptional(typeSchemas.queryParams?.schema),
+        }
+      : undefined,
+    headers: typeSchemas.headerParams?.name
+      ? {
+          type: typeSchemas.headerParams?.name,
+          optional: isOptional(typeSchemas.headerParams?.schema),
+        }
+      : undefined,
+    options: {
+      type: `
+{
+  query?: Partial<QueryObserverOptions<${[TData, typeSchemas.errors?.map((item) => item.name).join(' | ') || 'Error', 'TData', 'TQueryData', 'TQueryKey'].join(', ')}>>,
+  client?: ${typeSchemas.request?.name ? `Partial<RequestConfig<${typeSchemas.request?.name}>>` : 'Partial<RequestConfig>'}
+}
+`,
+      default: '{}',
+    },
+  })
 }
 
 export function Query({
-  factory,
-  optionsType,
-  hookName,
-  resultType,
-  Template = defaultTemplates.react,
-  QueryKeyTemplate = QueryKey.templates.react,
-  QueryOptionsTemplate = QueryOptions.templates.react,
-  ...props
+  name,
+  queryKeyTypeName,
+  queryOptionsName,
+  queryKeyName,
+  paramsType,
+  pathParamsType,
+  dataReturnType,
+  typeSchemas,
+  operation,
 }: Props): ReactNode {
-  const {
-    pluginManager,
-    plugin: {
-      key: pluginKey,
-      options: { dataReturnType, pathParamsType },
-    },
-  } = useApp<PluginReactQuery>()
+  const TData = dataReturnType === 'data' ? typeSchemas.response.name : `ResponseConfig<${typeSchemas.response.name}>`
+  const returnType = `UseQueryResult<${['TData', typeSchemas.errors?.map((item) => item.name).join(' | ') || 'Error'].join(', ')}> & { queryKey: TQueryKey }`
+  const generics = [`TData = ${TData}`, `TQueryData = ${TData}`, `TQueryKey extends QueryKey = ${queryKeyTypeName}`]
 
-  const operation = useOperation()
-  const { getSchemas, getName } = useOperationManager()
-
-  const schemas = getSchemas(operation, { pluginKey: [pluginTsName], type: 'type' })
-  const name = getName(operation, { type: 'function' })
-  const isV5 = new PackageManager().isValidSync(reactQueryDepRegex, '>=5')
-
-  const queryKey = pluginManager.resolveName({
-    name: [factory.name, props.infinite ? 'Infinite' : undefined, props.suspense ? 'Suspense' : undefined, 'QueryKey'].filter(Boolean).join(''),
-    pluginKey,
+  const queryKeyParams = QueryKey.getParams({
+    pathParamsType,
+    typeSchemas,
   })
-  const queryKeyType = pluginManager.resolveName({
-    name: [factory.name, props.infinite ? 'Infinite' : undefined, props.suspense ? 'Suspense' : undefined, 'QueryKey'].filter(Boolean).join(''),
-    type: 'type',
-    pluginKey,
+  const queryOptionsParams = QueryOptions.getParams({
+    paramsType,
+    pathParamsType,
+    typeSchemas,
   })
-  const queryOptions = pluginManager.resolveName({
-    name: [factory.name, props.infinite ? 'Infinite' : undefined, props.suspense ? 'Suspense' : undefined, 'QueryOptions'].filter(Boolean).join(''),
-    pluginKey,
+  const params = getParams({
+    paramsType,
+    pathParamsType,
+    dataReturnType,
+    typeSchemas,
   })
 
-  const generics = new FunctionParams()
-  const params = new FunctionParams()
-  const queryParams = new FunctionParams()
-  const queryKeyParams = new FunctionParams()
-  //TODO operationManager.getCleitn
-  const client = {
-    method: operation.method,
-    path: new URLPath(operation.path),
-    withQueryParams: !!schemas.queryParams?.name,
-    withData: !!schemas.request?.name,
-    withPathParams: !!schemas.pathParams?.name,
-    withHeaders: !!schemas.headerParams?.name,
-  }
-
-  generics.add([
-    {
-      type: 'TData',
-      default: props.infinite ? `InfiniteData<${factory.name}["response"]>` : `${factory.name}["response"]`,
-    },
-    props.suspense ? undefined : { type: 'TQueryData', default: `${factory.name}["response"]` },
-    { type: 'TQueryKey extends QueryKey', default: queryKeyType },
-  ])
-
-  const resultGenerics = ['TData', `${factory.name}['error']`]
-  // only needed for the options to override the useQuery options/params
-  // suspense is having 4 generics instead of 5, TQueryData is not needed because data will always be defined
-  const queryOptionsOverrideGenerics = props.suspense
-    ? [`${factory.name}['response']`, `${factory.name}['error']`, 'TData', 'TQueryKey']
-    : [`${factory.name}['response']`, `${factory.name}['error']`, 'TData', 'TQueryData', 'TQueryKey']
-
-  const queryOptionsGenerics = props.suspense ? ['TData'] : ['TData', 'TQueryData']
-
-  params.add([
-    ...(pathParamsType === 'object' ? [getASTParams(schemas.pathParams, { typed: true })] : getASTParams(schemas.pathParams, { typed: true })),
-    {
-      name: 'params',
-      type: `${factory.name}['queryParams']`,
-      enabled: client.withQueryParams,
-      required: isRequired(schemas.queryParams?.schema),
-    },
-    {
-      name: 'headers',
-      type: `${factory.name}['headerParams']`,
-      enabled: client.withHeaders,
-      required: isRequired(schemas.headerParams?.schema),
-    },
-    {
-      name: 'data',
-      type: `${factory.name}['request']`,
-      enabled: client.withData,
-      required: isRequired(schemas.request?.schema),
-    },
-    {
-      name: 'options',
-      type: `{
-    query?: Partial<${optionsType}<${queryOptionsOverrideGenerics.join(', ')}>>,
-    client?: ${factory.name}['client']['parameters']
-}`,
-      default: '{}',
-    },
-  ])
-
-  queryParams.add([
-    ...(pathParamsType === 'object' ? [getASTParams(schemas.pathParams)] : getASTParams(schemas.pathParams)),
-    {
-      name: 'params',
-      enabled: client.withQueryParams,
-      required: isRequired(schemas.queryParams?.schema),
-    },
-    {
-      name: 'headers',
-      enabled: client.withHeaders,
-      required: isRequired(schemas.headerParams?.schema),
-    },
-    {
-      name: 'data',
-      enabled: client.withData,
-      required: isRequired(schemas.request?.schema),
-    },
-    {
-      name: 'clientOptions',
-      required: false,
-    },
-  ])
-
-  queryKeyParams.add([
-    ...(pathParamsType === 'object' ? [getASTParams(schemas.pathParams)] : getASTParams(schemas.pathParams)),
-    {
-      name: 'params',
-      enabled: client.withQueryParams,
-      required: isRequired(schemas.queryParams?.schema),
-    },
-    {
-      name: 'data',
-      enabled: client.withData,
-      required: isRequired(schemas.request?.schema),
-    },
-  ])
-
-  const hook = {
-    name: hookName,
-    generics: [isV5 ? 'any' : `${factory.name}['data']`, `${factory.name}['error']`, 'TData', 'any'].join(', '),
-    queryOptions: isV5 ? `${queryOptions}(${queryParams.toString()})` : `${queryOptions}<${queryOptionsGenerics.join(', ')}>(${queryParams.toString()})`,
-    queryKey: `${queryKey}(${queryKeyParams.toString()})`,
-  }
+  const queryOptions = `${queryOptionsName}(${queryOptionsParams.toCall()}) as unknown as QueryObserverOptions`
 
   return (
-    <>
-      <QueryKey
-        keysFn={props.query ? props.query.queryKey : (keys: unknown[]) => keys}
-        Template={QueryKeyTemplate}
-        factory={factory}
-        name={queryKey}
-        typeName={queryKeyType}
-      />
-
-      {props.queryOptions && (
-        <QueryOptions
-          Template={QueryOptionsTemplate}
-          factory={factory}
-          resultType={optionsType}
-          dataReturnType={dataReturnType}
-          infinite={props.infinite}
-          suspense={props.suspense}
-        />
-      )}
-
-      {props.query && (
-        <Template
-          name={[name, props.infinite ? 'Infinite' : undefined, props.suspense ? 'Suspense' : undefined].filter(Boolean).join('')}
-          generics={generics.toString()}
-          JSDoc={{ comments: getComments(operation) }}
-          params={params.toString()}
-          returnType={`${resultType}<${resultGenerics.join(', ')}>`}
-          hook={hook}
-          infinite={props.infinite}
-          context={{
-            factory,
-            queryKey,
-          }}
-        />
-      )}
-    </>
-  )
-}
-
-type FileProps = {
-  /**
-   * This will make it possible to override the default behaviour.
-   */
-  templates?: {
-    query: typeof defaultTemplates
-    queryKey: typeof QueryKey.templates
-    queryOptions: typeof QueryOptions.templates
-    queryImports: typeof QueryImports.templates
-  }
-}
-
-Query.File = function ({ templates }: FileProps): ReactNode {
-  const {
-    plugin: {
-      options: {
-        client: { importPath },
-        infinite,
-        suspense,
-        query,
-        queryOptions,
-        parser,
-      },
-    },
-  } = useApp<PluginReactQuery>()
-
-  const { getSchemas, getFile, getName } = useOperationManager()
-  const operation = useOperation()
-
-  const schemas = getSchemas(operation, { pluginKey: [pluginTsName], type: 'type' })
-  const zodSchemas = getSchemas(operation, { pluginKey: [pluginZodName], type: 'function' })
-  const file = getFile(operation)
-  const fileType = getFile(operation, { pluginKey: [pluginTsName] })
-  const fileZodSchemas = getFile(operation, {
-    pluginKey: [pluginZodName],
-  })
-
-  const factoryName = getName(operation, { type: 'type' })
-
-  const importNames = getImportNames()
-  const isV5 = new PackageManager().isValidSync(reactQueryDepRegex, '>=5')
-  const Template = templates?.query['react'] || defaultTemplates['react']
-  const QueryOptionsTemplate = templates?.queryOptions['react'] || QueryOptions.templates['react']
-  const QueryKeyTemplate = templates?.queryKey['react'] || QueryKey.templates['react']
-  const Import = templates?.queryImports['react'] || QueryImports.templates['react']
+    <File.Source name={name} isExportable isIndexable>
+      <Function
+        name={name}
+        export
+        generics={generics.join(', ')}
+        params={params.toConstructor()}
+        JSDoc={{
+          comments: getComments(operation),
+        }}
+      >
+        {`
+       const { query: queryOptions, client: config = {} } = options ?? {}
+       const queryKey = queryOptions?.queryKey ?? ${queryKeyName}(${queryKeyParams.toCall()})
 
-  const factory = {
-    name: factoryName,
-  }
+       const query = useQuery({
+        ...${queryOptions},
+        queryKey,
+        ...queryOptions as unknown as Omit<QueryObserverOptions, "queryKey">
+       }) as ${returnType}
 
-  return (
-    <File<FileMeta> baseName={file.baseName} path={file.path} meta={file.meta}>
-      {parser === 'zod' && <File.Import name={[zodSchemas.response?.name]} root={file.path} path={fileZodSchemas.path} />}
-      <File.Import name={'client'} path={importPath} />
-      <File.Import name={['ResponseConfig']} path={importPath} isTypeOnly />
-      <File.Import
-        name={[
-          schemas.request?.name,
-          schemas.response.name,
-          schemas.pathParams?.name,
-          schemas.queryParams?.name,
-          schemas.headerParams?.name,
-          ...(schemas.errors?.map((error) => error.name) || []),
-        ].filter(Boolean)}
-        root={file.path}
-        path={fileType.path}
-        isTypeOnly
-      />
+       query.queryKey = queryKey as TQueryKey
 
-      <QueryImports hookPath={typeof query !== 'boolean' ? query.importPath : undefined} Template={Import} isInfinite={false} isSuspense={false} />
-      {!!infinite && (
-        <QueryImports hookPath={typeof query !== 'boolean' ? query.importPath : undefined} Template={Import} isInfinite={true} isSuspense={false} />
-      )}
-      {!!suspense && isV5 && (
-        <QueryImports hookPath={typeof query !== 'boolean' ? query.importPath : undefined} Template={Import} isInfinite={false} isSuspense={true} />
-      )}
-      <SchemaType />
-      <Query
-        factory={factory}
-        Template={Template}
-        QueryKeyTemplate={QueryKeyTemplate}
-        QueryOptionsTemplate={QueryOptionsTemplate}
-        infinite={false}
-        suspense={false}
-        query={query}
-        queryOptions={queryOptions}
-        hookName={importNames.query['react'].hookName}
-        resultType={importNames.query['react'].resultType}
-        optionsType={importNames.query['react'].optionsType}
-      />
-      {!!infinite && (
-        <Query
-          factory={factory}
-          Template={Template}
-          QueryKeyTemplate={QueryKeyTemplate}
-          QueryOptionsTemplate={QueryOptionsTemplate}
-          infinite={infinite}
-          suspense={false}
-          query={query}
-          queryOptions={queryOptions}
-          hookName={importNames.queryInfinite['react'].hookName}
-          resultType={importNames.queryInfinite['react'].resultType}
-          optionsType={importNames.queryInfinite['react'].optionsType}
-        />
-      )}
-      {!!suspense && isV5 && (
-        <Query
-          factory={factory}
-          Template={Template}
-          QueryKeyTemplate={QueryKeyTemplate}
-          QueryOptionsTemplate={QueryOptionsTemplate}
-          infinite={false}
-          suspense={suspense}
-          query={query}
-          queryOptions={queryOptions}
-          hookName={importNames.querySuspense['react'].hookName}
-          resultType={importNames.querySuspense['react'].resultType}
-          optionsType={importNames.querySuspense['react'].optionsType}
-        />
-      )}
-    </File>
+       return query
+       `}
+      </Function>
+    </File.Source>
   )
 }
 
-Query.templates = defaultTemplates
+Query.getParams = getParams