@dxos/util 0.3.11-next.ee2b64c → 0.4.0

package/src/defer.ts

@@ -0,0 +1,54 @@
+/**
+ * Run function on scope exit.
+ *
+ * @example
+ * ```ts
+ * {
+ *   using _ = defer(() => console.log('exiting'));
+ *
+ *   ...
+ * }
+ */
+//
+// Copyright 2024 DXOS.org
+//
+
+export const defer = (fn: () => void): Disposable => new DeferGuard(fn);
+
+class DeferGuard {
+  /**
+   * @internal
+   */
+  constructor(private readonly _fn: () => void) {}
+
+  [Symbol.dispose]() {
+    const result = this._fn();
+    if ((result as any) instanceof Promise) {
+      throw new Error('Async functions in defer are not supported. Use deferAsync instead.');
+    }
+  }
+}
+
+/**
+ * Run async function on scope exit.
+ *
+ * @example
+ * ```ts
+ * {
+ *   await using _ = deferAsync(async () => console.log('exiting'));
+ *
+ *   ...
+ * }
+ */
+export const deferAsync = (fn: () => Promise<void>): AsyncDisposable => new DeferAsyncGuard(fn);
+
+class DeferAsyncGuard implements AsyncDisposable {
+  /**
+   * @internal
+   */
+  constructor(private readonly _fn: () => Promise<void>) {}
+
+  async [Symbol.asyncDispose]() {
+    await this._fn();
+  }
+}

package/src/index.ts

@@ -28,3 +28,4 @@ export * from './sum';
 export * from './for-each-async';
 export * from './weak';
 export * from './map-values';
+export * from './defer';