@datatruck/cli 0.36.4 → 0.36.5

package/lib/utils/datatruck/repository-server.js

@@ -4,10 +4,12 @@ exports.createDatatruckRepositoryServer = exports.headerKey = void 0;
 const ConfigAction_1 = require("../../actions/ConfigAction");
 const cli_1 = require("../cli");
 const http_1 = require("../http");
+const math_1 = require("../math");
 const virtual_fs_1 = require("../virtual-fs");
 const fs_1 = require("fs");
 const promises_1 = require("fs/promises");
 const http_2 = require("http");
+const promises_2 = require("stream/promises");
 exports.headerKey = {
     user: "x-dtt-user",
     password: "x-dtt-password",
@@ -58,33 +60,31 @@ const getRemoteAddress = (req, options) => {
         : undefined) ?? req.socket.remoteAddress);
 };
 function createDatatruckRepositoryServer(inOptions, config = {}) {
+    const counter = new math_1.Counter();
     return (0, http_2.createServer)(async (req, res) => {
+        const url = req.url || "";
+        if (url === "/" || url === "/favicon.ico")
+            return res.end();
+        const id = counter.next();
+        let requestError;
+        let responseError;
+        req.on("error", (error) => (requestError = error));
+        res.on("error", (error) => (responseError = error));
         try {
-            if (req.url === "/" || req.url === "/favicon.ico")
-                return res.end();
-            const { repository, action, params } = parseUrl(req.url);
-            if (!repository || !action) {
-                res.statusCode = 404;
-                return res.end();
-            }
+            const { repository, action, params } = parseUrl(url);
+            if (!repository || !action)
+                return res.writeHead(404);
             const fileOptions = config.configPath
                 ? (await ConfigAction_1.ConfigAction.findAndParseFile(config.configPath)).server
                     ?.repository
                 : undefined;
             const options = fileOptions ?? inOptions;
             const backend = findRepositoryBackend(req, repository, options);
-            if (!backend) {
-                res.statusCode = 401;
-                return res.end();
-            }
+            if (!backend)
+                return res.writeHead(401);
             if (config.log)
-                (0, cli_1.logJson)("repository-server", "request", {
-                    repository,
-                    url: req.url,
-                });
-            const fs = new virtual_fs_1.LocalFs({
-                backend: backend.path,
-            });
+                (0, cli_1.logJson)("repository-server", "request", { id, repository, url });
+            const fs = new virtual_fs_1.LocalFs({ backend: backend.path });
             if (action === "comcheck") {
                 res.write(JSON.stringify({ success: true }));
             }
@@ -92,12 +92,7 @@ function createDatatruckRepositoryServer(inOptions, config = {}) {
                 const [target] = params;
                 const path = fs.resolvePath(target);
                 const file = (0, fs_1.createWriteStream)(path);
-                req.pipe(file);
-                await new Promise((resolve, reject) => {
-                    req.on("error", reject);
-                    file.on("error", reject);
-                    file.on("close", resolve);
-                });
+                await (0, promises_2.pipeline)(req, file);
             }
             else if (action === "download") {
                 const [target] = params;
@@ -105,13 +100,7 @@ function createDatatruckRepositoryServer(inOptions, config = {}) {
                 const file = (0, fs_1.createReadStream)(path);
                 const fileStat = await (0, promises_1.stat)(path);
                 res.setHeader("Content-Length", fileStat.size);
-                file.pipe(res);
-                await new Promise((resolve, reject) => {
-                    req.on("error", reject);
-                    file.on("error", reject);
-                    res.on("error", reject);
-                    res.on("close", resolve);
-                });
+                await (0, promises_2.pipeline)(file, res, { end: false });
             }
             else if (action === "writeFile") {
                 const data = await (0, http_1.readRequestData)(req);
@@ -127,21 +116,31 @@ function createDatatruckRepositoryServer(inOptions, config = {}) {
                     res.write(JSON.stringify(json));
             }
             if (config.log)
-                (0, cli_1.logJson)("repository-server", "request finished", {
-                    url: req.url,
-                });
-            res.end();
+                (0, cli_1.logJson)("repository-server", "request finished", { id });
         }
         catch (error) {
             if (config.log) {
-                (0, cli_1.logJson)("repository-server", "request failed", {
-                    url: req.url,
-                });
+                (0, cli_1.logJson)("repository-server", "request failed", { id });
                 console.error(error);
             }
-            res.statusCode = 500;
-            res.statusMessage = error.message;
-            res.end();
+            if (!res.headersSent)
+                res.writeHead(500, error.message);
+        }
+        finally {
+            if (requestError) {
+                (0, cli_1.logJson)("repository-server", "request error", { id });
+                console.error(requestError);
+            }
+            if (responseError) {
+                (0, cli_1.logJson)("repository-server", "response error", { id });
+                console.error(responseError);
+            }
+            if (requestError || responseError) {
+                res.destroy();
+            }
+            else {
+                res.end();
+            }
         }
     });
 }

package/lib/utils/math.d.ts

@@ -1 +1,7 @@
 export declare function progressPercent(total: number, current: number): number;
+export declare class Counter {
+    protected maxValue: number;
+    protected value: number;
+    constructor(maxValue?: number);
+    next(): number;
+}

package/lib/utils/math.js

@@ -1,9 +1,22 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.progressPercent = void 0;
+exports.Counter = exports.progressPercent = void 0;
 function progressPercent(total, current) {
     if (total === 0 && current === 0)
         return 0;
     return Number(((current / total) * 100).toFixed(2));
 }
 exports.progressPercent = progressPercent;
+class Counter {
+    maxValue;
+    value = 0;
+    constructor(maxValue = 4294967295) {
+        this.maxValue = maxValue;
+    }
+    next() {
+        if (this.maxValue && this.value >= this.maxValue)
+            this.value = 0;
+        return ++this.value;
+    }
+}
+exports.Counter = Counter;

package/package.json

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