chore: extract fixTypeBased function

Author: mpiniarski

lib/rules/define-props-declaration.js

@@ -6,6 +6,68 @@
 
 const utils = require('../utils')
 
+const PROPS_SEPARATOR = ', '
+
+/**
+ * @param {RuleFixer} fixer
+ * @param {CallExpression} node
+ * @param {ComponentProp[]} props
+ * @param {RuleContext} context
+ */
+function* fixTypeBased(fixer, node, props, context) {
+  try {
+    const sourceCode = context.getSourceCode()
+    const autoFixToSeparateInterface =
+      context.options[1]?.autoFixToSeparateInterface || false
+
+    const propTypes = props.map((prop) =>
+      getComponentPropData(prop, sourceCode)
+    )
+
+    const definePropsType = `{ ${propTypes
+      .map(
+        ({ name, type, required, defaultValue }) =>
+          `${name}${required === false || defaultValue ? '?' : ''}: ${type}`
+      )
+      .join(PROPS_SEPARATOR)} }`
+
+    // remove defineProps function parameters
+    yield fixer.replaceText(node.arguments[0], '')
+
+    // add type annotation
+    if (autoFixToSeparateInterface) {
+      const variableDeclarationNode = node.parent.parent
+      if (!variableDeclarationNode) {
+        return
+      }
+
+      yield fixer.insertTextBefore(
+        variableDeclarationNode,
+        `interface Props ${definePropsType.replace(/;/g, ',')}; `
+      )
+      yield fixer.insertTextAfter(node.callee, `<Props>`)
+    } else {
+      yield fixer.insertTextAfter(node.callee, `<${definePropsType}>`)
+    }
+
+    // add defaults if needed
+    const defaults = propTypes.filter(({ defaultValue }) => defaultValue)
+    if (defaults.length > 0) {
+      const defaultsCode = defaults
+        .map(
+          ({ name, defaultValue }) =>
+            `${name}: ${sourceCode.getText(defaultValue)}`
+        )
+        .join(PROPS_SEPARATOR)
+
+      yield fixer.insertTextBefore(node, `withDefaults(`)
+      yield fixer.insertTextAfter(node, `, { ${defaultsCode} })`)
+    }
+    return null
+  } catch (error) {
+    return null
+  }
+}
 const mapNativeType = (/** @type {string} */ nativeType) => {
   switch (nativeType) {
     case 'String': {
@@ -198,15 +260,12 @@ module.exports = {
   },
   /** @param {RuleContext} context */
   create(context) {
-    const sourceCode = context.getSourceCode()
-
     const scriptSetup = utils.getScriptSetupElement(context)
     if (!scriptSetup || !utils.hasAttribute(scriptSetup, 'lang', 'ts')) {
       return {}
     }
 
     const defineType = context.options[0] || 'type-based'
-    const autoFixToSeparateInterface = context.options[1]?.autoFixToSeparateInterface || false
 
     return utils.defineScriptSetupVisitor(context, {
       onDefinePropsEnter(node, props) {
@@ -217,67 +276,7 @@ module.exports = {
                 node,
                 messageId: 'hasArg',
                 *fix(fixer) {
-                  try {
-                    const propTypes = props.map((prop) =>
-                      getComponentPropData(prop, sourceCode)
-                    )
-
-                    const definePropsType = `{ ${propTypes
-                      .map(
-                        ({ name, type, required, defaultValue }) =>
-                          `${name}${
-                            required === false || defaultValue ? '?' : ''
-                          }: ${type}`
-                      )
-                      .join(', ')} }`
-
-                    // remove defineProps function parameters
-                    yield fixer.replaceText(node.arguments[0], '')
-
-                    // add type annotation
-                    if (autoFixToSeparateInterface) {
-                      const variableDeclarationNode = node.parent.parent
-                      if (!variableDeclarationNode) {
-                        return
-                      }
-
-                      yield fixer.insertTextBefore(
-                        variableDeclarationNode,
-                        `interface Props ${definePropsType.replace(
-                          /;/g,
-                          ','
-                        )}; `
-                      )
-                      yield fixer.insertTextAfter(node.callee, `<Props>`)
-                    } else {
-                      yield fixer.insertTextAfter(
-                        node.callee,
-                        `<${definePropsType}>`
-                      )
-                    }
-
-                    // add defaults if needed
-                    const defaults = propTypes.filter(
-                      ({ defaultValue }) => defaultValue
-                    )
-                    if (defaults.length > 0) {
-                      const defaultsCode = defaults
-                        .map(
-                          ({ name, defaultValue }) =>
-                            `${name}: ${sourceCode.getText(defaultValue)}`
-                        )
-                        .join(', ')
-
-                      yield fixer.insertTextBefore(node, `withDefaults(`)
-                      yield fixer.insertTextAfter(
-                        node,
-                        `, { ${defaultsCode} })`
-                      )
-                    }
-                    return null
-                  } catch (error) {
-                    return null
-                  }
+                  yield* fixTypeBased(fixer, node, props, context)
                 }
               })
             }