fastypest 1.3.10 → 1.4.1

package/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to this project will be documented in this file. See [standard-version](https://github.com/conventional-changelog/standard-version) for commit guidelines.
 
+### [1.4.1](https://github.com/juanjoGonDev/fastypest/compare/v1.4.0...v1.4.1) (2024-05-28)
+
+
+### Bug Fixes
+
+* **npm:** husky not found ([cfa9a49](https://github.com/juanjoGonDev/fastypest/commit/cfa9a49344177959fa13e72e996501ff61efaed5))
+
 ### [1.3.10](https://github.com/juanjoGonDev/fastypest/compare/v1.3.0...v1.3.10) (2024-05-21)
 
 

package/dist/core/config.d.ts

@@ -0,0 +1,2 @@
+import { DBType } from "./types";
+export declare const INDEX_OFFSET_CONFIG: Partial<Record<DBType, number>>;

package/dist/core/config.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.INDEX_OFFSET_CONFIG = void 0;
+exports.INDEX_OFFSET_CONFIG = {
+    postgres: 1,
+    cockroachdb: 0,
+};
+//# sourceMappingURL=config.js.map

package/dist/core/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/core/config.ts"],"names":[],"mappings":";;;AAEa,QAAA,mBAAmB,GAAoC;IAClE,QAAQ,EAAE,CAAC;IACX,WAAW,EAAE,CAAC;CACf,CAAC"}

package/dist/core/fastypest.d.ts

@@ -1,21 +1,26 @@
 import { Connection, DataSource, EntityManager } from "typeorm";
 import { SQLScript } from "./sql-script";
-interface Manager {
-    foreignKey: {
-        disable: () => Promise<string | object>;
-        enable: () => Promise<string | object>;
-    };
-    dependencyTree: () => Promise<Set<string> | undefined>;
-}
+import { Manager } from "./types";
 export declare class Fastypest extends SQLScript {
     private manager;
     private tables;
+    private tablesWithAutoIncrement;
+    private restoreInOder;
     constructor(connection: DataSource | Connection);
     init(): Promise<void>;
-    detectTables(em: EntityManager): Promise<void>;
+    private createTempTable;
+    private detectTablesWithAutoIncrement;
+    private processTable;
+    private getColumnsWithAutoIncrement;
+    private processColumn;
+    private getMaxColumnIndex;
+    private getSequenceName;
+    private updateTablesWithAutoIncrement;
     restoreData(): Promise<void>;
+    protected restoreManager(em: EntityManager): Promise<Manager>;
+    private calculateDependencyTables;
+    private detectTables;
     private restoreOrder;
-    restoreManager(em: EntityManager): Promise<Manager>;
-    clearTempTables(): Promise<void>;
+    private recreateData;
+    private resetAutoIncrementColumns;
 }
-export {};

package/dist/core/fastypest.js

@@ -10,60 +10,104 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Fastypest = void 0;
+const config_1 = require("./config");
 const sql_script_1 = require("./sql-script");
 class Fastypest extends sql_script_1.SQLScript {
     constructor(connection) {
         super(connection.options.type);
         this.tables = new Set();
+        this.tablesWithAutoIncrement = new Map();
+        this.restoreInOder = false;
         this.manager = connection.manager;
     }
     init() {
         return __awaiter(this, void 0, void 0, function* () {
             yield this.manager.transaction((em) => __awaiter(this, void 0, void 0, function* () {
                 yield this.detectTables(em);
-                yield Promise.all([...this.tables].map((tableName) => __awaiter(this, void 0, void 0, function* () {
-                    yield em.query(this.getQuery("createTempTable", { tableName }));
-                })));
+                yield this.calculateDependencyTables(em);
+                const tables = [...this.tables];
+                yield Promise.all([
+                    this.createTempTable(em, tables),
+                    this.detectTablesWithAutoIncrement(em, tables),
+                ]);
             }));
         });
     }
-    detectTables(em) {
+    createTempTable(em, tables) {
         return __awaiter(this, void 0, void 0, function* () {
-            (yield em.query(this.getQuery("getTables"))).map((row) => {
-                this.tables.add(row.name);
+            yield Promise.all(tables.map((tableName) => __awaiter(this, void 0, void 0, function* () {
+                yield this.execQuery(em, "createTempTable", { tableName });
+            })));
+        });
+    }
+    detectTablesWithAutoIncrement(em, tables) {
+        return __awaiter(this, void 0, void 0, function* () {
+            for (const tableName of tables) {
+                yield this.processTable(em, tableName);
+            }
+        });
+    }
+    processTable(em, tableName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const columns = yield this.getColumnsWithAutoIncrement(em, tableName);
+            if (!columns)
+                return;
+            for (const column of columns) {
+                yield this.processColumn(em, tableName, column);
+            }
+        });
+    }
+    getColumnsWithAutoIncrement(em, tableName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const columns = yield this.execQuery(em, "getColumnsWithAutoIncrement", { tableName });
+            return Array.isArray(columns) ? columns : null;
+        });
+    }
+    processColumn(em, tableName, column) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const stat = yield this.getMaxColumnIndex(em, tableName, column.column_name);
+            const sequenceName = this.getSequenceName(column.column_default);
+            if (!sequenceName)
+                return;
+            const index = Number(stat === null || stat === void 0 ? void 0 : stat.maxindex) || 0;
+            this.updateTablesWithAutoIncrement(tableName, {
+                column: column.column_name,
+                sequenceName,
+                index: String(index + ((_a = config_1.INDEX_OFFSET_CONFIG[this.getType()]) !== null && _a !== void 0 ? _a : 0)),
             });
         });
     }
+    getMaxColumnIndex(em, tableName, columnName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const [stat] = yield this.execQuery(em, "getMaxColumnIndex", {
+                tableName,
+                column_name: columnName,
+            });
+            return stat || null;
+        });
+    }
+    getSequenceName(columnDefault) {
+        var _a;
+        return ((_a = columnDefault.match(/'([^']+)'/)) === null || _a === void 0 ? void 0 : _a[1]) || null;
+    }
+    updateTablesWithAutoIncrement(tableName, data) {
+        var _a;
+        if (!this.tablesWithAutoIncrement.has(tableName)) {
+            this.tablesWithAutoIncrement.set(tableName, []);
+        }
+        (_a = this.tablesWithAutoIncrement.get(tableName)) === null || _a === void 0 ? void 0 : _a.push(data);
+    }
     restoreData() {
         return __awaiter(this, void 0, void 0, function* () {
             yield this.manager.transaction((em) => __awaiter(this, void 0, void 0, function* () {
-                const restoreManager = yield this.restoreManager(em);
-                yield restoreManager.foreignKey.disable();
-                const dependencyTree = yield restoreManager.dependencyTree();
-                const tables = [...(dependencyTree || this.tables)];
-                yield this.restoreOrder(em, tables, dependencyTree ? "sorted" : "random");
-                yield restoreManager.foreignKey.enable();
+                const { foreignKey, restoreOrder } = yield this.restoreManager(em);
+                yield foreignKey.disable();
+                yield restoreOrder();
+                yield foreignKey.enable();
             }));
         });
     }
-    restoreOrder(em_1, tables_1) {
-        return __awaiter(this, arguments, void 0, function* (em, tables, type = "random") {
-            switch (type) {
-                case "sorted":
-                    for (const tableName of tables) {
-                        yield em.query(this.getQuery("truncateTable", { tableName }));
-                        yield em.query(this.getQuery("restoreData", { tableName }));
-                    }
-                    break;
-                default:
-                    yield Promise.all(tables.map((tableName) => __awaiter(this, void 0, void 0, function* () {
-                        yield em.query(this.getQuery("truncateTable", { tableName }));
-                        yield em.query(this.getQuery("restoreData", { tableName }));
-                    })));
-                    break;
-            }
-        });
-    }
     restoreManager(em) {
         return __awaiter(this, void 0, void 0, function* () {
             if (this.tables.size === 0) {
@@ -71,29 +115,73 @@ class Fastypest extends sql_script_1.SQLScript {
             }
             const manager = {
                 foreignKey: {
-                    disable: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve({}); }),
-                    enable: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve({}); }),
+                    disable: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve(); }),
+                    enable: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve(); }),
                 },
-                dependencyTree: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve(undefined); }),
+                restoreOrder: () => __awaiter(this, void 0, void 0, function* () { return Promise.resolve(); }),
             };
-            switch (this.type) {
-                case "cockroachdb":
-                    const dependencyTree = yield em.query(this.getQuery("dependencyTree"));
-                    manager.dependencyTree = () => __awaiter(this, void 0, void 0, function* () {
-                        return new Set(dependencyTree.map((row) => row.table_name));
-                    });
-                    break;
-                case "mariadb":
-                case "mysql":
-                    manager.foreignKey.disable = () => __awaiter(this, void 0, void 0, function* () { return em.query(this.getQuery("foreignKey.disable")); });
-                    manager.foreignKey.enable = () => __awaiter(this, void 0, void 0, function* () { return em.query(this.getQuery("foreignKey.enable")); });
+            const typesWithForeignKey = ["postgres", "mariadb", "mysql"];
+            if (typesWithForeignKey.includes(this.getType())) {
+                manager.foreignKey.disable = () => __awaiter(this, void 0, void 0, function* () { return this.execQuery(em, "foreignKey.disable"); });
+                manager.foreignKey.enable = () => __awaiter(this, void 0, void 0, function* () { return this.execQuery(em, "foreignKey.enable"); });
             }
+            manager.restoreOrder = () => this.restoreOrder(em);
             return manager;
         });
     }
-    clearTempTables() {
+    calculateDependencyTables(em) {
         return __awaiter(this, void 0, void 0, function* () {
+            const dependencyTree = yield this.execQuery(em, "dependencyTree");
+            if (!dependencyTree.length) {
+                this.restoreInOder = false;
+                return;
+            }
+            const sortedTables = new Set(dependencyTree.map((row) => row.table_name));
             this.tables.clear();
+            this.tables = sortedTables;
+            this.restoreInOder = true;
+        });
+    }
+    detectTables(em) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const tables = yield this.execQuery(em, "getTables");
+            tables.forEach((row) => {
+                this.tables.add(row.name);
+            });
+        });
+    }
+    restoreOrder(em) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.restoreInOder) {
+                for (const tableName of this.tables) {
+                    yield this.recreateData(em, tableName);
+                }
+                return;
+            }
+            const tables = [...this.tables];
+            yield Promise.all(tables.map((tableName) => this.recreateData(em, tableName)));
+        });
+    }
+    recreateData(em, tableName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.execQuery(em, "truncateTable", { tableName });
+            yield this.execQuery(em, "restoreData", { tableName });
+            yield this.resetAutoIncrementColumns(em, tableName);
+        });
+    }
+    resetAutoIncrementColumns(em, tableName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const tables = this.tablesWithAutoIncrement.get(tableName);
+            if (!tables)
+                return;
+            for (const { column, sequenceName, index } of tables) {
+                yield this.execQuery(em, "resetAutoIncrementColumn", {
+                    tableName,
+                    column,
+                    sequenceName,
+                    index,
+                });
+            }
         });
     }
 }

package/dist/core/fastypest.js.map

@@ -1 +1 @@
-{"version":3,"file":"fastypest.js","sourceRoot":"","sources":["../../src/core/fastypest.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,6CAAyC;AAUzC,MAAa,SAAU,SAAQ,sBAAS;IAItC,YAAY,UAAmC;QAC7C,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAHzB,WAAM,GAAgB,IAAI,GAAG,EAAE,CAAC;QAItC,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;IACpC,CAAC;IAEK,IAAI;;YACR,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAO,EAAiB,EAAE,EAAE;gBACzD,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBAE5B,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAO,SAAS,EAAE,EAAE;oBACvC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;gBAClE,CAAC,CAAA,CAAC,CACH,CAAC;YACJ,CAAC,CAAA,CAAC,CAAC;QACL,CAAC;KAAA;IAEK,YAAY,CAAC,EAAiB;;YAClC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,CAC9C,CAAC,GAA2B,EAAE,EAAE;gBAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC,CACF,CAAC;QACJ,CAAC;KAAA;IAEK,WAAW;;YACf,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAO,EAAiB,EAAE,EAAE;gBACzD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;gBACrD,MAAM,cAAc,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBAC1C,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,cAAc,EAAE,CAAC;gBAC7D,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBAEpD,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAE1E,MAAM,cAAc,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC3C,CAAC,CAAA,CAAC,CAAC;QACL,CAAC;KAAA;IAEa,YAAY;6DACxB,EAAiB,EACjB,MAAgB,EAChB,OAA4B,QAAQ;YAEpC,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,QAAQ;oBACX,KAAK,MAAM,SAAS,IAAI,MAAM,EAAE,CAAC;wBAC/B,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;wBAC9D,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;oBAC9D,CAAC;oBACD,MAAM;gBACR;oBACE,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAO,SAAS,EAAE,EAAE;wBAC7B,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;wBAC9D,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;oBAC9D,CAAC,CAAA,CAAC,CACH,CAAC;oBACF,MAAM;YACV,CAAC;QACH,CAAC;KAAA;IAEK,cAAc,CAAC,EAAiB;;YACpC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;YAED,MAAM,OAAO,GAAY;gBACvB,UAAU,EAAE;oBACV,OAAO,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA,GAAA;oBACxC,MAAM,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA,GAAA;iBACxC;gBACD,cAAc,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA,GAAA;aACvD,CAAC;YAEF,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,aAAa;oBAChB,MAAM,cAAc,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,CAAC;oBACvE,OAAO,CAAC,cAAc,GAAG,GAAS,EAAE;wBAClC,OAAA,IAAI,GAAG,CACL,cAAc,CAAC,GAAG,CAAC,CAAC,GAA2B,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CACrD,CAAA;sBAAA,CAAC;oBACnB,MAAM;gBACR,KAAK,SAAS,CAAC;gBACf,KAAK,OAAO;oBACV,OAAO,CAAC,UAAU,CAAC,OAAO,GAAG,GAAS,EAAE,gDACtC,OAAA,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,CAAA,GAAA,CAAC;oBAChD,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,GAAS,EAAE,gDACrC,OAAA,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAA,GAAA,CAAC;YACnD,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;KAAA;IAEK,eAAe;;YACnB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACtB,CAAC;KAAA;CACF;AApGD,8BAoGC"}
+{"version":3,"file":"fastypest.js","sourceRoot":"","sources":["../../src/core/fastypest.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,qCAA+C;AAC/C,6CAAyC;AAUzC,MAAa,SAAU,SAAQ,sBAAS;IAMtC,YAAY,UAAmC;QAC7C,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QALzB,WAAM,GAAgB,IAAI,GAAG,EAAE,CAAC;QAChC,4BAAuB,GAAmC,IAAI,GAAG,EAAE,CAAC;QACpE,kBAAa,GAAY,KAAK,CAAC;QAIrC,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;IACpC,CAAC;IAEY,IAAI;;YACf,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAO,EAAiB,EAAE,EAAE;gBACzD,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBAC5B,MAAM,IAAI,CAAC,yBAAyB,CAAC,EAAE,CAAC,CAAC;gBACzC,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;gBAChC,MAAM,OAAO,CAAC,GAAG,CAAC;oBAChB,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,MAAM,CAAC;oBAChC,IAAI,CAAC,6BAA6B,CAAC,EAAE,EAAE,MAAM,CAAC;iBAC/C,CAAC,CAAC;YACL,CAAC,CAAA,CAAC,CAAC;QACL,CAAC;KAAA;IAEa,eAAe,CAC3B,EAAiB,EACjB,MAAgB;;YAEhB,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAO,SAAS,EAAE,EAAE;gBAC7B,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,iBAAiB,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;YAC7D,CAAC,CAAA,CAAC,CACH,CAAC;QACJ,CAAC;KAAA;IAEa,6BAA6B,CACzC,EAAiB,EACjB,MAAgB;;YAEhB,KAAK,MAAM,SAAS,IAAI,MAAM,EAAE,CAAC;gBAC/B,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;KAAA;IAEa,YAAY,CACxB,EAAiB,EACjB,SAAiB;;YAEjB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YACtE,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;KAAA;IAEa,2BAA2B,CACvC,EAAiB,EACjB,SAAiB;;YAEjB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAClC,EAAE,EACF,6BAA6B,EAC7B,EAAE,SAAS,EAAE,CACd,CAAC;YACF,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QACjD,CAAC;KAAA;IAEa,aAAa,CACzB,EAAiB,EACjB,SAAiB,EACjB,MAAgC;;;YAEhC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CACvC,EAAE,EACF,SAAS,EACT,MAAM,CAAC,WAAW,CACnB,CAAC;YACF,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACjE,IAAI,CAAC,YAAY;gBAAE,OAAO;YAE1B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC,6BAA6B,CAAC,SAAS,EAAE;gBAC5C,MAAM,EAAE,MAAM,CAAC,WAAW;gBAC1B,YAAY;gBACZ,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,CAAC,MAAA,4BAAmB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,mCAAI,CAAC,CAAC,CAAC;aAClE,CAAC,CAAC;QACL,CAAC;KAAA;IAEa,iBAAiB,CAC7B,EAAiB,EACjB,SAAiB,EACjB,UAAkB;;YAElB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,SAAS,CAAa,EAAE,EAAE,mBAAmB,EAAE;gBACvE,SAAS;gBACT,WAAW,EAAE,UAAU;aACxB,CAAC,CAAC;YACH,OAAO,IAAI,IAAI,IAAI,CAAC;QACtB,CAAC;KAAA;IAEO,eAAe,CAAC,aAAqB;;QAC3C,OAAO,CAAA,MAAA,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,0CAAG,CAAC,CAAC,KAAI,IAAI,CAAC;IACvD,CAAC;IAEO,6BAA6B,CACnC,SAAiB,EACjB,IAA6D;;QAE7D,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,MAAA,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,0CAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC;IAEY,WAAW;;YACtB,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAO,EAAiB,EAAE,EAAE;gBACzD,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;gBACnE,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC;gBAC3B,MAAM,YAAY,EAAE,CAAC;gBACrB,MAAM,UAAU,CAAC,MAAM,EAAE,CAAC;YAC5B,CAAC,CAAA,CAAC,CAAC;QACL,CAAC;KAAA;IAEe,cAAc,CAAC,EAAiB;;YAC9C,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;YAED,MAAM,OAAO,GAAY;gBACvB,UAAU,EAAE;oBACV,OAAO,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,EAAE,CAAA,GAAA;oBACtC,MAAM,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,EAAE,CAAA,GAAA;iBACtC;gBACD,YAAY,EAAE,GAAS,EAAE,gDAAC,OAAA,OAAO,CAAC,OAAO,EAAE,CAAA,GAAA;aAC5C,CAAC;YAEF,MAAM,mBAAmB,GAAa,CAAC,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACvE,IAAI,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBACjD,OAAO,CAAC,UAAU,CAAC,OAAO,GAAG,GAAwB,EAAE,gDACrD,OAAA,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,oBAAoB,CAAC,CAAA,GAAA,CAAC;gBAC3C,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,GAAwB,EAAE,gDACpD,OAAA,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAA,GAAA,CAAC;YAC5C,CAAC;YAED,OAAO,CAAC,YAAY,GAAG,GAAkB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAElE,OAAO,OAAO,CAAC;QACjB,CAAC;KAAA;IAEa,yBAAyB,CAAC,EAAiB;;YACvD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,SAAS,CACzC,EAAE,EACF,gBAAgB,CACjB,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC3B,OAAO;YACT,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC1E,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC;YAC3B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC5B,CAAC;KAAA;IAEa,YAAY,CAAC,EAAiB;;YAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAQ,EAAE,EAAE,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACrB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC,CAAC,CAAC;QACL,CAAC;KAAA;IAEa,YAAY,CAAC,EAAiB;;YAC1C,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvB,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;oBACpC,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;gBACzC,CAAC;gBAED,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;YAChC,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC,CAC5D,CAAC;QACJ,CAAC;KAAA;IAEa,YAAY,CACxB,EAAiB,EACjB,SAAiB;;YAEjB,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,eAAe,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;YACzD,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,aAAa,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;YACvD,MAAM,IAAI,CAAC,yBAAyB,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;KAAA;IAEa,yBAAyB,CACrC,EAAiB,EACjB,SAAiB;;YAEjB,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC3D,IAAI,CAAC,MAAM;gBAAE,OAAO;YAEpB,KAAK,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,MAAM,EAAE,CAAC;gBACrD,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,0BAA0B,EAAE;oBACnD,SAAS;oBACT,MAAM;oBACN,YAAY;oBACZ,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;QACH,CAAC;KAAA;CACF;AAvND,8BAuNC"}

package/dist/core/sql-script/queries/cockroachdb.json

@@ -7,6 +7,9 @@
     "getTables": "SELECT table_name AS name FROM information_schema.tables WHERE table_catalog = CURRENT_DATABASE() AND table_schema = 'public' AND table_type = 'BASE TABLE'",
     "createTempTable": "SET experimental_enable_temp_tables=on; CREATE TEMPORARY TABLE \"{{tableName}}_temp\" AS SELECT * FROM \"{{tableName}}\"",
     "dropTempTable": "DROP TABLE IF EXISTS \"{{tableName}}_temp\"",
-    "restoreData": "INSERT INTO \"{{tableName}}\" SELECT * FROM \"{{tableName}}_temp\"; SELECT setval(pg_get_serial_sequence('\"{{tableName}}\"', 'id'), (SELECT COALESCE(MAX(id), 0) FROM \"{{tableName}}\"))",
-    "truncateTable": "TRUNCATE TABLE \"{{tableName}}\" CASCADE; ALTER SEQUENCE {{tableName}}_id_seq RESTART WITH 1;"
+    "restoreData": "INSERT INTO \"{{tableName}}\" SELECT * FROM \"{{tableName}}_temp\";",
+    "truncateTable": "TRUNCATE TABLE \"{{tableName}}\" CASCADE;",
+    "getColumnsWithAutoIncrement": "SELECT column_name, column_default FROM information_schema.columns WHERE table_schema = 'public' AND column_default LIKE 'nextval%' AND table_name = '{{tableName}}'",
+    "resetAutoIncrementColumn": "SELECT setval('{{sequenceName}}',{{index}});",
+    "getMaxColumnIndex": "SELECT MAX({{column_name}}) AS maxindex FROM \"{{tableName}}\""
 }

package/dist/core/sql-script/queries/index.d.ts

@@ -10,6 +10,9 @@ export declare const DB_QUERIES: {
         dropTempTable: string;
         restoreData: string;
         truncateTable: string;
+        getColumnsWithAutoIncrement: string;
+        resetAutoIncrementColumn: string;
+        getMaxColumnIndex: string;
     };
     mariadb: {
         dependencyTree: string;
@@ -22,6 +25,9 @@ export declare const DB_QUERIES: {
         dropTempTable: string;
         restoreData: string;
         truncateTable: string;
+        getColumnsWithAutoIncrement: string;
+        resetAutoIncrementColumn: string;
+        getMaxColumnIndex: string;
     };
     postgres: {
         dependencyTree: string;
@@ -34,6 +40,9 @@ export declare const DB_QUERIES: {
         dropTempTable: string;
         restoreData: string;
         truncateTable: string;
+        getColumnsWithAutoIncrement: string;
+        resetAutoIncrementColumn: string;
+        getMaxColumnIndex: string;
     };
     cockroachdb: {
         dependencyTree: string;
@@ -46,6 +55,9 @@ export declare const DB_QUERIES: {
         dropTempTable: string;
         restoreData: string;
         truncateTable: string;
+        getColumnsWithAutoIncrement: string;
+        resetAutoIncrementColumn: string;
+        getMaxColumnIndex: string;
     };
 };
 export type AllowedDataBases = keyof typeof DB_QUERIES;

package/dist/core/sql-script/queries/mysql.query.json

@@ -8,5 +8,8 @@
     "createTempTable": "CREATE TEMPORARY TABLE {{tableName}}_temp AS SELECT * FROM {{tableName}}",
     "dropTempTable": "DROP TEMPORARY TABLE {{tableName}}_temp",
     "restoreData": "INSERT INTO {{tableName}} SELECT * FROM {{tableName}}_temp",
-    "truncateTable": "TRUNCATE TABLE {{tableName}}"
+    "truncateTable": "TRUNCATE TABLE {{tableName}}",
+    "getColumnsWithAutoIncrement": "-- NOT USED",
+    "resetAutoIncrementColumn": "-- NOT USED",
+    "getMaxColumnIndex": "-- NOT USED"
 }

package/dist/core/sql-script/queries/postgres.query.json

@@ -1,12 +1,15 @@
 {
     "dependencyTree": "-- NOT USED",
     "foreignKey": {
-        "disable": "-- NOT USED",
-        "enable": "-- NOT USED"
+        "disable": "DO $$ DECLARE table_name text; BEGIN FOR table_name IN (SELECT tablename FROM pg_tables WHERE schemaname = current_schema()) LOOP EXECUTE 'ALTER TABLE \"' || table_name || '\" DISABLE TRIGGER ALL;'; END LOOP; END $$;",
+        "enable": "DO $$ DECLARE table_name text; BEGIN FOR table_name IN (SELECT tablename FROM pg_tables WHERE schemaname = current_schema()) LOOP EXECUTE 'ALTER TABLE \"' || table_name || '\" ENABLE TRIGGER ALL;'; END LOOP; END $$;"
     },
     "getTables": "SELECT table_name AS name FROM information_schema.tables  WHERE table_schema = CURRENT_SCHEMA() AND table_type = 'BASE TABLE'",
     "createTempTable": "CREATE TEMPORARY TABLE \"{{tableName}}_temp\" AS SELECT * FROM \"{{tableName}}\"",
     "dropTempTable": "DROP TABLE IF EXISTS \"{{tableName}}_temp\"",
-    "restoreData": "INSERT INTO \"{{tableName}}\" SELECT * FROM \"{{tableName}}_temp\"; SELECT SETVAL(pg_get_serial_sequence('{{tableName}}', 'id'), (SELECT COALESCE(MAX(id), 0) FROM \"{{tableName}}\"))",
-    "truncateTable": "TRUNCATE TABLE \"{{tableName}}\" RESTART IDENTITY CASCADE"
+    "restoreData": "INSERT INTO \"{{tableName}}\" SELECT * FROM \"{{tableName}}_temp\";",
+    "truncateTable": "TRUNCATE TABLE \"{{tableName}}\" RESTART IDENTITY CASCADE",
+    "getColumnsWithAutoIncrement": "SELECT column_name, column_default FROM information_schema.columns WHERE table_schema = 'public' AND column_default LIKE 'nextval%' AND table_name = '{{tableName}}'",
+    "resetAutoIncrementColumn": "ALTER SEQUENCE \"{{sequenceName}}\" RESTART WITH {{index}};",
+    "getMaxColumnIndex": "SELECT MAX({{column_name}}) AS maxindex FROM \"{{tableName}}\""
 }

package/dist/core/sql-script/sql-script.d.ts

@@ -1,11 +1,12 @@
-import { DataSourceOptions } from "typeorm";
+import { DataSourceOptions, EntityManager } from "typeorm";
 import { Queries } from "./queries";
 import { QueryPath } from "./types";
 type DBTypes = DataSourceOptions["type"];
 export declare class SQLScript {
-    readonly type: DBTypes;
-    queries: Queries;
-    constructor(type: DBTypes);
-    getQuery(queryPath: QueryPath<Queries>, values?: Record<string, string>): string;
+    private readonly type;
+    private queries;
+    protected constructor(type: DBTypes);
+    protected getType(): DBTypes;
+    protected execQuery<T = void>(em: EntityManager, queryPath: QueryPath<Queries>, values?: Record<string, string>): T extends void ? Promise<void> : Promise<T[]>;
 }
 export {};

package/dist/core/sql-script/sql-script.js

@@ -10,15 +10,18 @@ class SQLScript {
         }
         this.queries = queries_1.DB_QUERIES[this.type];
     }
-    getQuery(queryPath, values) {
+    getType() {
+        return this.type;
+    }
+    execQuery(em, queryPath, values) {
         const queryObj = queryPath
             .split(".")
             .reduce((obj, key) => obj[key], this.queries);
-        let queryStr = queryObj;
+        let query = queryObj;
         for (const key in values) {
-            queryStr = queryStr.replace(new RegExp(`{{\\s*${key}\\s*}}`, "g"), values[key]);
+            query = query.replace(new RegExp(`{{\\s*${key}\\s*}}`, "g"), values[key]);
         }
-        return queryStr;
+        return em.query(query);
     }
 }
 exports.SQLScript = SQLScript;

package/dist/core/sql-script/sql-script.js.map

@@ -1 +1 @@
-{"version":3,"file":"sql-script.js","sourceRoot":"","sources":["../../../src/core/sql-script/sql-script.ts"],"names":[],"mappings":";;;AACA,uCAAkE;AAKlE,MAAa,SAAS;IAEpB,YAAqB,IAAa;QAAb,SAAI,GAAJ,IAAI,CAAS;QAChC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,oBAAU,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,kFAAkF,MAAM,CAAC,IAAI,CAC3F,oBAAU,CACX,EAAE,CACJ,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,oBAAU,CAAC,IAAI,CAAC,IAAwB,CAAC,CAAC;IAC3D,CAAC;IAED,QAAQ,CACN,SAA6B,EAC7B,MAA+B;QAE/B,MAAM,QAAQ,GAAG,SAAS;aACvB,KAAK,CAAC,GAAG,CAAC;aACV,MAAM,CAAC,CAAC,GAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAErD,IAAI,QAAQ,GAAG,QAAQ,CAAC;QAExB,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,QAAQ,GAAG,QAAQ,CAAC,OAAO,CACzB,IAAI,MAAM,CAAC,SAAS,GAAG,QAAQ,EAAE,GAAG,CAAC,EACrC,MAAM,CAAC,GAAG,CAAC,CACZ,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAjCD,8BAiCC"}
+{"version":3,"file":"sql-script.js","sourceRoot":"","sources":["../../../src/core/sql-script/sql-script.ts"],"names":[],"mappings":";;;AACA,uCAAkE;AAKlE,MAAa,SAAS;IAGpB,YAAuC,IAAa;QAAb,SAAI,GAAJ,IAAI,CAAS;QAClD,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,oBAAU,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,kFAAkF,MAAM,CAAC,IAAI,CAC3F,oBAAU,CACX,EAAE,CACJ,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,oBAAU,CAAC,IAAI,CAAC,IAAwB,CAAC,CAAC;IAC3D,CAAC;IAES,OAAO;QACf,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAES,SAAS,CACjB,EAAiB,EACjB,SAA6B,EAC7B,MAA+B;QAE/B,MAAM,QAAQ,GAAG,SAAS;aACvB,KAAK,CAAC,GAAG,CAAC;aACV,MAAM,CAAC,CAAC,GAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAErD,IAAI,KAAK,GAAG,QAAQ,CAAC;QAErB,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,SAAS,GAAG,QAAQ,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,EAAE,CAAC,KAAK,CAAC,KAAK,CAAkD,CAAC;IAC1E,CAAC;CACF;AApCD,8BAoCC"}

package/dist/core/types.d.ts

@@ -0,0 +1,28 @@
+import { DataSourceOptions } from "typeorm";
+export type Table = {
+    name: string;
+};
+export type DependencyTreeQueryOut = {
+    table_name: string;
+    level: number;
+};
+export type ColumnsWithAutoIncrement = {
+    column_name: string;
+    column_default: string;
+};
+export type Manager = {
+    foreignKey: {
+        disable: () => Promise<void>;
+        enable: () => Promise<void>;
+    };
+    restoreOrder: () => Promise<void>;
+};
+export type ColumnStat = {
+    maxindex: string | null;
+};
+export type DBType = DataSourceOptions["type"];
+export type IncrementDetail = {
+    column: string;
+    sequenceName: string;
+    index: string;
+};

package/dist/core/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/core/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fastypest",
-  "version": "1.3.10",
+  "version": "1.4.1",
   "description": "Restores the database automatically after each test. Allows serial execution of tests without having to delete and restore the database having to stop the application",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,15 +12,14 @@
     "node": "18.18.0",
     "yarn": "4.2.2"
   },
-  "installConfig": {
-    "pnp": false
-  },
   "scripts": {
+    "postinstall": "husky install",
     "prebuild": "rimraf dist",
     "build": "tsc",
     "eslint": "eslint --fix .",
     "pretest": "tsc",
-    "test": "jest --runInBand --config ./jest.config.json",
+    "test:ci": "jest --runInBand --forceExit --verbose --showSeed --config ./tests/jest.config.json",
+    "test": "yarn build && yarn test:ci",
     "docker": "ts-node -r tsconfig-paths/register scripts/prepare-docker.ts",
     "release": "standard-version --release-as patch"
   },
@@ -53,15 +52,16 @@
     "@types/cross-spawn": "^6.0.6",
     "@types/jest": "^29.5.12",
     "@types/node": "^20.12.12",
-    "@typescript-eslint/eslint-plugin": "^7.10.0",
+    "@typescript-eslint/eslint-plugin": "^7.11.0",
     "cross-spawn": "^7.0.3",
     "eslint": "^9.3.0",
     "eslint-config-standard-with-typescript": "^43.0.1",
     "eslint-plugin-import": "^2.29.1",
     "eslint-plugin-n": "^17.7.0",
     "eslint-plugin-prettier": "^5.1.3",
-    "eslint-plugin-promise": "^6.1.1",
+    "eslint-plugin-promise": "^6.2.0",
     "globals": "^15.3.0",
+    "husky": "^9.0.11",
     "jest": "^29.7.0",
     "mariadb": "^3.3.0",
     "mysql": "^2.18.1",
@@ -73,7 +73,7 @@
     "tsconfig-paths": "^4.2.0",
     "typeorm": "^0.3.20",
     "typescript": "^5.4.5",
-    "typescript-eslint": "^7.10.0"
+    "typescript-eslint": "^7.11.0"
   },
   "peerDependencies": {
     "typeorm": "^0.3.20"