@kubb/plugin-vue-query 0.0.0-canary-20241104172400

package/src/plugin.ts

@@ -0,0 +1,149 @@
+import path from 'node:path'
+
+import { FileManager, type Group, PluginManager, createPlugin } from '@kubb/core'
+import { camelCase, pascalCase } from '@kubb/core/transformers'
+import { OperationGenerator, pluginOasName } from '@kubb/plugin-oas'
+
+import { pluginTsName } from '@kubb/plugin-ts'
+import { pluginZodName } from '@kubb/plugin-zod'
+
+import type { Plugin } from '@kubb/core'
+import type { PluginOas } from '@kubb/plugin-oas'
+import { MutationKey, QueryKey } from './components'
+import { infiniteQueryGenerator, mutationGenerator, queryGenerator } from './generators'
+import type { PluginVueQuery } from './types.ts'
+
+export const pluginVueQueryName = 'plugin-vue-query' satisfies PluginVueQuery['name']
+
+export const pluginVueQuery = createPlugin<PluginVueQuery>((options) => {
+  const {
+    output = { path: 'hooks', barrelType: 'named' },
+    group,
+    exclude = [],
+    include,
+    override = [],
+    parser = 'client',
+    infinite,
+    transformers = {},
+    paramsType = 'inline',
+    pathParamsType = 'inline',
+    mutation = {},
+    query = {},
+    mutationKey = MutationKey.getTransformer,
+    queryKey = QueryKey.getTransformer,
+    generators = [queryGenerator, infiniteQueryGenerator, mutationGenerator].filter(Boolean),
+  } = options
+
+  return {
+    name: pluginVueQueryName,
+    options: {
+      output,
+      client: {
+        importPath: '@kubb/plugin-client/client',
+        dataReturnType: 'data',
+        pathParamsType: 'inline',
+        ...options.client,
+      },
+      infinite: infinite
+        ? {
+            queryParam: 'id',
+            initialPageParam: 0,
+            cursorParam: undefined,
+            ...infinite,
+          }
+        : false,
+      queryKey,
+      query: {
+        methods: ['get'],
+        importPath: '@tanstack/vue-query',
+        ...query,
+      },
+      mutationKey,
+      mutation: {
+        methods: ['post', 'put', 'patch', 'delete'],
+        importPath: '@tanstack/vue-query',
+        ...mutation,
+      },
+      paramsType,
+      pathParamsType: paramsType === 'object' ? 'object' : pathParamsType,
+      parser,
+    },
+    pre: [pluginOasName, pluginTsName, parser === 'zod' ? pluginZodName : undefined].filter(Boolean),
+    resolvePath(baseName, pathMode, options) {
+      const root = path.resolve(this.config.root, this.config.output.path)
+      const mode = pathMode ?? FileManager.getMode(path.resolve(root, output.path))
+
+      if (options?.tag && group?.type === 'tag') {
+        const groupName: Group['name'] = group?.name ? group.name : (ctx) => `${ctx.group}Controller`
+
+        return path.resolve(root, output.path, groupName({ group: camelCase(options.tag) }), baseName)
+      }
+
+      if (mode === 'single') {
+        /**
+         * when output is a file then we will always append to the same file(output file), see fileManager.addOrAppend
+         * Other plugins then need to call addOrAppend instead of just add from the fileManager class
+         */
+        return path.resolve(root, output.path)
+      }
+
+      return path.resolve(root, output.path, baseName)
+    },
+    resolveName(name, type) {
+      let resolvedName = camelCase(name)
+
+      if (type === 'file' || type === 'function') {
+        resolvedName = camelCase(name, {
+          isFile: type === 'file',
+        })
+      }
+      if (type === 'type') {
+        resolvedName = pascalCase(name)
+      }
+
+      if (type) {
+        return transformers?.name?.(resolvedName, type) || resolvedName
+      }
+
+      return resolvedName
+    },
+    async buildStart() {
+      const [swaggerPlugin]: [Plugin<PluginOas>] = PluginManager.getDependedPlugins<PluginOas>(this.plugins, [pluginOasName])
+
+      const oas = await swaggerPlugin.context.getOas()
+      const root = path.resolve(this.config.root, this.config.output.path)
+      const mode = FileManager.getMode(path.resolve(root, output.path))
+      const baseURL = await swaggerPlugin.context.getBaseURL()
+
+      if (baseURL) {
+        this.plugin.options.client.baseURL = baseURL
+      }
+      const operationGenerator = new OperationGenerator(this.plugin.options, {
+        oas,
+        pluginManager: this.pluginManager,
+        plugin: this.plugin,
+        contentType: swaggerPlugin.context.contentType,
+        exclude,
+        include,
+        override,
+        mode,
+      })
+
+      const files = await operationGenerator.build(...generators)
+      await this.addFile(...files)
+
+      const barrelFiles = await this.fileManager.getBarrelFiles({
+        type: output.barrelType ?? 'named',
+        root,
+        output,
+        files: this.fileManager.files,
+        meta: {
+          pluginKey: this.plugin.key,
+        },
+        logger: this.logger,
+      })
+
+      await this.addFile(...barrelFiles)
+    },
+  }
+})

package/src/types.ts

@@ -0,0 +1,159 @@
+import type { Group, Output, PluginFactoryOptions, ResolveNameParams } from '@kubb/core'
+
+import type { HttpMethod, Operation } from '@kubb/oas'
+import type { PluginClient } from '@kubb/plugin-client'
+import type { Exclude, Generator, Include, OperationSchemas, Override, ResolvePathOptions } from '@kubb/plugin-oas'
+import type { PluginReactQuery } from '@kubb/plugin-react-query'
+
+type TransformerProps = {
+  operation: Operation
+  schemas: OperationSchemas
+}
+
+export type Transformer = (props: TransformerProps) => unknown[]
+
+/**
+ * Customize the queryKey
+ */
+type QueryKey = Transformer
+
+/**
+ * Customize the mutationKey
+ */
+type MutationKey = Transformer
+
+type Query = {
+  /**
+   * Define which HttpMethods can be used for queries
+   * @default ['get']
+   */
+  methods: Array<HttpMethod>
+  /**
+   * Path to the useQuery that will be used to do the useQuery functionality.
+   * It will be used as `import { useQuery } from '${importPath}'`.
+   * It allows both relative and absolute path.
+   * the path will be applied as is, so relative path should be based on the file being generated.
+   * @default '@tanstack/react-query'
+   */
+  importPath?: string
+}
+
+type Mutation = {
+  /**
+   * Define which HttpMethods can be used for mutations
+   * @default ['post', 'put', 'delete']
+   */
+  methods: Array<HttpMethod>
+  /**
+   * Path to the useQuery that will be used to do the useQuery functionality.
+   * It will be used as `import { useQuery } from '${importPath}'`.
+   * It allows both relative and absolute path.
+   * the path will be applied as is, so relative path should be based on the file being generated.
+   * @default '@tanstack/react-query'
+   */
+  importPath?: string
+}
+
+export type Infinite = {
+  /**
+   * Specify the params key used for `pageParam`.
+   * @default `'id'`
+   */
+  queryParam: string
+  /**
+   * Which field of the data will be used, set it to undefined when no cursor is known.
+   */
+  cursorParam?: string | undefined
+  /**
+   * The initial value, the value of the first page.
+   * @default `0`
+   */
+  initialPageParam: unknown
+}
+
+export type Options = {
+  /**
+   * Specify the export location for the files and define the behavior of the output
+   * @default { path: 'hooks', barrelType: 'named' }
+   */
+  output?: Output
+  /**
+   * Group the @tanstack/query hooks based on the provided name.
+   */
+  group?: Group
+  client?: Pick<PluginClient['options'], 'dataReturnType' | 'importPath' | 'baseURL'>
+
+  /**
+   * Array containing exclude parameters to exclude/skip tags/operations/methods/paths.
+   */
+  exclude?: Array<Exclude>
+  /**
+   * Array containing include parameters to include tags/operations/methods/paths.
+   */
+  include?: Array<Include>
+  /**
+   * Array containing override parameters to override `options` based on tags/operations/methods/paths.
+   */
+  override?: Array<Override<ResolvedOptions>>
+  /**
+   * How to pass your params
+   * - 'object' will return the params and pathParams as an object.
+   * - 'inline' will return the params as comma separated params.
+   * @default 'inline'
+   */
+  paramsType?: 'object' | 'inline'
+  /**
+   * How to pass your pathParams.
+   * - 'object' will return the pathParams as an object.
+   * - 'inline' will return the pathParams as comma separated params.
+   * @default 'inline'
+   */
+  pathParamsType?: PluginClient['options']['pathParamsType']
+  /**
+   * When set, an infiniteQuery hooks will be added.
+   */
+  infinite?: Partial<Infinite> | false
+  queryKey?: QueryKey
+  /**
+   * Override some useQuery behaviours.
+   */
+  query?: Partial<Query> | false
+  mutationKey?: MutationKey
+  /**
+   * Override some useMutation behaviours.
+   */
+  mutation?: Mutation | false
+  /**
+   * Which parser should be used before returning the data to `@tanstack/query`.
+   * `'zod'` will use `@kubb/plugin-zod` to parse the data.
+   */
+  parser?: PluginClient['options']['parser']
+  transformers?: {
+    /**
+     * Customize the names based on the type that is provided by the plugin.
+     */
+    name?: (name: ResolveNameParams['name'], type?: ResolveNameParams['type']) => string
+  }
+  /**
+   * Define some generators next to the vue-query generators
+   */
+  generators?: Array<Generator<PluginVueQuery>>
+}
+
+type ResolvedOptions = {
+  output: Output
+  client: Required<Omit<NonNullable<PluginReactQuery['options']['client']>, 'baseURL'>> & { baseURL?: string }
+  parser: Required<NonNullable<Options['parser']>>
+  paramsType: NonNullable<Options['paramsType']>
+  pathParamsType: NonNullable<Options['pathParamsType']>
+  /**
+   * Only used of infinite
+   */
+  infinite: NonNullable<Infinite> | false
+  queryKey: QueryKey | undefined
+  query: NonNullable<Required<Query>> | false
+  mutationKey: MutationKey | undefined
+  mutation: NonNullable<Required<Mutation>> | false
+}
+
+export type PluginVueQuery = PluginFactoryOptions<'plugin-vue-query', Options, ResolvedOptions, never, ResolvePathOptions>