@datatruck/cli 0.34.3 → 0.34.5

package/lib/utils/temp.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GargabeCollector = exports.CleanupListener = exports.useTempFile = exports.useTempDir = exports.mkTmpDir = exports.tmpDir = exports.rmTmpDir = exports.isTmpDir = exports.ensureFreeDiskTempSpace = exports.sessionTmpDir = exports.parentTmpDir = void 0;
+exports.GargabeCollector = exports.useTempFile = exports.useTempDir = exports.mkTmpDir = exports.collectors = exports.tmpDir = exports.rmTmpDir = exports.isTmpDir = exports.ensureFreeDiskTempSpace = exports.sessionTmpDir = exports.parentTmpDir = void 0;
 const globalData_1 = __importDefault(require("../globalData"));
 const fs_1 = require("./fs");
 const crypto_1 = require("crypto");
@@ -44,12 +44,15 @@ exports.rmTmpDir = rmTmpDir;
 function tmpDir(...keys) {
     const id = (0, crypto_1.randomUUID)().slice(0, 8);
     const path = (0, path_1.join)(sessionTmpDir(), [...keys, id].map(encodeURIComponent).join("-"));
-    for (const listener of listeners)
-        listener.paths.push(path);
+    if (exports.collectors.size) {
+        const lastListener = [...exports.collectors.values()].at(exports.collectors.size - 1);
+        if (lastListener)
+            lastListener.paths.add(path);
+    }
     return path;
 }
 exports.tmpDir = tmpDir;
-const listeners = new Set();
+exports.collectors = new Set();
 async function mkTmpDir(...keys) {
     const path = tmpDir(...keys);
     await (0, promises_1.mkdir)(path, { recursive: true });
@@ -81,53 +84,57 @@ function useTempFile(path) {
     };
 }
 exports.useTempFile = useTempFile;
-class CleanupListener {
-    paths = [];
-    stop() {
-        listeners.delete(this);
-    }
-    async dispose() {
-        this.stop();
-        await rmTmpDir(this.paths);
-    }
-}
-exports.CleanupListener = CleanupListener;
 class GargabeCollector {
-    listeners = new Set();
-    get pending() {
-        return this.listeners.size > 0;
+    parent;
+    paths = new Set();
+    children = new Set();
+    constructor(parent) {
+        this.parent = parent;
+        exports.collectors.add(this);
     }
-    async cleanup(cb) {
-        try {
-            await cb?.();
-        }
-        finally {
-            for (const listener of this.listeners) {
-                this.listeners.delete(listener);
-                await listener.dispose();
+    pending() {
+        if (this.paths.size)
+            return true;
+        for (const child of this.children)
+            if (child.pending())
+                return true;
+        return false;
+    }
+    async cleanup() {
+        for (const path of this.paths) {
+            try {
+                await rmTmpDir(path);
+                this.paths.delete(path);
             }
+            catch (_) { }
         }
+        for (const child of this.children)
+            await child.cleanup();
     }
-    async cleanupOnFinish(cb) {
-        const cleanup = new CleanupListener();
-        try {
-            await cb();
-        }
-        finally {
-            cleanup.dispose();
-        }
+    async dispose() {
+        await this.cleanup();
+        exports.collectors.delete(this);
     }
-    async cleanupIfFail(cb) {
-        const cleanup = new CleanupListener();
+    async disposeIfFail(cb) {
         try {
-            await cb();
+            return await cb();
         }
         catch (error) {
-            await cleanup.dispose();
+            await this.dispose();
             throw error;
         }
-        this.listeners.add(cleanup);
-        return cleanup;
+    }
+    disposeOnFinish() {
+        return {
+            [Symbol.asyncDispose]: async () => {
+                return this.dispose();
+            },
+        };
+    }
+    create() {
+        const gc = new GargabeCollector();
+        this.children.add(gc);
+        return gc;
     }
 }
 exports.GargabeCollector = GargabeCollector;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datatruck/cli",
-  "version": "0.34.3",
+  "version": "0.34.5",
   "description": "Tool for creating and managing backups",
   "homepage": "https://github.com/swordev/datatruck#readme",
   "bugs": {