pqb 0.3.8 → 0.4.0

package/dist/index.js

@@ -783,6 +783,27 @@ const aggregateToSql = (ctx, model, item, quotedAs) => {
 
 class PormError extends Error {
 }
+class QueryError extends pg.DatabaseError {
+  get isUnique() {
+    return this.code === "23505";
+  }
+  get columns() {
+    var _a;
+    if (this.columnsCache)
+      return this.columnsCache;
+    const columns = {};
+    if (this.detail) {
+      const list = (_a = this.detail.match(/\((.*)\)=/)) == null ? void 0 : _a[1];
+      if (list) {
+        list.split(", ").forEach((item) => {
+          const column = item.startsWith('"') ? item.slice(1, -1) : item;
+          columns[column] = true;
+        });
+      }
+    }
+    return this.columnsCache = columns;
+  }
+}
 class NotFoundError extends PormError {
   constructor(message = "Record is not found") {
     super(message);
@@ -3195,13 +3216,11 @@ const queryMethodByReturnType = {
   rowCount: "arrays",
   void: "arrays"
 };
+let queryError = void 0;
 class Then {
-  then(resolve, reject) {
-    if (this.query.wrapInTransaction && !this.query.inTransaction) {
-      return this.transaction((q) => then(q, resolve, reject));
-    } else {
-      return then(this, resolve, reject);
-    }
+  get then() {
+    queryError = new Error();
+    return maybeWrappedThen;
   }
   async catch(fn) {
     return this.then(void 0, fn);
@@ -3210,6 +3229,13 @@ class Then {
 const handleResult = async (q, result) => {
   return parseResult(q, q.query.returnType || "all", result);
 };
+function maybeWrappedThen(resolve, reject) {
+  if (this.query.wrapInTransaction && !this.query.inTransaction) {
+    return this.transaction((q) => then(q, resolve, reject));
+  } else {
+    return then(this, resolve, reject);
+  }
+}
 const then = async (q, resolve, reject) => {
   let sql;
   let logData;
@@ -3249,13 +3275,38 @@ const then = async (q, resolve, reject) => {
       );
     }
     resolve == null ? void 0 : resolve(result);
-  } catch (error) {
+  } catch (err) {
+    const error = err instanceof pg.DatabaseError ? assignError(q, err) : err;
     if (q.query.log && sql && logData) {
       q.query.log.onError(error, sql, logData);
     }
     reject == null ? void 0 : reject(error);
   }
 };
+const assignError = (q, from) => {
+  const to = new q.error();
+  to.stack = queryError.stack;
+  to.message = from.message;
+  to.length = from.length;
+  to.name = from.name;
+  to.severity = from.severity;
+  to.code = from.code;
+  to.detail = from.detail;
+  to.hint = from.hint;
+  to.position = from.position;
+  to.internalPosition = from.internalPosition;
+  to.internalQuery = from.internalQuery;
+  to.where = from.where;
+  to.schema = from.schema;
+  to.table = from.table;
+  to.column = from.column;
+  to.dataType = from.dataType;
+  to.constraint = from.constraint;
+  to.file = from.file;
+  to.line = from.line;
+  to.routine = from.routine;
+  return to;
+};
 const parseResult = (q, returnType = "all", result) => {
   var _a, _b;
   switch (returnType) {
@@ -5544,7 +5595,7 @@ class Db {
     const defaultSelect = this.defaultSelectColumns.length === columns.length ? void 0 : this.defaultSelectColumns;
     const columnsParsers = {};
     let hasParsers = false;
-    let modifyQuery;
+    let modifyQuery = void 0;
     for (const key in shape) {
       const column = shape[key];
       if (column.parseFn) {
@@ -5565,6 +5616,8 @@ class Db {
     } : toSql;
     this.relations = {};
     modifyQuery == null ? void 0 : modifyQuery.forEach((cb) => cb(this));
+    this.error = class extends QueryError {
+    };
   }
 }
 applyMixins(Db, [QueryMethods]);
@@ -5673,6 +5726,7 @@ exports.PolygonColumn = PolygonColumn;
 exports.PormError = PormError;
 exports.PormInternalError = PormInternalError;
 exports.QueryCallbacks = QueryCallbacks;
+exports.QueryError = QueryError;
 exports.QueryGet = QueryGet;
 exports.QueryLog = QueryLog;
 exports.QueryMethods = QueryMethods;