From b4c3f9429f8810be6a86505d300b6a63cc0bf70d Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Sun, 18 Jul 2021 15:14:17 +0900
Subject: [PATCH 1/6] Add defineDocumentVisitor to parserServices

---
 README.md              |  1 +
 src/parser-services.ts | 70 +++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 70 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 73831c41..6e0d5992 100644
--- a/README.md
+++ b/README.md
@@ -203,6 +203,7 @@ See also to [here](https://github.com/vuejs/rfcs/blob/master/active-rfcs/0043-sf
     - `getTemplateBodyTokenStore()` ... returns ESLint `TokenStore` to get the tokens of `<template>`.
     - `getDocumentFragment()` ... returns the root `VDocumentFragment`.
     - `defineCustomBlocksVisitor(context, customParser, rule, scriptVisitor)` ... returns ESLint visitor that parses and traverses the contents of the custom block.
+    - `defineDocumentVisitor(documentVisitor, options)` ... returns ESLint visitor to traverses the document.
 - [ast.md](./docs/ast.md) is `<template>` AST specification.
 - [mustache-interpolation-spacing.js](https://github.com/vuejs/eslint-plugin-vue/blob/b434ff99d37f35570fa351681e43ba2cf5746db3/lib/rules/mustache-interpolation-spacing.js) is an example.
 
diff --git a/src/parser-services.ts b/src/parser-services.ts
index 845201c7..a1fda2e8 100644
--- a/src/parser-services.ts
+++ b/src/parser-services.ts
@@ -26,7 +26,7 @@ import {
 } from "./sfc/custom-block"
 import type { ParserOptions } from "./common/parser-options"
 import { isSFCFile } from "./common/parser-options"
-import { getLang } from "./common/ast-utils"
+import { getLang, isStyleElement } from "./common/ast-utils"
 
 //------------------------------------------------------------------------------
 // Helpers
@@ -56,6 +56,16 @@ export interface ParserServices {
         options?: { templateBodyTriggerSelector: "Program" | "Program:exit" },
     ): object
 
+    /**
+     * Define handlers to traverse the document.
+     * @param documentVisitor The document handlers.
+     * @param options The options. This is optional.
+     */
+    defineDocumentVisitor(
+        documentVisitor: { [key: string]: (...args: any) => void },
+        options?: { triggerSelector: "Program" | "Program:exit" },
+    ): object
+
     /**
      * Define handlers to traverse custom blocks.
      * @param context The rule context.
@@ -103,6 +113,8 @@ export function define(
     const templateBodyEmitters = new Map<string, EventEmitter>()
     const stores = new WeakMap<object, TokenStore>()
 
+    const documentEmitters = new Map<string, EventEmitter>()
+
     const customBlocksEmitters = new Map<
         | ESLintCustomBlockParser["parseForESLint"]
         | ESLintCustomBlockParser["parse"],
@@ -180,6 +192,62 @@ export function define(
             return scriptVisitor
         },
 
+        /**
+         * Define handlers to traverse the document.
+         * @param documentVisitor The document handlers.
+         * @param options The options. This is optional.
+         */
+        defineDocumentVisitor(
+            documentVisitor: { [key: string]: (...args: any) => void },
+            options?: { triggerSelector: "Program" | "Program:exit" },
+        ): object {
+            const scriptVisitor: { [key: string]: (...args: any) => void } = {}
+            if (!document) {
+                return scriptVisitor
+            }
+
+            const documentTriggerSelector =
+                options?.triggerSelector ?? "Program:exit"
+
+            let emitter = documentEmitters.get(documentTriggerSelector)
+
+            // If this is the first time, initialize the intermediate event emitter.
+            if (emitter == null) {
+                emitter = new EventEmitter()
+                emitter.setMaxListeners(0)
+                documentEmitters.set(documentTriggerSelector, emitter)
+
+                const programExitHandler =
+                    scriptVisitor[documentTriggerSelector]
+                scriptVisitor[documentTriggerSelector] = (node) => {
+                    try {
+                        if (typeof programExitHandler === "function") {
+                            programExitHandler(node)
+                        }
+
+                        // Traverse document.
+                        const generator = new NodeEventGenerator(emitter!, {
+                            visitorKeys: KEYS,
+                            fallback: getFallbackKeys,
+                        })
+                        traverseNodes(document, generator)
+                    } finally {
+                        // eslint-disable-next-line @mysticatea/ts/ban-ts-ignore
+                        // @ts-ignore
+                        scriptVisitor[styleTriggerSelector] = programExitHandler
+                        documentEmitters.delete(documentTriggerSelector)
+                    }
+                }
+            }
+
+            // Register handlers into the intermediate event emitter.
+            for (const selector of Object.keys(documentVisitor)) {
+                emitter.on(selector, documentVisitor[selector])
+            }
+
+            return scriptVisitor
+        },
+
         /**
          * Define handlers to traverse custom blocks.
          * @param context The rule context.

From d7c85b3e390f4d2a16afa660f8477cbb54640af4 Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Sun, 18 Jul 2021 15:20:47 +0900
Subject: [PATCH 2/6] fix

---
 src/parser-services.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/parser-services.ts b/src/parser-services.ts
index a1fda2e8..3cb447a4 100644
--- a/src/parser-services.ts
+++ b/src/parser-services.ts
@@ -26,7 +26,7 @@ import {
 } from "./sfc/custom-block"
 import type { ParserOptions } from "./common/parser-options"
 import { isSFCFile } from "./common/parser-options"
-import { getLang, isStyleElement } from "./common/ast-utils"
+import { getLang } from "./common/ast-utils"
 
 //------------------------------------------------------------------------------
 // Helpers

From 2f1f4a7809e3ba891535947276187bcf3fe680d5 Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Sun, 18 Jul 2021 15:48:04 +0900
Subject: [PATCH 3/6] fix

---
 src/parser-services.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/parser-services.ts b/src/parser-services.ts
index 3cb447a4..d3e832ef 100644
--- a/src/parser-services.ts
+++ b/src/parser-services.ts
@@ -234,7 +234,8 @@ export function define(
                     } finally {
                         // eslint-disable-next-line @mysticatea/ts/ban-ts-ignore
                         // @ts-ignore
-                        scriptVisitor[styleTriggerSelector] = programExitHandler
+                        scriptVisitor[documentTriggerSelector] =
+                            programExitHandler
                         documentEmitters.delete(documentTriggerSelector)
                     }
                 }

From 3d06ceeb473f6bf8339e1c5f6ebb28f47462b338 Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Sun, 18 Jul 2021 17:34:25 +0900
Subject: [PATCH 4/6] Changed getTemplateBodyTokenStore to work even if
 `<template>` is missing

---
 src/parser-services.ts | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/parser-services.ts b/src/parser-services.ts
index d3e832ef..192adf11 100644
--- a/src/parser-services.ts
+++ b/src/parser-services.ts
@@ -400,14 +400,13 @@ export function define(
          * @returns The token store of template body.
          */
         getTemplateBodyTokenStore(): TokenStore {
-            const ast = rootAST.templateBody
-            const key = ast || stores
+            const key = document || stores
             let store = stores.get(key)
 
             if (!store) {
                 store =
-                    ast != null
-                        ? new TokenStore(ast.tokens, ast.comments)
+                    document != null
+                        ? new TokenStore(document.tokens, document.comments)
                         : new TokenStore([], [])
                 stores.set(key, store)
             }

From 439ae87cc57cb04e6aca966134dafc0cb0cc43bd Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Tue, 27 Jul 2021 14:43:18 +0900
Subject: [PATCH 5/6] Add testcase

---
 test/define-document-visitor.js | 92 +++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 test/define-document-visitor.js

diff --git a/test/define-document-visitor.js b/test/define-document-visitor.js
new file mode 100644
index 00000000..53281d81
--- /dev/null
+++ b/test/define-document-visitor.js
@@ -0,0 +1,92 @@
+/**
+ * @author Yosuke Ota <https://github.com/ota-meshi>
+ */
+"use strict"
+
+//------------------------------------------------------------------------------
+// Requirements
+//------------------------------------------------------------------------------
+
+const assert = require("assert")
+const path = require("path")
+const eslint = require("eslint")
+const Linter = eslint.Linter
+
+//------------------------------------------------------------------------------
+// Helpers
+//------------------------------------------------------------------------------
+
+const PARSER_PATH = path.resolve(__dirname, "../src/index.ts")
+
+//------------------------------------------------------------------------------
+// Tests
+//------------------------------------------------------------------------------
+
+describe("parserServices.defineDocumentVisitor tests", () => {
+    it("should be able to visit the document using defineDocumentVisitor.", () => {
+        const code = `
+<template>
+{{forbidden}}
+{{foo()}}
+{{ok}}
+</template>
+<style>
+.ng {
+    font: v-bind(forbidden)
+}
+.call {
+    font: v-bind(foo())
+}
+.ok {
+    font: v-bind(ok)
+}
+</style>`
+
+        const linter = new Linter()
+
+        linter.defineParser(PARSER_PATH, require(PARSER_PATH))
+        linter.defineRule("test-no-forbidden", {
+            create(context) {
+                return context.parserServices.defineDocumentVisitor({
+                    'Identifier[name="forbidden"]'(node) {
+                        context.report({
+                            node,
+                            message: 'no "forbidden"',
+                        })
+                    },
+                })
+            },
+        })
+        linter.defineRule("test-no-call", {
+            create(context) {
+                return context.parserServices.defineDocumentVisitor({
+                    CallExpression(node) {
+                        context.report({
+                            node,
+                            message: "no call",
+                        })
+                    },
+                })
+            },
+        })
+        const messages = linter.verify(code, {
+            parser: PARSER_PATH,
+            parserOptions: {
+                ecmaVersion: 2018,
+            },
+            rules: {
+                "test-no-forbidden": "error",
+                "test-no-call": "error",
+            },
+        })
+        assert.strictEqual(messages.length, 4)
+        assert.strictEqual(messages[0].message, 'no "forbidden"')
+        assert.strictEqual(messages[0].line, 3)
+        assert.strictEqual(messages[1].message, "no call")
+        assert.strictEqual(messages[1].line, 4)
+        assert.strictEqual(messages[2].message, 'no "forbidden"')
+        assert.strictEqual(messages[2].line, 9)
+        assert.strictEqual(messages[3].message, "no call")
+        assert.strictEqual(messages[3].line, 12)
+    })
+})

From 7a69ddc8355193e61dd5afcace655b3b4271a318 Mon Sep 17 00:00:00 2001
From: yosuke ota <otameshiyo23@gmail.com>
Date: Tue, 27 Jul 2021 15:16:20 +0900
Subject: [PATCH 6/6] fix

---
 test/define-document-visitor.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/define-document-visitor.js b/test/define-document-visitor.js
index 53281d81..64c80ea2 100644
--- a/test/define-document-visitor.js
+++ b/test/define-document-visitor.js
@@ -35,7 +35,7 @@ describe("parserServices.defineDocumentVisitor tests", () => {
     font: v-bind(forbidden)
 }
 .call {
-    font: v-bind(foo())
+    font: v-bind('foo()')
 }
 .ok {
     font: v-bind(ok)