Merge pull request #183 from croutonn/fix-useSWRInfinite-import

Fix import `useSWRInfinite` statement

Author: eitoball

src/visitor.ts

@@ -124,8 +124,9 @@ export class SWRVisitor extends ClientSideBaseVisitor<
         this._additionalImports.push(
           `import type { SWRInfiniteConfiguration } from 'swr/infinite';`
         )
+        this._additionalImports.push(`import useSWR from 'swr';`)
         this._additionalImports.push(
-          `import useSWR, { useSWRInfinite } from 'swr';`
+          `import useSWRInfinite from 'swr/infinite';`
         )
       } else {
         this._additionalImports.push(
@@ -135,10 +136,10 @@ export class SWRVisitor extends ClientSideBaseVisitor<
       }
     } else if (this._enabledInfinite) {
       this._additionalImports.push(
-        `import useSWR, { useSWRInfinite, SWRConfiguration as SWRConfigInterface, Key as SWRKeyInterface } from 'swr';`
+        `import useSWR, { SWRConfiguration as SWRConfigInterface, Key as SWRKeyInterface } from 'swr';`
       )
       this._additionalImports.push(
-        `import { SWRInfiniteConfiguration } from 'swr/infinite';`
+        `import useSWRInfinite, { SWRInfiniteConfiguration } from 'swr/infinite';`
       )
     } else {
       this._additionalImports.push(

tests/swr.spec.ts

@@ -234,10 +234,10 @@ async function test() {
       const usage = basicUsage
       const output = await validate(content, config, docs, schema, usage)
       expect(content.prepend).toContain(
-        `import useSWR, { useSWRInfinite, SWRConfiguration as SWRConfigInterface, Key as SWRKeyInterface } from 'swr';`
+        `import useSWR, { SWRConfiguration as SWRConfigInterface, Key as SWRKeyInterface } from 'swr';`
       )
       expect(content.prepend).toContain(
-        `import { SWRInfiniteConfiguration } from 'swr/infinite';`
+        `import useSWRInfinite, { SWRInfiniteConfiguration } from 'swr/infinite';`
       )
       expect(output).toContain(readOutput('infinite'))
     })