feat: log qrcode in preview server (#112)

svitejs · Jul 10, 2023 · 2fc6cb5 · 2fc6cb5
1 parent 4cc7d72
commit 2fc6cb5
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/.changeset/lazy-laws-exercise.md b/.changeset/lazy-laws-exercise.md
@@ -0,0 +1,5 @@
+---
+'vite-plugin-qrcode': minor
+---
+
+Log qrcode in preview server for Vite 4.3.0 and above
diff --git a/packages/playground/basic/package.json b/packages/playground/basic/package.json
@@ -6,7 +6,7 @@
   "scripts": {
     "dev": "vite",
     "build": "vite build",
-    "serve": "vite preview"
+    "preview": "vite preview"
   },
   "devDependencies": {
     "vite": "^4.3.9",

diff --git a/packages/vite-plugin-qrcode/src/index.ts b/packages/vite-plugin-qrcode/src/index.ts
@@ -1,4 +1,4 @@
-import type { Plugin, ViteDevServer } from 'vite';
+import type { Plugin, PreviewServerForHook, ViteDevServer } from 'vite';
 import qr from 'qrcode-terminal';
 
 export function qrcode(options: PluginOptions = {}): Plugin {
@@ -19,11 +19,20 @@ export function qrcode(options: PluginOptions = {}): Plugin {
 				// eslint-disable-next-line prefer-rest-params
 				return _listen.apply(this, arguments);
 			};
+		},
+		configurePreviewServer(server) {
+			// Preview server has no restarts, so we can hook directly
+			// The `resolvedUrls` only exist in Vite >=4.3.0, so add a guard to prevent unnecessary hook
+			if ('resolvedUrls' in server) {
+				server.httpServer?.on('listening', () => {
+					setTimeout(() => logQrcode(server, options), 0);
+				});
+			}
 		}
 	};
 }
 
-function logQrcode(server: ViteDevServer, options: PluginOptions) {
+function logQrcode(server: ViteDevServer | PreviewServerForHook, options: PluginOptions) {
 	let networkUrls = server.resolvedUrls?.network;
 
 	if (!networkUrls) return;