Extract findTagName() function

Author: Turbo87

lib/index.js

@@ -105,19 +105,12 @@ async function runForPath(path, options = {}) {
     // find `tagName` property if it exists
     let properties = objectArg.get('properties');
 
-    let tagName = 'div';
-
-    let tagNameProperty = properties.filter(path => isProperty(path, 'tagName'))[0];
-    if (tagNameProperty) {
-      let tagNamePath = tagNameProperty.get('value');
-      if (tagNamePath.value.type !== 'StringLiteral') {
-        console.log(
-          chalk.yellow(`${dimPath}: Unexpected \`tagName\` value: ${j(tagNamePath).toSource()}`)
-        );
-        continue;
-      }
-
-      tagName = tagNamePath.value.value;
+    let tagName = findTagName(properties);
+    if (tagName.constructor.name === 'NodePath') {
+      console.log(
+        chalk.yellow(`${dimPath}: Unexpected \`tagName\` value: ${j(tagName).toSource()}`)
+      );
+      continue;
     }
 
     // skip tagless components (silent)
@@ -176,4 +169,18 @@ function isMethod(path, name) {
   return node.type === 'ObjectMethod' && node.key.type === 'Identifier' && node.key.name === name;
 }
 
+function findTagName(properties) {
+  let tagNameProperty = properties.filter(path => isProperty(path, 'tagName'))[0];
+  if (!tagNameProperty) {
+    return 'div';
+  }
+
+  let tagNamePath = tagNameProperty.get('value');
+  if (tagNamePath.value.type === 'StringLiteral') {
+    return tagNamePath.value.value;
+  }
+
+  return tagNamePath;
+}
+
 module.exports = { run };