add a fix to replace the path

Author: broccolinisoup

src/rules/__tests__/use-next-tooltip.test.js

@@ -13,20 +13,27 @@ const ruleTester = new RuleTester({
 
 ruleTester.run('use-next-tooltip', rule, {
   valid: [
-    `import {Tooltip} from '@primer/react/next'`,
+    `import {Tooltip} from '@primer/react/next';`,
     `import {UnderlineNav, Button} from '@primer/react';
    import {Tooltip} from '@primer/react/next';`,
     `import {UnderlineNav, Button} from '@primer/react';
    import {Tooltip, SelectPanel} from '@primer/react/next';`,
   ],
   invalid: [
     {
-      code: `import {Tooltip} from '@primer/react'`,
+      code: `import {Tooltip} from '@primer/react';`,
       errors: [{messageId: 'useNextTooltip'}],
+      output: `import {Tooltip} from '@primer/react/next';`,
     },
     {
-      code: `import {Tooltip, Button} from '@primer/react'`,
+      code: `import {Tooltip, Button} from '@primer/react';`,
       errors: [{messageId: 'useNextTooltip'}],
+      output: `import {Button} from '@primer/react';\nimport {Tooltip} from '@primer/react/next';`,
+    },
+    {
+      code: `import {ActionList, ActionMenu, Tooltip, Button} from '@primer/react';`,
+      errors: [{messageId: 'useNextTooltip'}],
+      output: `import {ActionList, ActionMenu, Button} from '@primer/react';\nimport {Tooltip} from '@primer/react/next';`,
     },
   ],
 })

src/rules/use-next-tooltip.js

@@ -8,7 +8,7 @@ module.exports = {
       category: 'Best Practices',
       recommended: true,
     },
-    fixable: null,
+    fixable: true,
     schema: [],
     messages: {
       useNextTooltip: 'Please use @primer/react/next Tooltip component that has accessibility improvements',
@@ -23,12 +23,36 @@ module.exports = {
         const hasTooltip = node.specifiers.some(
           specifier => specifier.imported && specifier.imported.name === 'Tooltip',
         )
+
+        const hasOtherImports = node.specifiers.some(
+          specifier => specifier.imported && specifier.imported.name !== 'Tooltip',
+        )
         if (!hasTooltip) {
           return
         }
         context.report({
           node,
           messageId: 'useNextTooltip',
+          fix(fixer) {
+            // If Tooltip is the only import, replace the whole import statement
+            if (!hasOtherImports) {
+              return fixer.replaceText(node.source, `'@primer/react/next'`)
+            } else {
+              // Otherwise, remove Tooltip from the import statement and add a new import statement with the correct path
+              const tooltipSpecifier = node.specifiers.find(
+                specifier => specifier.imported && specifier.imported.name === 'Tooltip',
+              )
+              return [
+                // remove tooltip specifier and the space and comma after it
+                fixer.removeRange([tooltipSpecifier.range[0], tooltipSpecifier.range[1] + 2]),
+                // fixer.remove(tooltipSpecifier),
+                fixer.insertTextAfterRange(
+                  [node.range[1], node.range[1]],
+                  `\nimport {Tooltip} from '@primer/react/next';`,
+                ),
+              ]
+            }
+          },
         })
       },
     }