From 58ba81badc3510f9d2cf4f2f97f2185c549d33fb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mi=C5=A1ko=20Hevery?= <misko@hevery.com>
Date: Thu, 2 Nov 2023 13:00:07 -0700
Subject: [PATCH] fix(qwik): Improve logging of vite plugin

The log now includes file location and code snippet.
---
 .vscode/launch.json                           |  9 ++++
 .../qwik/src/optimizer/src/plugins/vite.ts    | 46 +++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/.vscode/launch.json b/.vscode/launch.json
index 03611821246..66594547d82 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -22,6 +22,15 @@
       "program": "${workspaceFolder}/packages/docs/node_modules/vite/bin/vite.js",
       "args": ["build"]
     },
+    {
+      "type": "node",
+      "name": "docs build.server",
+      "request": "launch",
+      "skipFiles": ["<node_internals>/**"],
+      "cwd": "${workspaceFolder}/packages/docs",
+      "program": "${workspaceFolder}/packages/docs/node_modules/vite/bin/vite.js",
+      "args": ["build", "-c", "adapters/cloudflare-pages/vite.config.ts"]
+    },
     {
       "type": "node",
       "name": "vscode-jest-tests",
diff --git a/packages/qwik/src/optimizer/src/plugins/vite.ts b/packages/qwik/src/optimizer/src/plugins/vite.ts
index 7334e1157de..81030f9bbcc 100644
--- a/packages/qwik/src/optimizer/src/plugins/vite.ts
+++ b/packages/qwik/src/optimizer/src/plugins/vite.ts
@@ -657,11 +657,57 @@ export function qwikVite(qwikViteOpts: QwikVitePluginOptions = {}): any {
         }
       }
     },
+
+    onLog(level, log) {
+      if (log.plugin == ('vite-plugin-qwik' satisfies QwikVitePlugin['name'])) {
+        const color = LOG_COLOR[level] || ANSI_COLOR.White;
+        const frames = (log.frame || '')
+          .split('\n')
+          .map(
+            (line) =>
+              (line.match(/^\s*\^\s*$/) ? ANSI_COLOR.BrightWhite : ANSI_COLOR.BrightBlack) + line
+          );
+        // eslint-disable-next-line no-console
+        console[level](
+          `${color}%s\n${ANSI_COLOR.BrightWhite}%s\n%s${ANSI_COLOR.RESET}`,
+          `[${log.plugin}](${level}): ${log.message}\n`,
+          `  ${log?.loc?.file}:${log?.loc?.line}:${log?.loc?.column}\n`,
+          `  ${frames.join('\n  ')}\n`
+        );
+        return false;
+      }
+    },
   };
 
   return vitePlugin;
 }
 
+const ANSI_COLOR = {
+  Black: '\x1b[30m',
+  Red: '\x1b[31m',
+  Green: '\x1b[32m',
+  Yellow: '\x1b[33m',
+  Blue: '\x1b[34m',
+  Magenta: '\x1b[35m',
+  Cyan: '\x1b[36m',
+  White: '\x1b[37m',
+  BrightBlack: '\x1b[90m',
+  BrightRed: '\x1b[91m',
+  BrightGreen: '\x1b[92m',
+  BrightYellow: '\x1b[93m',
+  BrightBlue: '\x1b[94m',
+  BrightMagenta: '\x1b[95m',
+  BrightCyan: '\x1b[96m',
+  BrightWhite: '\x1b[97m',
+  RESET: '\x1b[0m',
+};
+
+const LOG_COLOR = {
+  warn: ANSI_COLOR.Yellow,
+  info: ANSI_COLOR.Cyan,
+  debug: ANSI_COLOR.BrightBlack,
+};
+
 function updateEntryDev(code: string) {
   code = code.replace(/["']@builder.io\/qwik["']/g, `'${VITE_CLIENT_MODULE}'`);
   return code;