1 files changed, 46 insertions, 0 deletions

diff --git a/packages/main/src/utils.ts b/packages/main/src/utils.ts
new file mode 100644
index 0000000..11c78e9
--- /dev/null
+++ b/packages/main/src/utils.ts
@@ -0,0 +1,46 @@
+/*
+ * 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
+ */
+
+export type Disposable = Disposer | DisposableObject;
+
+export type Disposer = () => void;
+
+export interface DisposableObject {
+  dispose(): void;
+}
+
+export class DisposeHelper implements DisposableObject {
+  private readonly disposers: Disposer[] = [];
+
+  protected registerDisposable<T extends Disposable>(disposable: T): T {
+    if (typeof disposable === 'object') {
+      this.disposers.push(() => disposable.dispose());
+    } else {
+      this.disposers.push(disposable);
+    }
+    return disposable;
+  }
+
+  dispose(): void {
+    for (let i = this.disposers.length - 1; i >= 0; i -= 1) {
+      this.disposers[i]();
+    }
+  }
+}