From 3e151cf72226c0611514d29d4f7db59760ea63d7 Mon Sep 17 00:00:00 2001
From: fi3ework <fi3ework@gmail.com>
Date: Wed, 2 Jun 2021 01:02:39 +0800
Subject: [PATCH] feat: support VLS in dev mode

---
 .../vue2-ts/src/components/HelloWorld.vue     |  2 +-
 pnpm-lock.yaml                                | 27 ++++++------
 presets/vls/package.json                      |  7 ++--
 presets/vls/src/cli.ts                        |  4 ++
 presets/vls/src/commands/diagnostics.ts       | 41 ++++++++++++++++---
 presets/vls/src/initParams.ts                 |  2 +-
 presets/vls/src/main.ts                       | 15 +++++--
 7 files changed, 73 insertions(+), 25 deletions(-)

diff --git a/examples/vue2-ts/src/components/HelloWorld.vue b/examples/vue2-ts/src/components/HelloWorld.vue
index a3d18200..3342ca35 100644
--- a/examples/vue2-ts/src/components/HelloWorld.vue
+++ b/examples/vue2-ts/src/components/HelloWorld.vue
@@ -1,6 +1,6 @@
 <template>
   <div class="hello">
-    <h1>{{ msg1 }}</h1>
+    <h1>{{ msg12 }}</h1>
     <p>
       For a guide and recipes on how to configure / customize this project,<br />
       check out the
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index f22ed185..bbff3a20 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -140,6 +140,7 @@ importers:
       '@types/babel__code-frame': ^7.0.2
       '@types/glob': ^7.1.3
       chalk: ^4.1.1
+      chokidar: ^3.5.1
       commander: ^7.2.0
       glob: ^7.1.7
       rollup: ^2.47.0
@@ -153,6 +154,7 @@ importers:
       vscode-languageserver-types: ^3.16.0
       vscode-uri: ^3.0.2
     dependencies:
+      chokidar: 3.5.1
       commander: 7.2.0
       vls: 0.7.2
     devDependencies:
@@ -1741,6 +1743,13 @@ packages:
       picomatch: 2.2.2
     dev: true
 
+  /anymatch/3.1.2:
+    resolution: {integrity: sha512-P43ePfOAIupkguHUycrc4qJ9kz8ZiuOUijaETwX7THt0Y/GNK7v0aa8rY816xWjZ7rJdA5XdMcpVFTKMq+RvWg==}
+    engines: {node: '>= 8'}
+    dependencies:
+      normalize-path: 3.0.0
+      picomatch: 2.3.0
+
   /argparse/1.0.10:
     resolution: {integrity: sha512-o5Roy6tNG4SL/FOkCAN6RzjiakZS25RLYFrcMttJqbdd8BWrnA+fGz57iN5Pb06pvBGvl5gQ0B48dJlslXvoTg==}
     dependencies:
@@ -2352,7 +2361,6 @@ packages:
   /binary-extensions/2.2.0:
     resolution: {integrity: sha512-jDctJ/IVQbZoJykoeHbhXpOlNBqGNcwXJKJog42E5HDPUwQTSdjCHdihjj0DlnheQ7blbT6dHOafNAiS8ooQKA==}
     engines: {node: '>=8'}
-    dev: true
 
   /binary/0.3.0:
     resolution: {integrity: sha1-n2BVO8XOjDOG87VTz/R0Yq3sqnk=}
@@ -2418,7 +2426,6 @@ packages:
     engines: {node: '>=8'}
     dependencies:
       fill-range: 7.0.1
-    dev: true
 
   /browser-process-hrtime/1.0.0:
     resolution: {integrity: sha512-9o5UecI3GhkpM6DrXr69PblIuWxPKk9Y0jHBRhdocZ2y7YECBFCsHm79Pr3OyR2AvjhDkabFJaDJMYRazHgsow==}
@@ -2639,7 +2646,7 @@ packages:
     resolution: {integrity: sha512-9+s+Od+W0VJJzawDma/gvBNQqkTiqYTWLuZoyAsivsI4AaWTCzHG06/TMjsf1cYe9Cb97UCEhjz7HvnPk2p/tw==}
     engines: {node: '>= 8.10.0'}
     dependencies:
-      anymatch: 3.1.1
+      anymatch: 3.1.2
       braces: 3.0.2
       glob-parent: 5.1.2
       is-binary-path: 2.1.0
@@ -2648,7 +2655,6 @@ packages:
       readdirp: 3.5.0
     optionalDependencies:
       fsevents: 2.3.2
-    dev: true
 
   /ci-info/2.0.0:
     resolution: {integrity: sha512-5tK7EtrZ0N+OLFMthtqOj4fI2Jeb88C4CAZPu25LDVUgXJ0A3Js4PMGqrn0JU1W0Mh1/Z8wZzYPxqUrXeBboCQ==}
@@ -3717,7 +3723,6 @@ packages:
     engines: {node: '>=8'}
     dependencies:
       to-regex-range: 5.0.1
-    dev: true
 
   /filter-obj/1.1.0:
     resolution: {integrity: sha1-mzERErxsYSehbgFsbF1/GeCAXFs=}
@@ -3821,7 +3826,6 @@ packages:
     resolution: {integrity: sha512-xiqMQR4xAeHTuB9uWm+fFRcIOgKBMiOBP+eXiyT7jsgVCq1bkVygt00oASowB7EdtpOHaaPgKt812P9ab+DDKA==}
     engines: {node: ^8.16.0 || ^10.6.0 || >=11.0.0}
     os: [darwin]
-    dev: true
     optional: true
 
   /fstream/1.0.12:
@@ -4412,7 +4416,6 @@ packages:
     engines: {node: '>=8'}
     dependencies:
       binary-extensions: 2.2.0
-    dev: true
 
   /is-boolean-object/1.1.0:
     resolution: {integrity: sha512-a7Uprx8UtD+HWdyYwnD1+ExtTgqQtD2k/1yJgtXP6wnMm8byhkoTZRl+95LLThpzNZJ5aEvi46cdH+ayMFRwmA==}
@@ -4555,7 +4558,6 @@ packages:
   /is-number/7.0.0:
     resolution: {integrity: sha512-41Cifkg6e8TylSpdtTpeLVMqvSBEVzTttHvERD741+pnZ8ANv0004MRL43QKPDlK9cGvNp6NZWZUBlbGXYxxng==}
     engines: {node: '>=0.12.0'}
-    dev: true
 
   /is-obj/2.0.0:
     resolution: {integrity: sha512-drqDG3cbczxxEJRoOXcOjtdp1J/lyp1mNn0xaznRs8+muBhgQcrnbspox5X5fOw0HnMnbfDzvnEMEtqDEJEo8w==}
@@ -5786,7 +5788,6 @@ packages:
   /normalize-path/3.0.0:
     resolution: {integrity: sha512-6eZs5Ls3WtCisHWp9S2GUy8dqkpGi4BVSz3GaqiE6ezub0512ESztXUwUB6C6IKbQkY2Pnb/mD4WYojCRwcwLA==}
     engines: {node: '>=0.10.0'}
-    dev: true
 
   /normalize-url/3.3.0:
     resolution: {integrity: sha512-U+JJi7duF1o+u2pynbp2zXDW2/PADgC30f0GsHZtRh+HOcXHnw137TrNlyxxRvWW5fjKd3bcLHPxofWuCjaeZg==}
@@ -6120,6 +6121,10 @@ packages:
     engines: {node: '>=8.6'}
     dev: true
 
+  /picomatch/2.3.0:
+    resolution: {integrity: sha512-lY1Q/PiJGC2zOv/z391WOTD+Z02bCgsFfvxoXXf6h7kv9o+WmsmzYqrAwY63sNgOxE4xEdq0WyUnXfKeBrSvYw==}
+    engines: {node: '>=8.6'}
+
   /pify/2.3.0:
     resolution: {integrity: sha1-7RQaasBDqEnqWISY59yosVMw6Qw=}
     engines: {node: '>=0.10.0'}
@@ -6514,8 +6519,7 @@ packages:
     resolution: {integrity: sha512-cMhu7c/8rdhkHXWsY+osBhfSy0JikwpHK/5+imo+LpeasTF8ouErHrlYkwT0++njiyuDvc7OFY5T3ukvZ8qmFQ==}
     engines: {node: '>=8.10.0'}
     dependencies:
-      picomatch: 2.2.2
-    dev: true
+      picomatch: 2.3.0
 
   /rechoir/0.6.2:
     resolution: {integrity: sha1-hSBLVNuoLVdC4oyWdW70OvUOM4Q=}
@@ -7443,7 +7447,6 @@ packages:
     engines: {node: '>=8.0'}
     dependencies:
       is-number: 7.0.0
-    dev: true
 
   /to-regex/3.0.2:
     resolution: {integrity: sha512-FWtleNAtZ/Ki2qtqej2CXTOayOH9bHDQF+Q48VpWyDXjbYxA4Yz8iDB31zXOBUlOHHKidDbqGVrTUvQMPmBGBw==}
diff --git a/presets/vls/package.json b/presets/vls/package.json
index 3f3d42ac..af07e787 100644
--- a/presets/vls/package.json
+++ b/presets/vls/package.json
@@ -30,6 +30,7 @@
   "bugs": "https://github.com/fi3ework/vite-plugin-ts-checker/issues",
   "homepage": "https://github.com/fi3ework/vite-plugin-ts-checker",
   "dependencies": {
+    "chokidar": "^3.5.1",
     "commander": "^7.2.0",
     "vls": "^0.7.2"
   },
@@ -38,15 +39,15 @@
     "vite-plugin-ts-checker": "^0.1.3"
   },
   "devDependencies": {
-    "typescript": "^4.2.2",
-    "vite-plugin-ts-checker": "^0.1.3",
-    "vite": "^2.1.3",
     "@babel/code-frame": "^7.12.13",
     "@types/babel__code-frame": "^7.0.2",
     "@types/glob": "^7.1.3",
     "chalk": "^4.1.1",
     "glob": "^7.1.7",
     "rollup": "^2.47.0",
+    "typescript": "^4.2.2",
+    "vite": "^2.1.3",
+    "vite-plugin-ts-checker": "^0.1.3",
     "vscode-languageclient": "^7.0.0",
     "vscode-languageserver": "^7.0.0",
     "vscode-languageserver-protocol": "^3.16.0",
diff --git a/presets/vls/src/cli.ts b/presets/vls/src/cli.ts
index 16623b2f..4dfb2efe 100644
--- a/presets/vls/src/cli.ts
+++ b/presets/vls/src/cli.ts
@@ -1,7 +1,9 @@
 import { Command, Option } from 'commander'
+
 import { diagnostics, LogLevel, logLevels } from './commands/diagnostics'
 
 function getVersion(): string {
+  // eslint-disable-next-line @typescript-eslint/no-require-imports
   const { version }: { version: string } = require('../package.json')
   return `v${version}`
 }
@@ -28,9 +30,11 @@ function validateLogLevel(logLevelInput: unknown): logLevelInput is LogLevel {
     )
     .action(async (workspace, options) => {
       const logLevelOption: unknown = options.logLevel
+
       if (!validateLogLevel(logLevelOption)) {
         throw new Error(`Invalid log level: ${logLevelOption}`)
       }
+
       await diagnostics(workspace, logLevelOption)
     })
 
diff --git a/presets/vls/src/commands/diagnostics.ts b/presets/vls/src/commands/diagnostics.ts
index 69fd6381..862dd62d 100644
--- a/presets/vls/src/commands/diagnostics.ts
+++ b/presets/vls/src/commands/diagnostics.ts
@@ -1,4 +1,5 @@
 import chalk from 'chalk'
+import chokidar from 'chokidar'
 import fs from 'fs'
 import glob from 'glob'
 import path from 'path'
@@ -11,6 +12,7 @@ import {
   Diagnostic,
   DiagnosticSeverity,
   DidOpenTextDocumentNotification,
+  DidChangeTextDocumentNotification,
   InitializeParams,
   InitializeRequest,
   InitializeResult,
@@ -34,7 +36,7 @@ const logLevel2Severity = {
   HINT: DiagnosticSeverity.Hint,
 }
 
-export async function diagnostics(workspace: string | null, logLevel: LogLevel) {
+export async function diagnostics(workspace: string | null, logLevel: LogLevel, watch = false) {
   console.log('====================================')
   console.log('Getting Vetur diagnostics')
   let workspaceUri
@@ -48,15 +50,19 @@ export async function diagnostics(workspace: string | null, logLevel: LogLevel)
     workspaceUri = URI.file(process.cwd())
   }
 
-  const errCount = await getDiagnostics(workspaceUri, logLevel2Severity[logLevel])
+  const errCount = await getDiagnostics(workspaceUri, logLevel2Severity[logLevel], watch)
   console.log('====================================')
 
   if (errCount === 0) {
     console.log(chalk.green(`VTI found no error`))
-    process.exit(0)
+    if (!watch) {
+      process.exit(0)
+    }
   } else {
     console.log(chalk.red(`VTI found ${errCount} ${errCount === 1 ? 'error' : 'errors'}`))
-    process.exit(1)
+    if (!watch) {
+      process.exit(1)
+    }
   }
 }
 
@@ -91,6 +97,13 @@ async function prepareClientConnection(workspaceUri: URI) {
     new StreamMessageReader(up),
     new StreamMessageWriter(down)
   )
+
+  // NOTE: hijack sendDiagnostics
+  serverConnection.sendDiagnostics = (diagnostic) => {
+    if (!diagnostic.diagnostics.length) return
+    console.log(chalk.red(JSON.stringify(diagnostic, null, 2)))
+  }
+
   const vls = new VLS(serverConnection as any)
 
   serverConnection.onInitialize(async (params: InitializeParams): Promise<InitializeResult> => {
@@ -127,7 +140,7 @@ function range2Location(range: Range): SourceLocation {
   }
 }
 
-async function getDiagnostics(workspaceUri: URI, severity: DiagnosticSeverity) {
+async function getDiagnostics(workspaceUri: URI, severity: DiagnosticSeverity, watch: boolean) {
   const clientConnection = await prepareClientConnection(workspaceUri)
 
   const files = glob.sync('**/*.vue', { cwd: workspaceUri.fsPath, ignore: ['node_modules/**'] })
@@ -144,6 +157,24 @@ async function getDiagnostics(workspaceUri: URI, severity: DiagnosticSeverity) {
 
   let errCount = 0
 
+  if (watch) {
+    chokidar
+      .watch(workspaceUri.fsPath, {
+        ignored: (path: string) => path.includes('node_modules'),
+        ignoreInitial: true,
+      })
+      .on('all', (event, path) => {
+        if (!path.endsWith('.vue')) return
+        clientConnection.sendNotification(DidChangeTextDocumentNotification.type, {
+          textDocument: {
+            uri: URI.file(path).toString(),
+            version: Date.now(),
+          },
+          contentChanges: [{ text: fs.readFileSync(path, 'utf-8') }],
+        })
+      })
+  }
+
   for (const absFilePath of absFilePaths) {
     const fileText = fs.readFileSync(absFilePath, 'utf-8')
     clientConnection.sendNotification(DidOpenTextDocumentNotification.type, {
diff --git a/presets/vls/src/initParams.ts b/presets/vls/src/initParams.ts
index 00ef7ae9..8cbe2ec3 100644
--- a/presets/vls/src/initParams.ts
+++ b/presets/vls/src/initParams.ts
@@ -29,7 +29,7 @@ export function getInitParams(workspaceUri: URI): InitializeParams {
   return init
 }
 
-function getDefaultVLSConfig() {
+export function getDefaultVLSConfig() {
   return {
     vetur: {
       useWorkspaceDependencies: false,
diff --git a/presets/vls/src/main.ts b/presets/vls/src/main.ts
index df4afd1d..63054ba2 100644
--- a/presets/vls/src/main.ts
+++ b/presets/vls/src/main.ts
@@ -1,13 +1,22 @@
 import type { UserConfig, ViteDevServer } from 'vite'
 import { CheckerFactory, CreateDiagnostic, PluginOptions } from 'vite-plugin-ts-checker'
+import { diagnostics, logLevels } from './commands/diagnostics'
 
 export const createDiagnostic: CreateDiagnostic = (userOptions = {}) => {
+  let overlay = true // Vite defaults to true
+
   return {
     config: (config: UserConfig) => {
-      // TODO:
+      const hmr = config.server?.hmr
+      const viteOverlay = !(typeof hmr === 'object' && hmr.overlay === false)
+
+      if (userOptions.overlay === false || !viteOverlay) {
+        overlay = false
+      }
     },
-    configureServer(server: ViteDevServer) {
-      // TODO:
+    async configureServer(server: ViteDevServer) {
+      const workDir: string = userOptions.root ?? server.config.root
+      await diagnostics(workDir, 'WARN', true)
     },
   }
 }