Support shorthand fragments in Components

Author: alexzherdev

lib/util/Components.js

@@ -11,6 +11,7 @@ const variableUtil = require('./variable');
 const pragmaUtil = require('./pragma');
 const astUtil = require('./ast');
 const propTypes = require('./propTypes');
+const jsxUtil = require('./jsx');
 
 function getId(node) {
   return node && node.range.join(':');
@@ -349,12 +350,12 @@ function componentRule(rule, context) {
       const returnsConditionalJSXConsequent =
         node[property] &&
         node[property].type === 'ConditionalExpression' &&
-        node[property].consequent.type === 'JSXElement'
+        jsxUtil.isJSX(node[property].consequent)
       ;
       const returnsConditionalJSXAlternate =
         node[property] &&
         node[property].type === 'ConditionalExpression' &&
-        node[property].alternate.type === 'JSXElement'
+        jsxUtil.isJSX(node[property].alternate)
       ;
       const returnsConditionalJSX =
         strict ?
@@ -363,7 +364,7 @@ function componentRule(rule, context) {
 
       const returnsJSX =
         node[property] &&
-        node[property].type === 'JSXElement'
+        jsxUtil.isJSX(node[property])
       ;
       const returnsReactCreateElement = this.isReactCreateElement(node[property]);
 

lib/util/jsx.js

@@ -9,7 +9,7 @@ const COMPAT_TAG_REGEX = /^[a-z]|\-/;
 
 /**
  * Checks if a node represents a DOM element.
- * @param {String} node - JSXOpeningElement to check.
+ * @param {object} node - JSXOpeningElement to check.
  * @returns {boolean} Whether or not the node corresponds to a DOM element.
  */
 function isDOMComponent(node) {
@@ -25,6 +25,16 @@ function isDOMComponent(node) {
   return COMPAT_TAG_REGEX.test(name);
 }
 
+/**
+ * Checks if a node represents a JSX element or fragment.
+ * @param {object} node - node to check.
+ * @returns {boolean} Whether or not the node if a JSX element or fragment.
+ */
+function isJSX(node) {
+  return ['JSXElement', 'JSXFragment'].indexOf(node.type) >= 0;
+}
+
 module.exports = {
-  isDOMComponent: isDOMComponent
+  isDOMComponent: isDOMComponent,
+  isJSX: isJSX
 };