New: services.getTemplateBodyTokenStore

Author: mysticatea

lib/get-token-store.js

@@ -0,0 +1,42 @@
+/**
+ * @author Toru Nagashima <https://github.com/mysticatea>
+ * @copyright 2017 Toru Nagashima. All rights reserved.
+ * See LICENSE file in root directory for full license.
+ */
+"use strict"
+
+//------------------------------------------------------------------------------
+// Requirements
+//------------------------------------------------------------------------------
+
+const TokenStore = require("eslint/lib/token-store")
+
+//------------------------------------------------------------------------------
+// Helpers
+//------------------------------------------------------------------------------
+
+const stores = new WeakMap()
+
+/**
+ * Get the token store of the template body.
+ * @param {RuleContext} context The rule context to get.
+ * @returns {TokenStore} The token store of template body.
+ */
+function getTokenStore(context) {
+    const ast = context.getSourceCode().ast.templateBody
+    if (stores.has(ast)) {
+        return stores.get(ast)
+    }
+    const store = (ast != null)
+        ? new TokenStore(ast.tokens, ast.comments)
+        : new TokenStore([], [])
+
+    stores.set(ast, store)
+    return store
+}
+
+//------------------------------------------------------------------------------
+// Exports
+//------------------------------------------------------------------------------
+
+module.exports = getTokenStore

lib/parse-component.js

@@ -11,6 +11,7 @@
 
 const path = require("path")
 const html = require("parse5")
+const getTemplateBodyTokenStore = require("./get-token-store")
 const registerTemplateBodyVisitor = require("./register-template-body-visitor")
 const ScriptParser = require("./script-parser")
 const TokenGenerator = require("./token-generator")
@@ -173,7 +174,10 @@ function parse(code, options) {
 
     return {
         ast,
-        services: {registerTemplateBodyVisitor},
+        services: {
+            getTemplateBodyTokenStore,
+            registerTemplateBodyVisitor,
+        },
     }
 }
 

test/fixtures/template-ast/attributes.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -1883,6 +1884,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/comments.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -961,6 +962,5 @@
                 "value": " comment 3 "
             }
         ]
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/directive-argument-and-modifiers.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -847,6 +848,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/directive-arguments.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -1269,6 +1270,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/directive-modifiers.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -1338,6 +1339,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/directive-shorthands.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -846,6 +847,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/directives.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -2569,6 +2570,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }

test/fixtures/template-ast/elements.ast.json

@@ -18,6 +18,7 @@
     ],
     "body": [],
     "sourceType": "script",
+    "comments": [],
     "tokens": [],
     "templateBody": {
         "range": [
@@ -815,6 +816,5 @@
             }
         ],
         "comments": []
-    },
-    "comments": []
+    }
 }