1 files changed, 66 insertions, 0 deletions

diff --git a/packages/main/src/utils/log.ts b/packages/main/src/utils/log.ts
new file mode 100644
index 0000000..c704797
--- /dev/null
+++ b/packages/main/src/utils/log.ts
@@ -0,0 +1,66 @@
+/*
+ * Copyright (C)  2021-2022 Kristóf Marussy <kristof@marussy.com>
+ *
+ * This file is part of Sophie.
+ *
+ * Sophie is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, version 3.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import chalk, { ChalkInstance } from 'chalk';
+import loglevel, { Logger } from 'loglevel';
+import prefix from 'loglevel-plugin-prefix';
+
+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,
+};
+
+function getColor(level: string): ChalkInstance {
+  return COLORS[level] ?? chalk.gray;
+}
+
+prefix.reg(loglevel);
+prefix.apply(loglevel, {
+  format(level, name, timestamp) {
+    const levelColor = getColor(level);
+    const timeStr = timestamp.toString();
+    const nameStr = typeof name === 'undefined'
+      ? levelColor(':')
+      : ` ${chalk.green(`${name}:`)}`;
+    return `${chalk.gray(`[${timeStr}]`)} ${levelColor(level)}${nameStr}`;
+  },
+});
+
+export function getLogger(loggerName: string): Logger {
+  return loglevel.getLogger(loggerName);
+}
+
+export function silenceLogger(): void {
+  loglevel.disableAll();
+  const loggers = loglevel.getLoggers();
+  Object.keys(loggers).forEach((loggerName) => {
+    loggers[loggerName].disableAll();
+  });
+}