Put all utilities to one file and add supporting of tag literals

Author: ezhlobo

lib/rules/uses-react.js

@@ -3,6 +3,8 @@
  * @author Eugene Zhlobo
  */
 
+const { isReactPugReference } = require('../utilities')
+
 //------------------------------------------------------------------------------
 // Rule Definition
 //------------------------------------------------------------------------------
@@ -21,9 +23,8 @@ module.exports = {
     return {
       TaggedTemplateExpression: function (node) {
         const pragma = 'React'
-        const { tag } = node
 
-        if (tag && tag.name === 'pug') {
+        if (isReactPugReference(node)) {
           context.markVariableAsUsed(pragma)
         }
       },

lib/rules/uses-vars.js

@@ -4,6 +4,7 @@
  */
 
 const { findVariablesInTemplate } = require('pug-uses-variables')
+const { isReactPugReference, getTemplate } = require('../utilities')
 
 //------------------------------------------------------------------------------
 // Rule Definition
@@ -20,14 +21,10 @@ module.exports = {
   },
 
   create: function (context) {
-    const sourceCode = context.getSourceCode()
-
     return {
       TaggedTemplateExpression: function (node) {
-        const { tag, quasi } = node
-
-        if (tag && tag.name === 'pug') {
-          const template = sourceCode.getText(quasi, -1, -1).trim()
+        if (isReactPugReference(node)) {
+          const template = getTemplate(node)
 
           findVariablesInTemplate(template).forEach(context.markVariableAsUsed)
         }

lib/utilities.js

@@ -0,0 +1,35 @@
+const getQuasiValue = ({ value }) => {
+  if (value && typeof value === 'object') {
+    return value.raw.trim()
+  }
+
+  return ''
+}
+
+const getInterpolatedTemplate = (template, interpolations) => template
+  .map((quasi, index) => {
+    const rawValue = getQuasiValue(quasi)
+
+    // We need to stringify interpolation so babel will understand the code
+    return interpolations[index] ? `${rawValue}` : rawValue
+  })
+  .join('')
+
+function isReactPugReference({ tag }) {
+  return tag && tag.name === 'pug'
+}
+
+function getTemplate({ quasi }) {
+  const { quasis, expressions } = quasi
+
+  if (expressions && expressions.length) {
+    return getInterpolatedTemplate(quasis, expressions)
+  }
+
+  return getQuasiValue(quasis[0])
+}
+
+module.exports = {
+  isReactPugReference,
+  getTemplate,
+}