feat: resolve body of default function arrow functions (#82)

closes #80

Author: elevatebart

src/script-handlers/__tests__/propHandler.ts

@@ -198,6 +198,21 @@ describe('propHandler', () => {
         defaultValue: { value: `"normal"` },
       })
     })
+
+    it('should return the body of the function as default value without parenthesis', () => {
+      const src = `
+        export default {
+          props: {
+            test: {
+              default: () => ({})
+            }
+          }
+        }
+        `
+      tester(src, {
+        defaultValue: { value: `{}` },
+      })
+    })
   })
 
   describe('description', () => {

src/script-handlers/propHandler.ts

@@ -136,13 +136,24 @@ export function describeDefault(
 ) {
   const defaultArray = propPropertiesPath.filter(p => p.node.key.name === 'default')
   if (defaultArray.length) {
-    const defaultPath = defaultArray[0].get('value')
+    let defaultPath = defaultArray[0].get('value')
+
+    let parenthesized = false
+    if (
+      bt.isArrowFunctionExpression(defaultPath.node) &&
+      bt.isObjectExpression(defaultPath.node.body) // if () => ({})
+    ) {
+      defaultPath = defaultPath.get('body')
+      const extra = (defaultPath.node as any).extra
+      if (extra && extra.parenthesized) {
+        parenthesized = true
+      }
+    }
 
-    const func =
-      bt.isArrowFunctionExpression(defaultPath.node) || bt.isFunctionExpression(defaultPath.node)
+    const rawValue = recast.print(defaultPath).code
     propDescriptor.defaultValue = {
-      func,
-      value: recast.print(defaultPath).code,
+      func: bt.isFunction(defaultPath.node),
+      value: parenthesized ? rawValue.slice(1, rawValue.length - 1) : rawValue,
     }
   }
 }