refactor: Improve logging

1 files changed, 20 insertions, 11 deletions

diff --git a/packages/main/src/utils/logging.ts b/packages/main/src/utils/logging.ts
index 5cb5d21..6c4cba2 100644
--- a/packages/main/src/utils/logging.ts
+++ b/packages/main/src/utils/logging.ts
@@ -18,28 +18,37 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
+import chalk, { ChalkInstance } from 'chalk';
 import loglevel, { Logger } from 'loglevel';
 import prefix from 'loglevel-plugin-prefix';
 
-const isDevelopment = import.meta.env.MODE === 'development';
+if (import.meta.env.DEV) {
+  loglevel.setLevel('debug');
+} else {
+  loglevel.setLevel('info');
+}
+
+const COLORS: Partial<Record<string, ChalkInstance>> = {
+  TRACE: chalk.magenta,
+  DEBUG: chalk.cyan,
+  INFO: chalk.blue,
+  WARN: chalk.yellow,
+  ERROR: chalk.red,
+  CRITICAL: chalk.red,
+};
 
-if (isDevelopment) {
-  loglevel.enableAll();
+function getColor(level: string): ChalkInstance {
+  return COLORS[level] ?? chalk.gray;
 }
 
 prefix.reg(loglevel);
 prefix.apply(loglevel, {
   format(level, name, timestamp) {
-    let shortName = 'global';
-    if (name !== undefined) {
-      const nameSegments = name.split(':');
-      const lastSegment = nameSegments.pop();
-      shortName = [...nameSegments.map((segment) => segment[0]), lastSegment].join(':');
-    }
-    return `[${timestamp}] ${level} (${shortName})`;
+    const levelColor = getColor(level);
+    return `${chalk.gray(`[${timestamp}]`)} ${levelColor(level)} ${chalk.green(`${name}:`)}`;
   },
 });
 
 export function getLogger(loggerName: string): Logger {
-  return loglevel.getLogger(`sophie:${loggerName}`);
+  return loglevel.getLogger(loggerName);
 }