1 files changed, 44 insertions, 61 deletions

diff --git a/packages/main/src/controllers/ConfigController.ts b/packages/main/src/controllers/ConfigController.ts
index 6690548..a28746c 100644
--- a/packages/main/src/controllers/ConfigController.ts
+++ b/packages/main/src/controllers/ConfigController.ts
@@ -25,56 +25,62 @@ import {
   IDisposer,
   onSnapshot,
 } from 'mobx-state-tree';
-import ms from 'ms';
 
-import { ConfigPersistenceService } from '../services/ConfigPersistenceService';
-import { Config, ConfigSnapshotOut } from '../stores/Config';
+import type { ConfigPersistenceService } from '../services/ConfigPersistenceService';
+import type { Config, ConfigSnapshotOut } from '../stores/Config';
 
-const DEFAULT_DEBOUNCE_TIME = ms('1s');
+export class ConfigController {
+  static inject = ['configPersistenceService', 'configDebounceTime'] as const;
 
-class ConfigController {
-  readonly #config: Config;
+  private config: Config | null = null;
 
-  readonly #persistenceService: ConfigPersistenceService;
+  private onSnapshotDisposer: IDisposer | null = null;
 
-  readonly #onSnapshotDisposer: IDisposer;
+  private lastSnapshotOnDisk: ConfigSnapshotOut | null = null;
 
-  readonly #watcherDisposer: IDisposer;
+  private writingConfig: boolean = false;
 
-  #lastSnapshotOnDisk: ConfigSnapshotOut | null = null;
+  constructor(
+    private readonly persistenceService: ConfigPersistenceService,
+    private readonly debounceTime: number,
+  ) {
+  }
 
-  #writingConfig: boolean = false;
+  async initConfig(config: Config): Promise<void> {
+    this.config = config;
 
-  #configMTime: Date | null = null;
+    const foundConfig: boolean = await this.readConfig();
+    if (!foundConfig) {
+      console.log('Creating new config file');
+      try {
+        await this.writeConfig();
+      } catch (err) {
+        console.error('Failed to initialize config');
+      }
+    }
 
-  constructor(
-    config: Config,
-    persistenceService: ConfigPersistenceService,
-    debounceTime: number,
-  ) {
-    this.#config = config;
-    this.#persistenceService = persistenceService;
-    this.#onSnapshotDisposer = onSnapshot(this.#config, debounce((snapshot) => {
+    this.onSnapshotDisposer = onSnapshot(this.config, debounce((snapshot) => {
       // We can compare snapshots by reference, since it is only recreated on store changes.
-      if (this.#lastSnapshotOnDisk !== snapshot) {
-        this.#writeConfig().catch((err) => {
+      if (this.lastSnapshotOnDisk !== snapshot) {
+        this.writeConfig().catch((err) => {
           console.log('Failed to write config on config change', err);
         })
       }
-    }, debounceTime));
-    this.#watcherDisposer = this.#persistenceService.watchConfig(async (mtime) => {
-      if (!this.#writingConfig && (this.#configMTime === null || mtime > this.#configMTime)) {
-        await this.#readConfig();
+    }, this.debounceTime));
+
+    this.persistenceService.watchConfig(async () => {
+      if (!this.writingConfig) {
+        await this.readConfig();
       }
-    }, debounceTime);
+    }, this.debounceTime);
   }
 
-  async #readConfig(): Promise<boolean> {
-    const result = await this.#persistenceService.readConfig();
+  private async readConfig(): Promise<boolean> {
+    const result = await this.persistenceService.readConfig();
     if (result.found) {
       try {
-        applySnapshot(this.#config, result.data);
-        this.#lastSnapshotOnDisk = getSnapshot(this.#config);
+        applySnapshot(this.config!, result.data);
+        this.lastSnapshotOnDisk = getSnapshot(this.config!);
         console.log('Loaded config');
       } catch (err) {
         console.error('Failed to read config', result.data, err);
@@ -83,42 +89,19 @@ class ConfigController {
     return result.found;
   }
 
-  async #writeConfig(): Promise<void> {
-    const snapshot = getSnapshot(this.#config);
-    this.#writingConfig = true;
+  private async writeConfig(): Promise<void> {
+    const snapshot = getSnapshot(this.config!);
+    this.writingConfig = true;
     try {
-      this.#configMTime = await this.#persistenceService.writeConfig(snapshot);
-      this.#lastSnapshotOnDisk = snapshot;
+      await this.persistenceService.writeConfig(snapshot);
+      this.lastSnapshotOnDisk = snapshot;
       console.log('Wrote config');
     } finally {
-      this.#writingConfig = false;
-    }
-  }
-
-  async initConfig(): Promise<void> {
-    const foundConfig: boolean = await this.#readConfig();
-    if (!foundConfig) {
-      console.log('Creating new config file');
-      try {
-        await this.#writeConfig();
-      } catch (err) {
-        console.error('Failed to initialize config');
-      }
+      this.writingConfig = false;
     }
   }
 
   dispose(): void {
-    this.#onSnapshotDisposer();
-    this.#watcherDisposer();
+    this.onSnapshotDisposer?.();
   }
 }
-
-export async function initConfig(
-  config: Config,
-  persistenceService: ConfigPersistenceService,
-  debounceTime: number = DEFAULT_DEBOUNCE_TIME,
-): Promise<IDisposer> {
-  const controller = new ConfigController(config, persistenceService, debounceTime);
-  await controller.initConfig();
-  return () => controller.dispose();
-}