Add useSWRInfinite option

Author: croutonn

src/config.ts

@@ -24,4 +24,5 @@ export interface RawSWRPluginConfig extends RawClientSideBasePluginConfig {
    */
   rawRequest?: boolean
   exclude?: string | string[]
+  useSWRInfinite?: string | string[]
 }

src/visitor.ts

@@ -14,6 +14,7 @@ import { RawSWRPluginConfig } from './config'
 export interface SWRPluginConfig extends ClientSideBasePluginConfig {
   rawRequest: boolean
   exclude: string | string[]
+  useSWRInfinite: string | string[]
 }
 
 export class SWRVisitor extends ClientSideBaseVisitor<
@@ -28,22 +29,44 @@ export class SWRVisitor extends ClientSideBaseVisitor<
     operationVariablesTypes: string
   }[] = []
 
+  private _enabledInfinite = false
+
   constructor(
     schema: GraphQLSchema,
     fragments: LoadedFragment[],
     rawConfig: RawSWRPluginConfig
   ) {
     super(schema, fragments, rawConfig, {
       exclude: rawConfig.exclude || null,
+      useSWRInfinite: rawConfig.useSWRInfinite || null,
     })
 
+    this._enabledInfinite =
+      (this.config.useSWRInfinite &&
+        typeof this.config.useSWRInfinite === 'string') ||
+      (Array.isArray(this.config.useSWRInfinite) &&
+        this.config.useSWRInfinite.length > 0)
+
     autoBind(this)
 
     if (this.config.useTypeImports) {
+      if (this._enabledInfinite) {
+        this._additionalImports.push(
+          `import type { ConfigInterface as SWRConfigInterface, keyInterface as SWRKeyInterface, SWRInfiniteConfigInterface } from 'swr';`
+        )
+        this._additionalImports.push(
+          `import useSWR, { useSWRInfinite } from 'swr';`
+        )
+      } else {
+        this._additionalImports.push(
+          `import type { ConfigInterface as SWRConfigInterface, keyInterface as SWRKeyInterface } from 'swr';`
+        )
+        this._additionalImports.push(`import useSWR from 'swr';`)
+      }
+    } else if (this._enabledInfinite) {
       this._additionalImports.push(
-        `import type { ConfigInterface as SWRConfigInterface, keyInterface as SWRKeyInterface } from 'swr';`
+        `import useSWR, { useSWRInfinite, ConfigInterface as SWRConfigInterface, keyInterface as SWRKeyInterface, SWRInfiniteConfigInterface } from 'swr';`
       )
-      this._additionalImports.push(`import useSWR from 'swr';`)
     } else {
       this._additionalImports.push(
         `import useSWR, { ConfigInterface as SWRConfigInterface, keyInterface as SWRKeyInterface } from 'swr';`
@@ -70,13 +93,15 @@ export class SWRVisitor extends ClientSideBaseVisitor<
   }
 
   public get sdkContent(): string {
+    const { exclude } = this.config
+    const disabledExclude =
+      !exclude || (Array.isArray(exclude) && !exclude.length)
     const allPossibleActions = this._operationsToInclude
       .filter((o) => {
         if (o.operationType !== 'Query') {
           return false
         }
-        const { exclude } = this.config
-        if (!exclude || (Array.isArray(exclude) && !exclude.length)) {
+        if (disabledExclude) {
           return true
         }
         const name = o.node.name.value
@@ -89,34 +114,87 @@ export class SWRVisitor extends ClientSideBaseVisitor<
           o.node.variableDefinitions.every(
             (v) => v.type.kind !== Kind.NON_NULL_TYPE || v.defaultValue
           )
+        const name = o.node.name.value
+        const enabledInfinite =
+          this._enabledInfinite &&
+          glob.isMatch(name, this.config.useSWRInfinite)
+        const codes: string[] = []
 
         if (this.config.rawRequest) {
-          return `use${pascalCase(
+          codes.push(`use${pascalCase(
             o.node.name.value
           )}(key: SWRKeyInterface, variables${optionalVariables ? '?' : ''}: ${
             o.operationVariablesTypes
-          }, config?: SWRConfigInterface<${o.operationResultType}>) {
-            return useSWR<{ data?: ${
+          }, config?: SWRConfigInterface<SWRRawResponse<${
+            o.operationResultType
+          }>}>) {
+            return useSWR<SWRRawResponse<${
               o.operationResultType
-            } | undefined; extensions?: any; headers: Headers; status: number; errors?: GraphQLError[] | undefined; }>(key, () => sdk.${
-            o.node.name.value
-          }(variables), config);
-        }`
+            }>>(key, () => sdk.${o.node.name.value}(variables), config);
+        }`)
+
+          if (enabledInfinite) {
+            codes.push(`use${pascalCase(
+              o.node.name.value
+            )}Infinite(getKey: SWRInfiniteKeyLoader<SWRRawResponse<${
+              o.operationResultType
+            }>>, variables${optionalVariables ? '?' : ''}: ${
+              o.operationVariablesTypes
+            }, config?: SWRInfiniteConfigInterface<SWRRawResponse<${
+              o.operationResultType
+            }>>) {
+            return useSWRInfinite<SWRRawResponse<${
+              o.operationResultType
+            }>>(getKey, () => sdk.${o.node.name.value}(variables), config);
+        }`)
+          }
+          return codes
         }
 
-        return `use${pascalCase(
+        codes.push(`use${pascalCase(
           o.node.name.value
         )}(key: SWRKeyInterface, variables${optionalVariables ? '?' : ''}: ${
           o.operationVariablesTypes
         }, config?: SWRConfigInterface<${o.operationResultType}>) {
   return useSWR<${o.operationResultType}>(key, () => sdk.${
           o.node.name.value
         }(variables), config);
-}`
+}`)
+
+        if (enabledInfinite) {
+          codes.push(`use${pascalCase(
+            o.node.name.value
+          )}Infinite(getKey: SWRInfiniteKeyLoader<${
+            o.operationResultType
+          }>, variables${optionalVariables ? '?' : ''}: ${
+            o.operationVariablesTypes
+          }, config?: SWRInfiniteConfigInterface<${o.operationResultType}>) {
+  return useSWRInfinite<${o.operationResultType}>(getKey, () => sdk.${
+            o.node.name.value
+          }(variables), config);
+}`)
+        }
+
+        return codes
       })
+      .reduce((p, c) => p.concat(c), [])
       .map((s) => indentMultiline(s, 2))
 
-    return `export function getSdkWithHooks(client: GraphQLClient, withWrapper: SdkFunctionWrapper = defaultWrapper) {
+    const types: string[] = []
+    if (this.config.rawRequest) {
+      types.push(
+        `type SWRRawResponse<Data = any> = { data?: Data | undefined; extensions?: any; headers: Headers; status: number; errors?: GraphQLError[] | undefined; };`
+      )
+    }
+    if (this._enabledInfinite) {
+      types.push(`export type SWRInfiniteKeyLoader<Data = any> = (
+  index: number,
+  previousPageData: Data | null
+) => string | any[] | null;`)
+    }
+
+    return `${types.join('\n')}
+export function getSdkWithHooks(client: GraphQLClient, withWrapper: SdkFunctionWrapper = defaultWrapper) {
   const sdk = getSdk(client, withWrapper);
   return {
     ...sdk,