@acrool/rtk-query-codegen-openapi 1.4.0-alpha.1 → 1.4.0-alpha.3

package/lib/index.js

@@ -489,12 +489,59 @@ export interface IRequestConfig {
   timeout?: number;
 }
 
-export type IRestFulEndpointsQueryReturn<TVariables> = TVariables extends void ? 
-  void | {fetchOptions?: IRequestConfig;}: 
+export type IRestFulEndpointsQueryReturn<TVariables> = TVariables extends void ?
+  void | {fetchOptions?: IRequestConfig;}:
   {
       variables: TVariables;
       fetchOptions?: IRequestConfig;
   };
+
+export interface RtkQueryBaseOptions {
+  skip?: boolean;
+  pollingInterval?: number;
+  refetchOnMountOrArgChange?: boolean | number;
+  refetchOnFocus?: boolean;
+  refetchOnReconnect?: boolean;
+}
+
+export interface RtkQueryResult<TData> {
+  data: TData | undefined;
+  currentData: TData | undefined;
+  isLoading: boolean;
+  isFetching: boolean;
+  isSuccess: boolean;
+  isError: boolean;
+  isUninitialized: boolean;
+  error: unknown;
+  refetch: () => void;
+}
+
+export type SimpleQueryHook<TData, TArg> = (
+  arg: TArg,
+  options?: RtkQueryBaseOptions
+) => RtkQueryResult<TData>;
+
+export type SimpleVoidQueryHook<TData> = (
+  options?: RtkQueryBaseOptions
+) => RtkQueryResult<TData>;
+
+export type SimpleLazyQueryHook<TData, TArg> = () => readonly [
+  (arg: TArg) => void,
+  RtkQueryResult<TData>
+];
+
+export type UseSimpleMutation<TRes, TArg = void> = () => readonly [
+  (arg: TArg) => Promise<TRes>,
+  {
+    data: TRes | undefined;
+    isLoading: boolean;
+    isSuccess: boolean;
+    isError: boolean;
+    isUninitialized: boolean;
+    error: unknown;
+    reset: () => void;
+  }
+];
 `;
 }
 
@@ -1107,10 +1154,42 @@ ${paramsLines}
   const hasTags = endpointInfos.some((info) => info.tags && info.tags.length > 0);
   const tagTypesImport = hasTags ? `import {ECacheTagTypes} from "../tagTypes";
 ` : "";
+  const hasVoidQuery = endpointInfos.some((info) => info.isQuery && info.isVoidArg);
+  const hasArgQuery = endpointInfos.some((info) => info.isQuery && !info.isVoidArg);
+  const hasLazyQuery = endpointInfos.some((info) => info.isQuery) && !!options.useLazyQueries;
+  const hasMutation = endpointInfos.some((info) => !info.isQuery);
+  const simpleTypeImports = [];
+  if (hasArgQuery) simpleTypeImports.push("SimpleQueryHook");
+  if (hasVoidQuery) simpleTypeImports.push("SimpleVoidQueryHook");
+  if (hasLazyQuery) simpleTypeImports.push("SimpleLazyQueryHook");
+  if (hasMutation) simpleTypeImports.push("UseSimpleMutation");
+  const simpleTypeImportStatement = simpleTypeImports.length > 0 ? `import type { ${simpleTypeImports.join(", ")} } from "../common-types";
+` : "";
+  const hookExports = endpointInfos.map((info) => {
+    const capitalizedOperationName = info.operationName.charAt(0).toUpperCase() + info.operationName.slice(1);
+    const argType = info.isVoidArg ? "void" : `${httpClientTypeName}<${info.argTypeName}>`;
+    const lines = [];
+    if (info.isQuery) {
+      const regularHook = `use${capitalizedOperationName}Query`;
+      if (info.isVoidArg) {
+        lines.push(`export const ${regularHook} = injectedRtkApi.${regularHook} as SimpleVoidQueryHook<${info.responseTypeName}>;`);
+      } else {
+        lines.push(`export const ${regularHook} = injectedRtkApi.${regularHook} as SimpleQueryHook<${info.responseTypeName}, ${argType}>;`);
+      }
+      if (options.useLazyQueries) {
+        const lazyHook = `useLazy${capitalizedOperationName}Query`;
+        lines.push(`export const ${lazyHook} = injectedRtkApi.${lazyHook} as unknown as SimpleLazyQueryHook<${info.responseTypeName}, ${argType}>;`);
+      }
+    } else {
+      const mutationHook = `use${capitalizedOperationName}Mutation`;
+      lines.push(`export const ${mutationHook} = injectedRtkApi.${mutationHook} as unknown as UseSimpleMutation<${info.responseTypeName}, ${argType}>;`);
+    }
+    return lines.join("\n");
+  }).join("\n");
   return `/* eslint-disable */
 // [Warning] Generated automatically - do not edit manually
 
-${apiImport}${httpClientImport}${tagTypesImport}
+${apiImport}${httpClientImport}${tagTypesImport}${simpleTypeImportStatement}
 ${typeImportStatement}
 
 
@@ -1120,24 +1199,7 @@ ${endpoints}
     }),
 });
 
-export const {
-${endpointInfos.map((info) => {
-    const capitalizedOperationName = info.operationName.charAt(0).toUpperCase() + info.operationName.slice(1);
-    if (info.isQuery) {
-      const regularHook = `use${capitalizedOperationName}Query`;
-      if (options.useLazyQueries) {
-        const lazyHook = `useLazy${capitalizedOperationName}Query`;
-        return `  ${regularHook},
-  ${lazyHook},`;
-      } else {
-        return `  ${regularHook},`;
-      }
-    } else {
-      const mutationHook = `use${capitalizedOperationName}Mutation`;
-      return `  ${mutationHook},`;
-    }
-  }).join("\n")}
-} = injectedRtkApi;
+${hookExports}
 
 export default injectedRtkApi;
 `;
@@ -1697,12 +1759,6 @@ var UnifiedCodeGenerator = class {
         );
         results.push(...schemaResults);
       }
-      const mainIndexContent = this.generateMainIndex(generatedGroups);
-      const mainIndexResult = await this.fileWriterService.writeFile(
-        import_node_path4.default.join(outputDir, "index.ts"),
-        mainIndexContent
-      );
-      results.push(mainIndexResult);
     } catch (error) {
       errors.push(error);
     }