feat(createLogger): allow custom prefix for logger (vitejs#2019)

viclm · Apr 17, 2021 · 344d77e · 344d77e
1 parent da11d43
commit 344d77e
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 6 deletions.
diff --git a/packages/vite/src/node/config.ts b/packages/vite/src/node/config.ts
@@ -239,7 +239,9 @@ export async function resolveConfig(
   }
 
   // Define logger
-  const logger = createLogger(config.logLevel, config.clearScreen)
+  const logger = createLogger(config.logLevel, {
+    allowClearScreen: config.clearScreen
+  })
 
   // user config may provide an alternative mode
   mode = config.mode || mode

diff --git a/packages/vite/src/node/index.ts b/packages/vite/src/node/index.ts
@@ -26,7 +26,13 @@ export type {
   DepOptimizationOptions
 } from './optimizer'
 export type { Plugin } from './plugin'
-export type { Logger, LogOptions, LogLevel, LogType } from './logger'
+export type {
+  Logger,
+  LogOptions,
+  LogLevel,
+  LogType,
+  LoggerOptions
+} from './logger'
 export type {
   AliasOptions,
   ResolverFunction,

diff --git a/packages/vite/src/node/logger.ts b/packages/vite/src/node/logger.ts
@@ -37,10 +37,17 @@ function clearScreen() {
   readline.clearScreenDown(process.stdout)
 }
 
+export interface LoggerOptions {
+  prefix?: string
+  allowClearScreen?: boolean
+}
+
 export function createLogger(
   level: LogLevel = 'info',
-  allowClearScreen = true
+  options: LoggerOptions = {}
 ): Logger {
+  const { prefix = '[vite]', allowClearScreen = true } = options
+
   const thresh = LogLevels[level]
   const clear =
     allowClearScreen && process.stdout.isTTY && !process.env.CI
@@ -54,10 +61,10 @@ export function createLogger(
         if (options.timestamp) {
           const tag =
             type === 'info'
-              ? chalk.cyan.bold(`[vite]`)
+              ? chalk.cyan.bold(prefix)
               : type === 'warn'
-              ? chalk.yellow.bold(`[vite]`)
-              : chalk.red.bold(`[vite]`)
+              ? chalk.yellow.bold(prefix)
+              : chalk.red.bold(prefix)
           return `${chalk.dim(new Date().toLocaleTimeString())} ${tag} ${msg}`
         } else {
           return msg