pqb 0.4.7 → 0.4.9

package/dist/index.js

@@ -70,6 +70,9 @@ class ColumnType {
     this.parseItem = fn;
     return this;
   }
+  as(column) {
+    return addColumnData(this, "as", column);
+  }
   toSQL() {
     return this.dataType;
   }
@@ -1953,6 +1956,14 @@ class DateBaseColumn extends ColumnType {
     this.data = {};
     this.operators = Operators.date;
   }
+  asNumber() {
+    return this.encode((input) => new Date(input)).parse(
+      Date.parse
+    );
+  }
+  asDate() {
+    return this.parse((input) => new Date(input));
+  }
 }
 assignMethodsToClass(DateBaseColumn, dateTypeMethods);
 class DateColumn extends DateBaseColumn {
@@ -4178,8 +4189,9 @@ const handleSelect = (q) => {
     q.query.select = ["*"];
   }
 };
-const processCreateItem = (item, rowIndex, ctx, columns, columnsMap) => {
+const processCreateItem = (item, rowIndex, ctx, columns, encoders, columnsMap, shape) => {
   Object.keys(item).forEach((key) => {
+    var _a;
     if (ctx.relations[key]) {
       if (ctx.relations[key].type === "belongsTo") {
         const foreignKey = ctx.relations[key].options.foreignKey;
@@ -4204,8 +4216,9 @@ const processCreateItem = (item, rowIndex, ctx, columns, columnsMap) => {
           item[key]
         ]);
       }
-    } else if (columnsMap[key] === void 0) {
+    } else if (columnsMap[key] === void 0 && (shape[key] || shape === anyShape)) {
       columnsMap[key] = columns.length;
+      encoders[key] = (_a = shape[key]) == null ? void 0 : _a.encodeFn;
       columns.push(key);
     }
   });
@@ -4232,30 +4245,37 @@ const getManyReturnType = (q) => {
     return "rowCount";
   }
 };
+const mapColumnValues = (columns, encoders, data) => {
+  return columns.map(
+    (key) => encoders[key] ? encoders[key](data[key]) : data[key]
+  );
+};
 const handleOneData = (q, data, ctx) => {
   const columns = [];
+  const encoders = {};
   const columnsMap = {};
   const defaults = q.query.defaults;
   if (defaults) {
     data = __spreadValues$4(__spreadValues$4({}, defaults), data);
   }
-  processCreateItem(data, 0, ctx, columns, columnsMap);
-  const values = [columns.map((key) => data[key])];
+  processCreateItem(data, 0, ctx, columns, encoders, columnsMap, q.shape);
+  const values = [mapColumnValues(columns, encoders, data)];
   return { columns, values };
 };
 const handleManyData = (q, data, ctx) => {
   const columns = [];
+  const encoders = {};
   const columnsMap = {};
   const defaults = q.query.defaults;
   if (defaults) {
     data = data.map((item) => __spreadValues$4(__spreadValues$4({}, defaults), item));
   }
   data.forEach((item, i) => {
-    processCreateItem(item, i, ctx, columns, columnsMap);
+    processCreateItem(item, i, ctx, columns, encoders, columnsMap, q.shape);
   });
   const values = Array(data.length);
   data.forEach((item, i) => {
-    values[i] = columns.map((key) => item[key]);
+    values[i] = mapColumnValues(columns, encoders, item);
   });
   return { columns, values };
 };
@@ -5127,7 +5147,7 @@ class Update {
     const data = args[0];
     const set = __spreadValues$1({}, data);
     pushQueryValue(this, "updateData", set);
-    const relations = this.relations;
+    const { relations, shape } = this;
     const prependRelations = {};
     const appendRelations = {};
     const originalReturnType = query.returnType || "all";
@@ -5145,6 +5165,12 @@ class Update {
           }
           appendRelations[key] = data[key];
         }
+      } else if (!shape[key] && shape !== anyShape) {
+        delete set[key];
+      } else {
+        const encode = shape[key].encodeFn;
+        if (encode)
+          set[key] = encode(set[key]);
       }
     }
     const state = {};
@@ -5609,8 +5635,9 @@ var __objRest = (source, exclude) => {
     }
   return target;
 };
+const anyShape = {};
 class Db {
-  constructor(adapter, queryBuilder, table = void 0, shape = {}, columnTypes, options) {
+  constructor(adapter, queryBuilder, table = void 0, shape = anyShape, columnTypes, options) {
     this.adapter = adapter;
     this.queryBuilder = queryBuilder;
     this.table = table;
@@ -5818,6 +5845,7 @@ exports.addWhere = addWhere;
 exports.addWhereIn = addWhereIn;
 exports.addWhereNot = addWhereNot;
 exports.aggregate1FunctionNames = aggregate1FunctionNames;
+exports.anyShape = anyShape;
 exports.applyMixins = applyMixins;
 exports.array = array;
 exports.arrayToEnum = arrayToEnum;