2 files changed, 57 insertions, 4 deletions

diff --git a/packages/service-preload/src/index.ts b/packages/service-preload/src/index.ts
index d1ea13c..2bbfefd 100644
--- a/packages/service-preload/src/index.ts
+++ b/packages/service-preload/src/index.ts
@@ -18,8 +18,12 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
-import { ipcRenderer, webFrame } from 'electron';
 import { ServiceToMainIpcMessage, webSource } from '@sophie/service-shared';
+import { ipcRenderer, webFrame } from 'electron';
+
+import { getLogger } from './utils/log';
+
+const log = getLogger('index');
 
 if (webFrame.parent === null) {
   // Inject CSS to simulate `browserView.setBackgroundColor`.
@@ -49,14 +53,14 @@ if (webFrame.parent === null) {
  * @see https://www.electronjs.org/docs/latest/api/web-contents#contentsexecutejavascriptinisolatedworldworldid-scripts-usergesture
  */
 async function fetchAndExecuteInjectScript(): Promise<void> {
-  const apiExposedResponse = await ipcRenderer.invoke(
+  const apiExposedResponse: unknown = await ipcRenderer.invoke(
     ServiceToMainIpcMessage.ApiExposedInMainWorld,
   );
   const injectSource = webSource.parse(apiExposedResponse);
   // Isolated world 0 is the main world.
-  return webFrame.executeJavaScriptInIsolatedWorld(0, [injectSource]);
+  await webFrame.executeJavaScriptInIsolatedWorld(0, [injectSource]);
 }
 
 fetchAndExecuteInjectScript().catch((err) => {
-  console.log('Failed to fetch inject source:', err);
+  log.error('Failed to fetch inject source:', err);
 });
diff --git a/packages/service-preload/src/utils/log.ts b/packages/service-preload/src/utils/log.ts
new file mode 100644
index 0000000..0c35319
--- /dev/null
+++ b/packages/service-preload/src/utils/log.ts
@@ -0,0 +1,49 @@
+/*
+ * Copyright (C)  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 loglevel, { Logger } from 'loglevel';
+import prefix from 'loglevel-plugin-prefix';
+
+if (import.meta.env?.DEV) {
+  loglevel.setLevel('debug');
+} else {
+  loglevel.setLevel('info');
+}
+
+prefix.reg(loglevel);
+prefix.apply(loglevel, {
+  format(level, name, timestamp) {
+    const timeStr = timestamp.toString();
+    const nameStr = typeof name === 'undefined' ? '' : ` ${name}`;
+    return `[${timeStr}] ${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();
+  });
+}