feat(prefetch): add prefetch query hook generation

closes: #91

README.md

@@ -71,13 +71,16 @@ $ openapi-rq -i ./petstore.yaml
 ```
 - openapi
   - queries
-    - index.ts <- main file that exports common types, variables, and hooks
+    - index.ts <- main file that exports common types, variables, and queries. Does not export suspense or prefetch hooks
     - common.ts <- common types
     - queries.ts <- generated query hooks
     - suspenses.ts <- generated suspense hooks
+    - prefetch.ts <- generated prefetch hooks learn more about prefetching in in link below
   - requests <- output code generated by @hey-api/openapi-ts
 ```
 
+- [Prefetching docs](https://tanstack.com/query/latest/docs/framework/react/guides/advanced-ssr#prefetching-and-dehydrating-data)
+
 ### In your app
 
 ```tsx

examples/react-app/src/main.tsx

@@ -4,10 +4,17 @@ import App from "./App";
 import "./index.css";
 import { QueryClientProvider } from "@tanstack/react-query";
 import { queryClient } from "./queryClient";
+import { useDefaultServiceFindPetsPrefetch } from "../openapi/queries/prefetch";
+
+const PrefetchData = () => {
+  useDefaultServiceFindPetsPrefetch(queryClient);
+  return null;
+};
 
 ReactDOM.createRoot(document.getElementById("root") as HTMLElement).render(
   <React.StrictMode>
     <QueryClientProvider client={queryClient}>
+      <PrefetchData />
       <App />
     </QueryClientProvider>
   </React.StrictMode>

src/constants.mts

@@ -4,3 +4,11 @@ export const requestsOutputPath = "requests";
 
 export const serviceFileName = "services.gen";
 export const modalsFileName = "types.gen";
+
+export const OpenApiRqFiles = {
+  queries: "queries",
+  common: "common",
+  suspense: "suspense",
+  index: "index",
+  prefetch: "prefetch",
+} as const;

src/createExports.mts

@@ -1,6 +1,7 @@
 import { createUseQuery } from "./createUseQuery.mjs";
 import { createUseMutation } from "./createUseMutation.mjs";
 import { Service } from "./service.mjs";
+import { createPrefetch } from "./createPrefetch.mjs";
 
 export const createExports = (service: Service) => {
   const { klasses } = service;
@@ -23,6 +24,7 @@ export const createExports = (service: Service) => {
   );
 
   const allGetQueries = allGet.map((m) => createUseQuery(m));
+  const allPrefetchQueries = allGet.map((m) => createPrefetch(m));
 
   const allPostMutations = allPost.map((m) => createUseMutation(m));
   const allPutMutations = allPut.map((m) => createUseMutation(m));
@@ -59,6 +61,12 @@ export const createExports = (service: Service) => {
 
   const suspenseExports = [...suspenseQueries];
 
+  const allPrefetches = allPrefetchQueries
+    .map(({ prefetchHook }) => [prefetchHook])
+    .flat();
+
+  const allPrefetchExports = [...allPrefetches];
+
   return {
     /**
      * Common types and variables between queries (regular and suspense) and mutations
@@ -72,5 +80,9 @@ export const createExports = (service: Service) => {
      * Suspense exports are the hooks that are used in the suspense components
      */
     suspenseExports,
+    /**
+     * Prefetch exports are the hooks that are used in the prefetch components
+     */
+    allPrefetchExports,
   };
 };

src/createPrefetch.mts

@@ -0,0 +1,160 @@
+import ts from "typescript";
+import { MethodDeclaration } from "ts-morph";
+import {
+  BuildCommonTypeName,
+  extractPropertiesFromObjectParam,
+  getNameFromMethod,
+} from "./common.mjs";
+import { type MethodDescription } from "./common.mjs";
+import {
+  createQueryKeyFromMethod,
+  getRequestParamFromMethod,
+  hookNameFromMethod,
+} from "./createUseQuery.mjs";
+import { addJSDocToNode } from "./util.mjs";
+
+/**
+ * Creates a custom hook for a query
+ * @param queryString The type of query to use from react-query
+ * @param suffix The suffix to append to the hook name
+ */
+function createPrefetchHook({
+  requestParams,
+  method,
+  className,
+}: {
+  requestParams: ts.ParameterDeclaration[];
+  method: MethodDeclaration;
+  className: string;
+}) {
+  const methodName = getNameFromMethod(method);
+  const customHookName = hookNameFromMethod({ method, className });
+  const queryKey = createQueryKeyFromMethod({ method, className });
+
+  // const
+  const hookExport = ts.factory.createVariableStatement(
+    // export
+    [ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)],
+    ts.factory.createVariableDeclarationList(
+      [
+        ts.factory.createVariableDeclaration(
+          ts.factory.createIdentifier(`${customHookName}Prefetch`),
+          undefined,
+          undefined,
+          ts.factory.createArrowFunction(
+            undefined,
+            undefined,
+            [
+              ts.factory.createParameterDeclaration(
+                undefined,
+                undefined,
+                "queryClient",
+                undefined,
+                ts.factory.createTypeReferenceNode(
+                  ts.factory.createIdentifier("QueryClient")
+                )
+              ),
+              ...requestParams,
+            ],
+            undefined,
+            ts.factory.createToken(ts.SyntaxKind.EqualsGreaterThanToken),
+            ts.factory.createCallExpression(
+              ts.factory.createIdentifier("queryClient.prefetchQuery"),
+              undefined,
+              [
+                ts.factory.createObjectLiteralExpression([
+                  ts.factory.createPropertyAssignment(
+                    ts.factory.createIdentifier("queryKey"),
+                    ts.factory.createArrayLiteralExpression(
+                      [
+                        BuildCommonTypeName(queryKey),
+                        method.getParameters().length
+                          ? ts.factory.createArrayLiteralExpression([
+                              ts.factory.createObjectLiteralExpression(
+                                method
+                                  .getParameters()
+                                  .map((param) =>
+                                    extractPropertiesFromObjectParam(param).map(
+                                      (p) =>
+                                        ts.factory.createShorthandPropertyAssignment(
+                                          ts.factory.createIdentifier(p.name)
+                                        )
+                                    )
+                                  )
+                                  .flat()
+                              ),
+                            ])
+                          : ts.factory.createArrayLiteralExpression([]),
+                      ],
+                      false
+                    )
+                  ),
+                  ts.factory.createPropertyAssignment(
+                    ts.factory.createIdentifier("queryFn"),
+                    ts.factory.createArrowFunction(
+                      undefined,
+                      undefined,
+                      [],
+                      undefined,
+                      ts.factory.createToken(
+                        ts.SyntaxKind.EqualsGreaterThanToken
+                      ),
+                      ts.factory.createCallExpression(
+                        ts.factory.createPropertyAccessExpression(
+                          ts.factory.createIdentifier(className),
+                          ts.factory.createIdentifier(methodName)
+                        ),
+                        undefined,
+                        method.getParameters().length
+                          ? [
+                              ts.factory.createObjectLiteralExpression(
+                                method
+                                  .getParameters()
+                                  .map((param) =>
+                                    extractPropertiesFromObjectParam(param).map(
+                                      (p) =>
+                                        ts.factory.createShorthandPropertyAssignment(
+                                          ts.factory.createIdentifier(p.name)
+                                        )
+                                    )
+                                  )
+                                  .flat()
+                              ),
+                            ]
+                          : undefined
+                      )
+                    )
+                  ),
+                ]),
+              ]
+            )
+          )
+        ),
+      ],
+      ts.NodeFlags.Const
+    )
+  );
+  return hookExport;
+}
+
+export const createPrefetch = ({
+  className,
+  method,
+  jsDoc,
+}: MethodDescription) => {
+  const requestParam = getRequestParamFromMethod(method);
+
+  const requestParams = requestParam ? [requestParam] : [];
+
+  const prefetchHook = createPrefetchHook({
+    requestParams,
+    method,
+    className,
+  });
+
+  const hookWithJsDoc = addJSDocToNode(prefetchHook, jsDoc);
+
+  return {
+    prefetchHook: hookWithJsDoc,
+  };
+};

src/createSource.mts

@@ -1,9 +1,10 @@
 import ts from "typescript";
+import { Project } from "ts-morph";
+import { join } from "path";
+import { OpenApiRqFiles } from "./constants.mjs";
 import { createImports } from "./createImports.mjs";
 import { createExports } from "./createExports.mjs";
 import { getServices } from "./service.mjs";
-import { Project } from "ts-morph";
-import { join } from "path";
 
 const createSourceFile = async (outputPath: string, serviceEndName: string) => {
   const project = new Project({
@@ -77,11 +78,18 @@ const createSourceFile = async (outputPath: string, serviceEndName: string) => {
     ts.NodeFlags.None
   );
 
+  const prefetchSource = ts.factory.createSourceFile(
+    [commonImport, ...imports, ...exports.allPrefetchExports],
+    ts.factory.createToken(ts.SyntaxKind.EndOfFileToken),
+    ts.NodeFlags.None
+  );
+
   return {
     commonSource,
     mainSource,
     suspenseSource,
     indexSource,
+    prefetchSource,
   };
 };
 
@@ -95,29 +103,37 @@ export const createSource = async ({
   serviceEndName: string;
 }) => {
   const queriesFile = ts.createSourceFile(
-    "queries.ts",
+    `${OpenApiRqFiles.queries}.ts`,
     "",
     ts.ScriptTarget.Latest,
     false,
     ts.ScriptKind.TS
   );
   const commonFile = ts.createSourceFile(
-    "common.ts",
+    `${OpenApiRqFiles.common}.ts`,
     "",
     ts.ScriptTarget.Latest,
     false,
     ts.ScriptKind.TS
   );
   const suspenseFile = ts.createSourceFile(
-    "suspense.ts",
+    `${OpenApiRqFiles.suspense}.ts`,
     "",
     ts.ScriptTarget.Latest,
     false,
     ts.ScriptKind.TS
   );
 
   const indexFile = ts.createSourceFile(
-    "index.ts",
+    `${OpenApiRqFiles.index}.ts`,
+    "",
+    ts.ScriptTarget.Latest,
+    false,
+    ts.ScriptKind.TS
+  );
+
+  const prefetchFile = ts.createSourceFile(
+    `${OpenApiRqFiles.prefetch}.ts`,
     "",
     ts.ScriptTarget.Latest,
     false,
@@ -129,8 +145,13 @@ export const createSource = async ({
     removeComments: false,
   });
 
-  const { commonSource, mainSource, suspenseSource, indexSource } =
-    await createSourceFile(outputPath, serviceEndName);
+  const {
+    commonSource,
+    mainSource,
+    suspenseSource,
+    indexSource,
+    prefetchSource,
+  } = await createSourceFile(outputPath, serviceEndName);
 
   const comment = `// generated with @7nohe/openapi-react-query-codegen@${version} \n\n`;
 
@@ -150,6 +171,10 @@ export const createSource = async ({
     comment +
     printer.printNode(ts.EmitHint.Unspecified, indexSource, indexFile);
 
+  const prefetchResult =
+    comment +
+    printer.printNode(ts.EmitHint.Unspecified, prefetchSource, prefetchFile);
+
   return [
     {
       name: "index.ts",
@@ -167,5 +192,9 @@ export const createSource = async ({
       name: "suspense.ts",
       content: suspenseResult,
     },
+    {
+      name: "prefetch.ts",
+      content: prefetchResult,
+    },
   ];
 };

src/createUseQuery.mts

@@ -200,7 +200,7 @@ export function createQueryKeyExport({
   );
 }
 
-function hookNameFromMethod({
+export function hookNameFromMethod({
   method,
   className,
 }: {
@@ -211,7 +211,7 @@ function hookNameFromMethod({
   return `use${className}${capitalizeFirstLetter(methodName)}`;
 }
 
-function createQueryKeyFromMethod({
+export function createQueryKeyFromMethod({
   method,
   className,
 }: {
@@ -228,7 +228,7 @@ function createQueryKeyFromMethod({
  * @param queryString The type of query to use from react-query
  * @param suffix The suffix to append to the hook name
  */
-function createQueryHook({
+export function createQueryHook({
   queryString,
   suffix,
   responseDataType,