From 5e8bbbbabeae8b255713efc8ea66b5f83ac86eed Mon Sep 17 00:00:00 2001
From: Favo Yang <favoyang@gmail.com>
Date: Thu, 14 May 2020 01:55:39 +0800
Subject: [PATCH] fix: make warning and error level output prefix consistent

---
 lib/core.js   |  4 ++--
 lib/logger.js | 12 +++++++++++-
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/lib/core.js b/lib/core.js
index d4981203..b0d4929c 100644
--- a/lib/core.js
+++ b/lib/core.js
@@ -183,13 +183,13 @@ const fetchPackageDependencies = async function({ name, version, deep }) {
         // handle version not exist
         if (!versions.find(x => x == entry.version)) {
           log.warn(
-            `[WARN] package ${entry.name}@${
+            `package ${entry.name}@${
               entry.version
             } is not a valid choice of ${versions.reverse().join(", ")}`
           );
           // eslint-disable-next-line require-atomic-updates
           validDep.version = entry.version = getLatestVersion(pkgInfo);
-          log.warn(`[WARN] fall back to ${entry.name}@${entry.version}`);
+          log.warn(`fall back to ${entry.name}@${entry.version}`);
         }
         // add dependencies to pending list
         if (validDep.self || deep) {
diff --git a/lib/logger.js b/lib/logger.js
index 90902291..28fc5f3d 100644
--- a/lib/logger.js
+++ b/lib/logger.js
@@ -1,14 +1,24 @@
 const log = require("loglevel");
 const prefix = require("loglevel-plugin-prefix");
 
+prefix.reg(log);
 if (process.env.NODE_ENV == "test") {
-  prefix.reg(log);
   prefix.apply(log, {
     // eslint-disable-next-line no-unused-vars
     format(level, name, timestamp) {
       return "[test]";
     }
   });
+} else {
+  prefix.apply(log, {
+    // eslint-disable-next-line no-unused-vars
+    format(level, name, timestamp) {
+      if (level == "ERROR" || level == "WARN") {
+        return `[${level}]`;
+      }
+      return "";
+    }
+  });
 }
 
 module.exports = log;