fix(inline-arrow): add parens for object expression

Author: sxzz

src/commands/_utils.ts

@@ -28,3 +28,13 @@ export function insideRange(node: Tree.Node, range: [number, number], includeSta
     && (includeEnd ? node.range[1] <= range[1] : node.range[1] < range[1])
   )
 }
+
+export function unwrapType(node: Tree.Node) {
+  if (node.type === 'TSAsExpression' // foo as number
+    || node.type === 'TSSatisfiesExpression' // foo satisfies T
+    || node.type === 'TSNonNullExpression' // foo!
+    || node.type === 'TSInstantiationExpression' // foo<string>
+    || node.type === 'TSTypeAssertion') // <number>foo
+    return node.expression
+  return node
+}

src/commands/inline-arrow.test.ts

@@ -56,4 +56,16 @@ run(
     `,
     errors: ['command-fix'],
   },
+  {
+    code: $`
+      /// inline-arrow
+      export const foo = () => {
+        return { a: 'b' } as any
+      }
+    `,
+    output: $`
+      export const foo = () => ({ a: 'b' } as any)
+    `,
+    errors: ['command-fix'],
+  },
 )

src/commands/inline-arrow.ts

@@ -1,4 +1,5 @@
 import type { Command, Tree } from '../types'
+import { unwrapType } from './_utils'
 
 export const inlineArrow: Command = {
   name: 'inline-arrow',
@@ -18,15 +19,20 @@ export const inlineArrow: Command = {
     )
       return ctx.reportError('Arrow function body must have a single statement')
     const statement = statements[0] as Tree.ReturnStatement | undefined
+    const argument: Tree.Node | null = statement?.argument ? unwrapType(statement.argument) : null
+    const isObject = (argument?.type === 'ObjectExpression')
 
     ctx.report({
       node: arrowFn,
       loc: body.loc,
       message: 'Inline arrow function',
       fix(fixer) {
-        return fixer.replaceTextRange(body.range, statement && statement.argument
+        let raw = statement && statement.argument
           ? ctx.getTextOf(statement.argument)
-          : 'undefined')
+          : 'undefined'
+        if (isObject)
+          raw = `(${raw})`
+        return fixer.replaceTextRange(body.range, raw)
       },
     })
   },